001/*
002 * Copyright (c) 2011 Nuxeo SA (http://nuxeo.com/) and others.
003 *
004 * All rights reserved. This program and the accompanying materials
005 * are made available under the terms of the Eclipse Public License v1.0
006 * which accompanies this distribution, and is available at
007 * http://www.eclipse.org/legal/epl-v10.html
008 *
009 * Contributors:
010 *     ldoguin
011 */
012package org.nuxeo.ecm.platform.comment.workflow;
013
014import org.nuxeo.ecm.automation.OperationContext;
015import org.nuxeo.ecm.automation.core.Constants;
016import org.nuxeo.ecm.automation.core.annotations.Context;
017import org.nuxeo.ecm.automation.core.annotations.Operation;
018import org.nuxeo.ecm.automation.core.annotations.OperationMethod;
019import org.nuxeo.ecm.automation.core.annotations.Param;
020import org.nuxeo.ecm.automation.core.collectors.DocumentModelCollector;
021import org.nuxeo.ecm.automation.task.CreateTask.OperationTaskVariableName;
022import org.nuxeo.ecm.core.api.CoreSession;
023import org.nuxeo.ecm.core.api.DocumentModel;
024import org.nuxeo.ecm.core.api.DocumentRef;
025import org.nuxeo.ecm.core.api.IdRef;
026import org.nuxeo.ecm.platform.comment.workflow.utils.CommentsConstants;
027import org.nuxeo.ecm.platform.comment.workflow.utils.FollowTransitionUnrestricted;
028import org.nuxeo.ecm.platform.task.Task;
029
030/**
031 * @since 5.5
032 */
033@Operation(id = ModerateCommentOperation.ID, category = Constants.CAT_DOCUMENT, label = "Follow publish or reject transition", description = "Follow publish if accept is true, reject otherwise.", addToStudio = false)
034public class ModerateCommentOperation {
035
036    public static final String ID = "Comment.Moderate";
037
038    @Context
039    protected OperationContext ctx;
040
041    @Context
042    protected CoreSession session;
043
044    @Param(name = "accept")
045    protected Boolean accept;
046
047    @OperationMethod(collector = DocumentModelCollector.class)
048    public DocumentModel run(DocumentModel doc) {
049        moderate(doc.getRef());
050        return doc;
051    }
052
053    @OperationMethod(collector = DocumentModelCollector.class)
054    public DocumentModel run(DocumentRef docRef) {
055        moderate(docRef);
056        return session.getDocument(docRef);
057    }
058
059    protected void moderate(DocumentRef docRef) {
060        DocumentModel taskDoc = (DocumentModel) ctx.get(OperationTaskVariableName.taskDocument.name());
061        Task task = taskDoc.getAdapter(Task.class);
062        DocumentRef targetDocRef = new IdRef(task.getVariable(CommentsConstants.COMMENT_ID));
063        FollowTransitionUnrestricted runner;
064        if (accept) {
065            runner = new FollowTransitionUnrestricted(session, targetDocRef,
066                    CommentsConstants.TRANSITION_TO_PUBLISHED_STATE);
067        } else {
068            runner = new FollowTransitionUnrestricted(session, targetDocRef, CommentsConstants.REJECT_STATE);
069        }
070        runner.runUnrestricted();
071    }
072
073}