Allow comments in interactive input.
[adu.git] / interactive.c
1 #include <ctype.h> /* isspace() */
2
3 #include "adu.h"
4 #include "format.h"
5 #include "select.h"
6 #include "string.h"
7 #include "error.h"
8 #include "cmdline.h"
9
10 struct interactive_command {
11 const char *name;
12 int (*handler)(char *);
13 const char *desc;
14 };
15
16 static struct uid_range *admissible_uids;
17 static struct format_info *fi;
18
19 #define INTERACTIVE_COMMANDS \
20 INTERACTIVE_COMMAND(set, "change the current configuration") \
21 INTERACTIVE_COMMAND(reset, "reset configuration to defaults") \
22 INTERACTIVE_COMMAND(help, "show list of commands and one-line descriptions") \
23 INTERACTIVE_COMMAND(run, "start the query according to the current configuration")
24
25
26 #define INTERACTIVE_COMMAND(name, desc) \
27 static int icom_ ## name (char *line);
28
29 INTERACTIVE_COMMANDS
30
31 #undef INTERACTIVE_COMMAND
32
33 #define INTERACTIVE_COMMAND(_name, _desc) \
34 { \
35 .name = #_name, \
36 .handler = icom_ ## _name, \
37 .desc = _desc \
38 },
39
40 struct interactive_command icmds[] = {
41 INTERACTIVE_COMMANDS
42 {.name = NULL}
43 };
44
45 #define FOR_EACH_COMMAND(c) for (c = icmds; c->name; c++)
46
47 static int read_input_line(char *line, size_t size)
48 {
49 return fgets(line, size, stdin)? 1 : -1;
50 }
51
52 static int icom_run(__a_unused char *line)
53 {
54 return run_select_query(admissible_uids, fi);
55 }
56
57 static int icom_help(__a_unused char *line)
58 {
59 struct interactive_command *c;
60
61 FOR_EACH_COMMAND(c)
62 fprintf(stdout, "%s\t%s\n", c->name, c->desc);
63 return 1;
64 }
65
66 void print_interactive_help(void)
67 {
68 icom_help(NULL);
69 }
70
71 static int icom_reset(__a_unused char *line)
72 {
73 free_format_info(fi);
74 fi = NULL;
75 free(admissible_uids);
76 admissible_uids = NULL;
77 select_cmdline_parser_init(&select_conf);
78 return 1;
79 }
80
81 static int icom_set(char *line)
82 {
83 struct select_cmdline_parser_params params = {
84 .override = 1,
85 .initialize = 0,
86 .check_required = 1,
87 .check_ambiguity = 0,
88 .print_errors = 1
89 };
90 if (!line) {
91 select_cmdline_parser_dump(stdout, &select_conf);
92 return 1;
93 }
94
95 free_format_info(fi);
96 fi = NULL;
97 free(admissible_uids);
98 admissible_uids = NULL;
99 return parse_select_options(line, &params, &admissible_uids, &fi);
100 }
101
102 static int exec_interactive_command(char *line)
103 {
104 const char const *delim = "\t\n\f\r\v ";
105 int i;
106 char *cmd, *args;
107 int ret = -E_SYNTAX;
108 size_t len;
109
110 if (!line || !*line)
111 return 1;
112 len = strlen(line);
113
114 while (len && isspace(line[len - 1])) {
115 line[len - 1] = '\0';
116 len--;
117 }
118 if (!len)
119 return 1;
120 line += strspn(line, delim); /* skip initial whitespace */
121 if (!*line)
122 return 1;
123 /* OK, we have a non-empty line */
124 if (*line == '#')
125 return 1;
126 cmd = adu_strdup(line);
127 args = cmd + strcspn(cmd, delim);
128 if (!*args)
129 args = NULL;
130 else {
131 *args = '\0';
132 args++;
133 /* let p point to the next non-whitespace char */
134 args += strspn(args, delim);
135 if (!*args)
136 args = NULL;
137 }
138 DEBUG_LOG("name: %s, args: %s.\n", cmd, args);
139 for (i = 0; icmds[i].name; i++) {
140 if (strcmp(icmds[i].name, cmd))
141 continue;
142 INFO_LOG("exec cmd: %s, args: %s\n", cmd, args);
143 ret = icmds[i].handler(args);
144 break;
145 }
146 free(cmd);
147 return ret;
148 }
149
150 int com_interactive(void)
151 {
152 char line[255];
153 int ret = 1;
154
155 select_cmdline_parser_init(&select_conf);
156 ret = parse_select_options(NULL, NULL, &admissible_uids, &fi);
157 while (read_input_line(line, sizeof(line)) >= 0) {
158 ret = exec_interactive_command(line);
159 if (ret < 0)
160 printf("%s\n", adu_strerror(-ret));
161 }
162 return ret;
163 }