use std::iter::FromIterator;

use tabled::settings::Alignment;
use testing_table::test_table;
use toml::{Table as TomlMap, Value};

use toml_to_table::{Orientation, TomlTable};

test_table!(
    test_string,
    toml_table(Value::String(String::from("123456789"))),
    "+-----------+\n| 123456789 |\n+-----------+"
);

test_table!(
    test_bool_true,
    toml_table(Value::Boolean(true)),
    "+------+\n| true |\n+------+"
);

test_table!(
    test_bool_false,
    toml_table(Value::Boolean(false)),
    "+-------+\n| false |\n+-------+"
);

test_table!(
    test_int,
    toml_table(Value::Integer(123456789)),
    "+-----------+\n| 123456789 |\n+-----------+"
);

test_table!(
    test_float,
    toml_table(Value::Float(123.456789)),
    "+------------+\n| 123.456789 |\n+------------+"
);

test_table!(
    test_sequence_0,
    toml_table(Value::Array(vec![
        Value::String(String::from("Hello")),
        Value::String(String::from("World")),
    ])),
    "+---------+"
    "|  Hello  |"
    "+---------+"
    "|  World  |"
    "+---------+"
);

test_table!(
    test_sequence_1,
    toml_table(Value::Array(vec![
        Value::Array(vec![
            Value::String(String::from("Hello")),
            Value::String(String::from("World")),
        ]),
        Value::String(String::from("Hello")),
        Value::Array(vec![
            Value::String(String::from("Hello")),
            Value::String(String::from("World")),
        ]),
        Value::String(String::from("World")),
        Value::Array(vec![
            Value::String(String::from("Hello")),
            Value::String(String::from("World")),
        ]),
    ])),
    "+-------------+"
    "| +---------+ |"
    "| |  Hello  | |"
    "| +---------+ |"
    "| |  World  | |"
    "| +---------+ |"
    "+-------------+"
    "|  Hello      |"
    "+-------------+"
    "| +---------+ |"
    "| |  Hello  | |"
    "| +---------+ |"
    "| |  World  | |"
    "| +---------+ |"
    "+-------------+"
    "|  World      |"
    "+-------------+"
    "| +---------+ |"
    "| |  Hello  | |"
    "| +---------+ |"
    "| |  World  | |"
    "| +---------+ |"
    "+-------------+"
);

test_table!(
    test_sequence_0_row,
    toml_table_orientation(
        Value::Array(vec![
            Value::String(String::from("Hello")),
            Value::String(String::from("World")),
        ]),
        Orientation::Row,
        Orientation::Column,
    ),
    "+---------+---------+"
    "|  Hello  |  World  |"
    "+---------+---------+"
);

test_table!(
    test_sequence_1_row,
    toml_table_orientation(
        Value::Array(vec![
            Value::Array(vec![
                Value::String(String::from("Hello")),
                Value::String(String::from("World")),
            ]),
            Value::String(String::from("Hello")),
            Value::Array(vec![
                Value::String(String::from("Hello")),
                Value::String(String::from("World")),
            ]),
            Value::String(String::from("World")),
            Value::Array(vec![
                Value::String(String::from("Hello")),
                Value::String(String::from("World")),
            ]),
        ]),
        Orientation::Row,
        Orientation::Column,
    ),
    "+-----------------------+---------+-----------------------+---------+-----------------------+"
    "| +---------+---------+ |  Hello  | +---------+---------+ |  World  | +---------+---------+ |"
    "| |  Hello  |  World  | |         | |  Hello  |  World  | |         | |  Hello  |  World  | |"
    "| +---------+---------+ |         | +---------+---------+ |         | +---------+---------+ |"
    "+-----------------------+---------+-----------------------+---------+-----------------------+"
);

test_table!(
    test_map_0,
    toml_table(Value::Table(TomlMap::from_iter([
        (
            String::from("Hello Key"),
            Value::String(String::from("World Value"))
        ),
        (
            String::from("Hello Key 2"),
            Value::String(String::from("Value 1")),
        ),
    ]))),
    "+-------------+---------------+"
    "| Hello Key   |  World Value  |"
    "+-------------+---------------+"
    "| Hello Key 2 |  Value 1      |"
    "+-------------+---------------+"
);

test_table!(
    test_map_1,
    toml_table(Value::Table(TomlMap::from_iter([
        (
            String::from("Hello Key"),
            Value::String(String::from("World Value")),
        ),
        (
            String::from("Hello Key 2"),
            Value::Array(vec![
                Value::String(String::from("Hello")),
                Value::String(String::from("World")),
            ]),
        ),
    ]))),
    "+-------------+---------------+"
    "| Hello Key   |  World Value  |"
    "+-------------+---------------+"
    "| Hello Key 2 | +---------+   |"
    "|             | |  Hello  |   |"
    "|             | +---------+   |"
    "|             | |  World  |   |"
    "|             | +---------+   |"
    "+-------------+---------------+"
);

test_table!(
    test_map_0_row,
    toml_table_orientation(
        Value::Table(TomlMap::from_iter([
            (
                String::from("Hello Key"),
                Value::String(String::from("World Value")),
            ),
            (
                String::from("Hello Key 2"),
                Value::String(String::from("Value 1")),
            ),
        ])),
        Orientation::Row,
        Orientation::Row,
    ),
    "+---------------+-------------+"
    "| Hello Key     | Hello Key 2 |"
    "+---------------+-------------+"
    "|  World Value  |  Value 1    |"
    "+---------------+-------------+"
);

test_table!(
    test_map_1_row,
    toml_table_orientation(
        Value::Table(TomlMap::from_iter([
            (
                String::from("Hello Key"),
                Value::String(String::from("World Value")),
            ),
            (
                String::from("Hello Key 2"),
                Value::Array(vec![
                    Value::String(String::from("Hello")),
                    Value::String(String::from("World")),
                ]),
            ),
        ])),
        Orientation::Row,
        Orientation::Row,
    ),
    "+---------------+-----------------------+"
    "| Hello Key     | Hello Key 2           |"
    "+---------------+-----------------------+"
    "|  World Value  | +---------+---------+ |"
    "|               | |  Hello  |  World  | |"
    "|               | +---------+---------+ |"
    "+---------------+-----------------------+"
);

test_table!(
    test_map_1_row_column,
    toml_table_orientation(
        Value::Table(TomlMap::from_iter([
            (
                String::from("Hello Key"),
                Value::String(String::from("World Value")),
            ),
            (
                String::from("Hello Key 2"),
                Value::Array(vec![
                    Value::String(String::from("Hello")),
                    Value::String(String::from("World")),
                ]),
            ),
        ])),
        Orientation::Column,
        Orientation::Row,
    ),
    "+---------------+-------------+"
    "| Hello Key     | Hello Key 2 |"
    "+---------------+-------------+"
    "|  World Value  | +---------+ |"
    "|               | |  Hello  | |"
    "|               | +---------+ |"
    "|               | |  World  | |"
    "|               | +---------+ |"
    "+---------------+-------------+"
);

test_table!(
    test_sequence_row_column,
    toml_table_orientation(
        Value::Array(vec![
            Value::Table(TomlMap::from_iter([
                (
                    String::from("Hello Key"),
                    Value::String(String::from("World Value")),
                ),
                (
                    String::from("Hello Key 2"),
                    Value::Array(vec![
                        Value::String(String::from("Hello")),
                        Value::String(String::from("World")),
                    ]),
                ),
            ])),
            Value::String(String::from("Hello Key 2")),
            Value::String(String::from("Hello")),
            Value::Array(vec![
                Value::String(String::from("Wold")),
                Value::String(String::from("Wod")),
                Value::String(String::from("Wo")),
                Value::String(String::from("Wo")),
            ]),
        ]),
        Orientation::Column,
        Orientation::Row,
    ),
    "+---------------------------------+"
    "| +---------------+-------------+ |"
    "| | Hello Key     | Hello Key 2 | |"
    "| +---------------+-------------+ |"
    "| |  World Value  | +---------+ | |"
    "| |               | |  Hello  | | |"
    "| |               | +---------+ | |"
    "| |               | |  World  | | |"
    "| |               | +---------+ | |"
    "| +---------------+-------------+ |"
    "+---------------------------------+"
    "|  Hello Key 2                    |"
    "+---------------------------------+"
    "|  Hello                          |"
    "+---------------------------------+"
    "| +--------+                      |"
    "| |  Wold  |                      |"
    "| +--------+                      |"
    "| |  Wod   |                      |"
    "| +--------+                      |"
    "| |  Wo    |                      |"
    "| +--------+                      |"
    "| |  Wo    |                      |"
    "| +--------+                      |"
    "+---------------------------------+"
);

test_table!(
    test_sequence_row_column_1,
    toml_table_orientation(
        Value::Array(vec![
            Value::Table(TomlMap::from_iter([
                (
                    String::from("Hello Key"),
                    Value::String(String::from("World Value")),
                ),
                (
                    String::from("Hello Key 2"),
                    Value::Array(vec![
                        Value::String(String::from("Hello")),
                        Value::String(String::from("World")),
                    ]),
                ),
            ])),
            Value::String(String::from("Hello Key 2")),
            Value::String(String::from("Hello")),
            Value::Array(vec![
                Value::String(String::from("Wold")),
                Value::String(String::from("Wod")),
                Value::String(String::from("Wo")),
                Value::String(String::from("Wo")),
            ]),
        ]),
        Orientation::Row,
        Orientation::Column,
    ),
    "+-----------------------------------------+---------------+---------+----------------------------------+"
    "| +-------------+-----------------------+ |  Hello Key 2  |  Hello  | +--------+-------+------+------+ |"
    "| | Hello Key   |  World Value          | |               |         | |  Wold  |  Wod  |  Wo  |  Wo  | |"
    "| +-------------+-----------------------+ |               |         | +--------+-------+------+------+ |"
    "| | Hello Key 2 | +---------+---------+ | |               |         |                                  |"
    "| |             | |  Hello  |  World  | | |               |         |                                  |"
    "| |             | +---------+---------+ | |               |         |                                  |"
    "| +-------------+-----------------------+ |               |         |                                  |"
    "+-----------------------------------------+---------------+---------+----------------------------------+"
);

test_table!(
    test_sequence_row_column_2,
    toml_table_orientation(
        Value::Array(vec![
            Value::Table(TomlMap::from_iter([
                (
                    String::from("Hello Key"),
                    Value::String(String::from("World Value")),
                ),
                (
                    String::from("Hello Key 2"),
                    Value::Array(vec![
                        Value::String(String::from("Hello")),
                        Value::String(String::from("World")),
                    ]),
                ),
            ])),
            Value::String(String::from("Hello Key 2")),
            Value::String(String::from("Hello")),
            Value::Array(vec![
                Value::String(String::from("Wold")),
                Value::String(String::from("Wod")),
                Value::String(String::from("Wo")),
                Value::String(String::from("Wo")),
            ]),
        ]),
        Orientation::Row,
        Orientation::Row,
    ),
    "+-------------------------------------------+---------------+---------+----------------------------------+"
    "| +---------------+-----------------------+ |  Hello Key 2  |  Hello  | +--------+-------+------+------+ |"
    "| | Hello Key     | Hello Key 2           | |               |         | |  Wold  |  Wod  |  Wo  |  Wo  | |"
    "| +---------------+-----------------------+ |               |         | +--------+-------+------+------+ |"
    "| |  World Value  | +---------+---------+ | |               |         |                                  |"
    "| |               | |  Hello  |  World  | | |               |         |                                  |"
    "| |               | +---------+---------+ | |               |         |                                  |"
    "| +---------------+-----------------------+ |               |         |                                  |"
    "+-------------------------------------------+---------------+---------+----------------------------------+"
);

test_table!(
    test_sequence_row_column_3,
    toml_table_orientation(
        Value::Array(vec![
            Value::Table(TomlMap::from_iter([
                (
                    String::from("Hello Key"),
                    Value::String(String::from("World Value")),
                ),
                (
                    String::from("Hello Key 2"),
                    Value::Array(vec![
                        Value::String(String::from("Hello")),
                        Value::String(String::from("World")),
                    ]),
                ),
            ])),
            Value::String(String::from("Hello Key 2")),
            Value::String(String::from("Hello")),
            Value::Array(vec![
                Value::String(String::from("Wold")),
                Value::String(String::from("Wod")),
                Value::String(String::from("Wo")),
                Value::String(String::from("Wo")),
            ]),
        ]),
        Orientation::Column,
        Orientation::Column,
    ),
    "+---------------------------------+"
    "| +-------------+---------------+ |"
    "| | Hello Key   |  World Value  | |"
    "| +-------------+---------------+ |"
    "| | Hello Key 2 | +---------+   | |"
    "| |             | |  Hello  |   | |"
    "| |             | +---------+   | |"
    "| |             | |  World  |   | |"
    "| |             | +---------+   | |"
    "| +-------------+---------------+ |"
    "+---------------------------------+"
    "|  Hello Key 2                    |"
    "+---------------------------------+"
    "|  Hello                          |"
    "+---------------------------------+"
    "| +--------+                      |"
    "| |  Wold  |                      |"
    "| +--------+                      |"
    "| |  Wod   |                      |"
    "| +--------+                      |"
    "| |  Wo    |                      |"
    "| +--------+                      |"
    "| |  Wo    |                      |"
    "| +--------+                      |"
    "+---------------------------------+"
);

test_table!(
    test_alignment_center,
    TomlTable::new(Value::Array(vec![
            Value::Table(TomlMap::from_iter([
                (
                    String::from("Hello Key"),
                    Value::String(String::from("World Value")),
                ),
                (
                    String::from("Hello Key 2"),
                    Value::Array(vec![
                        Value::String(String::from("Hello")),
                        Value::String(String::from("World")),
                    ]),
                ),
            ])),
            Value::String(String::from("Hello Key 2")),
            Value::String(String::from("Hello")),
            Value::Array(vec![
                Value::String(String::from("Wold")),
                Value::String(String::from("Wod")),
                Value::String(String::from("Wo")),
                Value::String(String::from("Wo")),
            ]),
        ]))
        .with(Alignment::center_vertical())
        .with(Alignment::center())
        .to_string(),
    "+---------------------------------+"
    "| +-------------+---------------+ |"
    "| |  Hello Key  |  World Value  | |"
    "| +-------------+---------------+ |"
    "| |             |  +---------+  | |"
    "| |             |  |  Hello  |  | |"
    "| | Hello Key 2 |  +---------+  | |"
    "| |             |  |  World  |  | |"
    "| |             |  +---------+  | |"
    "| +-------------+---------------+ |"
    "+---------------------------------+"
    "|           Hello Key 2           |"
    "+---------------------------------+"
    "|              Hello              |"
    "+---------------------------------+"
    "|           +--------+            |"
    "|           |  Wold  |            |"
    "|           +--------+            |"
    "|           |  Wod   |            |"
    "|           +--------+            |"
    "|           |   Wo   |            |"
    "|           +--------+            |"
    "|           |   Wo   |            |"
    "|           +--------+            |"
    "+---------------------------------+"
);

fn toml_table(value: Value) -> String {
    TomlTable::new(value).to_string()
}

fn toml_table_orientation(value: Value, seq: Orientation, map: Orientation) -> String {
    TomlTable::new(value)
        .seq_orientation(seq)
        .map_orientation(map)
        .to_string()
}