Completion works. Unfortunately, we have to access a couple of internal

symbols of libreadline :-(

3 files changed, 193 insertions, 20 deletions

diff --git a/client/client.c b/client/client.c
index b9e3ab8..9c6b4ae 100644
--- a/client/client.c
+++ b/client/client.c
@@ -71,6 +71,11 @@ parse_args(int argc, char **argv)
 static void server_send(char *);
 static void io_loop(int);
 
+/* HACK: libreadline internals we need to access */
+extern int _rl_vis_botlin;
+extern void _rl_move_vert(int);
+extern Function *rl_last_func;
+
 static void
 got_line(char *cmd_buffer)
 {
@@ -91,10 +96,40 @@ got_line(char *cmd_buffer)
   free(cmd_buffer);
 }
 
+void
+input_start_list(void)			/* Leave the currently edited line and make space for listing */
+{
+  _rl_move_vert(_rl_vis_botlin);
+  crlf();
+}
+
+void
+input_stop_list(void)			/* Reprint the currently edited line after listing */
+{
+  rl_on_new_line();
+  rl_redisplay();
+}
+
 static int
 input_complete(int arg, int key)
 {
-  ding();
+  static int complete_flag;
+  char buf[256];
+
+  if (rl_last_func != input_complete)
+    complete_flag = 0;
+  switch (cmd_complete(rl_line_buffer, rl_point, buf, complete_flag))
+    {
+    case 0:
+      complete_flag = 1;
+      break;
+    case 1:
+      rl_insert_text(buf);
+      break;
+    default:
+      complete_flag = 1;
+      ding();
+    }
   return 0;
 }
 
@@ -103,22 +138,26 @@ input_help(int arg, int key)
 {
   int i = 0;
 
-  if (rl_point != rl_end || arg != 1)
+  if (arg != 1)
     return rl_insert(arg, '?');
-  while (i < rl_end)
+  while (i < rl_point)
     {
       if (rl_line_buffer[i++] == '"')
 	do
 	  {
-	    if (i >= rl_end)		/* `?' inside quoted string -> insert */
+	    if (i >= rl_point)		/* `?' inside quoted string -> insert */
 	      return rl_insert(1, '?');
 	  }
         while (rl_line_buffer[i++] != '"');
     }
-  puts("?");
-  cmd_help(rl_line_buffer, rl_end);
-  rl_on_new_line();
+  rl_begin_undo_group();		/* HACK: We want to display `?' at point position */
+  rl_insert_text("?");
   rl_redisplay();
+  rl_end_undo_group();
+  input_start_list();
+  cmd_help(rl_line_buffer, rl_point);
+  rl_undo_command(1, 0);
+  input_stop_list();
   return 0;
 }
 
diff --git a/client/client.h b/client/client.h
index a9c9903..f63c75c 100644
--- a/client/client.h
+++ b/client/client.h
@@ -9,8 +9,11 @@
 /* client.c */
 
 void cleanup(void);
+void input_start_list(void);
+void input_stop_list(void);
 
 /* commands.c */
 
 void cmd_build_tree(void);
 void cmd_help(char *cmd, int len);
+int cmd_complete(char *cmd, int len, char *buf, int again);
diff --git a/client/commands.c b/client/commands.c
index ceb036b..09c5c34 100644
--- a/client/commands.c
+++ b/client/commands.c
@@ -7,6 +7,8 @@
  */
 
 #include <stdio.h>
+#include <string.h>
+#include <ctype.h>
 
 #include "nest/bird.h"
 #include "lib/resource.h"
@@ -27,7 +29,7 @@ static struct cmd_info command_table[] = {
 
 struct cmd_node {
   struct cmd_node *sibling, *son, **plastson;
-  struct cmd_info *cmd;
+  struct cmd_info *cmd, *help;
   int len;
   char token[1];
 };
@@ -58,26 +60,28 @@ cmd_build_tree(void)
 	      break;
 	  if (!new)
 	    {
-	      new = xmalloc(sizeof(struct cmd_node) + c-d);
+	      int size = sizeof(struct cmd_node) + c-d;
+	      new = xmalloc(size);
+	      bzero(new, size);
 	      *old->plastson = new;
 	      old->plastson = &new->sibling;
-	      new->sibling = new->son = NULL;
 	      new->plastson = &new->son;
-	      new->cmd = NULL;
 	      new->len = c-d;
 	      memcpy(new->token, d, c-d);
-	      new->token[c-d] = 0;
 	    }
 	  old = new;
 	  while (*c == ' ')
 	    c++;
 	}
-      old->cmd = cmd;
+      if (cmd->is_real_cmd)
+	old->cmd = cmd;
+      else
+	old->help = cmd;
     }
 }
 
 static void
-cmd_display_help(struct cmd_info *c)
+cmd_do_display_help(struct cmd_info *c)
 {
   char buf[strlen(c->command) + strlen(c->args) + 4];
 
@@ -85,11 +89,21 @@ cmd_display_help(struct cmd_info *c)
   printf("%-45s  %s\n", buf, c->help);
 }
 
+static void
+cmd_display_help(struct cmd_info *c1, struct cmd_info *c2)
+{
+  if (c1)
+    cmd_do_display_help(c1);
+  else if (c2)
+    cmd_do_display_help(c2);
+}
+
 static struct cmd_node *
-cmd_find_abbrev(struct cmd_node *root, char *cmd, int len)
+cmd_find_abbrev(struct cmd_node *root, char *cmd, int len, int *pambiguous)
 {
   struct cmd_node *m, *best = NULL, *best2 = NULL;
 
+  *pambiguous = 0;
   for(m=root->son; m; m=m->sibling)
     {
       if (m->len == len && !memcmp(m->token, cmd, len))
@@ -100,7 +114,22 @@ cmd_find_abbrev(struct cmd_node *root, char *cmd, int len)
 	  best = m;
 	}
     }
-  return best2 ? NULL : best;
+  if (best2)
+    {
+      *pambiguous = 1;
+      return NULL;
+    }
+  return best;
+}
+
+static void
+cmd_list_ambiguous(struct cmd_node *root, char *cmd, int len)
+{
+  struct cmd_node *m;
+
+  for(m=root->son; m; m=m->sibling)
+    if (m->len > len && !memcmp(m->token, cmd, len))	
+      cmd_display_help(m->help, m->cmd);
 }
 
 void
@@ -109,6 +138,7 @@ cmd_help(char *cmd, int len)
   char *end = cmd + len;
   struct cmd_node *n, *m;
   char *z;
+  int ambig;
 
   n = &cmd_root;
   while (cmd < end)
@@ -121,13 +151,114 @@ cmd_help(char *cmd, int len)
       z = cmd;
       while (cmd < end && *cmd != ' ' && *cmd != '\t')
 	cmd++;
-      m = cmd_find_abbrev(n, z, cmd-z);
+      m = cmd_find_abbrev(n, z, cmd-z, &ambig);
+      if (ambig)
+	{
+	  cmd_list_ambiguous(n, z, cmd-z);
+	  return;
+	}
       if (!m)
 	break;
       n = m;
     }
-  if (n->cmd && n->cmd->is_real_cmd)
-    cmd_display_help(n->cmd);
+  cmd_display_help(n->cmd, NULL);
   for (m=n->son; m; m=m->sibling)
-    cmd_display_help(m->cmd);
+    cmd_display_help(m->help, m->cmd);
+}
+
+static int
+cmd_find_common_match(struct cmd_node *root, char *cmd, int len, int *pcount, char *buf)
+{
+  struct cmd_node *m;
+  int best, i;
+
+  *pcount = 0;
+  best = -1;
+  for(m=root->son; m; m=m->sibling)
+    {
+      if (m->len < len || memcmp(m->token, cmd, len))
+	continue;
+      (*pcount)++;
+      if (best < 0)
+	{
+	  strcpy(buf, m->token + len);
+	  best = m->len - len;
+	}
+      else
+	{
+	  i = 0;
+	  while (i < best && i < m->len - len && buf[i] == m->token[len+i])
+	    i++;
+	  best = i;
+	}
+    }
+  return best;
+}
+
+int
+cmd_complete(char *cmd, int len, char *buf, int again)
+{
+  char *start = cmd;
+  char *end = cmd + len;
+  char *fin;
+  struct cmd_node *n, *m;
+  char *z;
+  int ambig, cnt = 0, common;
+
+  /* Find the last word we want to complete */
+  for(fin=end; fin > start && !isspace(fin[-1]); fin--)
+    ;
+
+  /* Find the context */
+  n = &cmd_root;
+  while (cmd < fin && n->son)
+    {
+      if (*cmd == ' ' || *cmd == '\t')
+	{
+	  cmd++;
+	  continue;
+	}
+      z = cmd;
+      while (cmd < fin && !isspace(*cmd))
+	cmd++;
+      m = cmd_find_abbrev(n, z, cmd-z, &ambig);
+      if (ambig)
+	{
+	  if (!again)
+	    return -1;
+	  input_start_list();
+	  cmd_list_ambiguous(n, z, cmd-z);
+	  input_stop_list();
+	  return 0;
+	}
+      if (!m)
+	return -1;
+      n = m;
+    }
+
+  /* Completion of parameters is not yet supported */
+  if (!n->son)
+    return -1;
+
+  /* We know the context, let's try to complete */
+  common = cmd_find_common_match(n, fin, end-fin, &cnt, buf);
+  if (!cnt)
+    return -1;
+  if (cnt == 1)
+    {
+      buf[common++] = ' ';
+      buf[common] = 0;
+      return 1;
+    }
+  if (common > 0)
+    {
+      buf[common] = 0;
+      return 1;
+    }
+  if (!again)
+    return -1;
+  input_start_list();
+  cmd_list_ambiguous(n, fin, end-fin);
+  input_stop_list();
+  return 0;
 }