if (completed == NULL)
return NULL;
- TAILQ_INIT(&completed->elts);
- completed->count_match = 0;
+ TAILQ_INIT(&completed->nodes);
return completed;
}
-struct ec_completed *
+/* XXX on error, states are not freed ?
+ * they can be left in a bad state and should not be reused */
+int
ec_node_complete_child(struct ec_node *node,
- struct ec_parsed *state,
+ struct ec_completed *completed,
+ struct ec_parsed *parsed_state,
const struct ec_strvec *strvec)
{
- struct ec_completed *completed;
- struct ec_parsed *child;
+ struct ec_parsed *child_state = NULL;
int ret;
/* build the node if required */
if (node->type->build != NULL) {
if ((node->flags & EC_NODE_F_BUILT) == 0) {
ret = node->type->build(node);
- if (ret < 0) {
- errno = -ret;
- return NULL;
- }
+ if (ret < 0)
+ return ret;
}
}
node->flags |= EC_NODE_F_BUILT;
- if (node->type->complete == NULL) {
- errno = ENOTSUP;
- return NULL;
- }
+ if (node->type->complete == NULL)
+ return -ENOTSUP;
- child = ec_parsed();
- if (child == NULL)
- return NULL;
+ child_state = ec_parsed();
+ if (child_state == NULL)
+ return -ENOMEM;
+ child_state->node = node;
+ ec_parsed_add_child(parsed_state, child_state);
- child->node = node;
- ec_parsed_add_child(state, child);
- completed = node->type->complete(node, child, strvec);
+ ret = ec_completed_add_node(completed, node);
+ if (ret < 0)
+ return ret;
+
+ ret = node->type->complete(node, completed,
+ child_state, strvec);
+ if (ret < 0)
+ return ret;
#if 0 // XXX dump
printf("----------------------------------------------------------\n");
ec_node_dump(stdout, node);
ec_strvec_dump(stdout, strvec);
ec_completed_dump(stdout, completed);
- ec_parsed_dump(stdout, state);
+ ec_parsed_dump(stdout, parsed_state);
#endif
- ec_parsed_del_child(state, child);
- assert(TAILQ_EMPTY(&child->children));
- ec_parsed_free(child);
+ ec_parsed_del_child(parsed_state, child_state);
+ assert(TAILQ_EMPTY(&child_state->children));
+ ec_parsed_free(child_state);
- return completed;
+ return 0;
}
struct ec_completed *ec_node_complete_strvec(struct ec_node *node,
const struct ec_strvec *strvec)
{
- struct ec_parsed *state = ec_parsed();
- struct ec_completed *completed;
+ struct ec_parsed *parsed_state = NULL;
+ struct ec_completed *completed = NULL;
+ int ret;
- if (state == NULL)
- return NULL;
+ parsed_state = ec_parsed();
+ if (parsed_state == NULL)
+ goto fail;
- completed = ec_node_complete_child(node, state, strvec);
- ec_parsed_free(state);
+ completed = ec_completed();
+ if (completed == NULL)
+ goto fail;
+
+ ret = ec_node_complete_child(node, completed,
+ parsed_state, strvec);
+ if (ret < 0)
+ goto fail;
+
+ ec_parsed_free(parsed_state);
return completed;
+
+fail:
+ ec_parsed_free(parsed_state);
+ ec_completed_free(completed);
+ return NULL;
}
struct ec_completed *ec_node_complete(struct ec_node *node,
return NULL;
}
-/* default completion function: return a no-match element */
-struct ec_completed *ec_node_default_complete(const struct ec_node *gen_node,
- struct ec_parsed *state,
- const struct ec_strvec *strvec)
-{
- struct ec_completed *completed;
-
- (void)strvec;
- (void)state;
-
- completed = ec_completed();
- if (completed == NULL)
- return NULL;
-
- if (ec_strvec_len(strvec) != 1)
- return completed;
-
- if (ec_completed_add_elt(completed, state, gen_node, NULL) < 0) {
- ec_completed_free(completed);
- return NULL;
- }
-
- return completed;
-}
-
/* count the number of identical chars at the beginning of 2 strings */
static size_t strcmp_count(const char *s1, const char *s2)
{
return i;
}
-static struct ec_completed_elt *
-ec_completed_elt(struct ec_parsed *parsed,
- const struct ec_node *node, const char *add)
+static struct ec_completed_node *
+ec_completed_node(const struct ec_node *node)
{
- struct ec_completed_elt *elt = NULL;
+ struct ec_completed_node *compnode = NULL;
- elt = ec_calloc(1, sizeof(*elt));
- if (elt == NULL)
+ compnode = ec_calloc(1, sizeof(*compnode));
+ if (compnode == NULL)
return NULL;
- if (parsed != NULL) {
- struct ec_parsed *p;
- size_t len;
+ compnode->node = node;
+ TAILQ_INIT(&compnode->items);
- /* get path len */
- for (p = parsed, len = 0; p != NULL;
- p = ec_parsed_get_parent(p), len++)
- ;
+ return compnode;
+}
- elt->path = ec_calloc(len, sizeof(*elt->path));
- if (elt->path == NULL)
- goto fail;
+static struct ec_completed_item *
+ec_completed_item(struct ec_parsed *state, const struct ec_node *node)
+{
+ struct ec_completed_item *item = NULL;
+ struct ec_parsed *p;
+ size_t len;
- elt->pathlen = len;
+ item = ec_calloc(1, sizeof(*item));
+ if (item == NULL)
+ return NULL;
- /* write path in array */
- for (p = parsed, len = 0; p != NULL;
- p = ec_parsed_get_parent(p), len++)
- elt->path[len] = p->node;
- }
+ /* get path len */
+ for (p = state, len = 0; p != NULL;
+ p = ec_parsed_get_parent(p), len++)
+ ;
+ /* allocate room for path */
+ item->path = ec_calloc(len, sizeof(*item->path));
+ if (item->path == NULL)
+ goto fail;
+ item->pathlen = len;
+ /* write path in array */
+ for (p = state, len = 0; p != NULL;
+ p = ec_parsed_get_parent(p), len++)
+ item->path[len] = p->node;
- elt->node = node;
- if (add != NULL) {
- elt->add = ec_strdup(add);
- if (elt->add == NULL)
- goto fail;
- }
+ item->type = EC_NO_MATCH;
+ item->node = node;
- return elt;
+ return item;
fail:
- if (elt != NULL) {
- ec_free(elt->path);
- ec_free(elt->add);
+ if (item != NULL) {
+ ec_free(item->path);
+ ec_free(item->str);
}
- ec_completed_elt_free(elt);
+ ec_completed_item_free(item);
return NULL;
}
-static int __ec_completed_add_elt(struct ec_completed *completed,
- struct ec_completed_elt *elt)
+static int
+__ec_completed_add_match(enum ec_completed_type type,
+ struct ec_completed *completed,
+ struct ec_parsed *parsed_state,
+ const struct ec_node *node, const char *str)
{
- size_t n;
+ struct ec_completed_node *compnode = NULL;
+ struct ec_completed_item *item = NULL;
+ int ret = -ENOMEM;
- TAILQ_INSERT_TAIL(&completed->elts, elt, next);
- completed->count++;
- if (elt->add != NULL) {
+ /* find the compnode entry corresponding to this node */
+ TAILQ_FOREACH(compnode, &completed->nodes, next) {
+ if (compnode->node == node)
+ break;
+ }
+ if (compnode == NULL)
+ return -ENOENT;
+
+ item = ec_completed_item(parsed_state, node);
+ if (item == NULL)
+ goto fail;
+ item->type = type;
+ if (str != NULL) {
+ item->str = ec_strdup(str);
+ if (item->str == NULL)
+ goto fail;
+ }
+
+ if (item->str != NULL)
completed->count_match++;
- if (completed->smallest_start == NULL) {
- completed->smallest_start = ec_strdup(elt->add);
- } else {
- n = strcmp_count(elt->add,
- completed->smallest_start);
- completed->smallest_start[n] = '\0';
- }
+
+ TAILQ_INSERT_TAIL(&compnode->items, item, next);
+ completed->count++;
+
+ return 0;
+
+fail:
+ ec_completed_item_free(item);
+ return ret;
+}
+
+int
+ec_completed_add_match(struct ec_completed *completed,
+ struct ec_parsed *parsed_state,
+ const struct ec_node *node, const char *str)
+{
+ struct ec_completed_node *compnode = NULL;
+ struct ec_completed_item *item = NULL;
+ int ret = -ENOMEM;
+
+ /* find the compnode entry corresponding to this node */
+ TAILQ_FOREACH(compnode, &completed->nodes, next) {
+ if (compnode->node == node)
+ break;
}
+ if (compnode == NULL)
+ return -ENOENT;
+
+ item = ec_completed_item(parsed_state, node);
+ if (item == NULL)
+ goto fail;
+ item->type = EC_MATCH;
+ if (str != NULL) {
+ item->str = ec_strdup(str);
+ if (item->str == NULL)
+ goto fail;
+ }
+
+ if (item->str != NULL)
+ completed->count_match++;
+
+ TAILQ_INSERT_TAIL(&compnode->items, item, next);
+ completed->count++;
return 0;
+
+fail:
+ ec_completed_item_free(item);
+ return ret;
+}
+
+int
+ec_completed_add_no_match(struct ec_completed *completed,
+ struct ec_parsed *parsed_state,
+ const struct ec_node *node)
+{
+ return __ec_completed_add_match(EC_NO_MATCH, completed, parsed_state,
+ node, NULL);
}
-int ec_completed_add_elt(struct ec_completed *completed,
- struct ec_parsed *parsed,
- const struct ec_node *node, const char *add)
+int
+ec_completed_add_partial_match(struct ec_completed *completed,
+ struct ec_parsed *parsed_state,
+ const struct ec_node *node, const char *str)
{
- struct ec_completed_elt *elt;
+ return __ec_completed_add_match(EC_PARTIAL_MATCH, completed, parsed_state,
+ node, str);
+}
- elt = ec_completed_elt(parsed, node, add);
- if (elt == NULL)
+int
+ec_completed_add_node(struct ec_completed *completed,
+ const struct ec_node *node)
+{
+ struct ec_completed_node *compnode = NULL;
+
+ compnode = ec_completed_node(node);
+ if (compnode == NULL)
return -ENOMEM;
- return __ec_completed_add_elt(completed, elt);
+ TAILQ_INSERT_TAIL(&completed->nodes, compnode, next);
+ return 0;
}
-void ec_completed_elt_free(struct ec_completed_elt *elt)
+void ec_completed_item_free(struct ec_completed_item *match)
{
- ec_free(elt->add);
- ec_free(elt->path);
- ec_free(elt);
+ ec_free(match->str);
+ ec_free(match->path);
+ ec_free(match);
}
-void ec_completed_merge(struct ec_completed *completed1,
- struct ec_completed *completed2)
+/* default completion function: return a no-match element */
+int
+ec_node_default_complete(const struct ec_node *gen_node,
+ struct ec_completed *completed,
+ struct ec_parsed *parsed_state,
+ const struct ec_strvec *strvec)
{
- struct ec_completed_elt *elt;
+ int ret;
- assert(completed1 != NULL);
- assert(completed2 != NULL);
+ if (ec_strvec_len(strvec) != 1)
+ return 0;
- while (!TAILQ_EMPTY(&completed2->elts)) {
- elt = TAILQ_FIRST(&completed2->elts);
- TAILQ_REMOVE(&completed2->elts, elt, next);
- __ec_completed_add_elt(completed1, elt);
- }
+ ret = ec_completed_add_no_match(completed, parsed_state, gen_node);
+ if (ret < 0)
+ return ret;
- ec_completed_free(completed2);
+ return 0;
}
void ec_completed_free(struct ec_completed *completed)
{
- struct ec_completed_elt *elt;
+ struct ec_completed_node *compnode;
+ struct ec_completed_item *item;
if (completed == NULL)
return;
- while (!TAILQ_EMPTY(&completed->elts)) {
- elt = TAILQ_FIRST(&completed->elts);
- TAILQ_REMOVE(&completed->elts, elt, next);
- ec_completed_elt_free(elt);
+ while (!TAILQ_EMPTY(&completed->nodes)) {
+ compnode = TAILQ_FIRST(&completed->nodes);
+ TAILQ_REMOVE(&completed->nodes, compnode, next);
+
+ while (!TAILQ_EMPTY(&compnode->items)) {
+ item = TAILQ_FIRST(&compnode->items);
+ TAILQ_REMOVE(&compnode->items, item, next);
+ ec_completed_item_free(item);
+ }
+ ec_free(compnode);
}
- ec_free(completed->smallest_start);
ec_free(completed);
}
void ec_completed_dump(FILE *out, const struct ec_completed *completed)
{
- struct ec_completed_elt *elt;
+ struct ec_completed_node *compnode;
+ struct ec_completed_item *item;
if (completed == NULL || completed->count == 0) {
fprintf(out, "no completion\n");
return;
}
- fprintf(out, "completion: count=%u match=%u smallest_start=<%s>\n",
- completed->count, completed->count_match,
- completed->smallest_start);
+ fprintf(out, "completion: count=%u match=%u\n",
+ completed->count, completed->count_match);
+
+ TAILQ_FOREACH(compnode, &completed->nodes, next) {
+ fprintf(out, "node=%p, node_type=%s\n",
+ compnode->node, compnode->node->type->name);
+ TAILQ_FOREACH(item, &compnode->items, next) {
+ const char *typestr;
+
+ switch (item->type) {
+ case EC_NO_MATCH: typestr = "no-match"; break;
+ case EC_MATCH: typestr = "match"; break;
+ case EC_PARTIAL_MATCH: typestr = "partial-match"; break;
+ default: typestr = "unknown"; break;
+ }
- TAILQ_FOREACH(elt, &completed->elts, next) {
- fprintf(out, "add=<%s>, node=%p, node_type=%s\n",
- elt->add, elt->node, elt->node->type->name);
+ fprintf(out, " type=%s str=<%s>\n", typestr, item->str);
+ }
}
}
-const char *ec_completed_smallest_start(
- const struct ec_completed *completed)
+char *ec_completed_smallest_start(const struct ec_completed *completed)
{
- if (completed == NULL || completed->smallest_start == NULL)
- return "";
+ struct ec_completed_node *compnode;
+ struct ec_completed_item *item;
+ char *smallest_start = NULL;
+ size_t n;
+
+ if (completed == NULL)
+ goto fail;
+
+ TAILQ_FOREACH(compnode, &completed->nodes, next) {
+ TAILQ_FOREACH(item, &compnode->items, next) {
+ if (item->str == NULL)
+ continue;
+ if (smallest_start == NULL) {
+ smallest_start = ec_strdup(item->str);
+ if (smallest_start == NULL)
+ goto fail;
+ } else {
+ n = strcmp_count(item->str, smallest_start);
+ smallest_start[n] = '\0';
+ }
+ }
+ }
+
+ return smallest_start;
- return completed->smallest_start;
+fail:
+ ec_free(smallest_start);
+ return NULL;
}
unsigned int ec_completed_count(
const struct ec_completed *completed,
- enum ec_completed_filter_flags flags)
+ enum ec_completed_type type)
{
unsigned int count = 0;
if (completed == NULL)
return count;
- if (flags & EC_MATCH)
+ if (type & EC_MATCH)
count += completed->count_match;
- if (flags & EC_NO_MATCH)
+ if (type & EC_NO_MATCH)
count += (completed->count - completed->count_match); //XXX
return count;
struct ec_completed_iter *
ec_completed_iter(struct ec_completed *completed,
- enum ec_completed_filter_flags flags)
+ enum ec_completed_type type)
{
struct ec_completed_iter *iter;
return NULL;
iter->completed = completed;
- iter->flags = flags;
- iter->cur = NULL;
+ iter->type = type;
+ iter->cur_node = NULL;
+ iter->cur_match = NULL;
return iter;
}
-const struct ec_completed_elt *ec_completed_iter_next(
+const struct ec_completed_item *ec_completed_iter_next(
struct ec_completed_iter *iter)
{
- if (iter->completed == NULL)
- return NULL;
-
- do {
- if (iter->cur == NULL)
- iter->cur = TAILQ_FIRST(&iter->completed->elts);
- else
- iter->cur = TAILQ_NEXT(iter->cur, next);
+ const struct ec_completed *completed = iter->completed;
+ const struct ec_completed_node *cur_node;
+ const struct ec_completed_item *cur_match;
- if (iter->cur == NULL)
- break;
+ if (completed == NULL)
+ return NULL;
- if (iter->cur->add == NULL &&
- (iter->flags & EC_NO_MATCH))
- break;
+ cur_node = iter->cur_node;
+ cur_match = iter->cur_match;
- if (iter->cur->add != NULL &&
- (iter->flags & EC_MATCH))
- break;
+ /* first call */
+ if (cur_node == NULL) {
+ TAILQ_FOREACH(cur_node, &completed->nodes, next) {
+ TAILQ_FOREACH(cur_match, &cur_node->items, next) {
+ if (cur_match != NULL &&
+ cur_match->type & iter->type)
+ goto found;
+ }
+ }
+ return NULL;
+ } else {
+ cur_match = TAILQ_NEXT(cur_match, next);
+ if (cur_match != NULL &&
+ cur_match->type & iter->type)
+ goto found;
+ cur_node = TAILQ_NEXT(cur_node, next);
+ while (cur_node != NULL) {
+ cur_match = TAILQ_FIRST(&cur_node->items);
+ if (cur_match != NULL &&
+ cur_match->type & iter->type)
+ goto found;
+ cur_node = TAILQ_NEXT(cur_node, next);
+ }
+ return NULL;
+ }
- } while (iter->cur != NULL);
+found:
+ iter->cur_node = cur_node;
+ iter->cur_match = cur_match;
- return iter->cur;
+ return iter->cur_match;
}
void ec_completed_iter_free(struct ec_completed_iter *iter)