module vweb

struct RoutePair {
    url   string
    route string
}

fn (rp RoutePair) test() ?[]string {
    url := rp.url.split('/').filter(it != '')
    route := rp.route.split('/').filter(it != '')
    return route_matches(url, route)
}

fn (rp RoutePair) test_match() {
    rp.test() or { panic('should match: ${rp}') }
}

fn (rp RoutePair) test_no_match() {
    rp.test() or { return }
    panic('should not match: ${rp}')
}

fn (rp RoutePair) test_param(expected []string) {
    res := rp.test() or { panic('should match: ${rp}') }
    assert res == expected
}

fn test_route_no_match() {
    tests := [
        RoutePair{
            url: '/a'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/b'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/b/'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/c/b'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/c/b/'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/b/c/d'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/b/c'
            route: '/'
        },
    ]
    for test in tests {
        test.test_no_match()
    }
}

fn test_route_exact_match() {
    tests := [
        RoutePair{
            url: '/a/b/c'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a/b/c/'
            route: '/a/b/c'
        },
        RoutePair{
            url: '/a'
            route: '/a'
        },
        RoutePair{
            url: '/'
            route: '/'
        },
    ]
    for test in tests {
        test.test_match()
    }
}

fn test_route_params_match() {
    RoutePair{
        url: '/a/b/c'
        route: '/:a/b/c'
    }.test_match()

    RoutePair{
        url: '/a/b/c'
        route: '/a/:b/c'
    }.test_match()

    RoutePair{
        url: '/a/b/c'
        route: '/a/b/:c'
    }.test_match()

    RoutePair{
        url: '/a/b/c'
        route: '/:a/b/:c'
    }.test_match()

    RoutePair{
        url: '/a/b/c'
        route: '/:a/:b/:c'
    }.test_match()

    RoutePair{
        url: '/one/two/three'
        route: '/:a/:b/:c'
    }.test_match()

    RoutePair{
        url: '/one/b/c'
        route: '/:a/b/c'
    }.test_match()

    RoutePair{
        url: '/one/two/three'
        route: '/:a/b/c'
    }.test_no_match()

    RoutePair{
        url: '/one/two/three'
        route: '/:a/:b/c'
    }.test_no_match()

    RoutePair{
        url: '/one/two/three'
        route: '/:a/b/:c'
    }.test_no_match()

    RoutePair{
        url: '/a/b/c/d'
        route: '/:a/:b/:c'
    }.test_no_match()

    RoutePair{
        url: '/1/2/3/4'
        route: '/:a/:b/:c'
    }.test_no_match()

    RoutePair{
        url: '/a/b'
        route: '/:a/:b/:c'
    }.test_no_match()

    RoutePair{
        url: '/1/2'
        route: '/:a/:b/:c'
    }.test_no_match()
}

fn test_route_params() {
    RoutePair{
        url: '/a/b/c'
        route: '/:a/b/c'
    }.test_param(['a'])

    RoutePair{
        url: '/one/b/c'
        route: '/:a/b/c'
    }.test_param(['one'])

    RoutePair{
        url: '/one/two/c'
        route: '/:a/:b/c'
    }.test_param(['one', 'two'])

    RoutePair{
        url: '/one/two/three'
        route: '/:a/:b/:c'
    }.test_param(['one', 'two', 'three'])

    RoutePair{
        url: '/one/b/three'
        route: '/:a/b/:c'
    }.test_param(['one', 'three'])
}

fn test_route_params_array_match() {
    // array can only be used on the last word (TODO: add parsing / tests to ensure this)

    RoutePair{
        url: '/a/b/c'
        route: '/a/b/:c...'
    }.test_match()

    RoutePair{
        url: '/a/b/c/d'
        route: '/a/b/:c...'
    }.test_match()

    RoutePair{
        url: '/a/b/c/d/e'
        route: '/a/b/:c...'
    }.test_match()

    RoutePair{
        url: '/one/b/c/d/e'
        route: '/:a/b/:c...'
    }.test_match()

    RoutePair{
        url: '/one/two/c/d/e'
        route: '/:a/:b/:c...'
    }.test_match()

    RoutePair{
        url: '/one/two/three/four/five'
        route: '/:a/:b/:c...'
    }.test_match()

    RoutePair{
        url: '/a/b'
        route: '/:a/:b/:c...'
    }.test_no_match()

    RoutePair{
        url: '/a/b/'
        route: '/:a/:b/:c...'
    }.test_no_match()
}

fn test_route_params_array() {
    RoutePair{
        url: '/a/b/c'
        route: '/a/b/:c...'
    }.test_param(['c'])

    RoutePair{
        url: '/a/b/c/d'
        route: '/a/b/:c...'
    }.test_param(['c/d'])

    RoutePair{
        url: '/a/b/c/d/'
        route: '/a/b/:c...'
    }.test_param(['c/d'])

    RoutePair{
        url: '/a/b/c/d/e'
        route: '/a/b/:c...'
    }.test_param(['c/d/e'])

    RoutePair{
        url: '/one/b/c/d/e'
        route: '/:a/b/:c...'
    }.test_param(['one', 'c/d/e'])

    RoutePair{
        url: '/one/two/c/d/e'
        route: '/:a/:b/:c...'
    }.test_param(['one', 'two', 'c/d/e'])

    RoutePair{
        url: '/one/two/three/d/e'
        route: '/:a/:b/:c...'
    }.test_param(['one', 'two', 'three/d/e'])
}

fn test_route_index_path() {
    RoutePair{
        url: '/'
        route: '/:path...'
    }.test_param(['/'])

    RoutePair{
        url: '/foo/bar'
        route: '/:path...'
    }.test_param(['/foo/bar'])
}