lib/main-readline.c

   1 /*
   2  * Copyright (c) 2016, Olivier MATZ <zer0@droids-corp.org>
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above copyright
  10  *       notice, this list of conditions and the following disclaimer in the
  11  *       documentation and/or other materials provided with the distribution.
  12  *     * Neither the name of the University of California, Berkeley nor the
  13  *       names of its contributors may be used to endorse or promote products
  14  *       derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #define _GNU_SOURCE /* for asprintf */
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <assert.h>
  32
  33 #include <readline/readline.h>
  34 #include <readline/history.h>
  35
  36 #include <ecoli_node.h>
  37 #include <ecoli_parsed.h>
  38 #include <ecoli_completed.h>
  39 #include <ecoli_keyval.h>
  40 #include <ecoli_node_str.h>
  41 #include <ecoli_node_seq.h>
  42 #include <ecoli_node_space.h>
  43 #include <ecoli_node_or.h>
  44 #include <ecoli_node_sh_lex.h>
  45 #include <ecoli_node_int.h>
  46 #include <ecoli_node_option.h>
  47 #include <ecoli_node_cmd.h>
  48
  49 static struct ec_node *commands;
  50
  51 static char *my_completion_entry(const char *s, int state)
  52 {
  53         static struct ec_completed *c;
  54         static struct ec_completed_iter *iter;
  55         static const struct ec_completed_elt *elt;
  56         char *out_string;
  57
  58
  59         if (state == 0) {
  60                 char *line;
  61
  62                 ec_completed_free(c);
  63
  64                 line = strdup(rl_line_buffer);
  65                 if (line == NULL)
  66                         return NULL;
  67                 line[rl_point] = '\0';
  68
  69                 c = ec_node_complete(commands, line);
  70                 free(line);
  71                 if (c == NULL)
  72                         return NULL;
  73
  74                 ec_completed_iter_free(iter);
  75                 iter = ec_completed_iter(c, EC_MATCH);
  76                 if (iter == NULL)
  77                         return NULL;
  78         }
  79
  80         elt = ec_completed_iter_next(iter);
  81         if (elt == NULL)
  82                 return NULL;
  83
  84         if (asprintf(&out_string, "%s%s", s, elt->add) < 0)
  85                 return NULL;
  86
  87         return out_string;
  88 }
  89
  90 static char **my_attempted_completion(const char *text, int start, int end)
  91 {
  92         (void)start;
  93         (void)end;
  94
  95         /* remove default file completion */
  96         rl_attempted_completion_over = 1;
  97
  98         return rl_completion_matches(text, my_completion_entry);
  99 }
 100
 101 /* this function builds the help string */
 102 static char *get_tk_help(const struct ec_node *node)
 103 {
 104         const struct ec_node *node2;
 105         char *help = NULL;
 106         char *tk_help = NULL;
 107
 108         for (node2 = node;
 109              node2 != NULL && tk_help == NULL;
 110              node2 = ec_node_parent(node2))
 111                 tk_help = ec_keyval_get(ec_node_attrs(node2), "help");
 112
 113         if (tk_help == NULL)
 114                 tk_help = "";
 115
 116         if (asprintf(&help, "%-20s %s", ec_node_desc(node), tk_help) < 0)
 117                 return NULL;
 118
 119         return help;
 120 }
 121
 122 static int show_help(int ignore, int invoking_key)
 123 {
 124         const struct ec_completed_elt *elt;
 125         struct ec_completed_iter *iter;
 126         struct ec_completed *c;
 127         char *line;
 128         unsigned int count, i;
 129         char **helps = NULL;
 130
 131         (void)ignore;
 132         (void)invoking_key;
 133
 134         line = strdup(rl_line_buffer);
 135         if (line == NULL)
 136                 return 1;
 137         line[rl_point] = '\0';
 138
 139         c = ec_node_complete(commands, line);
 140         free(line);
 141         if (c == NULL)
 142                 return 1;
 143         //ec_completed_dump(stdout, c);
 144
 145         count = ec_completed_count(c, EC_MATCH | EC_NO_MATCH);
 146         helps = calloc(count + 1, sizeof(char *));
 147         if (helps == NULL)
 148                 return 1;
 149
 150         iter = ec_completed_iter(c, EC_MATCH | EC_NO_MATCH);
 151         if (iter == NULL)
 152                 goto fail;
 153
 154         /* strangely, rl_display_match_list() expects first index at 1 */
 155         for (i = 1, elt = ec_completed_iter_next(iter);
 156              i <= count && elt != NULL;
 157              i++, elt = ec_completed_iter_next(iter)) {
 158                 helps[i] = get_tk_help(elt->node);
 159         }
 160
 161         ec_completed_free(c);
 162
 163         rl_display_match_list(helps, count, 1000);
 164
 165         rl_forced_update_display();
 166
 167         return 0;
 168
 169 fail:
 170         free(helps);
 171         // free helps[n] XXX
 172         return 1;
 173 }
 174
 175 static int create_commands(void)
 176 {
 177         struct ec_node *cmdlist = NULL, *cmd = NULL;
 178
 179         cmdlist = ec_node("or", NULL);
 180         if (cmdlist == NULL)
 181                 goto fail;
 182
 183         cmd = EC_NODE_SEQ(NULL,
 184                 ec_node_str(NULL, "hello"),
 185                 EC_NODE_OR("name",
 186                         ec_node_str(NULL, "john"),
 187                         ec_node_str(NULL, "johnny"),
 188                         ec_node_str(NULL, "mike")
 189                 ),
 190                 ec_node_option(NULL, ec_node_int("int", 0, 10, 10))
 191         );
 192         if (cmd == NULL)
 193                 goto fail;
 194         ec_keyval_set(ec_node_attrs(cmd), "help",
 195                 "say hello to someone several times", NULL);
 196         ec_keyval_set(ec_node_attrs(ec_node_find(cmd, "name")),
 197                 "help", "the name of the person", NULL);
 198         ec_keyval_set(ec_node_attrs(ec_node_find(cmd, "int")),
 199                 "help", "an integer", NULL);
 200         if (ec_node_or_add(cmdlist, cmd) < 0)
 201                 goto fail;
 202
 203 #if 0
 204         cmd = EC_NODE_CMD(NULL, "good morning john|johnny|mike [count]",
 205                         ec_node_int("count", 0, 10, 10));
 206         if (cmd == NULL)
 207                 goto fail;
 208         ec_keyval_set(ec_node_attrs(cmd), "help",
 209                 "say good morning to someone several times", NULL);
 210         if (ec_node_or_add(cmdlist, cmd) < 0)
 211                 goto fail;
 212 #endif
 213
 214         cmd = EC_NODE_SEQ(NULL,
 215                 ec_node_str(NULL, "bye")
 216         );
 217         ec_keyval_set(ec_node_attrs(cmd), "help", "say bye to someone", NULL);
 218         if (ec_node_or_add(cmdlist, cmd) < 0)
 219                 goto fail;
 220
 221         commands = ec_node_sh_lex(NULL, cmdlist);
 222         if (commands == NULL)
 223                 goto fail;
 224
 225         return 0;
 226
 227  fail:
 228         fprintf(stderr, "cannot initialize nodes\n");
 229         ec_node_free(cmdlist);
 230         return -1;
 231 }
 232
 233 int main(void)
 234 {
 235         struct ec_parsed *p;
 236 //      const char *name;
 237         char *line;
 238
 239
 240         if (create_commands() < 0)
 241                 return 1;
 242
 243         rl_bind_key('?', show_help);
 244         rl_attempted_completion_function = my_attempted_completion;
 245
 246         while (1) {
 247                 line = readline("> ");
 248                 if (line == NULL)
 249                         break;
 250
 251                 p = ec_node_parse(commands, line);
 252                 ec_parsed_dump(stdout, p);
 253                 add_history(line);
 254                 ec_parsed_free(p);
 255         }
 256
 257
 258         ec_node_free(commands);
 259         return 0;
 260
 261 }