#pragma once

#include <napi/env.h>

#if _MSC_VER
#pragma warning(disable : 4324) // 'bx::DirectoryReader': structure was padded due to alignment specifier
#endif

#include <bx/allocator.h>
#include <bimg/bimg.h>
#include <Babylon/JsRuntime.h>
#include <Babylon/Graphics/DeviceContext.h>
#include <Babylon/Graphics/Platform.h>
#include <vector>

namespace Babylon::Plugins::Internal
{
    class TestUtils final : public Napi::ObjectWrap<TestUtils>
    {
    public:
        static inline constexpr const char* JS_INSTANCE_NAME{"TestUtils"};

        using ParentT = Napi::ObjectWrap<TestUtils>;

        static void CreateInstance(Napi::Env env, Graphics::WindowT window)
        {
            Napi::Function func = ParentT::DefineClass(
                env,
                "TestUtilsClass",
                {
                    ParentT::InstanceMethod("exit", &TestUtils::Exit),
                    ParentT::InstanceMethod("updateSize", &TestUtils::UpdateSize),
                    ParentT::InstanceMethod("setTitle", &TestUtils::SetTitle),
                    ParentT::InstanceMethod("getGraphicsApiName", &TestUtils::GetGraphicsApiName),
                    ParentT::InstanceMethod("writePNG", &TestUtils::WritePNG),
                    ParentT::InstanceMethod("decodeImage", &TestUtils::DecodeImage),
                    ParentT::InstanceMethod("getImageData", &TestUtils::GetImageData),
                    ParentT::InstanceMethod("getOutputDirectory", &TestUtils::GetOutputDirectory),
                    ParentT::InstanceMethod("getFrameBufferData", &TestUtils::GetFrameBufferData),
                },
                &window);

            env.Global().Set(JS_INSTANCE_NAME, func.New({}));
        }

        TestUtils(const Napi::CallbackInfo& info)
            : TestUtils(info, JsRuntime::GetFromJavaScript(info.Env()), *static_cast<Graphics::WindowT*>(info.Data()))
        {
        }

        explicit TestUtils(const Napi::CallbackInfo& info, JsRuntime& runtime, Graphics::WindowT window)
            : ParentT{info}
            , m_runtime{runtime}
            , m_deviceContext{Graphics::DeviceContext::GetFromJavaScript(info.Env())}
            , m_window{window}
        {
        }

    private:
        static inline Napi::FunctionReference constructor{};

        void Exit(const Napi::CallbackInfo& info);
        void UpdateSize(const Napi::CallbackInfo& info);
        void SetTitle(const Napi::CallbackInfo& info);
        Napi::Value GetOutputDirectory(const Napi::CallbackInfo& info);

        Napi::Value GetGraphicsApiName(const Napi::CallbackInfo& info);
        void WritePNG(const Napi::CallbackInfo& info);
        Napi::Value DecodeImage(const Napi::CallbackInfo& info);
        Napi::Value GetImageData(const Napi::CallbackInfo& info);
        void GetFrameBufferData(const Napi::CallbackInfo& info);

        JsRuntime& m_runtime;
        Graphics::DeviceContext& m_deviceContext;
        Graphics::WindowT m_window;

        struct Image
        {
            Image() = default;
            ~Image()
            {
                if (m_Image)
                {
                    bimg::imageFree(m_Image);
                    m_Image = nullptr;
                }
            }
            bimg::ImageContainer* m_Image{};
        };
    };
}