package org.apache.s4.tools;

import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.I0Itec.zkclient.ZkClient;
import org.apache.helix.ConfigScope;
import org.apache.helix.ConfigScopeBuilder;
import org.apache.helix.HelixAdmin;
import org.apache.helix.manager.zk.ZKHelixAdmin;
import org.apache.helix.model.IdealState;
import org.apache.helix.model.IdealState.IdealStateModeProperty;
import org.apache.helix.model.InstanceConfig;
import org.apache.helix.tools.IdealStateCalculatorByShuffling;
import org.apache.s4.deploy.DistributedDeploymentManager;

import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters;

public class DeployApp extends S4ArgsBase {
    public static void main(String[] args) {
        DeployAppArgs deployArgs = new DeployAppArgs();

        Tools.parseArgs(deployArgs, args);

        HelixAdmin admin = new ZKHelixAdmin(deployArgs.zkConnectionString);
        ConfigScopeBuilder builder = new ConfigScopeBuilder();
        ConfigScope scope = builder.forCluster(deployArgs.clusterName).forResource(deployArgs.appName).build();
        Map<String, String> properties = new HashMap<String, String>();
        properties.put(DistributedDeploymentManager.S4R_URI, new File(deployArgs.s4rPath).toURI().toString());
        properties.put("type", "App");
        admin.setConfig(scope, properties);

        IdealState is = admin.getResourceIdealState(deployArgs.clusterName, deployArgs.appName);
        if (is == null) {
            is = new IdealState(deployArgs.appName);
        }
        is.setNumPartitions(1);
        is.setIdealStateMode(IdealStateModeProperty.CUSTOMIZED.toString());
        is.setStateModelDefRef("OnlineOffline");
        List<String> instancesInGroup = new ArrayList<String>();
        List<String> instancesInCluster = admin.getInstancesInCluster(deployArgs.clusterName);
        for (String instanceName : instancesInCluster) {
            InstanceConfig instanceConfig = admin.getInstanceConfig(deployArgs.clusterName, instanceName);
            String nodeGroup = instanceConfig.getRecord().getSimpleField("GROUP");
            if (nodeGroup.equals(deployArgs.nodeGroup)) {
                instancesInGroup.add(instanceName);
            }
        }
        for (String instanceName : instancesInGroup) {
            is.setPartitionState(deployArgs.appName, instanceName, "ONLINE");
        }

        admin.setResourceIdealState(deployArgs.clusterName, deployArgs.appName, is);
    }

    @Parameters(commandNames = "newStreamProcessor", separators = "=", commandDescription = "Create a new stream processor")
    static class DeployAppArgs extends S4ArgsBase {

        @Parameter(names = "-zk", description = "ZooKeeper connection string")
        String zkConnectionString = "localhost:2181";

        @Parameter(names = { "-c", "-cluster" }, description = "Logical name of the S4 cluster", required = true)
        String clusterName;

        @Parameter(names = "-s4r", description = "Path to existing s4r file", required = true)
        String s4rPath;

        @Parameter(names = { "-generatedS4R", "-g" }, description = "Location of generated s4r (incompatible with -s4r option). By default, s4r is generated in a temporary directory on the local file system. In a distributed environment, you probably want to specify a location accessible through a distributed file system like NFS. That's the purpose of this option.", required = false)
        String generatedS4R;

        @Parameter(names = { "-appName" }, description = "Name of the App", required = true, arity = 1)
        String appName;

        @Parameter(names = { "-ng", "-nodeGroup" }, description = "Node group name where the App needs to be deployed", required = false, arity = 1)
        String nodeGroup = "default";

    }
}