import arrays

type ConcatValue = []u8 | string

fn get() []int {
    return [1, 2, 3, 4, 5]
}

fn get2() [][]int {
    return [[0], [1, 2, 3, 4, 5]]
}

fn receive(a ...int) {
}

fn test_main() {
    assert arrays.concat([0], ...get().map(it)) == [0, 1, 2, 3, 4, 5]
    assert arrays.concat[int]([], ...get().map(it)) == [1, 2, 3, 4, 5]
    assert arrays.concat[[]int]([[0]], ...[get().map(it)]) == [
        [0],
        [1, 2, 3, 4, 5],
    ]
    assert arrays.concat[int](...get2()) == [0, 1, 2, 3, 4, 5]
}

fn test_decompose_variadic_generic_sumtype_array_arg() {
    mut params := []ConcatValue{}
    ids_bin := [
        [u8(`a`), `b`],
        [u8(`c`), `d`],
    ]
    params = arrays.concat(params, ...ids_bin)
    assert params.len == 2
    assert params[0] is []u8
    assert params[1] is []u8
    assert (params[0] as []u8).bytestr() == 'ab'
    assert (params[1] as []u8).bytestr() == 'cd'
}

fn test_decompose_variadic_nested_string_array_arg() {
    a := [['0']]
    b := [['aa', 'bb']]
    c := arrays.concat(a, ...b)
    assert c == [['0'], ['aa', 'bb']]
    assert c.str() == "[['0'], ['aa', 'bb']]"
}