X-Git-Url: http://git.droids-corp.org/?a=blobdiff_plain;f=lib%2Fecoli_node_sh_lex.c;h=e27f21bdb1135635c3a5a83aca54751fe2571dab;hb=15718f19cbf3868bf6fba79f4e26b5254c57873e;hp=53122c478362364a3439a899521a53bd7cfc7b01;hpb=044f2e33e83ec36df6601e3950820d07facde0a7;p=protos%2Flibecoli.git

diff --git a/lib/ecoli_node_sh_lex.c b/lib/ecoli_node_sh_lex.c
index 53122c4..e27f21b 100644
--- a/lib/ecoli_node_sh_lex.c
+++ b/lib/ecoli_node_sh_lex.c
@@ -343,12 +343,38 @@ static void ec_node_sh_lex_free_priv(struct ec_node *gen_node)
 	ec_node_free(node->child);
 }
 
+static size_t
+ec_node_sh_lex_get_children_count(const struct ec_node *gen_node)
+{
+	struct ec_node_sh_lex *node = (struct ec_node_sh_lex *)gen_node;
+
+	if (node->child)
+		return 1;
+	return 0;
+}
+
+static int
+ec_node_sh_lex_get_child(const struct ec_node *gen_node, size_t i,
+	struct ec_node **child, unsigned int *refs)
+{
+	struct ec_node_sh_lex *node = (struct ec_node_sh_lex *)gen_node;
+
+	if (i >= 1)
+		return -1;
+
+	*refs = 1;
+	*child = node->child;
+	return 0;
+}
+
 static struct ec_node_type ec_node_sh_lex_type = {
 	.name = "sh_lex",
 	.parse = ec_node_sh_lex_parse,
 	.complete = ec_node_sh_lex_complete,
 	.size = sizeof(struct ec_node_sh_lex),
 	.free_priv = ec_node_sh_lex_free_priv,
+	.get_children_count = ec_node_sh_lex_get_children_count,
+	.get_child = ec_node_sh_lex_get_child,
 };
 
 EC_NODE_TYPE_REGISTER(ec_node_sh_lex_type);
@@ -360,7 +386,7 @@ struct ec_node *ec_node_sh_lex(const char *id, struct ec_node *child)
 	if (child == NULL)
 		return NULL;
 
-	node = (struct ec_node_sh_lex *)__ec_node(&ec_node_sh_lex_type, id);
+	node = (struct ec_node_sh_lex *)ec_node_from_type(&ec_node_sh_lex_type, id);
 	if (node == NULL) {
 		ec_node_free(child);
 		return NULL;