#pragma once

#include <dmf-node/node.hpp>
#include <mxl/flow.h>
#include <mxl/time.h>

#include <DeckLinkAPI.h>

#include <atomic>
#include <mutex>
#include <optional>
#include <thread>

namespace dmf_node {

class DeckLinkInNode : public Node {
public:
    DeckLinkInNode() = default;
    ~DeckLinkInNode();

    std::string type() const override { return "decklink-in"; }

    std::vector<PortDef> input_ports() const override { return {}; }

    std::vector<PortDef> output_ports() const override {
        return {{"video_out", PortDirection::Output, MediaType::VideoV210}};
    }

    void configure(const nlohmann::json& params) override;
    void on_add_writer(const std::string& port_id, mxlFlowWriter writer) override;
    void on_add_reader(const std::string& port_id, mxlFlowReader reader) override {}
    void on_remove_writer(const std::string& port_id) override;
    void on_remove_reader(const std::string& port_id) override {}

    void process() override;
    nlohmann::json status() const override;

private:
    bool open_device();
    void close_device();

    std::optional<mxlFlowWriter> writer_;
    mxlRational grain_rate_{25, 1};
    uint64_t write_index_ = 0;
    uint64_t grains_written_ = 0;

    int device_index_ = 0;
    BMDDisplayMode display_mode_ = bmdModeHD1080i50;
    BMDVideoConnection input_connection_ = bmdVideoConnectionUnspecified;

    IDeckLink* decklink_ = nullptr;
    IDeckLinkInput* input_ = nullptr;

    std::atomic<bool> capturing_{false};
    bool is_interlaced_ = false;
    std::mutex frame_mutex_;
    void* frame_data_ = nullptr;
    long frame_row_bytes_ = 0;
    long frame_width_ = 1920;
    long frame_height_ = 1080;
    bool frame_ready_ = false;

    class CaptureCallback : public IDeckLinkInputCallback {
    public:
        CaptureCallback(DeckLinkInNode& owner) : owner_(owner) {}

        HRESULT STDMETHODCALLTYPE QueryInterface(REFIID, void**) override { return E_NOINTERFACE; }
        ULONG STDMETHODCALLTYPE AddRef() override { return 1; }
        ULONG STDMETHODCALLTYPE Release() override { return 1; }

        HRESULT STDMETHODCALLTYPE VideoInputFormatChanged(
            BMDVideoInputFormatChangedEvents, IDeckLinkDisplayMode*, BMDDetectedVideoInputFormatFlags) override;

        HRESULT STDMETHODCALLTYPE VideoInputFrameArrived(
            IDeckLinkVideoInputFrame* videoFrame, IDeckLinkAudioInputPacket*) override;

    private:
        DeckLinkInNode& owner_;
    };

    std::unique_ptr<CaptureCallback> callback_;
};

} // namespace dmf_node