GetSegmentsServlet.java 5.15 KB
/*
 * Decompiled with CFR 0_118.
 * 
 * Could not load the following classes:
 *  javax.servlet.ServletException
 *  org.apache.felix.scr.annotations.Component
 *  org.apache.felix.scr.annotations.Reference
 *  org.apache.felix.scr.annotations.sling.SlingServlet
 *  org.apache.sling.api.SlingHttpServletRequest
 *  org.apache.sling.api.SlingHttpServletResponse
 *  org.apache.sling.api.resource.Resource
 *  org.apache.sling.api.servlets.SlingSafeMethodsServlet
 *  org.apache.sling.commons.json.JSONException
 *  org.apache.sling.commons.json.JSONObject
 *  org.slf4j.Logger
 *  org.slf4j.LoggerFactory
 */
package com.adobe.cq.aam;

import com.adobe.cq.aam.client.spi.AudienceManagerAccessDenied;
import com.adobe.cq.aam.client.spi.AudienceManagerClient;
import com.adobe.cq.aam.client.spi.AudienceManagerConfiguration;
import com.adobe.cq.aam.client.spi.AudienceManagerItem;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Reference;
import org.apache.felix.scr.annotations.sling.SlingServlet;
import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.api.SlingHttpServletResponse;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.servlets.SlingSafeMethodsServlet;
import org.apache.sling.commons.json.JSONException;
import org.apache.sling.commons.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletException;
import java.io.IOException;
import java.util.Arrays;
import java.util.Map;

@Component(metatype=0, immediate=1)
@SlingServlet(extensions={"json"}, methods={"GET"}, resourceTypes={"cq/personalization/audiencemanager/components/cloudservices/audiencemanagerpage"}, selectors={"segments"}, generateComponent=0)
public class GetSegmentsServlet
extends SlingSafeMethodsServlet {
    private static final long serialVersionUID = 5745142649831823327L;
    private static final Logger LOGGER = LoggerFactory.getLogger(GetSegmentsServlet.class);
    @Reference
    private AudienceManagerClient audienceManagerClient;

    protected void doGet(SlingHttpServletRequest request, SlingHttpServletResponse response) throws ServletException, IOException {
        Resource configResource = request.getResource();
        AudienceManagerConfiguration config = new AudienceManagerConfiguration(configResource);
        if (!this.audienceManagerClient.isValidForUse(config)) {
            response.sendError(404, "AAM Configuration not valid.");
            return;
        }
        Object[] traits = request.getParameterValues("t");
        boolean fullList = "1".equals(request.getParameter("f"));
        if (traits == null || traits.length == 0) {
            response.sendError(400, "No Traits supplied.");
            return;
        }
        try {
            response.setContentType("application/json; charset=utf-8");
            JSONObject jsonOut = new JSONObject();
            JSONObject jsegments = new JSONObject();
            if (fullList) {
                Iterable<AudienceManagerItem> segments = this.audienceManagerClient.getSegments(config, (String[])traits);
                jsonOut.put("segments", (Object)jsegments);
                if (segments != null) {
                    for (AudienceManagerItem segment : segments) {
                        if (segment != null) {
                            JSONObject segmentAsJson = new JSONObject();
                            for (Map.Entry<String, Object> p : segment.getEntrySet()) {
                                segmentAsJson.put(p.getKey(), p.getValue());
                            }
                            jsegments.put(String.valueOf(segment.getItemId()), (Object)segmentAsJson);
                            continue;
                        }
                        LOGGER.debug("Invalid segment found ");
                    }
                } else {
                    LOGGER.debug("No Segments returned for trait query: {} ", (Object)Arrays.toString(traits));
                }
            } else {
                Iterable<String> segments = this.audienceManagerClient.getSegmentIds(config, (String[])traits);
                jsonOut.put("segments", (Object)jsegments);
                if (segments != null) {
                    for (String segment : segments) {
                        jsegments.put(segment, true);
                    }
                } else {
                    LOGGER.debug("No Segments returned for trait query: {} ", (Object)Arrays.toString(traits));
                }
            }
            response.getWriter().write(jsonOut.toString());
        }
        catch (JSONException e) {
            throw new ServletException(e.getMessage(), (Throwable)e);
        }
        catch (AudienceManagerAccessDenied e) {
            response.sendError(403, "Access Forbidden, please check cloud configuration");
        }
    }

    protected void bindAudienceManagerClient(AudienceManagerClient audienceManagerClient) {
        this.audienceManagerClient = audienceManagerClient;
    }

    protected void unbindAudienceManagerClient(AudienceManagerClient audienceManagerClient) {
        if (this.audienceManagerClient == audienceManagerClient) {
            this.audienceManagerClient = null;
        }
    }
}