profanity/src/parser.c

/*
 * parser.c
 *
 * Copyright (C) 2012 James Booth <boothj5@gmail.com>
 *
 * This file is part of Profanity.
 *
 * Profanity 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, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Profanity 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 Profanity.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

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

#include <glib.h>

gchar **
parse_args(const char * const inp, int min, int max, int *num)
{
    if (inp == NULL) {
        *num = 0;
        return NULL;
    }

    // copy and strip input of leading/trailing whitepsace
    char *copy = strdup(inp);
    g_strstrip(copy);

    int inp_size = strlen(copy);
    gboolean in_token = FALSE;
    char *token_start = &copy[0];
    int token_size = 0;
    GSList *tokens = NULL;

    // add tokens to GSList
    int i;
    for (i = 0; i <= inp_size; i++) {
        if (!in_token) {
            if (copy[i] == ' ') {
                continue;
            } else {
                in_token = TRUE;
                token_start = &copy[i];
                token_size++;
            }
        } else {
            if (copy[i] == ' ' || copy[i] == '\0') {
                tokens = g_slist_append(tokens, g_strndup(token_start,
                    token_size));
                token_size = 0;
                in_token = FALSE;
            } else {
                token_size++;
            }
        }
    }

    *num = g_slist_length(tokens) - 1;

    // if num args not valid return NULL
    if ((*num < min) || (*num > max)) {
        g_slist_free_full(tokens, free);
        free(copy);
        *num = 0;
        return NULL;

    // otherwise return args array
    } else {
        gchar **args = malloc((*num + 1) * sizeof(*args));
        GSList *token = tokens;
        token = g_slist_next(token);
        int arg_count = 0;

        while (token != NULL) {
            args[arg_count++] = strdup(token->data);
            token = g_slist_next(token);
        }

        args[arg_count] = NULL;
        g_slist_free_full(tokens, free);
        free(copy);

        return args;
    }
}

gchar **
parse_args_with_freetext(const char * const inp, int min, int max, int *num)
{
    if (inp == NULL) {
        *num = 0;
        return NULL;
    }

    // copy and strip input of leading/trailing whitepsace
    char *copy = strdup(inp);
    g_strstrip(copy);

    int inp_size = strlen(copy);
    gboolean in_token = FALSE;
    gboolean in_freetext = FALSE;
    char *token_start = &copy[0];
    int token_size = 0;
    int num_tokens = 0;
    GSList *tokens = NULL;

    // add tokens to GSList
    int i;
    for (i = 0; i <= inp_size; i++) {
        if (!in_token) {
            if (copy[i] == ' ') {
                continue;
            } else {
                in_token = TRUE;
                num_tokens++;
                if (num_tokens == max + 1) {
                    in_freetext = TRUE;
                }
                token_start = &copy[i];
                token_size++;
            }
        } else {
            if ((!in_freetext && copy[i] == ' ') || copy[i] == '\0') {
                tokens = g_slist_append(tokens, g_strndup(token_start,
                    token_size));
                token_size = 0;
                in_token = FALSE;
            } else {
                token_size++;
            }
        }
    }

    *num = g_slist_length(tokens) - 1;

    // if num args not valid return NULL
    if ((*num < min) || (*num > max)) {
        g_slist_free_full(tokens, free);
        free(copy);
        *num = 0;
        return NULL;

    // otherwise return args array
    } else {
        gchar **args = malloc((*num + 1) * sizeof(*args));
        GSList *token = tokens;
        token = g_slist_next(token);
        int arg_count = 0;

        while (token != NULL) {
            args[arg_count++] = strdup(token->data);
            token = g_slist_next(token);
        }

        args[arg_count] = NULL;
        g_slist_free_full(tokens, free);
        free(copy);

        return args;
    }
}