1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#include <stdlib.h>
#include <stdio.h>
#include <stdbool.h>
#include <stdarg.h>
#include <readline/readline.h>
#include <readline/history.h>
#include "rl.h"
#include "cmd.h"
#include "parse.h"
void rl_printf(const char *fmt, ...) {
// save line
char* saved_line = rl_copy_text(0, rl_end);
int saved_point = rl_point;
int saved_end = rl_end;
// clear line
rl_save_prompt();
rl_replace_line("", 0);
rl_redisplay();
// printf
va_list args;
va_start(args, fmt);
vprintf(fmt, args);
va_end(args);
// restore line
rl_restore_prompt();
rl_replace_line(saved_line, 0);
rl_point = saved_point;
rl_end = saved_end;
rl_redisplay();
free(saved_line);
}
static void cli_cmd(char* cmd) {
cmd += strspn(cmd, IFS); // skip leading whitespace
char* line = consume_token(cmd, IFS);
for (size_t i = 0; i < cmds_length; i++) {
if (strncmp(cmds[i].name, cmd, strlen(cmd)) != 0)
continue;
cmds[i].handle(line);
return;
}
printf("unknown command!\n");
}
static char* rl_completion_entries(const char *text, int state) {
static size_t i = 0;
if (state == 0) i = 0;
while (i < cmds_length) {
struct cmd cmd = cmds[i];
i++;
if (strncmp(text, cmd.name, strlen(text)) == 0) {
return strdup(cmd.name);
}
}
return NULL;
}
static char** rl_attempted_completion(const char * text, int start, int end) {
// do not suggest filenames
rl_attempted_completion_over = 1;
// if first word in line buffer -> complete commands from cmds[]
size_t cmd_start = strspn(rl_line_buffer, IFS);
if (start == cmd_start)
return rl_completion_matches(text, rl_completion_entries);
// else, check specialized completion functions
size_t cmd_len = strcspn(rl_line_buffer + cmd_start, IFS);
for (size_t i = 0; i < cmds_length; i++) {
cmd_t cmd = cmds[i];
if (cmd.complete == NULL) continue;
if (strncmp(cmd.name, rl_line_buffer + cmd_start, cmd_len) != 0) continue;
return cmd.complete(rl_line_buffer, start, end);
}
// else, no completion available
return NULL;
}
int cli_main() {
char* input = NULL;
rl_attempted_completion_function = rl_attempted_completion;
while (1) {
if (input != NULL) free(input);
input = readline(CLI_PROMPT);
if (input == NULL) return EXIT_SUCCESS; // exit on ^D (EOF)
if (*input == '\0') continue; // ignore empty lines
add_history(input);
cli_cmd(input);
}
return EXIT_SUCCESS;
}
|