1 files changed, 93 insertions, 0 deletions
diff --git a/cli/src/input.c b/cli/src/input.c
new file mode 100644
index 00000000000..5ac1a20edb1
--- /dev/null
+++ b/cli/src/input.c
@@ -0,0 +1,93 @@
+/*
+   Copyright (c) 2010-2012 Red Hat, Inc. <http://www.redhat.com>
+   This file is part of GlusterFS.
+
+   This file is licensed to you under your choice of the GNU Lesser
+   General Public License, version 3 or any later version (LGPLv3 or
+   later), or the GNU General Public License, version 2 (GPLv2), in all
+   cases as published by the Free Software Foundation.
+*/
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <pthread.h>
+
+#include "cli.h"
+#include "cli-mem-types.h"
+
+#define CMDBUFSIZ 1024
+
+void *
+cli_batch(void *d)
+{
+    struct cli_state *state = NULL;
+    int ret = 0;
+
+    state = d;
+
+    ret = cli_cmd_process(state, state->argc, state->argv);
+
+    gf_log("", GF_LOG_INFO, "Exiting with: %d", ret);
+    exit(-ret);
+
+    return NULL;
+}
+
+void *
+cli_input(void *d)
+{
+    struct cli_state *state = NULL;
+    int ret = 0;
+    char cmdbuf[CMDBUFSIZ];
+    char *cmd = NULL;
+    size_t len = 0;
+
+    state = d;
+
+    fprintf(stderr,
+            "Welcome to gluster prompt, type 'help' to see the available "
+            "commands.\n");
+    for (;;) {
+        printf("%s", state->prompt);
+
+        cmd = fgets(cmdbuf, CMDBUFSIZ, stdin);
+        if (!cmd)
+            break;
+        len = strlen(cmd);
+        if (len > 0 && cmd[len - 1] == '\n')  // strip trailing \n
+            cmd[len - 1] = '\0';
+        ret = cli_cmd_process_line(state, cmd);
+        if (ret != 0 && state->mode & GLUSTER_MODE_ERR_FATAL)
+            break;
+    }
+
+    exit(-ret);
+
+    return NULL;
+}
+
+int
+cli_input_init(struct cli_state *state)
+{
+    int ret = 0;
+
+    if (state->argc) {
+        ret = pthread_create(&state->input, NULL, cli_batch, state);
+        return ret;
+    }
+
+    if (isatty(STDIN_FILENO)) {
+        state->prompt = "gluster> ";
+
+        cli_rl_enable(state);
+    } else {
+        state->prompt = "";
+        state->mode |= GLUSTER_MODE_SCRIPT | GLUSTER_MODE_ERR_FATAL;
+    }
+
+    if (!state->rl_enabled)
+        ret = pthread_create(&state->input, NULL, cli_input, state);
+
+    return ret;
+}