001/*
002 *  Copyright 2022 Anyware Services
003 *
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package org.ametys.cms.workflow;
017
018import java.util.Map;
019
020import javax.jcr.RepositoryException;
021
022import org.ametys.cms.repository.WorkflowAwareContent;
023import org.ametys.cms.repository.WorkflowAwareContentHelper;
024
025import com.opensymphony.module.propertyset.PropertySet;
026import com.opensymphony.workflow.WorkflowException;
027import com.opensymphony.workflow.spi.WorkflowEntry;
028
029/**
030 * OS workflow function to set the current step id
031 */
032public class ResetWorkflowIdFunction extends AbstractContentFunction
033{
034    @Override
035    public void execute(Map transientVars, Map args, PropertySet ps) throws WorkflowException
036    {
037        WorkflowAwareContent content = getContent(transientVars);
038        
039        try
040        {
041            // Add the lock token to the current session if the content is locked.
042            _addLockToken(content);
043            
044            // Save the current workflow ID.
045            if (content != null)
046            {
047                long workflowId = ((WorkflowEntry) transientVars.get("entry")).getId();
048                // Remove the old workflow (used on re-initialization of the workflow
049                WorkflowAwareContentHelper.removeWorkflowId(content);
050                content.setWorkflowId(workflowId);
051                content.saveChanges();
052            }
053        }
054        catch (RepositoryException e)
055        {
056            throw new WorkflowException("Unable to add a lock token to the current session.", e);
057        }
058    }
059}