From 7326933065cb9f800d25d6c88dd3f4cb0f470d48 Mon Sep 17 00:00:00 2001
From: c9s <yoanlin93@gmail.com>
Date: Thu, 15 May 2014 18:26:41 +0800
Subject: [PATCH] supporting endpoint

---
 include/node.h    |  4 ++++
 src/node.c        | 45 ++++++++++++++++++++++++++++++++++++++++++++-
 tests/test_tree.c |  7 +++++++
 3 files changed, 55 insertions(+), 1 deletion(-)

diff --git a/include/node.h b/include/node.h
index b5c80e8..1535fd6 100644
--- a/include/node.h
+++ b/include/node.h
@@ -37,6 +37,10 @@ void rnode_append_edge(rnode *n, redge *child);
 
 rnode * rnode_insert_tokens(rnode * tree, token_array * tokens);
 
+rnode * rnode_insert_route(rnode *tree, char *route);
+
+rnode * rnode_insert_routel(rnode *tree, char *route, int route_len);
+
 void rnode_dump(rnode * n, int level);
 
 redge * redge_create(char * pattern, int pattern_len, rnode * child);
diff --git a/src/node.c b/src/node.c
index 42459db..d26b0ea 100644
--- a/src/node.c
+++ b/src/node.c
@@ -12,6 +12,7 @@
 // Judy array
 #include <Judy.h>
 
+#include "str.h"
 #include "node.h"
 #include "token.h"
 
@@ -22,6 +23,8 @@ struct _rnode {
 
     /* the combined regexp pattern string from pattern_tokens */
     char* combined_pattern;
+
+    int      endpoint;
 };
 
 struct _redge {
@@ -32,12 +35,18 @@ struct _redge {
 
 // String value as the index http://judy.sourceforge.net/doc/JudySL_3x.htm
 
+
+
+/**
+ * Create a rnode object
+ */
 rnode * rnode_create(int cap) {
     rnode * n = (rnode*) calloc( sizeof(rnode) , 1);
 
     n->children = (redge**) malloc( sizeof(redge*) * 10 );
     n->children_len = 0;
     n->children_cap = 10;
+    n->endpoint = 0;
     // n->edge_patterns = token_array_create(10);
     return n;
 }
@@ -94,6 +103,28 @@ redge * rnode_find_edge(rnode * n, char * pat) {
     return NULL;
 }
 
+
+rnode * rnode_lookup(rnode * tree, char * path, int path_len) {
+    token_array * tokens = split_route_pattern(path, path_len);
+
+    rnode * n = tree;
+    redge * e = NULL;
+    for ( int i = 0 ; i < tokens->len ; i++ ) {
+        e = rnode_find_edge(n, token_array_fetch(tokens, i) );
+        if (!e) {
+            return NULL;
+        }
+        n = e->child;
+    }
+    if (n->endpoint) {
+        return n;
+    }
+    return NULL;
+}
+
+
+
+
 rnode * rnode_insert_tokens(rnode * tree, token_array * tokens) {
     rnode * n = tree;
     redge * e = NULL;
@@ -105,12 +136,24 @@ rnode * rnode_insert_tokens(rnode * tree, token_array * tokens) {
         }
         // insert node
         rnode * child = rnode_create(3);
-        rnode_add_child(n, token_array_fetch(tokens,i) , child);
+        rnode_add_child(n, strdup(token_array_fetch(tokens,i)) , child);
         n = child;
     }
+    n->endpoint++;
     return n;
 }
 
+rnode * rnode_insert_route(rnode *tree, char *route)
+{
+    return rnode_insert_routel(tree, route, strlen(route) );
+}
+
+rnode * rnode_insert_routel(rnode *tree, char *route, int route_len)
+{
+    token_array * t = split_route_pattern(route, strlen(route));
+    return rnode_insert_tokens(tree, t);
+}
+
 void rnode_dump(rnode * n, int level) {
     if ( n->children_len ) {
         print_indent(level);
diff --git a/tests/test_tree.c b/tests/test_tree.c
index d0511c8..eb89713 100644
--- a/tests/test_tree.c
+++ b/tests/test_tree.c
@@ -66,6 +66,13 @@ START_TEST (test_rnode_insert_tokens)
     rnode_dump(n, 0);
 
 
+    fail_if( rnode_lookup(n, "/a/jj/kk", strlen("/a/jj/kk") ) == NULL );
+    fail_if( rnode_lookup(n, "/a/jj", strlen("/a/jj") ) != NULL );
+    fail_if( rnode_lookup(n, "/a/jj/kk/ll", strlen("/a/jj/kk/ll") ) != NULL );
+
+    fail_if( rnode_lookup(n, "/xxxx", strlen("xxxx") ) != NULL );
+
+
     // fail_if( rnode_find_edge(n, "/add") == NULL );
     // fail_if( rnode_find_edge(n, "/bar") != NULL );