#include <stdio.h>
#include <check.h>
#include "str.h"
#include "node.h"
#include "token.h"

START_TEST (test_ltrim_slash)
{
    fail_if( strcmp( ltrim_slash("/blog") , "blog" ) != 0 );
    fail_if( strcmp( ltrim_slash("blog") , "blog" ) != 0 );
}
END_TEST

START_TEST (test_rnode_construct_uniq)
{
    rnode * n = rnode_create(10);

    rnode * child = rnode_create(3);

    // fail_if( rnode_add_child(n, strdup("/add") , child) != NULL );
    // fail_if( rnode_add_child(n, strdup("/add") , child) != NULL );

    rnode_free(n);
}
END_TEST

START_TEST (test_rnode_find_edge)
{
    rnode * n = rnode_create(10);

    rnode * child = rnode_create(3);

    fail_if( rnode_add_child(n, strdup("/add") , child) == FALSE );

    fail_if( rnode_find_edge(n, "/add") == NULL );
    fail_if( rnode_find_edge(n, "/bar") != NULL );

    rnode_free(n);
}
END_TEST


START_TEST (test_compile)
{
    token_array *t;
    rnode * n;
    n = rnode_create(10);



    rnode_insert_routel(n, "/zoo", strlen("/zoo") );
    rnode_insert_routel(n, "/foo", strlen("/foo") );
    rnode_insert_routel(n, "/bar", strlen("/bar") );
    rnode_compile(n);
    fail_if( n->combined_pattern );
    fail_if( NULL == rnode_find_edge_str(n, "/", strlen("/") ) );

    rnode_insert_routel(n, "/{id}", strlen("/{id}") );
    rnode_compile(n);
    rnode_dump(n, 0);
    fail_if(n->edges[0]->child->combined_pattern == NULL);

    redge *e = rnode_find_edge_str(n, "/", strlen("/") );
    fail_if( NULL == e );

    /*
    printf( "%s\n", e->pattern );
    printf( "%s\n", e->child->combined_pattern );
    printf( "%s\n", n->edges[0]->child->combined_pattern);
    printf( "%s\n", n->combined_pattern );
    */
    rnode *m = rnode_match( e->child , "foo", strlen("foo") );
    fail_if( NULL == m );

    m = rnode_match( n , "/foo", strlen("/foo") );
    fail_if( NULL == m );

    m = rnode_match( n , "/zoo", strlen("/zoo") );
    fail_if( NULL == m );

    m = rnode_match( n , "/bar", strlen("/bar") );
    fail_if( NULL == m );

    m = rnode_match( n , "/zzz", strlen("/zzz") );
    fail_if( NULL == m );
}
END_TEST

START_TEST (test_compile_slug)
{
    /*
    char * pattern;
    pattern = compile_slug("{id}", strlen("{id}"));
    ck_assert_str_eq( pattern, "([^/]+)" );
    free(pattern);

    pattern = compile_slug("/{id}", strlen("/{id}"));
    ck_assert_str_eq( pattern, "/([^/]+)" );
    free(pattern);

    pattern = compile_slug("-{id}", strlen("-{id}"));
    ck_assert_str_eq( pattern, "-([^-]+)" );
    free(pattern);

    pattern = compile_slug("{id}-{title}", strlen("{id}-{title}"));
    ck_assert_str_eq( pattern, "([^/]+)-([^-]+)" );
    free(pattern);


    pattern = compile_slug("{id:[a-z]+}", strlen("{id:[a-z]+}") );
    ck_assert_str_eq( pattern, "([a-z]+)" );
    free(pattern);


    pattern = compile_slug("/path/{id:[a-z]+}", strlen("/path/{id:[a-z]+}") );
    ck_assert_str_eq( pattern, "/path/([a-z]+)" );
    free(pattern);
    */

    /*
    char * p = malloc(sizeof(char) * 10);
    strncat(p, "foo", 3);
    free(p);
    */





}
END_TEST


START_TEST (test_rnode_insert_routel)
{
    rnode * n = rnode_create(10);

    // printf("Inserting /foo/bar\n");
    rnode_insert_routel(n, "/foo/bar", strlen("/foo/bar") );
    // rnode_dump(n, 0);

    // printf("Inserting /foo/zoo\n");
    rnode_insert_routel(n, "/foo/zoo", strlen("/foo/zoo") );
    // rnode_dump(n, 0);

    // printf("Inserting /f/id\n");
    rnode_insert_routel(n, "/f/id", strlen("/f/id") );
    // rnode_dump(n, 0);

    // printf("Inserting /post/{id}\n");
    rnode_insert_routel(n, "/post/{id}", strlen("/post/{id}") );
    // rnode_dump(n, 0);

    // printf("Inserting /post/{handle}\n");
    rnode_insert_routel(n, "/post/{handle}", strlen("/post/{handle}") );
    // rnode_dump(n, 0);

    // printf("Inserting /post/{handle}-{id}\n");
    rnode_insert_routel(n, "/post/{handle}-{id}", strlen("/post/{handle}-{id}") );
    rnode_compile(n);
    // 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 );

    rnode_free(n);
}
END_TEST




START_TEST (test_route_split)
{
    token_array *t;

    t = split_route_pattern("/blog", strlen("/blog") );
    fail_if( t == NULL );
    token_array_dump(t);
    token_array_free(t);

    t = split_route_pattern("/foo/{id}", strlen("/foo/{id}") );
    fail_if( t == NULL );
    token_array_dump(t);
    fail_if( t->len != 2 );
    token_array_free(t);

    t = split_route_pattern("/foo/bar/{id}", strlen("/foo/bar/{id}") );
    fail_if( t == NULL );
    token_array_dump(t);
    fail_if( t->len != 3 );
    token_array_free(t);

    t = split_route_pattern("/{title}", strlen("/{title}") );
    fail_if( t == NULL );
    token_array_dump(t);
    fail_if( t->len != 1 );
    token_array_free(t);

    t = split_route_pattern("/", strlen("/") );
    fail_if( t == NULL );
    token_array_dump(t);
    fail_if( t->len != 1 );
    token_array_free(t);

}
END_TEST

START_TEST (test_token_array)
{
    token_array * l = token_array_create(3);
    fail_if( l == NULL );

    fail_if( FALSE == token_array_append(l, strdup("abc") ) );
    fail_if( l->len != 1 );

    fail_if( FALSE == token_array_append(l, strdup("foo") ) );
    fail_if( l->len != 2 );

    fail_if( FALSE == token_array_append(l, strdup("bar") ) );
    fail_if( l->len != 3 );

    fail_if( FALSE == token_array_append(l, strdup("zoo") ) );
    fail_if( l->len != 4 );

    fail_if( FALSE == token_array_resize(l, l->cap * 2) );

    token_array_free(l);
}
END_TEST

Suite* r3_suite (void) {
        Suite *suite = suite_create("blah");

        TCase *tcase = tcase_create("testcase");
        tcase_add_test(tcase, test_route_split);
        tcase_add_test(tcase, test_token_array);
        tcase_add_test(tcase, test_ltrim_slash);
        tcase_add_test(tcase, test_rnode_construct_uniq);
        tcase_add_test(tcase, test_rnode_find_edge);
        tcase_add_test(tcase, test_rnode_insert_routel);
        tcase_add_test(tcase, test_compile_slug);
        tcase_add_test(tcase, test_compile);

        suite_add_tcase(suite, tcase);

        return suite;
}

int main (int argc, char *argv[]) {
        int number_failed;
        Suite *suite = r3_suite();
        SRunner *runner = srunner_create(suite);
        srunner_run_all(runner, CK_NORMAL);
        number_failed = srunner_ntests_failed(runner);
        srunner_free(runner);
        return number_failed;
}