#include "asl/format.hpp"
#include "asl/testing/testing.hpp"
#include "asl/float.hpp"
#include "asl/tests/test_types.hpp"

static_assert(asl::formattable<decltype("Hello")>);

ASL_TEST(format_args)
{
    StringSink sink;

    // @Todo Introduce ASL_TEST_EXPECT_EQ, or ASL_TEST_EXPECT_STREQ

    asl::format(&sink, "Hello, world!");
    ASL_TEST_EXPECT(sink.str() == "Hello, world!"_sv);

    sink.reset();
    asl::format(&sink, "");
    ASL_TEST_EXPECT(sink.str() == ""_sv);

    sink.reset();
    asl::format(&sink, "Hello, {}!", "world");
    ASL_TEST_EXPECT(sink.str() == "Hello, world!"_sv);

    sink.reset();
    asl::format(&sink, "Hello, {}! {}", "world");
    ASL_TEST_EXPECT(sink.str() == "Hello, world! <ERROR>"_sv);

    sink.reset();
    asl::format(&sink, "Hello, pup!", "world");
    ASL_TEST_EXPECT(sink.str() == "Hello, pup!"_sv);

    sink.reset();
    asl::format(&sink, "{}", "CHEESE");
    ASL_TEST_EXPECT(sink.str() == "CHEESE"_sv);

    sink.reset();
    asl::format(&sink, "{   ", "CHEESE");
    ASL_TEST_EXPECT(sink.str() == "<ERROR>   "_sv);

    sink.reset();
    asl::format(&sink, "{", "CHEESE");
    ASL_TEST_EXPECT(sink.str() == "<ERROR>"_sv);

    sink.reset();
    asl::format(&sink, "a{{b");
    ASL_TEST_EXPECT(sink.str() == "a{b"_sv);

    sink.reset();
    asl::format(&sink, "{{{}}} }", "CHEESE");
    ASL_TEST_EXPECT(sink.str() == "{CHEESE} }"_sv);
}

ASL_TEST(format_integers)
{
    StringSink sink;
    
    sink.reset();
    asl::format(&sink, "{} {} {}", 0, 1, 2);
    ASL_TEST_EXPECT(sink.str() == "0 1 2"_sv);

    sink.reset();
    asl::format(&sink, "{} {} {}", 10, 11, 12);
    ASL_TEST_EXPECT(sink.str() == "10 11 12"_sv);

    sink.reset();
    asl::format(&sink, "{} {} {}", 100, 101, 102);
    ASL_TEST_EXPECT(sink.str() == "100 101 102"_sv);

    sink.reset();
    asl::format(&sink, "{} {} {}", 1000, 1001, 1002);
    ASL_TEST_EXPECT(sink.str() == "1000 1001 1002"_sv);

    sink.reset();
    asl::format(&sink, "{} {} {} {}", -1, -23, -456, -7890);
    ASL_TEST_EXPECT(sink.str() == "-1 -23 -456 -7890"_sv);
}

ASL_TEST(format_floats)
{
    StringSink sink;
    
    sink.reset();
    asl::format(&sink, "{} {} {}", 0.0F, 1.0, 2.0F);
    ASL_TEST_EXPECT(sink.str() == "0 1 2"_sv);
    
    sink.reset();
    asl::format(&sink, "{} {} {}", 0.1F, 0.001F, 0.123F);
    ASL_TEST_EXPECT(sink.str() == "0.1 0.001 0.123"_sv);

    sink.reset();
    asl::format(&sink, "{} {}", 1.25F, -22.3);
    ASL_TEST_EXPECT(sink.str() == "1.25 -22.3"_sv);

    sink.reset();
    asl::format(&sink, "{}", 1e32);
    ASL_TEST_EXPECT(sink.str() == "100000000000000000000000000000000"_sv);

    sink.reset();
    asl::format(&sink, "{}", 123e-8);
    ASL_TEST_EXPECT(sink.str() == "0.00000123"_sv);

    sink.reset();
    asl::format(&sink, "{} {}", asl::infinity<float>(), -asl::infinity<double>());
    ASL_TEST_EXPECT(sink.str() == "Infinity -Infinity"_sv);

    sink.reset();
    asl::format(&sink, "{}", asl::nan<float>());
    ASL_TEST_EXPECT(sink.str() == "NaN"_sv);
}

ASL_TEST(format_boolean)
{
    StringSink sink;
    
    sink.reset();
    asl::format(&sink, "{} {}", true, false);
    ASL_TEST_EXPECT(sink.str() == "true false"_sv);
}

struct CustomFormat
{
    int x;
    friend void AslFormat(asl::Formatter&, const CustomFormat&);
};

void AslFormat(asl::Formatter& f, const CustomFormat& c)
{
    f.write("("_sv);
    AslFormat(f, c.x);
    f.write(")"_sv);
}

static_assert(asl::formattable<CustomFormat>);

ASL_TEST(format_custom)
{
    StringSink sink;
    
    asl::format(&sink, "{}", CustomFormat{37});
    ASL_TEST_EXPECT(sink.str() == "(37)"_sv);
}