/*
 * Copyright (c) 2018, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * WSO2 Inc. licenses this file to you 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.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.builders;

import org.wso2.carbon.siddhi.editor.core.util.designview.beans.EventFlow;
import org.wso2.carbon.siddhi.editor.core.util.designview.beans.configs.Edge;
import org.wso2.carbon.siddhi.editor.core.util.designview.beans.configs.SiddhiAppConfig;
import org.wso2.carbon.siddhi.editor.core.util.designview.beans.configs.siddhielements.partition.PartitionConfig;
import org.wso2.carbon.siddhi.editor.core.util.designview.beans.configs.siddhielements.query.QueryConfig;
import org.wso2.carbon.siddhi.editor.core.util.designview.beans.configs.siddhielements.sourcesink.SourceSinkConfig;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.AggregationConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.AnnotationConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.EdgesGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.FunctionConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.PartitionConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.SourceSinkConfigsGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.StreamDefinitionConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.TableConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.TriggerConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.WindowConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.commentspreserver.OuterScopeCommentsPreserver;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.commentspreserver.PartitionScopeCommentsPreserver;
import org.wso2.carbon.siddhi.editor.core.util.designview.designgenerator.generators.query.QueryConfigGenerator;
import org.wso2.carbon.siddhi.editor.core.util.designview.exceptions.DesignGenerationException;
import org.wso2.siddhi.core.SiddhiAppRuntime;
import org.wso2.siddhi.core.stream.input.source.Source;
import org.wso2.siddhi.core.stream.output.sink.Sink;
import org.wso2.siddhi.query.api.SiddhiApp;
import org.wso2.siddhi.query.api.annotation.Annotation;
import org.wso2.siddhi.query.api.definition.AbstractDefinition;
import org.wso2.siddhi.query.api.definition.AggregationDefinition;
import org.wso2.siddhi.query.api.definition.FunctionDefinition;
import org.wso2.siddhi.query.api.definition.StreamDefinition;
import org.wso2.siddhi.query.api.definition.TableDefinition;
import org.wso2.siddhi.query.api.definition.TriggerDefinition;
import org.wso2.siddhi.query.api.definition.WindowDefinition;
import org.wso2.siddhi.query.api.execution.ExecutionElement;
import org.wso2.siddhi.query.api.execution.partition.Partition;
import org.wso2.siddhi.query.api.execution.query.Query;

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

/**
 * Builder to create EventFlow
 */
public class EventFlowBuilder {
    private String siddhiAppString;
    private SiddhiApp siddhiApp;
    private SiddhiAppRuntime siddhiAppRuntime;

    private SiddhiAppConfig siddhiAppConfig;
    private Set<Edge> edges;

    public EventFlowBuilder(String siddhiAppString, SiddhiApp siddhiApp, SiddhiAppRuntime siddhiAppRuntime) {
        this.siddhiAppString = siddhiAppString;
        this.siddhiApp = siddhiApp;
        this.siddhiAppRuntime = siddhiAppRuntime;
        siddhiAppConfig = new SiddhiAppConfig();
        edges = new HashSet<>();
    }

    /**
     * Creates an EventFlow object with loaded elements
     *
     * @return EventFlow object
     */
    public EventFlow create() {
        return new EventFlow(siddhiAppConfig, edges);
    }

    /**
     * Loads App level Annotations from the Siddhi app
     *
     * @return A reference to this object
     */
    public EventFlowBuilder loadAppAnnotations() {
        String siddhiAppName = "";
        String siddhiAppDescription = "";
        List<String> appAnnotations = new ArrayList<>();
        AnnotationConfigGenerator annotationConfigGenerator = new AnnotationConfigGenerator();
        for (Annotation annotation : siddhiApp.getAnnotations()) {
            if (annotation.getName().equalsIgnoreCase("NAME")) {
                siddhiAppName = annotation.getElements().get(0).getValue();
                annotationConfigGenerator.preserveCodeSegment(annotation);
            } else if (annotation.getName().equalsIgnoreCase("DESCRIPTION")) {
                siddhiAppDescription = annotation.getElements().get(0).getValue();
                annotationConfigGenerator.preserveCodeSegment(annotation);
            } else {
                appAnnotations.add(
                        "@App:" + annotationConfigGenerator.generateAnnotationConfig(annotation).split("@")[1]);
            }
        }
        siddhiAppConfig.setSiddhiAppName(siddhiAppName);
        siddhiAppConfig.setSiddhiAppDescription(siddhiAppDescription);
        siddhiAppConfig.setAppAnnotationList(appAnnotations);
        siddhiAppConfig.addElementCodeSegments(annotationConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Triggers from the SiddhiApp
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while loading elements
     */
    public EventFlowBuilder loadTriggers() throws DesignGenerationException {
        TriggerConfigGenerator triggerConfigGenerator =
                new TriggerConfigGenerator(siddhiAppString, siddhiAppRuntime.getStreamDefinitionMap());
        for (TriggerDefinition triggerDefinition : siddhiApp.getTriggerDefinitionMap().values()) {
            siddhiAppConfig.add(triggerConfigGenerator.generateTriggerConfig(triggerDefinition));
        }
        siddhiAppConfig.addElementCodeSegments(triggerConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Returns the availability of a Trigger with the given name, in the given Siddhi app
     *
     * @param streamName Name of the Stream
     * @param siddhiApp  Siddhi app in which, availability of Trigger is searched
     * @return Availability of Trigger with the given name, in given Siddhi app
     */
    private boolean isTriggerDefined(String streamName, SiddhiApp siddhiApp) {
        return (siddhiApp.getTriggerDefinitionMap().size() != 0 &&
                siddhiApp.getTriggerDefinitionMap().containsKey(streamName));
    }

    /**
     * Loads Streams from the SiddhiAppRuntime
     *
     * @return A reference to this object
     */
    public EventFlowBuilder loadStreams() {
        StreamDefinitionConfigGenerator streamDefinitionConfigGenerator = new StreamDefinitionConfigGenerator();
        Map<String, StreamDefinition> streamDefinitionMap = siddhiAppRuntime.getStreamDefinitionMap();
        for (Map.Entry<String, StreamDefinition> streamDefinitionEntry : streamDefinitionMap.entrySet()) {
            if (!isTriggerDefined(streamDefinitionEntry.getKey(), siddhiApp)) {
                siddhiAppConfig.add(
                        streamDefinitionConfigGenerator
                                .generateStreamConfig(streamDefinitionEntry.getValue()));
            }
        }
        siddhiAppConfig.addElementCodeSegments(streamDefinitionConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Sources from the SiddhiAppRuntime
     *
     * @return A reference to this object
     */
    public EventFlowBuilder loadSources() throws DesignGenerationException {
        SourceSinkConfigsGenerator sourceConfigsGenerator = new SourceSinkConfigsGenerator();
        for (List<Source> sourceList : siddhiAppRuntime.getSources()) {
            for (SourceSinkConfig sourceConfig : sourceConfigsGenerator.generateSourceConfigs(sourceList)) {
                siddhiAppConfig.addSource(sourceConfig);
            }
        }
        siddhiAppConfig.addElementCodeSegments(sourceConfigsGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Sinks from the SiddhiAppRuntime
     *
     * @return A reference to this object
     */
    public EventFlowBuilder loadSinks() throws DesignGenerationException {
        SourceSinkConfigsGenerator sinkConfigsGenerator = new SourceSinkConfigsGenerator();
        for (List<Sink> sinkList : siddhiAppRuntime.getSinks()) {
            for (SourceSinkConfig sinkConfig : sinkConfigsGenerator.generateSinkConfigs(sinkList)) {
                siddhiAppConfig.addSink(sinkConfig);
            }
        }
        siddhiAppConfig.addElementCodeSegments(sinkConfigsGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Tables from the Siddhi App
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while loading elements
     */
    public EventFlowBuilder loadTables() throws DesignGenerationException {
        TableConfigGenerator tableConfigGenerator = new TableConfigGenerator();
        for (TableDefinition tableDefinition : siddhiApp.getTableDefinitionMap().values()) {
            siddhiAppConfig.add(tableConfigGenerator.generateTableConfig(tableDefinition));
        }
        siddhiAppConfig.addElementCodeSegments(tableConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Defined Windows from the SiddhiAppRuntime
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while loading elements
     */
    public EventFlowBuilder loadWindows() throws DesignGenerationException {
        WindowConfigGenerator windowConfigGenerator = new WindowConfigGenerator(siddhiAppString);
        for (WindowDefinition windowDefinition : siddhiApp.getWindowDefinitionMap().values()) {
            siddhiAppConfig.add(windowConfigGenerator.generateWindowConfig(windowDefinition));
        }
        siddhiAppConfig.addElementCodeSegments(windowConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Aggregations from the SiddhiApp
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while loading elements
     */
    public EventFlowBuilder loadAggregations() throws DesignGenerationException {
        AggregationConfigGenerator aggregationConfigGenerator = new AggregationConfigGenerator(siddhiAppString);
        for (AggregationDefinition aggregationDefinition : siddhiApp.getAggregationDefinitionMap().values()) {
            siddhiAppConfig.add(aggregationConfigGenerator.generateAggregationConfig(aggregationDefinition));
        }
        siddhiAppConfig.addElementCodeSegments(aggregationConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Functions from the siddhi app
     *
     * @return A reference to this object
     */
    public EventFlowBuilder loadFunctions() {
        FunctionConfigGenerator functionConfigGenerator = new FunctionConfigGenerator();
        for (FunctionDefinition functionDefinition : siddhiApp.getFunctionDefinitionMap().values()) {
            siddhiAppConfig.add(functionConfigGenerator.generateFunctionConfig(functionDefinition));
        }
        siddhiAppConfig.addElementCodeSegments(functionConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads Execution Elements from the Siddhi App
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while loading elements
     */
    public EventFlowBuilder loadExecutionElements() throws DesignGenerationException {
        Map<String, Map<String, AbstractDefinition>> partitionedInnerStreamDefinitions =
                siddhiAppRuntime.getPartitionedInnerStreamDefinitionMap();
        QueryConfigGenerator queryConfigGenerator = new QueryConfigGenerator(siddhiAppString, siddhiApp);
        PartitionConfigGenerator partitionConfigGenerator =
                new PartitionConfigGenerator(siddhiAppString, siddhiApp, partitionedInnerStreamDefinitions);
        int partitionCounter = 0;
        for (ExecutionElement executionElement : siddhiApp.getExecutionElementList()) {
            if (executionElement instanceof Query) {
                QueryConfig queryConfig = queryConfigGenerator.generateQueryConfig((Query) executionElement);
                siddhiAppConfig.addQuery(QueryConfigGenerator.getQueryListType(queryConfig), queryConfig);
            } else if (executionElement instanceof Partition) {
                String partitionId = (String) partitionedInnerStreamDefinitions.keySet().toArray()[partitionCounter];
                siddhiAppConfig.addPartition(
                        partitionConfigGenerator
                                .generatePartitionConfig(
                                        ((Partition) executionElement),
                                        partitionId));
                partitionCounter++;
            } else {
                throw new DesignGenerationException("Unable create config for execution element of type unknown");
            }
        }
        siddhiAppConfig.addElementCodeSegments(queryConfigGenerator.getPreservedCodeSegments());
        siddhiAppConfig.addElementCodeSegments(partitionConfigGenerator.getPreservedCodeSegments());
        return this;
    }

    /**
     * Loads generated Edges that represent connections between SiddhiElementConfigs, into SiddhiAppConfig object
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while loading edges
     */
    public EventFlowBuilder loadEdges() throws DesignGenerationException {
        EdgesGenerator edgesGenerator = new EdgesGenerator(siddhiAppConfig);
        edges = edgesGenerator.generateEdges();
        return this;
    }

    /**
     * Loads Comments, that were preserved by each Siddhi Element Config generator
     *
     * @return A reference to this object
     * @throws DesignGenerationException Error while generating Comment Code Segments
     */
    public EventFlowBuilder loadComments() throws DesignGenerationException {
        OuterScopeCommentsPreserver outerScopeCommentsPreserver =
                new OuterScopeCommentsPreserver(siddhiAppString, siddhiAppConfig.getElementCodeSegments());
        siddhiAppConfig.assignCommentCodeSegments(outerScopeCommentsPreserver.generateCommentCodeSegments());
        siddhiAppConfig =
                outerScopeCommentsPreserver.bindCommentsToElements(
                        siddhiAppConfig.getCommentCodeSegments(),
                        siddhiAppConfig);
        // Preserve Comments of Partitions
        for (PartitionConfig partitionConfig : siddhiAppConfig.getPartitionList()) {
            PartitionScopeCommentsPreserver partitionScopeCommentsPreserver =
                    new PartitionScopeCommentsPreserver(
                            siddhiAppString, partitionConfig, siddhiAppConfig.getElementCodeSegments());
            siddhiAppConfig.clearCommentCodeSegments();
            siddhiAppConfig.assignCommentCodeSegments(partitionScopeCommentsPreserver.generateCommentCodeSegments());
            siddhiAppConfig =
                    partitionScopeCommentsPreserver.bindCommentsToElements(
                            siddhiAppConfig.getCommentCodeSegments(),
                            siddhiAppConfig);
        }
        return this;
    }
}
