/*
    * Graphics.hpp
    * SYS_FBINFO, SYS_FBMAP, SYS_TERMSIZE, SYS_TERMSCALE syscalls
    * Copyright (c) 2026 Daniel Hammer
*/

#pragma once
#include <cstdint>
#include <Sched/Scheduler.hpp>
#include <Memory/Paging.hpp>
#include <Memory/HHDM.hpp>
#include <Graphics/Framebuffer.hpp>
#include <Drivers/Graphics/IntelGPU.hpp>
#include <Terminal/Terminal.hpp>

#include "Syscall.hpp"
#include "Common.hpp"
#include "../Libraries/flanterm/src/flanterm.h"

namespace montauk::abi {

    static void Sys_FbInfo(FbInfo* out) {
        if (out == nullptr) return;
        out->width    = Graphics::Framebuffer::GetWidth();
        out->height   = Graphics::Framebuffer::GetHeight();
        out->pitch    = Graphics::Framebuffer::GetPitch();
        out->bpp      = 32;
        out->userAddr = 0;
    }

    static uint64_t Sys_FbMap() {
        auto* proc = Sched::GetCurrentProcessPtr();
        if (proc == nullptr) return 0;

        uint32_t* fbBase = Graphics::Framebuffer::GetBase();
        if (fbBase == nullptr) return 0;

        uint64_t fbPhys = Memory::SubHHDM((uint64_t)fbBase);
        uint64_t width = Graphics::Framebuffer::GetWidth();
        uint64_t height = Graphics::Framebuffer::GetHeight();
        uint64_t pitch = Graphics::Framebuffer::GetPitch();
        if (width == 0 || height == 0 || pitch == 0
            || width > (~0ULL / 4) || pitch < width * 4
            || height > (~0ULL / pitch)) {
            Kt::KernelLogStream(Kt::ERROR, "FbMap")
                << "Invalid or overflowing framebuffer geometry";
            return 0;
        }
        uint64_t fbSize = height * pitch;
        uint64_t numPages = (fbSize + 0xFFF) / 0x1000;

        Kt::KernelLogStream(Kt::INFO, "FbMap") << "fbPhys=" << kcp::hex << fbPhys
            << " size=" << kcp::dec << fbSize
            << " pages=" << numPages
            << " (" << Graphics::Framebuffer::GetWidth()
            << "x" << Graphics::Framebuffer::GetHeight()
            << " pitch=" << Graphics::Framebuffer::GetPitch() << ")";

        // Map in the dedicated framebuffer arena, away from the heap,
        // shared libraries, reusable IPC surfaces, and the user stack.
        constexpr uint64_t userVa = Sched::UserFramebufferBase;
        uint64_t mappedBytes = numPages * 0x1000ULL;
        bool flipSupported = Drivers::Graphics::IntelGPU::FlipSupported();
        uint64_t buffers = flipSupported ? 2 : 1;
        if (mappedBytes > (Sched::UserFramebufferLimit - userVa) / buffers) {
            Kt::KernelLogStream(Kt::ERROR, "FbMap")
                << "Framebuffer mappings exceed reserved user VA range";
            return 0;
        }

        // Never expose more of buffer 1 than the GPU driver allocated. This
        // independently contains a future geometry-contract regression before
        // any partial framebuffer mapping is installed in the process.
        if (flipSupported
            && (Drivers::Graphics::IntelGPU::GetBufferPhys(1) == 0
                || Drivers::Graphics::IntelGPU::GetBufferPageCount(1) < numPages)) {
            Kt::KernelLogStream(Kt::ERROR, "FbMap")
                << "Page-flip buffer is smaller than firmware framebuffer";
            return 0;
        }

        for (uint64_t i = 0; i < numPages; i++) {
            if (!Memory::VMM::Paging::MapUserInWC(
                proc->pml4Phys,
                fbPhys + i * 0x1000,
                userVa + i * 0x1000
            )) {
                return 0;
            }
        }

        // When page flipping is available, map the second scanout buffer
        // immediately after the first so a double-buffered client can reach
        // both. Single-buffer clients never touch the extra pages.
        if (flipSupported) {
            uint64_t buf1Phys = Drivers::Graphics::IntelGPU::GetBufferPhys(1);
            for (uint64_t i = 0; i < numPages; i++) {
                if (!Memory::VMM::Paging::MapUserInWC(
                    proc->pml4Phys,
                    buf1Phys + i * 0x1000,
                    userVa + (numPages + i) * 0x1000
                )) {
                    return 0;
                }
            }
        }

        return userVa;
    }

    // SYS_FBFLIP: point scanout at buffer `index` (0 or 1). The hardware
    // latches the new surface at vblank, so the flip is tear-free.
    // index == -1 queries support without side effects.
    // index == -2 acquires flip ownership and returns the live front buffer.
    static int64_t Sys_FbFlip(uint64_t index, uint64_t flags) {
        namespace GPU = Drivers::Graphics::IntelGPU;

        if ((int64_t)index == -1) {
            return GPU::FlipSupported() ? 1 : 0;
        }
        if (!GPU::FlipSupported()) return -1;

        auto* proc = Sched::GetCurrentProcessPtr();
        if ((int64_t)index == -2) {
            return proc != nullptr ? GPU::AcquireFlip(proc->pid) : -1;
        }
        if (proc == nullptr) return -1;
        return GPU::Flip((int)index, (flags & 1) != 0, proc->pid);
    }

    static int64_t Sys_DisplayInfo(DisplayInfo* out) {
        return Drivers::Graphics::IntelGPU::GetDisplayInfo(out) ? 0 : -1;
    }

    static int64_t Sys_DisplayModes(DisplayModeInfo* out, int maxCount) {
        return Drivers::Graphics::IntelGPU::GetDisplayModes(out, maxCount);
    }

    static int64_t Sys_DisplaySetMode(int modeIndex) {
        return Drivers::Graphics::IntelGPU::SetDisplayMode(modeIndex);
    }

    static int64_t Sys_DisplayBrightness(int percent) {
        return Drivers::Graphics::IntelGPU::Brightness(percent);
    }

    static uint64_t Sys_TermSize() {
        // If the process is redirected to a GUI terminal, return those dimensions
        auto* proc = Sched::GetCurrentProcessPtr();
        if (proc && proc->redirected) {
            if (proc->termCols > 0 && proc->termRows > 0) {
                return ((uint64_t)proc->termRows << 32) | ((uint64_t)proc->termCols & 0xFFFFFFFF);
            }
            auto* target = GetRedirTarget(proc);
            if (target && target->termCols > 0 && target->termRows > 0) {
                return ((uint64_t)target->termRows << 32) | ((uint64_t)target->termCols & 0xFFFFFFFF);
            }
        }
        size_t cols = 0, rows = 0;
        flanterm_get_dimensions(Kt::ctx, &cols, &rows);
        return (rows << 32) | (cols & 0xFFFFFFFF);
    }

    static int64_t Sys_TermScale(uint64_t scale_x, uint64_t scale_y) {
        if (scale_x == 0) {
            return (int64_t)((Kt::GetFontScaleY() << 32) | (Kt::GetFontScaleX() & 0xFFFFFFFF));
        }
        Kt::Rescale((size_t)scale_x, (size_t)scale_y);
        size_t cols = 0, rows = 0;
        flanterm_get_dimensions(Kt::ctx, &cols, &rows);
        return (int64_t)((rows << 32) | (cols & 0xFFFFFFFF));
    }
};