/*
 * Copyright 2009-2012 by The Regents of the University of California
 * 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 from
 * 
 *     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 edu.uci.ics.asterix.installer.error;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import edu.uci.ics.asterix.event.schema.cluster.Cluster;
import edu.uci.ics.asterix.event.schema.cluster.Node;
import edu.uci.ics.asterix.installer.driver.InstallerDriver;
import edu.uci.ics.asterix.installer.driver.InstallerUtil;
import edu.uci.ics.asterix.installer.model.AsterixInstance;
import edu.uci.ics.asterix.installer.model.AsterixInstance.State;
import edu.uci.ics.asterix.installer.model.AsterixRuntimeState;
import edu.uci.ics.asterix.installer.model.ProcessInfo;

public class VerificationUtil {

    private static final String VERIFY_SCRIPT_PATH = InstallerDriver.getManagixHome() + File.separator
            + InstallerDriver.MANAGIX_INTERNAL_DIR + File.separator + "scripts" + File.separator + "verify.sh";

    public static AsterixRuntimeState getAsterixRuntimeState(AsterixInstance instance) throws Exception {

        Cluster cluster = instance.getCluster();
        List<String> args = new ArrayList<String>();
        args.add(instance.getName());
        args.add(instance.getCluster().getMasterNode().getIp());
        for (Node node : cluster.getNode()) {
            args.add(node.getIp());
            args.add(instance.getName() + "_" + node.getId());
        }

        String output = InstallerUtil.executeLocalScript(VERIFY_SCRIPT_PATH, args);
        boolean ccRunning = true;
        List<String> failedNCs = new ArrayList<String>();
        String[] infoFields;
        ProcessInfo pInfo;
        List<ProcessInfo> processes = new ArrayList<ProcessInfo>();

        for (String line : output.split("\n")) {
            infoFields = line.split(":");
            try {
                int pid = Integer.parseInt(infoFields[3]);
                pInfo = new ProcessInfo(infoFields[0], infoFields[1], pid);
                processes.add(pInfo);
            } catch (Exception e) {
                if (infoFields[0].equalsIgnoreCase("CC")) {
                    ccRunning = false;
                } else {
                    failedNCs.add(infoFields[1]);
                }
            }
        }
        return new AsterixRuntimeState(processes, failedNCs, ccRunning);
    }

    public static void updateInstanceWithRuntimeDescription(AsterixInstance instance, AsterixRuntimeState state,
            boolean expectedRunning) {
        StringBuffer summary = new StringBuffer();
        if (expectedRunning) {
            if (!state.isCcRunning()) {
                summary.append("Cluster Controller not running at " + instance.getCluster().getMasterNode().getIp()
                        + "\n");
                instance.setState(State.UNUSABLE);
            }
            if (state.getFailedNCs() != null && !state.getFailedNCs().isEmpty()) {
                summary.append("Node Controller not running at the following nodes" + "\n");
                for (String failedNC : state.getFailedNCs()) {
                    summary.append(failedNC + "\n");
                }
                instance.setState(State.UNUSABLE);
            }
            if (!(instance.getState().equals(State.UNUSABLE))) {
                instance.setState(State.ACTIVE);
            }
        } else {
            if (state.getProcesses() != null && state.getProcesses().size() > 0) {
                summary.append("Following process still running " + "\n");
                for (ProcessInfo pInfo : state.getProcesses()) {
                    summary.append(pInfo + "\n");
                }
                instance.setState(State.UNUSABLE);
            } else {
                instance.setState(State.INACTIVE);
            }
        }
        state.setSummary(summary.toString());
        instance.setAsterixRuntimeStates(state);
    }

    public static void verifyBackupRestoreConfiguration(String hdfsUrl, String hadoopVersion, String hdfsBackupDir)
            throws Exception {
        StringBuffer errorCheck = new StringBuffer();
        if (hdfsUrl == null || hdfsUrl.length() == 0) {
            errorCheck.append("\n HDFS Url not configured");
        }
        if (hadoopVersion == null || hadoopVersion.length() == 0) {
            errorCheck.append("\n HDFS version not configured");
        }
        if (hdfsBackupDir == null || hdfsBackupDir.length() == 0) {
            errorCheck.append("\n HDFS backup directory not configured");
        }
        if (errorCheck.length() > 0) {
            throw new Exception("Incomplete hdfs configuration in " + InstallerDriver.getManagixHome() + File.separator
                    + InstallerDriver.MANAGIX_CONF_XML + errorCheck);
        }
    }
}