picoshell.cpp

/* *************************************************************************** *
 *                   Tiny sample shell using msh routines.
 * *************************************************************************** */

#include "picoshell.h"
#include "picoshell_termesc.h"

#define PUTS_BLUE_BACK(charp) \
{ \
    pico_puts(TERMESC_BACK_BLUE); \
    pico_puts(charp); \
    pico_puts(TERMESC_FONT_NORMAL);\
}


#define PUTS_GREEN_BACK(charp) \
{ \
    pico_puts(TERMESC_BACK_GREEN); \
    pico_puts(charp); \
    pico_puts(TERMESC_FONT_NORMAL);\
}

/*
 * Prototypes for our commands
 */
msh_declare_command( help );
msh_declare_command( myecho );
msh_declare_command( exit );


const msh_command_entry my_commands[] = {
    msh_define_command( help ),
    msh_define_command( myecho ),
    msh_define_command( exit ),
    MSH_COMMAND_TERMINATOR
};


/*
 * Define some original commands over built-ins
 *     We have to define cmd_help by yourself...
 */

msh_define_help( help, "display help for available commands",
        "Usage: help [command]\n"
        "    Displays help for 'command', or all commands and their\n"
        "    short descriptions.\n" );
int cmd_help(int argc, const char** argv)
{
    if ( argc == 1) {
        msh_print_cmdlist(my_commands);
        msh_print_cmdlist(msh_builtin_commands);
    }
    else
    {
        const char* usage;
        usage = msh_get_command_usage(my_commands, argv[1]);

        if ( ! usage ) {
            usage = msh_get_command_usage(msh_builtin_commands, argv[1]);
        }
        if ( ! usage  ) {
            printf("No such command: '%s'\n", argv[1]);
        }
        else
        {
            pico_puts(usage);
        }
    }
    return 0;
}


msh_define_help( myecho, "display a line of text",
        "Usage: myecho [STRING]...\n");
int cmd_myecho(int argc, const char** argv)
{
    int i;
    if ( argc < 1 ) {
        return 0;
    }
    for ( i = 1;  i < argc;  i++ ) {
        pico_puts(argv[i]);
        pico_putchar('_'); /* '_' instead of ' ' */
    }
    pico_putchar('\n');
    return 0;
}


msh_define_help( exit, "exit the msh sample",
        "Usage: exit [EXIT_CODE]\n" );
int cmd_exit(int argc, const char** argv)
{
    if ( argc > 2 ) {
        pico_puts("exit: too many arguments\n");
        return 1;
    } else {
        pico_puts("(noop)");
    }
    return 0;
}


/*
 * main() just loopingly read input line, parse, and execute.
 */

int shell(void)
{
    char  linebuf[MSH_CMDLINE_CHAR_MAX];

    msh_set_prompt("SAMPLE> ");


    /*
     * Loop for reading line (forever).
     */
    int   argc;
    char* argv[MSH_CMDARGS_MAX];
    char  argbuf[MSH_CMDLINE_CHAR_MAX];
    while ( 1 ) {
        /*
         * Read a input line
         */
        if (  msh_get_cmdline(linebuf) == 0 ) {
            /* got empty input */
            continue;
        }

        /*
         * Loop for parse line and executing commands.
         */
        const char *linebufp = linebuf;
        while ( 1 ) {
            int i;
            const char* ret_parse;
            int ret_command;

            ret_parse = msh_parse_line(linebufp, argbuf, &argc, argv);

            if ( ! ret_parse ) {
                pico_puts("Syntax error\n");
                break; /* discard this line */
            }
            if ( strlen(argv[0]) <= 0 ) {
                break; /* empty input line */
            }

            pico_puts(">> ");
            for ( i = argc-1; i > 0; i-- ) {
                PUTS_GREEN_BACK(argv[i]);
                if ( i > 1 ) {
                    pico_puts(", ");
                }
            }
            pico_puts("\r\n");

            ret_command = msh_do_command(my_commands, argc, (const char**)argv);
            if ( ret_command < 0 ) {
                /* If the command not found amoung my_commands[], search the
                 * buildin */
                ret_command =
                    msh_do_command(msh_builtin_commands, argc, (const char**)argv);
            }
            if ( ret_command < 0 ) {
                pico_puts("command not found: \'");
                pico_puts(argv[0]);
                pico_puts("'\r\n");
            }

            /*
             * Do we have more sentents remained in linebuf
             * separated by a ';' or a '\n' ?
             */
            if ( ret_parse == linebufp ) {
                /* No, we don't */
                break;
            } else {
                /* Yes, we have. We have to parse rest of lines,
                 * which begins with char* ret_parse; */
                linebufp = ret_parse;
            }
        }
    }

    return 0;
}