/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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.apache.tuscany.geronimoace;

import org.apache.tuscany.sca.node.SCAContribution;
import org.apache.tuscany.sca.node.SCANode2;
import org.apache.tuscany.sca.node.SCANode2Factory;

import java.util.ArrayList;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;


public class ManageNodes {

    Map<String, LocalNode> nodeList = new Hashtable<String, LocalNode>();
    List<LocalNode> nodeSet = new ArrayList<LocalNode>();

    private SCANode2Factory factory;

    public ManageNodes() {
        factory = SCANode2Factory.newInstance();
    }

    public Map<String, LocalNode> getNodeList() {
        return nodeList;
    }

    public List<LocalNode> getNodeSet() {
        return nodeSet;
    }

    public void startNode(String nodeName) {
        LocalNode lnode = nodeList.get(nodeName);
        if (!lnode.getIsStarted()) {
            nodeList.get(nodeName).getNode().start();
            lnode.setIsStarted(true);
            System.out.println("Node " + nodeName + " is now started");
        } else {
            System.out.println("Node " + nodeName + " is already started");
        }
    }

    public void stopNode(String nodeName) {
        LocalNode lnode = nodeList.get(nodeName);
        if (lnode.getIsStarted()) {
            nodeList.get(nodeName).getNode().stop();
            lnode.setIsStarted(false);
            System.out.println("Node " + nodeName + " is now stopped");
        } else {
            System.out.println("Node " + nodeName + " is already stopped");
        }
    }

    public void serviceRequest(String loc, String nodeName, String composite) {
        if (nodeList.containsKey(nodeName)) {
            LocalNode topNode = nodeList.get(nodeName);
            SCANode2 tempNode = topNode.getNode();
            tempNode.stop();
            topNode.setIsStarted(false);

            SCAContribution contribution = new SCAContribution(nodeName, "file:" + loc);
            ArrayList<SCAContribution> contributionList = topNode.getContributionList();
            contributionList.add(contribution);
            SCAContribution[] tempArray = new SCAContribution[contributionList.size()];

            for (int i = 0; i < contributionList.size(); i++) {
                tempArray[i] = contributionList.get(i);
            }

            SCANode2 node = factory.createSCANode(composite, tempArray);
            LocalNode tempLocalNode = new LocalNode(node);
            tempLocalNode.setName(nodeName);
            tempLocalNode.setContributionList(contributionList);
            nodeList.remove(nodeName);
            nodeSet.remove(topNode);
            nodeList.put(nodeName, tempLocalNode);
            nodeSet.add(tempLocalNode);
            node.start();
            tempLocalNode.setIsStarted(true);

        } else {
            SCAContribution contribution = new SCAContribution(nodeName, "file:" + loc);
            SCANode2 node = factory.createSCANode(composite, new SCAContribution[]{contribution});
            LocalNode tempLocalNode = new LocalNode(node);
            tempLocalNode.setName(nodeName);
            tempLocalNode.addContribution(contribution);

            nodeList.put(nodeName, tempLocalNode);
            nodeSet.add(tempLocalNode);
            node.start();
            tempLocalNode.setIsStarted(true);
        }
    }


}