#include "Output.hpp"
#include "../helpers/Log.hpp"
#include "hyprlock.hpp"

void COutput::create(WP<COutput> pSelf, SP<CCWlOutput> pWlOutput, uint32_t _name) {
    m_ID       = _name;
    m_wlOutput = pWlOutput;
    m_self     = pSelf;

    m_wlOutput->setDescription([this](CCWlOutput* r, const char* description) {
        stringDesc = description ? std::string{description} : "";
        Log::logger->log(Log::INFO, "output {} description {}", m_ID, stringDesc);
    });

    m_wlOutput->setName([this](CCWlOutput* r, const char* name) {
        stringName = std::string{name} + stringName;
        stringPort = std::string{name};
        Log::logger->log(Log::INFO, "output {} name {}", name, name);
    });

    m_wlOutput->setScale([this](CCWlOutput* r, int32_t sc) { scale = sc; });

    m_wlOutput->setDone([this](CCWlOutput* r) {
        done = true;
        Log::logger->log(Log::INFO, "output {} done", m_ID);
        if (g_pHyprlock->m_lockAquired && !m_sessionLockSurface) {
            Log::logger->log(Log::INFO, "output {} creating a new lock surface", m_ID);
            createSessionLockSurface();
        }
    });

    m_wlOutput->setMode([this](CCWlOutput* r, uint32_t flags, int32_t width, int32_t height, int32_t refresh) {
        // handle portrait mode and flipped cases
        if (transform % 2 == 1)
            size = {height, width};
        else
            size = {width, height};
    });

    m_wlOutput->setGeometry(
        [this](CCWlOutput* r, int32_t x, int32_t y, int32_t physical_width, int32_t physical_height, int32_t subpixel, const char* make, const char* model, int32_t transform_) {
            transform = (wl_output_transform)transform_;

            Log::logger->log(Log::INFO, "output {} make {} model {}", m_ID, make ? make : "", model ? model : "");
        });
}

void COutput::createSessionLockSurface() {
    if (!m_self.valid()) {
        Log::logger->log(Log::ERR, "output {} dead??", m_ID);
        return;
    }

    if (m_sessionLockSurface) {
        Log::logger->log(Log::ERR, "output {} already has a session lock surface", m_ID);
        return;
    }

    if (size == Vector2D{0, 0}) {
        Log::logger->log(Log::WARN, "output {} refusing to create a lock surface with size 0x0", m_ID);
        return;
    }

    m_sessionLockSurface = makeUnique<CSessionLockSurface>(m_self.lock());
}

Vector2D COutput::getViewport() const {
    return (m_sessionLockSurface) ? m_sessionLockSurface->size : size;
}