Merge pull request #23 from lukechampine/betterprompt

refactor command argument parsing

src/prompt.c

@@ -22,40 +22,39 @@ static char prompt_buf[MAX_STR_SIZE] = {0};
 static int prompt_buf_pos = 0;
 
 /* commands */
-void cmd_accept(ToxWindow *, Tox *m, char **);
+void cmd_accept(ToxWindow *, Tox *m, int, char **);
-void cmd_add(ToxWindow *, Tox *m, char **);
+void cmd_add(ToxWindow *, Tox *m, int, char **);
-void cmd_clear(ToxWindow *, Tox *m, char **);
+void cmd_clear(ToxWindow *, Tox *m, int, char **);
-void cmd_connect(ToxWindow *, Tox *m, char **);
+void cmd_connect(ToxWindow *, Tox *m, int, char **);
-void cmd_help(ToxWindow *, Tox *m, char **);
+void cmd_help(ToxWindow *, Tox *m, int, char **);
-void cmd_msg(ToxWindow *, Tox *m, char **);
+void cmd_msg(ToxWindow *, Tox *m, int, char **);
-void cmd_myid(ToxWindow *, Tox *m, char **);
+void cmd_myid(ToxWindow *, Tox *m, int, char **);
-void cmd_nick(ToxWindow *, Tox *m, char **);
+void cmd_nick(ToxWindow *, Tox *m, int, char **);
-void cmd_mynick(ToxWindow *, Tox *m, char **);
+void cmd_mynick(ToxWindow *, Tox *m, int, char **);
-void cmd_quit(ToxWindow *, Tox *m, char **);
+void cmd_quit(ToxWindow *, Tox *m, int, char **);
-void cmd_status(ToxWindow *, Tox *m, char **);
+void cmd_status(ToxWindow *, Tox *m, int, char **);
-void cmd_statusmsg(ToxWindow *, Tox *m, char **);
+void cmd_statusmsg(ToxWindow *, Tox *m, int, char **);
 
 #define NUM_COMMANDS 14
 
 static struct {
     char *name;
-    int numargs;
+    void (*func)(ToxWindow *, Tox *m, int, char **);
-    void (*func)(ToxWindow *, Tox *m, char **);
 } commands[] = {
-    { "accept",    1, cmd_accept    },
+    { "accept",    cmd_accept    },
-    { "add",       1, cmd_add       },
+    { "add",       cmd_add       },
-    { "clear",     0, cmd_clear     },
+    { "clear",     cmd_clear     },
-    { "connect",   3, cmd_connect   },
+    { "connect",   cmd_connect   },
-    { "exit",      0, cmd_quit      },
+    { "exit",      cmd_quit      },
-    { "help",      0, cmd_help      },
+    { "help",      cmd_help      },
-    { "msg",       2, cmd_msg       },
+    { "msg",       cmd_msg       },
-    { "myid",      0, cmd_myid      },
+    { "myid",      cmd_myid      },
-    { "nick",      1, cmd_nick      },
+    { "nick",      cmd_nick      },
-    { "mynick",    0, cmd_mynick    },
+    { "mynick",    cmd_mynick    },
-    { "q",         0, cmd_quit      },
+    { "q",         cmd_quit      },
-    { "quit",      0, cmd_quit      },
+    { "quit",      cmd_quit      },
-    { "status",    2, cmd_status    },
+    { "status",    cmd_status    },
-    { "statusmsg", 1, cmd_statusmsg },
+    { "statusmsg", cmd_statusmsg },
 };
 
 // XXX:
@@ -80,15 +79,24 @@ unsigned char *hex_string_to_bin(char hex_string[])
     return val;
 }
 
-void cmd_accept(ToxWindow *self, Tox *m, char **args)
+/* command functions */
+void cmd_accept(ToxWindow *self, Tox *m, int argc, char **argv)
 {
-    int num = atoi(args[1]);
+    int num;
 
-    if (num >= num_requests) {
+    /* check arguments */
+    if (argc != 1) {
       wprintw(self->window, "Invalid syntax.\n");
       return;
     }
 
+    num = atoi(argv[1]);
+
+    if (num < 0 || num >= num_requests) {
+        wprintw(self->window, "No pending request with that number.\n");
+        return;
+    }
+
     num = tox_addfriend_norequest(m, pending_requests[num]);
 
     if (num == -1)
@@ -99,29 +107,33 @@ void cmd_accept(ToxWindow *self, Tox *m, char **args)
     }
 }
 
-void cmd_add(ToxWindow *self, Tox *m, char **args)
+void cmd_add(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     uint8_t id_bin[TOX_FRIEND_ADDRESS_SIZE];
     char xx[3];
     uint32_t x;
-    char *id = args[1];
+    char *id;
-    char *msg = args[2];
+    char *msg;
+    int i, num;
 
-    if (!id) {
+    /* check arguments */
-        wprintw(self->window, "Invalid command: add expected at least one argument.\n");
+    if (argv[2] && argv[2][0] != '\"') {
+        wprintw(self->window, "Strings must be enclosed in quotes.\n");
+        return;
+    }
+    if (argc != 1 && argc != 2) {
+        wprintw(self->window, "Invalid syntax.\n");
         return;
     }
 
-    if (!msg)
+    id = argv[1];
-        msg = "";
+    msg = (argc == 2) ? argv[2] : "";
 
     if (strlen(id) != 2 * TOX_FRIEND_ADDRESS_SIZE) {
         wprintw(self->window, "Invalid ID length.\n");
         return;
     }
 
-    int i;
-
     for (i = 0; i < TOX_FRIEND_ADDRESS_SIZE; ++i) {
         xx[0] = id[2 * i];
         xx[1] = id[2 * i + 1];
@@ -139,7 +151,7 @@ void cmd_add(ToxWindow *self, Tox *m, char **args)
         id[i] = toupper(id[i]);
     }
 
-    int num = tox_addfriend(m, id_bin, (uint8_t *) msg, strlen(msg) + 1);
+    num = tox_addfriend(m, id_bin, (uint8_t *) msg, strlen(msg) + 1);
 
     switch (num) {
         case TOX_FAERR_TOOLONG:
@@ -177,17 +189,25 @@ void cmd_add(ToxWindow *self, Tox *m, char **args)
     }
 }
 
-void cmd_clear(ToxWindow *self, Tox *m, char **args)
+void cmd_clear(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     wclear(self->window);
 }
 
-void cmd_connect(ToxWindow *self, Tox *m, char **args)
+void cmd_connect(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     tox_IP_Port dht;
-    char *ip = args[1];
+    char *ip, *port, *key;
-    char *port = args[2];
+
-    char *key = args[3];
+    /* check arguments */
+    if (argc != 3) {
+      wprintw(self->window, "Invalid syntax.\n");
+      return;
+    }
+
+    ip = argv[1];
+    port = argv[2];
+    key = argv[3];
 
     if (atoi(port) == 0) {
         wprintw(self->window, "Invalid syntax.\n");
@@ -207,13 +227,13 @@ void cmd_connect(ToxWindow *self, Tox *m, char **args)
     free(binary_string);
 }
 
-void cmd_quit(ToxWindow *self, Tox *m, char **args)
+void cmd_quit(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     endwin();
     exit(0);
 }
 
-void cmd_help(ToxWindow *self, Tox *m, char **args)
+void cmd_help(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     wclear(self->window);
     wattron(self->window, COLOR_PAIR(2) | A_BOLD);
@@ -233,16 +253,25 @@ void cmd_help(ToxWindow *self, Tox *m, char **args)
     wprintw(self->window, "      clear                     : Clear this window\n");
 
     wattron(self->window, A_BOLD);
+    wprintw(self->window, "NOTE: Strings must be enclosed in quotation marks.\n");
     wprintw(self->window, "TIP: Use the TAB key to navigate through the tabs.\n\n");
     wattroff(self->window, A_BOLD);
 
     wattroff(self->window, COLOR_PAIR(2));
 }
 
-void cmd_msg(ToxWindow *self, Tox *m, char **args)
+void cmd_msg(ToxWindow *self, Tox *m, int argc, char **argv)
 {
-    char *id = args[1];
+    char *id, *msg;
-    char *msg = args[2];
+
+    /* check arguments */
+    if (argc != 2) {
+      wprintw(self->window, "Invalid syntax.\n");
+      return;
+    }
+
+    id = argv[1];
+    msg = argv[2];
 
     if (tox_sendmessage(m, atoi(id), (uint8_t *) msg, strlen(msg) + 1) == 0)
         wprintw(self->window, "Error occurred while sending message.\n");
@@ -250,7 +279,7 @@ void cmd_msg(ToxWindow *self, Tox *m, char **args)
         wprintw(self->window, "Message successfully sent.\n");
 }
 
-void cmd_myid(ToxWindow *self, Tox *m, char **args)
+void cmd_myid(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     char id[TOX_FRIEND_ADDRESS_SIZE * 2 + 1] = {0};
     size_t i;
@@ -266,9 +295,18 @@ void cmd_myid(ToxWindow *self, Tox *m, char **args)
     wprintw(self->window, "%s\n", id);
 }
 
-void cmd_nick(ToxWindow *self, Tox *m, char **args)
+void cmd_nick(ToxWindow *self, Tox *m, int argc, char **argv)
 {
-    char *nick = args[1];
+    char *nick;
+
+    /* check arguments */
+    if (argc != 1) {
+      wprintw(self->window, "Invalid syntax.\n");
+      return;
+    }
+
+    nick = argv[1];
+
     tox_setname(m, (uint8_t *) nick, strlen(nick) + 1);
     wprintw(self->window, "Nickname set to: %s\n", nick);
 
@@ -277,7 +315,7 @@ void cmd_nick(ToxWindow *self, Tox *m, char **args)
     }
 }
 
-void cmd_mynick(ToxWindow *self, Tox *m, char **args)
+void cmd_mynick(ToxWindow *self, Tox *m, int argc, char **argv)
 {
     uint8_t *nick = malloc(TOX_MAX_NAME_LENGTH);
     tox_getselfname(m, nick, TOX_MAX_NAME_LENGTH);
@@ -285,10 +323,21 @@ void cmd_mynick(ToxWindow *self, Tox *m, char **args)
     free(nick);
 }
 
-void cmd_status(ToxWindow *self, Tox *m, char **args)
+void cmd_status(ToxWindow *self, Tox *m, int argc, char **argv)
 {
-    char *status = args[1];
+    char *status, *status_text, *msg;
-    char *status_text;
+
+    /* check arguments */
+    if (argv[2] && argv[2][0] != '\"') {
+        wprintw(self->window, "Strings must be enclosed in quotes.\n");
+        return;
+    }
+    if (argc != 1 && argc != 2) {
+        wprintw(self->window, "Wrong number of arguments.\n");
+        return;
+    }
+
+    status = argv[1];
 
     TOX_USERSTATUS status_kind;
 
@@ -306,7 +355,7 @@ void cmd_status(ToxWindow *self, Tox *m, char **args)
         return;
     }
 
-    char *msg = args[2];
+    msg = argv[2];
 
     if (msg == NULL) {
         tox_set_userstatus(m, status_kind);
@@ -318,9 +367,22 @@ void cmd_status(ToxWindow *self, Tox *m, char **args)
     }
 }
 
-void cmd_statusmsg(ToxWindow *self, Tox *m, char **args)
+void cmd_statusmsg(ToxWindow *self, Tox *m, int argc, char **argv)
 {
-    char *msg = args[1];
+    char *msg;
+
+    /* check arguments */
+    if (argv[1] && argv[1][0] != '\"') {
+        wprintw(self->window, "Strings must be enclosed in quotes.\n");
+        return;
+    }
+    if (argc != 1) {
+        wprintw(self->window, "Wrong number of arguments.\n");
+        return;
+    }
+
+    msg = argv[1];
+
     tox_set_statusmessage(m, (uint8_t *) msg, strlen(msg) + 1);
     wprintw(self->window, "Status message set to: %s\n", msg);
 }
@@ -355,76 +417,47 @@ static void execute(ToxWindow *self, Tox *m, char *u_cmd)
 
     /* insert \0 at argument boundaries */
     int numargs = 0;
-
     for (i = 0; i < MAX_STR_SIZE; i++) {
-        char quote_chr;
+        if (cmd[i] == ' ') {
-        if (cmd[i] == '\"' || cmd[i] == '\'') {
+            cmd[i] = '\0';
-            quote_chr = cmd[i];
+            numargs++;
-            while (cmd[++i] != quote_chr && i < MAX_STR_SIZE); /* skip over strings */
+        }
-            /* Check if got qoute character */
+        /* skip over strings */
-            if (cmd[i] != quote_chr) {
+        else if (cmd[i] == '\"') {
-                wprintw(self->window, "Missing terminating %c character\n", quote_chr);
+            while (cmd[++i] != '\"') {
+                if (cmd[i] == '\0') {
+                    wprintw(self->window, "Invalid command: did you forget a closing \"?\n");
                     return;
                 }
             }
-
-        if (cmd[i] == ' ') {
-            cmd[i] = '\0';
-
-            int j = i;
-
-            while (++j < MAX_STR_SIZE && isspace(cmd[j]));
-
-            i = j - 1;
-
-            numargs++;
         }
     }
 
-    /* excessive arguments */
+    /* read arguments into array */
-    if (numargs > 3) {
+    char **cmdargs = malloc((numargs + 1) * sizeof(char *));
+    if (!cmdargs) {
       wprintw(self->window, "Invalid command: too many arguments.\n");
       return;
     }
 
-    /* read arguments into array */
-    char *cmdargs[5];
     int pos = 0;
     
-    for (i = 0; i < 5; i++) {
+    for (i = 0; i < numargs + 1; i++) {
         cmdargs[i] = cmd + pos;
         pos += strlen(cmdargs[i]) + 1;
-
+        /* replace empty strings with NULL for easier error checking */
-        while (isspace(cmd[pos]) && pos < MAX_STR_SIZE)
+        if (strlen(cmdargs[i]) == 0)
-            ++pos;
+            cmdargs[i] = NULL;
     }
 
     /* no input */
-    if (strlen(cmdargs[0]) == 0)
+    if (!cmdargs[0])
         return;
 
     /* match input to command list */
     for (i = 0; i < NUM_COMMANDS; i++) {
         if (!strcmp(cmdargs[0], commands[i].name)) {
-            /* check for missing arguments */
+            (commands[i].func)(self, m, numargs, cmdargs);
-            int j;
-
-            for (j = 0; j <= commands[i].numargs; j++) {
-                if (strlen(cmdargs[j]) == 0) {
-                    wprintw(self->window, "Invalid command: %s expected %d arguments, got %d.\n",
-                            commands[i].name, commands[i].numargs, j - 1);
-                    return;
-                }
-            }
-
-            /* check for excess arguments */
-            if (strcmp(cmdargs[0], "add") && strlen(cmdargs[j]) != 0) {
-                wprintw(self->window, "Invalid command: too many arguments to %s.\n", commands[i].name);
-                return;
-            }
-
-            /* pass arguments to command function */
-            (commands[i].func)(self, m, cmdargs);
             return;
         }
     }
@@ -493,7 +526,7 @@ static void prompt_onDraw(ToxWindow *self, Tox *m)
 static void prompt_onInit(ToxWindow *self, Tox *m)
 {
     scrollok(self->window, 1);
-    cmd_help(self, m, NULL);
+    cmd_help(self, m, 0, NULL);
     wclrtoeol(self->window);
 }