#include <iostream>
#include <functional>

#ifdef _WIN32
#define needUTF8() (SetConsoleOutputCP(65001) && setvbuf(stdout, nullptr, _IOFBF, 1000))
#else
#define needUTF8()
#endif

namespace lpstd
{
    namespace testing
    {

        struct Testingresults
        {
            int passed = 0;
            int failed = 0;
        };

        Testingresults results;

        void drawResults()
        {

            needUTF8();

            std::cout << std::endl;
            if (results.failed == 0)
            {
                std::cout << "=== ✅ All tests passed ===" << std::endl;
            }
            else
            {
                std::cout << "=== ❌ " << results.failed << " tests failed ===" << std::endl;
            }
        }

        template <typename T>
        class Expect
        {
        public:
            Expect(T value)
            {

                needUTF8();

                this->value = value;
            }

            bool toBe(T expected)
            {
                bool result = this->value == expected;
                if (result)
                {
                    std::cout << "✅ Expected " << this->value << " to be " << expected << std::endl;
                    results.passed++;
                }
                else
                {
                    std::cout << "❌ Expected " << expected << " but got " << this->value << std::endl;
                    results.failed++;
                }
                return result;
            }

            bool toBeGreaterThan(T expected)
            {
                bool result = this->value > expected;
                if (result)
                {
                    std::cout << "✅ Expected " << this->value << " to be greater than " << expected << std::endl;
                    results.passed++;
                }
                else
                {
                    std::cout << "❌ Expected " << this->value << " to be greater than " << expected << std::endl;
                    results.failed++;
                }

                return result;
            }

            bool toBeLessThan(T expected)
            {
                bool result = this->value < expected;
                if (result)
                {
                    std::cout << "✅ Expected " << this->value << " to be less than " << expected << std::endl;
                    results.passed++;
                }
                else
                {
                    std::cout << "❌ Expected " << this->value << " to be less than " << expected << std::endl;
                    results.failed++;
                }
                return result;
            }

            bool toBeTrue()
            {
                bool result = this->value;
                if (result)
                {
                    std::cout << "✅ Expected " << this->value << " to be true" << std::endl;
                    results.passed++;
                }
                else
                {
                    std::cout << "❌ Expected " << this->value << " to be true" << std::endl;
                    results.failed++;
                }
                return result;
            }

            bool toBeFalse()
            {
                bool result = !this->value;
                if (result)
                {
                    std::cout << "✅ Expected " << this->value << " to be false" << std::endl;
                    results.passed++;
                }
                else
                {
                    std::cout << "❌ Expected " << this->value << " to be false" << std::endl;
                    results.failed++;
                }
                return result;
            }

            bool toBeCloseTo(T expected, T delta = 0.01)
            {
                bool result = this->value >= expected - delta && this->value <= expected + delta;
                if (result)
                {
                    std::cout << "✅ Expected " << this->value << " to be close to " << expected << std::endl;
                    results.passed++;
                }
                else
                {
                    std::cout << "❌ Expected " << this->value << " to be close to " << expected << std::endl;
                    results.failed++;
                }
                return result;
            }

        private:
            T value;
        };

        template <typename T>
        Expect<T> expect(T value)
        {
            return Expect<T>(value);
        }

        // void test(std::string description, void (*test)())
        // {
        //     std::cout << description << std::endl;
        //     test();
        // }

        // Custom test function to execute a single test case
        void test(std::string description, std::function<void()> testCase)
        {
            std::cout << description << std::endl;
            testCase();
            std::cout << std::endl;
        }

        // Custom describe function to group related test cases
        void describe(std::string description, std::function<void()> testSuite)
        {
            std::cout << "Describing " << description << std::endl;
            testSuite();
            drawResults();
            std::cout << std::endl;                
        }

        void testThrow(std::function<void()> testCase)
        {
            try
            {
                testCase();
            }
            catch (std::exception &e)
            {
                std::cout << "✅ Expected exception thrown: " << e.what() << std::endl;
                results.passed++;
                return;
            }
            std::cout << "❌ Expected exception not thrown" << std::endl;
            results.failed++;
        }

    }
}