2 * Copyright (c) 2016, Olivier MATZ <zer0@droids-corp.org>
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
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.
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.
28 #define _GNU_SOURCE /* for asprintf */
29 #include <sys/types.h>
38 #include <ecoli_log.h>
39 #include <ecoli_malloc.h>
40 #include <ecoli_test.h>
41 #include <ecoli_strvec.h>
42 #include <ecoli_string.h>
43 #include <ecoli_node.h>
44 #include <ecoli_parsed.h>
45 #include <ecoli_completed.h>
46 #include <ecoli_node_file.h>
53 ec_node_file_parse(const struct ec_node *gen_node,
54 struct ec_parsed *state,
55 const struct ec_strvec *strvec)
60 if (ec_strvec_len(strvec) == 0)
61 return EC_PARSED_NOMATCH;
67 * Almost the same than dirname (3) and basename (3) except that:
68 * - it always returns a substring of the given path, which can
70 * - the behavior is different when the path finishes with a '/'
71 * - the path argument is not modified
72 * - the outputs are allocated and must be freed with ec_free().
74 * path dirname basename split_path
75 * /usr/lib /usr lib /usr/ lib
82 static int split_path(const char *path, char **dname_p, char **bname_p)
91 last_slash = strrchr(path, '/');
92 if (last_slash == NULL)
95 dirlen = last_slash - path + 1;
97 dname = ec_strdup(path);
100 dname[dirlen] = '\0';
102 bname = ec_strdup(path + dirlen);
115 ec_node_file_complete(const struct ec_node *gen_node,
116 struct ec_completed *completed,
117 struct ec_parsed *state,
118 const struct ec_strvec *strvec)
120 struct ec_completed_item *item = NULL;
124 struct dirent *de = NULL;
126 char *dname = NULL, *bname = NULL, *effective_dir;
127 char *comp_str = NULL;
128 char *disp_str = NULL;
133 * Example with this file tree:
144 * Input Output completions
145 * / [dir1/, dir2/, file5]
148 * /dir1/ [file1, file2, subdir/]
154 if (ec_strvec_len(strvec) != 1)
157 input = ec_strvec_val(strvec, 0);
158 ret = split_path(input, &dname, &bname);
160 ec_completed_free(completed);
165 if (strcmp(dname, "") == 0)
168 effective_dir = dname;
170 ret = lstat(effective_dir, &st);
175 if (!S_ISDIR(st.st_mode))
178 dir = opendir(effective_dir);
182 bname_len = strlen(bname);
188 if (!ec_str_startswith(de->d_name, bname))
190 if (bname[0] != '.' && de->d_name[0] == '.')
193 /* add '/' if it's a dir */
194 if (de->d_type == DT_DIR) {
196 } else if (de->d_type == DT_UNKNOWN) { // XXX todo
201 item = ec_completed_item(state, gen_node);
208 if (asprintf(&comp_str, "%s%s/", input,
209 &de->d_name[bname_len]) < 0) {
213 if (asprintf(&disp_str, "%s/", de->d_name) < 0) {
217 ret = ec_completed_item_set(item, EC_PARTIAL_MATCH,
222 if (asprintf(&comp_str, "%s%s", input,
223 &de->d_name[bname_len]) < 0) {
227 if (asprintf(&disp_str, "%s", de->d_name) < 0) {
231 ret = ec_completed_item_set(item, EC_MATCH,
236 ret = ec_completed_item_set_display(item, disp_str);
239 ret = ec_completed_item_add(completed, item);
252 ec_completed_item_free(item);
263 static struct ec_node_type ec_node_file_type = {
265 .parse = ec_node_file_parse,
266 .complete = ec_node_file_complete,
267 .size = sizeof(struct ec_node_file),
270 EC_NODE_TYPE_REGISTER(ec_node_file_type);
272 /* LCOV_EXCL_START */
273 static int ec_node_file_testcase(void)
275 struct ec_node *node;
278 node = ec_node("file", NULL);
280 ec_log(EC_LOG_ERR, "cannot create node\n");
283 /* any string matches */
284 ret |= EC_TEST_CHECK_PARSE(node, 1, "foo");
285 ret |= EC_TEST_CHECK_PARSE(node, 1, "/tmp/bar");
286 ret |= EC_TEST_CHECK_PARSE(node, -1);
288 /* test completion */
289 #if 0 // XXX how to properly test file completion?
290 ret |= EC_TEST_CHECK_COMPLETE(node,
293 ret |= EC_TEST_CHECK_COMPLETE(node,
296 ret |= EC_TEST_CHECK_COMPLETE(node,
297 "/", EC_NODE_ENDLIST,
299 ret |= EC_TEST_CHECK_COMPLETE(node,
300 "/tmp", EC_NODE_ENDLIST,
302 ret |= EC_TEST_CHECK_COMPLETE(node,
303 "/tmp/", EC_NODE_ENDLIST,
305 ret |= EC_TEST_CHECK_COMPLETE(node,
306 "/tmp/.", EC_NODE_ENDLIST,
315 static struct ec_test ec_node_file_test = {
317 .test = ec_node_file_testcase,
320 EC_TEST_REGISTER(ec_node_file_test);