RolloutCommand.java 9.07 KB
/*
 * Decompiled with CFR 0_118.
 * 
 * Could not load the following classes:
 *  com.day.cq.commons.servlets.HtmlStatusResponseHelper
 *  com.day.cq.i18n.I18n
 *  com.day.cq.wcm.api.Page
 *  com.day.cq.wcm.api.PageManager
 *  com.day.cq.wcm.api.commands.WCMCommand
 *  com.day.cq.wcm.api.commands.WCMCommandContext
 *  com.day.cq.wcm.msm.api.RolloutManager
 *  com.day.cq.wcm.msm.api.RolloutManager$RolloutParams
 *  com.day.cq.wcm.msm.api.RolloutManager$RolloutProgress
 *  javax.servlet.ServletOutputStream
 *  javax.servlet.http.HttpServletRequest
 *  org.apache.commons.lang.ArrayUtils
 *  org.apache.felix.scr.annotations.Component
 *  org.apache.felix.scr.annotations.Reference
 *  org.apache.felix.scr.annotations.Service
 *  org.apache.sling.api.SlingHttpServletRequest
 *  org.apache.sling.api.SlingHttpServletResponse
 *  org.apache.sling.api.resource.Resource
 *  org.apache.sling.api.resource.ResourceResolver
 *  org.apache.sling.api.servlets.HtmlResponse
 *  org.apache.sling.bgservlets.RuntimeState
 *  org.slf4j.Logger
 *  org.slf4j.LoggerFactory
 */
package com.day.cq.wcm.msm.impl.commands;

import com.day.cq.commons.servlets.HtmlStatusResponseHelper;
import com.day.cq.i18n.I18n;
import com.day.cq.wcm.api.Page;
import com.day.cq.wcm.api.PageManager;
import com.day.cq.wcm.api.commands.WCMCommand;
import com.day.cq.wcm.api.commands.WCMCommandContext;
import com.day.cq.wcm.msm.api.RolloutManager;
import java.io.IOException;
import java.util.Arrays;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import org.apache.commons.lang.ArrayUtils;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Reference;
import org.apache.felix.scr.annotations.Service;
import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.api.SlingHttpServletResponse;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ResourceResolver;
import org.apache.sling.api.servlets.HtmlResponse;
import org.apache.sling.bgservlets.RuntimeState;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Component
@Service
public class RolloutCommand
implements WCMCommand {
    private final Logger log = LoggerFactory.getLogger(RolloutCommand.class);
    public static final String TYPE_DEEP = "deep";
    public static final String TYPE_PAGE = "page";
    public static final String TYPE_DELETE = "delete";
    public static final String TYPE_PARAM = "type";
    public static final String PARAS_PARAM = "paras";
    public static final String RESET_PARAM = "reset";
    @Reference
    private RolloutManager rolloutManager = null;

    public String getCommandName() {
        return "rollout";
    }

    public HtmlResponse performCommand(WCMCommandContext ctx, SlingHttpServletRequest request, SlingHttpServletResponse response, PageManager pageManager) {
        final RuntimeState runtimeState = (RuntimeState)request.getAttribute(RuntimeState.class.getName());
        boolean backgroundExecution = runtimeState != null;
        try {
            Object[] paths = request.getParameterValues("path");
            if (ArrayUtils.isEmpty((Object[])paths)) {
                if (backgroundExecution) {
                    runtimeState.setProgressMessage(I18n.get((HttpServletRequest)request, (String)"Roll-out Command incomplete: source path missing"));
                }
                return HtmlStatusResponseHelper.createStatusResponse((boolean)false, (String)I18n.get((HttpServletRequest)request, (String)"No path provided"));
            }
            String[] targets = request.getParameterValues("msm:targetPath");
            RolloutManager.RolloutProgress rolloutProgress = null;
            if (backgroundExecution) {
                response.setContentType("text/plain; charset=UTF-8");
                final ServletOutputStream out = response.getOutputStream();
                rolloutProgress = new RolloutManager.RolloutProgress(){

                    public void reportProgress(String path, String operation) {
                        try {
                            String msg = String.format("%s : %s", operation, path);
                            runtimeState.setProgressMessage(msg);
                            out.println(msg);
                        }
                        catch (IOException e) {
                            RolloutCommand.this.log.error("Trying to commit status change: {}", (Throwable)e);
                        }
                    }
                };
                Object[] arrobject = new Object[2];
                arrobject[0] = Arrays.toString(paths);
                arrobject[1] = targets == null ? I18n.get((HttpServletRequest)request, (String)"all LiveCopies") : Arrays.asList(targets);
                String description = I18n.get((HttpServletRequest)request, (String)"Rollout of {0} to {1}", (String)null, (Object[])arrobject);
                rolloutProgress.reportProgress(description, I18n.get((HttpServletRequest)request, (String)"STARTING"));
            }
            long startTime = System.currentTimeMillis();
            String type = request.getParameter("type");
            if ("delete".equals(type) && !ArrayUtils.isEmpty((Object[])request.getParameterValues("paras"))) {
                ResourceResolver resolver = request.getResourceResolver();
                for (String parPath : request.getParameterValues("paras")) {
                    Resource resource = resolver.getResource(parPath);
                    if (resource == null) continue;
                    resolver.delete(resource);
                    this.log.debug("Deleted Component at {}", (Object)parPath);
                    if (rolloutProgress == null) continue;
                    rolloutProgress.reportProgress("Deleted Component at " + parPath, I18n.get((HttpServletRequest)request, (String)"ROLLOUT"));
                }
                resolver.commit();
            }
            Object lastPath = null;
            RolloutManager.RolloutParams params = new RolloutManager.RolloutParams();
            for (Object path : paths) {
                Page masterPage = pageManager.getPage((String)path);
                if (masterPage == null) {
                    this.log.debug("Received request to roll-out non-existing Blueprint at {}: ignore", path);
                    continue;
                }
                lastPath = path;
                params.rolloutProgress = rolloutProgress;
                params.master = masterPage;
                params.reset = "true".equals(request.getParameter("reset"));
                if ("page".equals(type) || "deep".equals(type)) {
                    params.isDeep = "deep".equals(type);
                } else {
                    params.isDeep = request.getParameter("shallow") == null || Boolean.valueOf(request.getParameter("shallow")) == false;
                    params.paragraphs = request.getParameterValues("paras");
                }
                if (targets == null) {
                    this.rolloutManager.rollout(params);
                    continue;
                }
                for (String target : targets) {
                    params.targets = new String[]{target};
                    this.rolloutManager.rollout(params);
                }
            }
            HtmlResponse statusResponse = null;
            if (lastPath == null) {
                String msg = "Roll-out Command incomplete: source path missing";
                if (backgroundExecution) {
                    runtimeState.setProgressMessage(I18n.get((HttpServletRequest)request, (String)msg));
                } else {
                    statusResponse = HtmlStatusResponseHelper.createStatusResponse((boolean)false, (String)I18n.get((HttpServletRequest)request, (String)msg));
                }
            } else {
                String msg = I18n.get((HttpServletRequest)request, (String)"Page rolled-out: {0}. Execution time={1} ms", (String)null, (Object[])new Object[]{Arrays.toString(paths), System.currentTimeMillis() - startTime});
                if (backgroundExecution) {
                    params.rolloutProgress.reportProgress(msg, I18n.get((HttpServletRequest)request, (String)"DONE"));
                } else {
                    statusResponse = HtmlStatusResponseHelper.createStatusResponse((boolean)true, (String)msg, (String)lastPath);
                }
            }
            return statusResponse;
        }
        catch (Exception e) {
            this.log.error("Error during roll-out.", (Throwable)e);
            if (backgroundExecution) {
                runtimeState.setProgressMessage(I18n.get((HttpServletRequest)request, (String)"Failed to Roll-out: {}", (String)e.getMessage()));
                return null;
            }
            return HtmlStatusResponseHelper.createStatusResponse((boolean)false, (String)(e.getMessage() == null ? e.getClass().getSimpleName() : e.getMessage()));
        }
    }

    protected void bindRolloutManager(RolloutManager rolloutManager) {
        this.rolloutManager = rolloutManager;
    }

    protected void unbindRolloutManager(RolloutManager rolloutManager) {
        if (this.rolloutManager == rolloutManager) {
            this.rolloutManager = null;
        }
    }

}