// --- BEGIN COPYRIGHT BLOCK ---
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; version 2 of the License.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
// (C) 2013 Red Hat, Inc.
// All rights reserved.
// --- END COPYRIGHT BLOCK ---
package com.netscape.cmstools.system;

import java.util.Collection;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.Option;

import com.netscape.certsrv.base.ResourceNotFoundException;
import com.netscape.certsrv.system.TPSConnectorCollection;
import com.netscape.certsrv.system.TPSConnectorData;
import com.netscape.cmstools.cli.CLI;
import com.netscape.cmstools.cli.MainCLI;

/**
 * @author Ade Lee
 */
public class TPSConnectorFindCLI extends CLI {
    public TPSConnectorCLI tpsConnectorCLI;

    public TPSConnectorFindCLI(TPSConnectorCLI tpsConnectorCLI) {
        super("find", "Find TPS connector details on TKS", tpsConnectorCLI);
        this.tpsConnectorCLI = tpsConnectorCLI;
    }

    public void printHelp() {
        formatter.printHelp(getFullName() + " [OPTIONS...]", options);
    }

    public void execute(String[] args) throws Exception {
        Option option = new Option(null, "host", true, "TPS host");
        option.setArgName("host");
        options.addOption(option);

        option = new Option(null, "port", true, "TPS port");
        option.setArgName("port");
        options.addOption(option);

        CommandLine cmd = null;

        try {
            cmd = parser.parse(options, args);

        } catch (Exception e) {
            System.err.println("Error: " + e.getMessage());
            printHelp();
            System.exit(1);
        }

        String tpsHost = cmd.getOptionValue("host");
        String tpsPort = cmd.getOptionValue("port");

        if (tpsHost != null) {
            if (tpsPort == null)
                tpsPort = "443";
            try {
                TPSConnectorData data = tpsConnectorCLI.tpsConnectorClient.getConnector(
                        tpsHost, tpsPort);
                TPSConnectorCLI.printConnectorInfo(data);
            } catch (ResourceNotFoundException e) {
                System.out.println("  TPS connector not found.");
                return;
            }
        } else {
            TPSConnectorCollection result = tpsConnectorCLI.tpsConnectorClient.listConnectors();
            Collection<TPSConnectorData> conns = result.getEntries();

            if (conns.isEmpty()) {
                System.out.println("  No TPS connectors found.");
                return;
            }

            MainCLI.printMessage(conns.size() + " TPS connector(s) matched");
            boolean first = true;
            for (TPSConnectorData data: conns) {
                if (first) {
                    first = false;
                } else {
                    System.out.println();
                }

                TPSConnectorCLI.printConnectorInfo(data);
            }

            MainCLI.printMessage("Number of entries returned " + conns.size());
        }
    }
}