add more test (3)

[cmore.git] / cmore.c

/* depend: */
/* cflags: */
/* linker: cmd.o debug.o tui.o -lcurses */
/* winlnk: cmd.o debug.o tui.o -lpdcurses */

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

#include "cmd.h"
#include "debug.h"
#include "tui.h"

/* static variables */
char *progname = NULL;
char *version = "0.1";

char *input = NULL;
char *command = NULL;
char **lines = NULL;
int second = 0;

#define MAX_LENGTH 48

void display (char **lines, int xmax, int ymax, int keep_skip)
{
    int stop = 0;

    assert (lines);

#ifdef __PDCURSES__
    statusmsg ("Press 'q' or escape to quit");
#else
    statusmsg ("Press 'q' or double-escape to quit");
#endif

    static int skip;
    if (!keep_skip) {
        skip = 0;
    }

    while (!stop) {
        clsbody ();
        int i = 0;
        int eol = 0;
        for (i = 0; i < ymax - 1; i++) {
            if (lines [skip + i] == NULL) {
                break;
            }
            if (eol) {
                bodymsg ("\n");
            }
            bodymsg (lines[skip + i]);
            eol = (lines[skip + i][xmax - 1 ] == '\0');
        }
        if (lines [skip + i] != NULL) {
            if (eol) {
                bodymsg ("\n");
            }
            char *str = (char *) calloc (1, xmax + 1);
            strcpy (str, lines[skip + i]);
            if (str[xmax - 1] != '\0') {
                VERBOSE (DEBUG, fprintf (stdout, "last line too long\n"));
                strcpy (str + xmax - 3, ">>");
            }
            bodymsg (str);
            free (str);
        }

        if (second > 0) {
            DoExit ();
            break;
        }
        int key = waitforkey ();
        VERBOSE (DEBUG, fprintf (stdout, "key pressed: 0x%04x (%c)\n", key, (key >= ' ') && (key < 256) ? key : '.'));
        switch (key) {
        case ':':
            switch (waitforkey ()) {
            case ':':
                DoExit ();
                /* fallthrough */
            case 'f':
            case 'h':
                stop = 1;
                break;
            }
            break;
        case KEY_ESC:
#ifndef __PDCURSES__
            if (waitforkey () != KEY_ESC) {
                break;
            }
#endif /* __PDCURSES__ */
            /* fallthrough */
        case 'q':
            DoExit ();
            stop = 1;
            break;
        case 'i':
        case KEY_UP:
            skip = (skip > 0) ? skip - 1 : skip;
            break;
        case 'k':
        case KEY_DOWN:
            skip = (lines[skip + 1] != NULL) ? skip + 1 : skip;
            break;
#ifdef __PDCURSES__
        case ALT_F:
        case ALT_H:
            stop = 1;
            break;
#endif
        }
    }

    rmstatus();
}

void process (void)
{
    int xmax, ymax;
    WINDOW *wbody = bodywin ();
    getmaxyx (wbody, ymax, xmax);

    char *buffer = (char *)-1;
    static int is_stdin = 0;
    if (command) {
        buffer = exec_cmd (command);
    } else if (input) {
        buffer = load_file (input);
    } else if (!lines) {
        if (!is_stdin) {
            buffer = read_stdin ();
            is_stdin = 1;
        }
    }

    if (buffer == NULL) {
        char msg[MAX_LENGTH + 13 + 2 + 1] = {0};
        sprintf (msg, "%s: %s", command ? "can't execute" : input ? "can't load" : "can't read", command ? command : input ? input : "stdin");
        VERBOSE (WARNING, fprintf (stderr, "%s\n", msg));
        errormsg (msg);
        return;
    }

    free_lines (lines);
    lines = split_lines (buffer, xmax);
    free (buffer);
    display (lines, xmax, ymax, 0);

    if (second > 0) {
        VERBOSE (DEBUG, fprintf (stdout, "sleeping for %d''\n", second));
        sleep (second);
        cleanup ();
        exit (0);
    }
}

void follow (void)
{
    int xmax, ymax;
    WINDOW *wbody = bodywin ();
    getmaxyx (wbody, ymax, xmax);

    display (lines, xmax, ymax, 1);
}

void about (void)
{
    char *message = "CMore: a curses more\n \nThis program is a pager with a text user interface.\n- To move displayed text or in menu, use arrow keys or vim keys (i, j, k, l).\n- To access menus, use Alt-f of Alt-h.\n- To quit displayed text, press Escape or 'q'.";

    int xmax, ymax;
    WINDOW *wbody = bodywin ();
    getmaxyx (stdscr, ymax, xmax);

    wmove (stdscr, ymax / 4, xmax / 4);
    WINDOW *whelp = winputbox (stdscr, ymax / 2, xmax / 2);

    int i, j;
    for (i = 1; i < ymax / 2 - 1; i++) {
        char c[2] = {0};
        for (j = 1; j < xmax / 2 - 1; j++) {
            *c = *message++;
            if ((*c == '\n') && (j == 1)) {
                *c = *message++;
            }
            if ((*c == '\n') || (*c == '\0')) {
                break;
            }
            wmove (whelp, i, j);
            waddstr (whelp, c);
        }
        if (*c == '\0') {
            break;
        }
    }
    wrefresh (whelp);

    waitforkey ();

    delwin (whelp);
    touchwin (wbody);
    wrefresh (wbody);
}

void chcmd (void)
{
    char buffer[MAX_LENGTH + 1] = {0};
    char *desc[] = {"New command", NULL};
    char *buf[] = {buffer, NULL};

    if (command) {
        strncpy (buffer, command, MAX_LENGTH);
    }
    int c = getstrings (desc, buf, MAX_LENGTH);

    if ((c == '\n') && (strlen (buffer) > 0)) {
        free (command);
        free (input);
        command = strdup (buffer);
        input = NULL;
        process ();
    }
}

void chfile (void)
{
    char buffer[MAX_LENGTH + 1] = {0};
    char *desc[] = {"New file", NULL};
    char *buf[] = {buffer, NULL};

    if (input) {
        strncpy (buffer, input, MAX_LENGTH);
    }
    int c = getstrings (desc, buf, MAX_LENGTH);

    if ((c == '\n') && (strlen (buffer) > 0)) {
        free (command);
        free (input);
        command = NULL;
        input = strdup (buffer);
        process ();
    }
}

menu SubMenu0[] =
{
    { "Command", chcmd, "Change command"},
    { "File", chfile, "Change file"},
    { "Display", follow, "Continue"},
    { "Refresh", process, "Refresh"},
    { "Quit", DoExit, "Terminate program" },
    { "", (FUNC)0, "" }
};

void sub0 (void)
{
    domenu (SubMenu0);
}

menu SubMenu1[] =
{
    { "About", about, "About..." },
    { "", (FUNC)0, "" }
};

void sub1 (void)
{
    domenu (SubMenu1);
}

menu MainMenu[] =
{
    { "File", sub0, "File menu" },
    { "Help", sub1, "Help menu" },
    { "", (FUNC)0, "" }   /* always add this as the last item! */
};

/* help message */
int usage (int ret)
{
    FILE *fd = ret ? stderr : stdout;
    fprintf (fd, "usage: %s [-c command] [-f file] [-h] [-s sec] [-v]\n", progname);
    fprintf (fd, " -c: command\n");
    fprintf (fd, " -i: input file\n");
    fprintf (fd, " -h: help message\n");
    fprintf (fd, " -s: stoptime\n");
    fprintf (fd, " -v: verbose level (%d)\n", verbose);

    return ret;
}

/* main function */
int main (int argc, char *argv[])
{

    /* get basename */
    char *pt = progname = argv[0];
    while (*pt) {
        if ((*pt == '/') || (*pt == '\\')) {
           progname = pt + 1;
        }
        pt++;
    }

    /* process argument */
     while (argc-- > 1) {
        char *arg = *(++argv);
        if (arg[0] != '-') {
            VERBOSE (ERROR, fprintf (stderr, "%s: invalid option -- %s\n", progname, arg));
            return usage (1);
        }
        char c = arg[1];
        switch (c) {
        case 'c':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg) {
                free (command);
                free (input);
                command = strdup (arg);
                input = NULL;
            } else {
                VERBOSE (ERROR, fprintf (stderr, "%s: error for command\n", progname));
                return usage (1);
            }
            break;
        case 'f':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg) {
                free (command);
                free (input);
                command = NULL;
                input = strdup (arg);
            } else {
                VERBOSE (ERROR, fprintf (stderr, "%s: error for file\n", progname));
                return usage (1);
            }
            break;
        case 's':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg == NULL) {
                VERBOSE (ERROR, fprintf (stderr, "%s: missing number of second\n", progname));
                return usage (1);
            }
            second = atoi (arg);
            break;
        case 'v':
            arg = (arg[2]) ? arg + 2 : (--argc > 0) ? *(++argv) : NULL;
            if (arg == NULL) {
                VERBOSE (ERROR, fprintf (stderr, "%s: missing verbose level\n", progname));
                return usage (1);
            }
            verbose = atoi (arg);
            break;
        case 'h':
        default:
            return usage (c != 'h');
        }
    }

    char titre[256] = {0};
    sprintf (titre, "Application: %s", command ? command : input ? input : "stdin");

    startmenu (MainMenu, titre, process);

    return 0;
}

/* test: cmore.exe -c 2>&1 | grep error */
/* test: echo :fi | cmore.exe -v 1 -c notAnExistingCommand 2>&1 | grep can\'t */
/* test: cmore.exe -f 2>&1 | grep error */
/* test: echo :fi | cmore.exe -v 1 -f notAnExistingFile 2>&1 | grep can\'t */
/* test: cmore.exe -h | grep usage */
/* test: cmore.exe -Z 2>&1 | grep usage */
/* test: cmore.exe -v 2>&1 | grep missing */
/* test: cmore.exe -s 2>&1 | grep missing */
/* test: cmore.exe _ 2>&1 | grep invalid */
/* test: cat tui.c | cmore.exe -v 3 -s 2*/
/* test: (sleep 1; echo -n q) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo -ne '\e\e') | cmore.exe -f tui.c */
/* test: (sleep 1; echo -ne '\ex'; sleep 1; echo -ne '\e\e') | cmore.exe -f tui.c */
/* test: (sleep 1; echo -n :x; sleep 1; echo -n ::) | cmore.exe -f tui.c */
/* test: (sleep 1; echo -n kkkkk; sleep 1; echo -n i; sleep 1; echo -n q) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo -n q) | cmore.exe -c 'echo -n a; for i in $(seq 1 '$(( $COLUMNS*($LINES-4)-2 ))'); do echo -n .; done; echo -n z' -v 3 | grep -q 'last line too long' */
/* test: (sleep 1; echo -n q) | cmore.exe -c 'echo -n a; for i in $(seq 1 '$(( $COLUMNS*($LINES-4)-3 ))'); do echo -n .; done; echo -n z' -v 3 | grep -q 'last line too long'; test $? -eq 1 */
/* test: (sleep 1; echo :f; sleep 1; echo i) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fkkk; sleep 1; echo :fiii) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fjljjkkikkkk) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :flk; sleep 1; echo xji) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fllk; sleep 1; echo; sleep 1; echo :fi) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo -n :fjjkll; sleep 1; echo -n :; sleep 1; echo -ne '\e'; sleep 1; echo -n kq) | cmore.exe -c 'ip addr' */

/* test: (sleep 1; echo :fk; sleep 1; echo date; sleep 1; echo :fi) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; sleep 1; echo -ne '\e\e'; sleep 1; echo i) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; echo not_a_command; sleep 1; echo i) | cmore.exe -c 'ip addr' -v 3 2>&1 | grep -q "can't execute" */
/* test: (sleep 1; echo :fkk; sleep 1; echo; sleep 1; echo ii) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; sleep 1; echo -e '\elljll\eroute'; sleep 1; echo q) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; sleep 1; echo -e '\elljll\nroute'; sleep 1; echo q) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; sleep 1; echo -en '\ek'; sleep 1; echo i) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; sleep 1; echo -e '\eidate'; sleep 1; echo q) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fk; sleep 1; echo -e '\elll\e\t-j x\b'; sleep 1; echo q) | cmore.exe -c 'ip addr' */

/* test: (sleep 1; echo :fkk; sleep 1; echo tui.c; sleep 1; echo :fii) | cmore.exe -f tui.h */
/* test: (sleep 1; echo :fkk; sleep 1; echo cmore.c; sleep 1; echo :fii) | cmore.exe -c 'ip addr' */
/* test: (sleep 1; echo :fkk; echo not_a_file; sleep 1; echo ii) | cmore.exe -c 'ip addr' -v 3 2>&1 | grep -q "can't load" */
/* test: (sleep 1; echo :fkk; sleep 1; echo; sleep 1; echo ii) | cmore.exe -c 'ip addr' */

/* vim: set ts=4 sw=4 et: */