/*
 *  Copyright 2023 Anyware Services
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.ametys.plugins.workflow.readers;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.cocoon.environment.Request;

import org.ametys.plugins.workflow.dao.WorkflowStepDAO;
import org.ametys.runtime.i18n.I18nizableText;

import com.opensymphony.workflow.loader.ActionDescriptor;
import com.opensymphony.workflow.loader.ResultDescriptor;
import com.opensymphony.workflow.loader.StepDescriptor;
import com.opensymphony.workflow.loader.WorkflowDescriptor;

/**
 * Read step mindMap from plantUML
 */
public class PlantUmlMindMapStepSVGReader extends AbstractPlantUMLMindmapSVGReader 
{
    @Override
    protected String _getPlantUMLStyle()
    {
        StringBuilder style = new StringBuilder();
        style.append("<style> \n");
        style.append("mindmapDiagram { \n");
        style.append("boxless { \n");
        style.append("HyperLinkColor #0a7fb2 \n");
        style.append("hyperlinkUnderlineThickness 0 \n"); //workaround until HyperlinkUnderline false is fixed
        style.append("} \n");
        style.append("} \n");
        style.append("</style> \n");
        return style.toString();
    }

    @Override
    protected String _getPlantUMLGraphContent(Request request, WorkflowDescriptor workflowDescriptor)
    {
        StringBuilder content = new StringBuilder();
        String workflowName = workflowDescriptor.getName();
        int stepId = Integer.valueOf((String) request.get("stepId"));
        String stepIdAsString = stepId != 0 ? String.valueOf(stepId) : WorkflowStepDAO.INITIAL_STEP_ID;
        
        // Graph for central step node
        String nodeContent = _getMindMapNodeContent(
            "+",
            _getStepNodeLabel(workflowDescriptor, stepId),
            _workflowStepDAO.getStepIconPathAsBase64(workflowDescriptor, stepId),
            __MAIN_STEP_NODE_COLOR,
            null,
            null,
            false
        );
        
        content.append(nodeContent);
        
        // Graph for incoming actions
        Set<ActionDescriptor> incomingActions = _getIncomingActions(stepId, workflowDescriptor);
        for (ActionDescriptor action : incomingActions)
        {
            String firstParentStepId = _getFirstParentStepId(stepId, workflowDescriptor.getSteps(), action.getId());
            String actionNode = _getMindMapNodeContent(
                "--",
                _getActionLabel(action),
                _workflowStepDAO.getActionIconPathAsBase64(action),
                null,
                _getJsFunction(workflowName, firstParentStepId, String.valueOf(action.getId())),
                _i18nUtils.translate(new I18nizableText("plugin.workflow", "PLUGIN_WORKFLOW_LINK_SEE_TRANSITION")),
                true
            );
            content.append(actionNode);
        }
        
        // Graph for outgoing action
        List<ActionDescriptor> outgoingActions = _getOutgoingActions(stepId, workflowDescriptor);
        for (ActionDescriptor action : outgoingActions)
        {
            String actionNode = _getMindMapNodeContent(
                "++",
                _getActionLabel(action),
                _workflowStepDAO.getActionIconPathAsBase64(action),
                null,
                _getJsFunction(workflowName, stepIdAsString, String.valueOf(action.getId())),
                _i18nUtils.translate(new I18nizableText("plugin.workflow", "PLUGIN_WORKFLOW_LINK_SEE_TRANSITION")),
                true
            );
            content.append(actionNode);
        }
        return content.toString();
    }
    
    /**
     * Get a set of actions incoming to current step
     * @param stepId id of current step
     * @param workflow current workflow
     * @return the set of outgoing actions
     */
    @SuppressWarnings("unchecked")
    protected Set<ActionDescriptor> _getIncomingActions(int stepId , WorkflowDescriptor workflow)
    {
        Set<ActionDescriptor> incomingActions = new HashSet<>();
        if (stepId != 0)
        {
            incomingActions.addAll(_getIncomingActionsFromList(stepId, workflow.getInitialActions()));
            List<StepDescriptor> steps = workflow.getSteps();
            for (StepDescriptor otherSteps : steps)
            {
                if (otherSteps.getId() != stepId)
                {
                    List<ActionDescriptor> actions = otherSteps.getActions();
                    incomingActions.addAll(_getIncomingActionsFromList(stepId, actions));
                }
            }
        }
        return incomingActions;
    }
    
    /**
     * Get a list of actions outgoing from current step
     * @param stepId id of current step
     * @param workflow current workflow
     * @return the list of outgoing actions
     */
    protected List<ActionDescriptor> _getOutgoingActions(int stepId, WorkflowDescriptor workflow)
    {
        return stepId != 0 
                ? workflow.getStep(stepId).getActions()
                : workflow.getInitialActions();
    }    
    
    /**
     * Get id of the first step having current action, INITIAL_STEP_ID if current action is an initial action
     * @param stepId id of current step
     * @param steps list of all the steps in current workflow
     * @param actionId id of current action
     * @return the id of the first found step having current action
     */
    protected String _getFirstParentStepId(int stepId, List<StepDescriptor> steps, Integer actionId)
    {
        String firstParentStepId = "";
        int i = 0;
        do 
        {
            StepDescriptor otherStep = steps.get(i);
            if (otherStep.getId() != stepId && otherStep.getAction(actionId) != null)
            {
                firstParentStepId = String.valueOf(otherStep.getId());
            }
            i++;
        } while (firstParentStepId.isEmpty() && i < steps.size());
        return firstParentStepId.isBlank() ? WorkflowStepDAO.INITIAL_STEP_ID : firstParentStepId;
    }

    /**
     * Get a set of incoming actions if present in actions list
     * @param stepId id of current step
     * @param actions list of other step's actions
     * @return a list containing other step's outgoing actions that are incoming to current step
     */
    protected Set<ActionDescriptor> _getIncomingActionsFromList(int stepId, List<ActionDescriptor> actions)
    {
        Set<ActionDescriptor> incoming = new HashSet<>();
        for (ActionDescriptor action : actions)
        {
            ResultDescriptor unconditionalResult = action.getUnconditionalResult();
            if (unconditionalResult.getStep() == stepId)
            {
                incoming.add(action);
            }
            else
            {
                boolean leadToStep = false;
                List<ResultDescriptor> conditionalResults = action.getConditionalResults();
                int indexResult = 0;
                while (!leadToStep && indexResult < conditionalResults.size())
                {
                    if (conditionalResults.get(indexResult).getStep() == stepId)
                    {
                        incoming.add(action);
                        leadToStep = true;
                    }
                    indexResult++;
                }
            }
        } 
        return incoming;
    }
}