use EC_NO_ID instead of NULL
[protos/libecoli.git] / lib / main-readline.c
index 6133c54..304b074 100644 (file)
 #define _GNU_SOURCE /* for asprintf */
 #include <stdlib.h>
 #include <stdio.h>
+#include <errno.h>
 #include <assert.h>
 
 #include <readline/readline.h>
 #include <readline/history.h>
 
+#include <ecoli_init.h>
 #include <ecoli_node.h>
 #include <ecoli_parsed.h>
 #include <ecoli_completed.h>
@@ -55,19 +57,21 @@ static char *my_completion_entry(const char *s, int state)
 {
        static struct ec_completed *c;
        static struct ec_completed_iter *iter;
-       static const struct ec_completed_item *item;
+       const struct ec_completed_item *item;
+       enum ec_completed_type item_type;
+       const char *item_str, *item_display;
 
        (void)s;
 
-       /* don't append a quote */
+       /* Don't append a quote. Note: there are still some bugs when
+        * completing a quoted token. */
        rl_completion_suppress_quote = 1;
-       rl_basic_quote_characters = "";
+       rl_completer_quote_characters = "\"'";
 
        if (state == 0) {
                char *line;
 
                ec_completed_free(c);
-
                line = strdup(rl_line_buffer);
                if (line == NULL)
                        return NULL;
@@ -79,7 +83,7 @@ static char *my_completion_entry(const char *s, int state)
                        return NULL;
 
                ec_completed_iter_free(iter);
-               iter = ec_completed_iter(c, EC_MATCH | EC_PARTIAL_MATCH);
+               iter = ec_completed_iter(c, EC_COMP_FULL | EC_COMP_PARTIAL);
                if (iter == NULL)
                        return NULL;
        }
@@ -88,18 +92,26 @@ static char *my_completion_entry(const char *s, int state)
        if (item == NULL)
                return NULL;
 
-       /* don't add the trailing space for partial completions */
-       if (state == 0) {
-               if (item->type == EC_MATCH)
-                       rl_completion_suppress_append = 0;
-               else
-                       rl_completion_suppress_append = 1;
+       item_str = ec_completed_item_get_str(item);
+       if (c->count_match == 1) {
+
+               /* don't add the trailing space for partial completions */
+               if (state == 0) {
+                       item_type = ec_completed_item_get_type(item);
+                       if (item_type == EC_COMP_FULL)
+                               rl_completion_suppress_append = 0;
+                       else
+                               rl_completion_suppress_append = 1;
+               }
+
+               return strdup(item_str);
+       } else if (rl_completion_type == '?') {
+               /* on second try only show the display part */
+               item_display = ec_completed_item_get_display(item);
+               return strdup(item_display);
        }
 
-       //XXX see printable_part() and rl_display_match_list()
-       //XXX or: if match count > 1, strip beginning (in node_file)
-
-       return strdup(item->str);
+       return strdup(item_str);
 }
 
 static char **my_attempted_completion(const char *text, int start, int end)
@@ -116,14 +128,19 @@ static char **my_attempted_completion(const char *text, int start, int end)
 /* this function builds the help string */
 static char *get_node_help(const struct ec_completed_item *item)
 {
+       const struct ec_completed_group *grp;
+       struct ec_parsed *state; // XXX keep const with macro
        const struct ec_node *node;
        char *help = NULL;
        const char *node_help = NULL;
        const char *node_desc = NULL;
-       size_t i;
 
-       for (i = 0; i < item->pathlen; i++) {
-               node = item->path[i];
+       grp = ec_completed_item_get_grp(item);
+       state = grp->state;
+       ec_parsed_dump(stdout, ec_parsed_get_root(state));
+       for (state = grp->state; state != NULL;
+            state = ec_parsed_get_parent(state)) {
+               node = ec_parsed_get_node(state);
                if (node_help == NULL)
                        node_help = ec_keyval_get(ec_node_attrs(node), "help");
                if (node_desc == NULL)
@@ -131,7 +148,7 @@ static char *get_node_help(const struct ec_completed_item *item)
        }
 
        if (node_help == NULL)
-               node_help = "";
+               node_help = "-";
        if (node_desc == NULL)
                return NULL;
 
@@ -143,13 +160,13 @@ static char *get_node_help(const struct ec_completed_item *item)
 
 static int show_help(int ignore, int invoking_key)
 {
-       const struct ec_completed_node *compnode;
-//     const struct ec_completed_item *item;
-//     struct ec_completed_iter *iter;
+       struct ec_completed_iter *iter;
+       const struct ec_completed_group *grp, *prev_grp = NULL;
+       const struct ec_completed_item *item;
        struct ec_completed *c;
        struct ec_parsed *p;
-       char *line;
-       unsigned int count, i;
+       char *line = NULL;
+       unsigned int count;
        char **helps = NULL;
        int match = 0;
 
@@ -158,80 +175,75 @@ static int show_help(int ignore, int invoking_key)
 
        line = strdup(rl_line_buffer);
        if (line == NULL)
-               return 1;
+               goto fail;
 
        /* check if the current line matches */
        p = ec_node_parse(commands, line);
        if (ec_parsed_matches(p))
                match = 1;
        ec_parsed_free(p);
+       p = NULL;
 
        /* complete at current cursor position */
        line[rl_point] = '\0';
        c = ec_node_complete(commands, line);
-       //ec_completed_dump(stdout, c);
        free(line);
+       line = NULL;
        if (c == NULL)
-               return 1;
-
-#if 0 // old method
-       count = ec_completed_count(c, EC_MATCH | EC_NO_MATCH |
-                               EC_PARTIAL_MATCH);
-       helps = calloc(count + match + 1, sizeof(char *));
-       if (helps == NULL)
-               return 1;
+               goto fail;
 
-       if (match)
-               helps[1] = "<return>";
+       ec_completed_dump(stdout, c);
 
-       iter = ec_completed_iter(c, EC_MATCH | EC_NO_MATCH |
-                               EC_PARTIAL_MATCH);
+       /* let's display one contextual help per node */
+       count = 0;
+       iter = ec_completed_iter(c,
+               EC_COMP_UNKNOWN | EC_COMP_FULL | EC_COMP_PARTIAL);
        if (iter == NULL)
                goto fail;
 
        /* strangely, rl_display_match_list() expects first index at 1 */
-       for (i = match + 1, item = ec_completed_iter_next(iter);
-            i < count + match + 1 && item != NULL;
-            i++, item = ec_completed_iter_next(iter)) {
-               helps[i] = get_node_help(item);
-       }
-#else
-       count = 0;
-       TAILQ_FOREACH(compnode, &c->nodes, next) {
-               if (TAILQ_EMPTY(&compnode->items))
-                       continue;
-               count++;
-       }
-
-       helps = calloc(count + match + 1, sizeof(char *));
+       helps = calloc(match + 1, sizeof(char *));
        if (helps == NULL)
-               return 1;
-
+               goto fail;
        if (match)
                helps[1] = "<return>";
 
-       /* strangely, rl_display_match_list() expects first index at 1 */
-       i = match + 1;
-       TAILQ_FOREACH(compnode, &c->nodes, next) {
-               if (TAILQ_EMPTY(&compnode->items))
+       while ((item = ec_completed_iter_next(iter)) != NULL) {
+               char **tmp;
+
+               /* keep one help per group, skip other items  */
+               grp = ec_completed_item_get_grp(item);
+               if (grp == prev_grp)
                        continue;
-               // we should pass compnode instead
-               helps[i++] = get_node_help(TAILQ_FIRST(&compnode->items)); //XXX
+
+               prev_grp = grp;
+
+               tmp = realloc(helps, (count + match + 2) * sizeof(char *));
+               if (tmp == NULL)
+                       goto fail;
+               helps = tmp;
+               helps[count + match + 1] = get_node_help(item);
+               count++;
        }
-#endif
 
-       ec_completed_dump(stdout, c);
+       ec_completed_iter_free(iter);
        ec_completed_free(c);
-
        rl_display_match_list(helps, count + match, 1000); /* XXX 1000 */
-
        rl_forced_update_display();
 
        return 0;
 
-//fail:
+fail:
+       ec_completed_iter_free(iter);
+       ec_parsed_free(p);
+       free(line);
+       ec_completed_free(c);
+       if (helps != NULL) {
+               while (count--)
+                       free(helps[count + match + 1]);
+       }
        free(helps);
-       // free helps[n] XXX
+
        return 1;
 }
 
@@ -239,19 +251,19 @@ static int create_commands(void)
 {
        struct ec_node *cmdlist = NULL, *cmd = NULL;
 
-       cmdlist = ec_node("or", NULL);
+       cmdlist = ec_node("or", EC_NO_ID);
        if (cmdlist == NULL)
                goto fail;
 
 
-       cmd = EC_NODE_SEQ(NULL,
-               ec_node_str(NULL, "hello"),
+       cmd = EC_NODE_SEQ(EC_NO_ID,
+               ec_node_str(EC_NO_ID, "hello"),
                EC_NODE_OR("name",
                        ec_node_str("john", "john"),
-                       ec_node_str(NULL, "johnny"),
-                       ec_node_str(NULL, "mike")
+                       ec_node_str(EC_NO_ID, "johnny"),
+                       ec_node_str(EC_NO_ID, "mike")
                ),
-               ec_node_option(NULL, ec_node_int("int", 0, 10, 10))
+               ec_node_option(EC_NO_ID, ec_node_int("int", 0, 10, 10))
        );
        if (cmd == NULL)
                goto fail;
@@ -262,12 +274,12 @@ static int create_commands(void)
        ec_keyval_set(ec_node_attrs(ec_node_find(cmd, "name")),
                "help", "the name of the person", NULL);
        ec_keyval_set(ec_node_attrs(ec_node_find(cmd, "int")),
-               "help", "an integer", NULL);
+               "help", "an integer (0-10)", NULL);
        if (ec_node_or_add(cmdlist, cmd) < 0)
                goto fail;
 
 
-       cmd = EC_NODE_CMD(NULL, "good morning name [count]",
+       cmd = EC_NODE_CMD(EC_NO_ID, "good morning name [count]",
                        EC_NODE_CMD("name", "bob|bobby|michael"),
                        ec_node_int("count", 0, 10, 10));
        if (cmd == NULL)
@@ -277,12 +289,12 @@ static int create_commands(void)
        ec_keyval_set(ec_node_attrs(ec_node_find(cmd, "name")), "help",
                "the person to greet", NULL);
        ec_keyval_set(ec_node_attrs(ec_node_find(cmd, "count")), "help",
-               "how many times to greet", NULL);
+               "how many times to greet (0-10)", NULL);
        if (ec_node_or_add(cmdlist, cmd) < 0)
                goto fail;
 
 
-       cmd = EC_NODE_CMD(NULL,
+       cmd = EC_NODE_CMD(EC_NO_ID,
                        "buy potatoes,carrots,pumpkins");
        if (cmd == NULL)
                goto fail;
@@ -292,14 +304,14 @@ static int create_commands(void)
                goto fail;
 
 
-       cmd = EC_NODE_CMD(NULL, "eat vegetables",
+       cmd = EC_NODE_CMD(EC_NO_ID, "eat vegetables",
                        ec_node_many("vegetables",
-                               EC_NODE_OR(NULL,
-                                       ec_node_str(NULL, "potatoes"),
-                                       ec_node_once(NULL,
-                                               ec_node_str(NULL, "carrots")),
-                                       ec_node_once(NULL,
-                                               ec_node_str(NULL, "pumpkins"))),
+                               EC_NODE_OR(EC_NO_ID,
+                                       ec_node_str(EC_NO_ID, "potatoes"),
+                                       ec_node_once(EC_NO_ID,
+                                               ec_node_str(EC_NO_ID, "carrots")),
+                                       ec_node_once(EC_NO_ID,
+                                               ec_node_str(EC_NO_ID, "pumpkins"))),
                        1, 0));
        if (cmd == NULL)
                goto fail;
@@ -309,24 +321,24 @@ static int create_commands(void)
                goto fail;
 
 
-       cmd = EC_NODE_SEQ(NULL,
-               ec_node_str(NULL, "bye")
+       cmd = EC_NODE_SEQ(EC_NO_ID,
+               ec_node_str(EC_NO_ID, "bye")
        );
        ec_keyval_set(ec_node_attrs(cmd), "help", "say bye", NULL);
        if (ec_node_or_add(cmdlist, cmd) < 0)
                goto fail;
 
 
-       cmd = EC_NODE_SEQ(NULL,
-               ec_node_str(NULL, "load"),
-               ec_node("file", NULL)
+       cmd = EC_NODE_SEQ(EC_NO_ID,
+               ec_node_str(EC_NO_ID, "load"),
+               ec_node("file", EC_NO_ID)
        );
        ec_keyval_set(ec_node_attrs(cmd), "help", "load a file", NULL);
        if (ec_node_or_add(cmdlist, cmd) < 0)
                goto fail;
 
 
-       commands = ec_node_sh_lex(NULL, cmdlist);
+       commands = ec_node_sh_lex(EC_NO_ID, cmdlist);
        if (commands == NULL)
                goto fail;
 
@@ -343,6 +355,11 @@ int main(void)
        struct ec_parsed *p;
        char *line;
 
+       if (ec_init() < 0) {
+               fprintf(stderr, "cannot init ecoli: %s\n", strerror(errno));
+               return 1;
+       }
+
        if (create_commands() < 0)
                return 1;