toxic/src/execute.c

/*
 * Toxic -- Tox Curses Client
 */

#include <stdlib.h>
#include <string.h>

#include "toxic_windows.h"
#include "execute.h"
#include "chat_commands.h"
#include "global_commands.h"

struct cmd_func {
    const char *name;
    void (*func)(WINDOW *, ToxWindow *, Tox *m, int argc, char (*argv)[MAX_STR_SIZE]);
};

#define GLOBAL_NUM_COMMANDS 12

static struct cmd_func global_commands[] = {
    { "/accept",    cmd_accept    },
    { "/add",       cmd_add       },
    { "/clear",     cmd_clear     },
    { "/connect",   cmd_connect   },
    { "/exit",      cmd_quit      },
    { "/groupchat", cmd_groupchat },
    { "/myid",      cmd_myid      },
    { "/nick",      cmd_nick      },
    { "/note",      cmd_note      },
    { "/q",         cmd_quit      },
    { "/quit",      cmd_quit      },
    { "/status",    cmd_status    },
};

#define CHAT_NUM_COMMANDS 5

static struct cmd_func chat_commands[] = {
    { "/help",      cmd_chat_help   },
    { "/invite",    cmd_groupinvite },
    { "/join",      cmd_join_group  },
    { "/savefile",  cmd_savefile    },
    { "/sendfile",  cmd_sendfile    },
};

/* Parses input command and puts args into arg array. 
   Returns number of arguments on success, -1 on failure. */
static int parse_command(WINDOW *w, char *cmd, char (*args)[MAX_STR_SIZE])
{
    int num_args = 0;
    bool cmd_end = false;    // flags when we get to the end of cmd
    char *end;               // points to the end of the current arg

    /* characters wrapped in double quotes count as one arg */
    while (!cmd_end && num_args < MAX_NUM_ARGS) {
        if (*cmd == '\"') {
            end = strchr(cmd+1, '\"');

            if (end++ == NULL) {    /* Increment past the end quote */
                wprintw(w, "Invalid argument. Did you forget a closing \"?\n");
                return -1;
            }

            cmd_end = *end == '\0';
        } else {
            end = strchr(cmd, ' ');
            cmd_end = end == NULL;
        }

        if (!cmd_end)
            *end++ = '\0';    /* mark end of current argument */

        /* Copy from start of current arg to where we just inserted the null byte */
        strcpy(args[num_args++], cmd);
        cmd = end;
    }

    return num_args;
}

/* Matches command to respective function. Returns 0 on match, 1 on no match */
static int do_command(WINDOW *w, ToxWindow *prompt, Tox *m, int num_args, int num_cmds,
                      struct cmd_func *commands, char (*args)[MAX_STR_SIZE])
{
    int i;

    for (i = 0; i < num_cmds; ++i) {
        if (strcmp(args[0], commands[i].name) == 0) {
            (commands[i].func)(w, prompt, m, num_args-1, args);
            return 0;
        }
    }

    return 1;
}

void execute(WINDOW *w, ToxWindow *prompt, Tox *m, char *cmd, int mode)
{
    if (string_is_empty(cmd))
        return;

    char args[MAX_NUM_ARGS][MAX_STR_SIZE] = {0};
    int num_args = parse_command(w, cmd, args);

    if (num_args == -1)
        return;

    /* Try to match input command to command functions. If non-global command mode is specified, 
       try specified mode's commands first, then upon failure try global commands. */
    switch (mode) {
    case CHAT_COMMAND_MODE:
        if (do_command(w, prompt, m, num_args, CHAT_NUM_COMMANDS, chat_commands, args) == 0)
            return;
        break;

    case GROUPCHAT_COMMAND_MODE:
        break;
    }

    if (do_command(w, prompt, m, num_args, GLOBAL_NUM_COMMANDS, global_commands, args) == 0)
        return;

    wprintw(w, "Invalid command.\n");
}