1 files changed, 126 insertions, 0 deletions
diff --git a/base/java-tools/src/com/netscape/cmstools/tps/connector/ConnectorModifyCLI.java b/base/java-tools/src/com/netscape/cmstools/tps/connector/ConnectorModifyCLI.java
new file mode 100644
index 000000000..f92e7c153
--- /dev/null
+++ b/base/java-tools/src/com/netscape/cmstools/tps/connector/ConnectorModifyCLI.java
@@ -0,0 +1,126 @@
+// --- 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.tps.connector;
+
+import java.io.BufferedReader;
+import java.io.FileReader;
+import java.io.PrintWriter;
+import java.io.StringWriter;
+import java.util.Arrays;
+
+import org.apache.commons.cli.CommandLine;
+import org.apache.commons.cli.Option;
+
+import com.netscape.certsrv.tps.connection.ConnectionData;
+import com.netscape.cmstools.cli.CLI;
+import com.netscape.cmstools.cli.MainCLI;
+
+/**
+ * @author Endi S. Dewata
+ */
+public class ConnectorModifyCLI extends CLI {
+
+    public ConnectorCLI connectorCLI;
+
+    public ConnectorModifyCLI(ConnectorCLI connectorCLI) {
+        super("mod", "Modify connector", connectorCLI);
+        this.connectorCLI = connectorCLI;
+
+        createOptions();
+    }
+
+    public void printHelp() {
+        formatter.printHelp(getFullName() + " <Connector ID> [OPTIONS...]", options);
+    }
+
+    public void createOptions() {
+        Option option = new Option(null, "action", true, "Action: update (default), approve, reject, enable, disable.");
+        option.setArgName("action");
+        options.addOption(option);
+
+        option = new Option(null, "input", true, "Input file containing connector properties.");
+        option.setArgName("file");
+        options.addOption(option);
+    }
+
+    public void execute(String[] args) throws Exception {
+        // Always check for "--help" prior to parsing
+        if (Arrays.asList(args).contains("--help")) {
+            // Display usage
+            printHelp();
+            System.exit(0);
+        }
+
+        CommandLine cmd = null;
+
+        try {
+            cmd = parser.parse(options, args);
+
+        } catch (Exception e) {
+            System.err.println("Error: " + e.getMessage());
+            printHelp();
+            System.exit(-1);
+        }
+
+        String[] cmdArgs = cmd.getArgs();
+
+        if (cmdArgs.length != 1) {
+            System.err.println("Error: No Connector ID specified.");
+            printHelp();
+            System.exit(-1);
+        }
+
+        String connectorID = cmdArgs[0];
+        String action = cmd.getOptionValue("action", "update");
+        String input = cmd.getOptionValue("input");
+
+        ConnectionData connectionData;
+
+        if (action.equals("update")) {
+
+            if (input == null) {
+                System.err.println("Error: Missing input file");
+                printHelp();
+                System.exit(-1);
+                return;
+            }
+
+            try (BufferedReader in = new BufferedReader(new FileReader(input));
+                StringWriter sw = new StringWriter();
+                PrintWriter out = new PrintWriter(sw, true)) {
+
+                String line;
+                while ((line = in.readLine()) != null) {
+                    out.println(line);
+                }
+
+                connectionData = ConnectionData.valueOf(sw.toString());
+            }
+
+            connectionData = connectorCLI.connectionClient.updateConnection(connectorID, connectionData);
+
+        } else { // other actions
+            connectionData = connectorCLI.connectionClient.changeConnectionStatus(connectorID, action);
+        }
+
+        MainCLI.printMessage("Modified connector \"" + connectorID + "\"");
+
+        ConnectorCLI.printConnectionData(connectionData, true);
+    }
+}