Real zombies are undead, not the kind you see on TV.
[citadel.git] / ctdlsh / main.c
index 6fc29315c51ab7d35cc5f1ae3f16297689e7dffe..fd6663ecf6fb0a4e517ace1f6b38a0d42f3294dc 100644 (file)
@@ -1,13 +1,14 @@
 /*
- * (c) 2009-2012 by Art Cancro and citadel.org
+ * (c) 2009-2017 by Art Cancro and citadel.org
  * This program is released under the terms of the GNU General Public License v3.
  */
 
 #include "ctdlsh.h"
 
 
-int cmd_quit(int sock, char *cmdbuf) {
-       return(cmdret_exit);
+int cmd_quit(int sock, char *cmdbuf)
+{
+       return (cmdret_exit);
 }
 
 
@@ -21,32 +22,55 @@ typedef struct {
 } COMMAND;
 
 COMMAND commands[] = {
-       {       "?",            cmd_help,       "Display this message"                  },
-       {       "help",         cmd_help,       "Display this message"                  },
-       {       "date",         cmd_datetime,   "Print the server's date and time"      },
-       {       "config",       cmd_config,     "Configure the Citadel server"          },
-       {       "export",       cmd_export,     "Export all Citadel databases"          },
-       {       "shutdown",     cmd_shutdown,   "Shut down the Citadel server"          },
-       {       "time",         cmd_datetime,   "Print the server's date and time"      },
-       {       "passwd",       cmd_passwd,     "Set or change an account password"     },
-       {       "who",          cmd_who,        "Display a list of online users"        },
-       {       "exit",         cmd_quit,       "Quit using ctdlsh"                     },
-       {       "quit",         cmd_quit,       "Quit using ctdlsh"                     },
-       {       NULL,           NULL,           NULL                                    }
+       {"?", cmd_help, "Display this message"},
+       {"help", cmd_help, "Display this message"},
+       {"date", cmd_datetime, "Print the server's date and time"},
+       {"config", cmd_config, "Configure the Citadel server"},
+       {"export", cmd_export, "Export all Citadel databases"},
+       {"shutdown", cmd_shutdown, "Shut down the Citadel server"},
+       {"time", cmd_datetime, "Print the server's date and time"},
+       {"passwd", cmd_passwd, "Set or change an account password"},
+       {"who", cmd_who, "Display a list of online users"},
+       {"exit", cmd_quit, "Quit using ctdlsh"},
+       {"quit", cmd_quit, "Quit using ctdlsh"},
+       {"mailq", cmd_mailq, "Show the outbound email queue"},
+       {NULL, NULL, NULL}
+};
+
+
+
+struct wow {
+       char *cmd[5];                   // increase this if we need to have larger commands
+       char *description;
+};
+
+struct wow wows[] = {
+       {{ "show", "eggs" }                             , "show how many eggs are available for serving"        },
+       {{ "kill", "mark", "zuckerberg" }               , "die motherfucker die"                                },
+       {{ "show", "undead", "zombies", "real" }        , "show how many zombies are actually undead"           },
+       {{ "show", "undead", "zombies", "hollywood" }   , "show how many zombies are hollywood communists"      }
 };
 
 
-int cmd_help(int sock, char *cmdbuf) {
+
+
+
+
+
+
+int cmd_help(int sock, char *cmdbuf)
+{
        int i;
 
-       for (i=0; commands[i].func != NULL; ++i) {
+       for (i = 0; commands[i].func != NULL; ++i) {
                printf("%-10s %s\n", commands[i].name, commands[i].doc);
        }
 }
 
 
 /* Auto-completer function */
-char *command_generator(const char *text, int state) {
+char *command_generator(const char *text, int state)
+{
        static int list_index;
        static int len;
        char *name;
@@ -60,22 +84,23 @@ char *command_generator(const char *text, int state) {
                ++list_index;
 
                if (!strncmp(name, text, len)) {
-                       return(strdup(name));
+                       return (strdup(name));
                }
        }
 
-       return(NULL);
+       return (NULL);
 }
 
 
 /* Auto-completer function */
-char **ctdlsh_completion(const char *text, int start, int end) {
+char **ctdlsh_completion(const char *text, int start, int end)
+{
        char **matches = (char **) NULL;
 
+       rl_completer_word_break_characters = " ";
        if (start == 0) {
                matches = rl_completion_matches(text, command_generator);
-       }
-       else {
+       } else {
                rl_bind_key('\t', rl_abort);
        }
 
@@ -83,7 +108,21 @@ char **ctdlsh_completion(const char *text, int start, int end) {
 }
 
 
-void do_main_loop(int server_socket) {
+int do_one_command(int server_socket, char *cmd)
+{
+       int i;
+       int ret;
+       for (i = 0; commands[i].func != NULL; ++i) {
+               if (!strncasecmp(cmd, commands[i].name, strlen(commands[i].name))) {
+                       ret = (*commands[i].func) (server_socket, cmd);
+               }
+       }
+       return ret;
+}
+
+
+void do_main_loop(int server_socket)
+{
        char *cmd = NULL;
        char prompt[1024];
        char buf[1024];
@@ -91,6 +130,7 @@ void do_main_loop(int server_socket) {
        int i;
        int ret = (-1);
 
+
        strcpy(prompt, "> ");
 
        /* Do an INFO command and learn the hostname for the prompt */
@@ -98,7 +138,7 @@ void do_main_loop(int server_socket) {
        sock_getln(server_socket, buf, sizeof buf);
        if (buf[0] == '1') {
                i = 0;
-               while(sock_getln(server_socket, buf, sizeof buf), strcmp(buf, "000")) {
+               while (sock_getln(server_socket, buf, sizeof buf), strcmp(buf, "000")) {
                        if (i == 1) {
                                sprintf(prompt, "\n%s> ", buf);
                        }
@@ -114,12 +154,13 @@ void do_main_loop(int server_socket) {
 
                if ((cmd) && (*cmd)) {
                        add_history(cmd);
+                       ret = do_one_command(server_socket, cmd);
 
-                       for (i=0; commands[i].func != NULL; ++i) {
-                               if (!strncasecmp(cmd, commands[i].name, strlen(commands[i].name))) {
-                                       ret = (*commands[i].func) (server_socket, cmd);
-                               }
-                       }
+                       //for (i=0; commands[i].func != NULL; ++i) {
+                       //if (!strncasecmp(cmd, commands[i].name, strlen(commands[i].name))) {
+                       //ret = (*commands[i].func) (server_socket, cmd);
+                       //}
+                       //}
 
                }
 
@@ -135,33 +176,50 @@ int main(int argc, char **argv)
 {
        int server_socket = 0;
        char buf[1024];
-       int c;
+       int i;
        char *ctdldir = CTDLDIR;
+       char cmd[1024] = { 0 };
+       int exitcode = 0;
 
-       printf("\nCitadel administration shell (c) 2009-2016 by citadel.org\n"
-               "This is open source software made available to you under the terms\n"
-               "of the GNU General Public License v3.  All other rights reserved.\n"
-       );
-
-       opterr = 0;
-       while ((c = getopt (argc, argv, "h:")) != -1) {
-               switch(c) {
-               case 'h':
-                       ctdldir = optarg;
-                       break;
-               case '?':
-                       if (optopt == 'h') {
-                               fprintf(stderr, "Option -%c requires an argument\n", optopt);
-                       }
-                       else {
-                               fprintf(stderr, "Unknown option '-%c'\n", optopt);
-                               fprintf(stderr, "usage: %s [-h citadel_dir]\n", argv[0]);
+
+       int num_wows = sizeof(wows) / sizeof(struct wow);
+       int j;
+       for (i=0; i<num_wows; ++i) {
+               printf("%s\n", wows[i].description);
+               for (j=0; j<5; ++j) {
+                       printf("%d '%s'\n", j, wows[i].cmd[j]);
+               }
+       }
+
+
+exit(0);
+
+
+
+
+
+
+
+       for (i = 1; i < argc; ++i) {
+               if (!strcmp(argv[i], "-h")) {
+                       ctdldir = argv[++i];
+               } else {
+                       if (strlen(cmd) > 0) {
+                               strcat(cmd, " ");
                        }
-                       exit(1);
+                       strcat(cmd, argv[i]);
                }
        }
 
-       printf("Trying %s...\n", ctdldir);
+       int is_interactive = ((strlen(cmd) == 0) ? 1 : 0);
+
+       if (is_interactive) {
+               printf("\nCitadel administration shell (c) 2009-2017 by citadel.org\n"
+                      "This is open source software made available to you under the terms\n"
+                      "of the GNU General Public License v3.  All other rights reserved.\n");
+               printf("Trying %s...\n", ctdldir);
+       }
+
        sprintf(buf, "%s/citadel-admin.socket", ctdldir);
        server_socket = uds_connectsock(buf);
        if (server_socket < 0) {
@@ -170,13 +228,19 @@ int main(int argc, char **argv)
 
        sock_getln(server_socket, buf, sizeof buf);
        if (buf[0] == '2') {
-               printf("Connected: %s\n", buf);
-               do_main_loop(server_socket);
+               if (is_interactive) {
+                       printf("Connected: %s\n", buf);
+                       do_main_loop(server_socket);
+               } else {
+                       exitcode = do_one_command(server_socket, cmd);
+               }
        }
 
        sock_puts(server_socket, "QUIT");
        sock_getln(server_socket, buf, sizeof buf);
-       printf("%s\n", buf);
+       if (is_interactive) {
+               printf("%s\n", buf);
+       }
        close(server_socket);
-       exit(0);
+       exit(exitcode);
 }