com.activitystream.aspects.PageviewAspect Maven / Gradle / Ivy

Go to download
package com.activitystream.aspects;

import com.activitystream.Aspect;
import com.activitystream.EntityRelation;
import com.activitystream.underware.Factories;

import java.util.*;

public class PageviewAspect implements Aspect {
    private String path;
    private Map pathProps = Factories.getMap();
    private List keywords;
    private RequestMethod method;
    private Integer responseCode;
    private Integer size;
    private String protocol;
    private List pageContent;
    private String referrer;
    private Map referrerProperties;

    public PageviewAspect(String url) {
        path = url;
    }

    public PageviewAspect pathProperties(Map pathProps) {
        this.pathProps = pathProps;
        return this;
    }

    public PageviewAspect keywords(String... keywords) {
        this.keywords = new ArrayList();
        Collections.addAll(this.keywords, keywords);
        return this;
    }

    public PageviewAspect method(RequestMethod method) {
        this.method = method;
        return this;
    }

    public PageviewAspect responseCode(Integer responseCode) {
        this.responseCode = responseCode;
        return this;
    }

    public PageviewAspect size(Integer size) {
        this.size = size;
        return this;
    }

    public PageviewAspect protocol(String protocol) {
        this.protocol = protocol;
        return this;
    }

    public PageviewAspect pageContent(EntityRelation... pageContent) {
        this.pageContent = new ArrayList<>();
        Collections.addAll(this.pageContent, pageContent);
        return this;
    }

    public PageviewAspect referrer(String referrer) {
        this.referrer = referrer;
        return this;
    }

    public PageviewAspect referrerProperties(Map referrerProperties) {
        this.referrerProperties = referrerProperties;
        return this;
    }

    @Override
    public void addToObject(Map jsonObject, Set processed) {
        Map result = Factories.getMap();
        if (path != null) result.put("path", path);
        if (pathProps != null && pathProps.size() > 0) result.put("path_properties", pathProps);
        if (keywords != null) result.put("keywords", keywords);
        if (method != null) result.put("method", method.toString());
        if (responseCode != null) result.put("response_code", responseCode);
        if (size != null) result.put("size", size);
        if (protocol != null) result.put("protocol", protocol);
        if (referrer != null) result.put("referrer", referrer);
        if (referrerProperties != null && referrerProperties.size() > 0)
            result.put("referrer_properties", referrerProperties);
        if (pageContent != null) {
            List pageContentItems = new ArrayList();
            for (EntityRelation entityRelation : pageContent) {
                pageContentItems.add(entityRelation.toJson(processed));
            }
            result.put("page_content", pageContentItems);
        }
        jsonObject.put("pageview", result);
    }
}