Add readline completion for tincctl config and tincctl info.

This commit is contained in:
Guus Sliepen 2012-08-03 13:23:07 +02:00
parent 8af2f3f5a4
commit 36c6afede3
3 changed files with 90 additions and 1 deletions

View file

@ -33,7 +33,7 @@ void logger(int level, int priority, const char *format, ...) {
va_end(ap);
}
static char *strip_weight(char *netstr) {
char *strip_weight(char *netstr) {
int len = strlen(netstr);
if(len >= 3 && !strcmp(netstr + len - 3, "#10"))
netstr[len - 3] = 0;

View file

@ -21,6 +21,7 @@
#define __TINC_INFO_H__
extern int info(int fd, const char *item);
extern char *strip_weight(char *);
#endif

View file

@ -1752,6 +1752,88 @@ static char *complete_dump(const char *text, int state) {
return NULL;
}
static char *complete_config(const char *text, int state) {
const char *sub[] = {"get", "set", "add", "del"};
static int i;
if(!state) {
i = 0;
if(!strchr(rl_line_buffer + 7, ' '))
i = -4;
else {
bool found = false;
for(int i = 0; i < 4; i++) {
if(!strncasecmp(rl_line_buffer + 7, sub[i], strlen(sub[i])) && rl_line_buffer[7 + strlen(sub[i])] == ' ') {
found = true;
break;
}
}
if(!found)
return NULL;
}
} else {
i++;
}
while(i < 0 || variables[i].name) {
if(i < 0 && !strncasecmp(sub[i + 4], text, strlen(text)))
return xstrdup(sub[i + 4]);
if(i >= 0) {
char *dot = strchr(text, '.');
if(dot) {
if((variables[i].type & VAR_HOST) && !strncasecmp(variables[i].name, dot + 1, strlen(dot + 1))) {
char *match;
xasprintf(&match, "%.*s.%s", dot - text, text, variables[i].name);
return match;
}
} else {
if(!strncasecmp(variables[i].name, text, strlen(text)))
return xstrdup(variables[i].name);
}
}
i++;
}
return NULL;
}
static char *complete_info(const char *text, int state) {
static int i;
if(!state) {
i = 0;
if(!connect_tincd(false))
return NULL;
// Check the list of nodes
sendline(fd, "%d %d", CONTROL, REQ_DUMP_NODES);
sendline(fd, "%d %d", CONTROL, REQ_DUMP_SUBNETS);
}
while(recvline(fd, line, sizeof line)) {
char item[4096];
int n = sscanf(line, "%d %d %s", &code, &req, item);
if(n == 2) {
i++;
if(i >= 2)
break;
else
continue;
}
if(n != 3) {
fprintf(stderr, "Unable to parse dump from tincd, n = %d, i = %d.\n", n, i);
break;
}
if(!strncmp(item, text, strlen(text)))
return xstrdup(strip_weight(item));
}
return NULL;
}
static char *complete_nothing(const char *text, int state) {
return NULL;
}
static char **completion (const char *text, int start, int end) {
char **matches = NULL;
@ -1759,6 +1841,10 @@ static char **completion (const char *text, int start, int end) {
matches = rl_completion_matches(text, complete_command);
else if(!strncasecmp(rl_line_buffer, "dump ", 5))
matches = rl_completion_matches(text, complete_dump);
else if(!strncasecmp(rl_line_buffer, "config ", 7))
matches = rl_completion_matches(text, complete_config);
else if(!strncasecmp(rl_line_buffer, "info ", 5))
matches = rl_completion_matches(text, complete_info);
return matches;
}
@ -1779,6 +1865,7 @@ static int cmd_shell(int argc, char *argv[]) {
#ifdef HAVE_READLINE
rl_readline_name = "tinc";
rl_completion_entry_function = complete_nothing;
rl_attempted_completion_function = completion;
rl_filename_completion_desired = 0;
char *copy = NULL;
@ -1789,6 +1876,7 @@ static int cmd_shell(int argc, char *argv[]) {
if(tty) {
free(copy);
free(line);
rl_basic_word_break_characters = "\t\n ";
line = readline(prompt);
if(line)
copy = xstrdup(line);