#include <stdint.h>
#include <stddef.h>
#include <string.h>
#include <intrin.h>
#include "bink.h"

// ---------------------------------------------------------------------------
// Win32 Boilerplate & API declarations
// ---------------------------------------------------------------------------
typedef void* HWND;
typedef void* HDC;
typedef unsigned long  DWORD;
typedef unsigned short WORD;
typedef unsigned int   UINT;
typedef long           LONG;
typedef int            BOOL;

#define BI_RGB          0L
#define DIB_RGB_COLORS  0
#define SRCCOPY         0x00CC0020

#define STD_OUTPUT_HANDLE ((DWORD)-11)

typedef struct _RECT {
    LONG left, top, right, bottom;
} RECT;

typedef struct tagPAINTSTRUCT {
    HDC  hdc;
    BOOL fErase;
    RECT rcPaint;
    BOOL fRestore;
    BOOL fIncUpdate;
    char rgbReserved[32];
} PAINTSTRUCT;

typedef struct tagBITMAPINFOHEADER {
    DWORD biSize;
    LONG  biWidth;
    LONG  biHeight;
    WORD  biPlanes;
    WORD  biBitCount;
    DWORD biCompression;
    DWORD biSizeImage;
    LONG  biXPelsPerMeter;
    LONG  biYPelsPerMeter;
    DWORD biClrUsed;
    DWORD biClrImportant;
} BITMAPINFOHEADER;

typedef struct tagRGBQUAD {
    unsigned char rgbBlue, rgbGreen, rgbRed, rgbReserved;
} RGBQUAD;

typedef struct tagBITMAPINFO {
    BITMAPINFOHEADER bmiHeader;
    RGBQUAD          bmiColors[1];
} BITMAPINFO;

// Kernel/Win32 Exports
void* GetStdHandle(DWORD nStdHandle);
int WriteConsoleA(void* hConsoleOutput, const void* lpBuffer, unsigned int nNumberOfCharsToWrite, unsigned int* lpNumberOfCharsWritten, void* lpReserved);
_Noreturn void ExitProcess(unsigned int code);
void* HeapAlloc(void* heap, unsigned long flags, size_t size);
void* CreateFileA(const char* name, unsigned long access, unsigned long share, void* sa, unsigned long disp, unsigned long flags, void* templ);
unsigned long GetFileSize(void* handle, unsigned long* high);
int ReadFile(void* handle, void* buf, unsigned long toRead, unsigned long* read, void* overlapped);
BOOL CloseHandle(void* hObject);

// New GDI/User32 Exports
HDC BeginPaint(HWND hwnd, PAINTSTRUCT* lpPaint);
BOOL EndPaint(HWND hwnd, const PAINTSTRUCT* lpPaint);
int StretchDIBits(HDC hdc, int xDest, int yDest, int DestWidth, int DestHeight,
                  int xSrc, int ySrc, int SrcWidth, int SrcHeight,
                  const void *lpBits, const BITMAPINFO *lpbmi,
                  UINT iUsage, DWORD rop);


int main(const char* cmdline) {
    void* h = GetStdHandle(STD_OUTPUT_HANDLE);
    WriteConsoleA(h, "video: starting\n", 16, 0, 0);

    WriteConsoleA(h, cmdline, (unsigned int)strlen(cmdline), 0, 0);
    WriteConsoleA(h, "\n", 1, 0, 0);

    void* f = CreateFileA("video.bk2", 0, 0, 0, 3, 0, 0);
    if (!f) {
        WriteConsoleA(h, "video: failed to open video.bk2\n", 32, 0, 0);
        ExitProcess(1);
    }

    unsigned long s = GetFileSize(f, 0);
    void* buf = HeapAlloc(0, 0, s);
    if (!buf) {
        WriteConsoleA(h, "video: HeapAlloc failed\n", 24, 0, 0);
        ExitProcess(1);
    }

    int r = ReadFile(f, buf, s, 0, 0);
    if (!r) {
        WriteConsoleA(h, "video: ReadFile failed\n", 23, 0, 0);
        ExitProcess(1);
    }
    WriteConsoleA(h, "video: video.bk2 loaded\n", 24, 0, 0);
    
    CloseHandle(h);

    HBINK b = BinkOpen(buf, BINKFROMMEMORY);
    if (!b) {
        WriteConsoleA(h, "video: BinkOpen failed: ", 24, 0, 0);
        const char* err = BinkGetError();
        WriteConsoleA(h, err, (unsigned int)strlen(err), 0, 0);
        WriteConsoleA(h, "\n", 1, 0, 0);
        ExitProcess(1);
    }

    uint32_t width  = b->Width;
    uint32_t height = b->Height;

    uint32_t* fb = HeapAlloc(0, 0, width * height * sizeof(uint32_t));
    if (!fb) {
        WriteConsoleA(h, "video: framebuffer alloc failed\n", 32, 0, 0);
        ExitProcess(1);
    }

    // Prepare the Win32 BITMAPINFO structure
    BITMAPINFO bmi;
    memset(&bmi, 0, sizeof(bmi));
    bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    bmi.bmiHeader.biWidth = width;
    bmi.bmiHeader.biHeight = -(LONG)height; // Negative means top-down drawing!
    bmi.bmiHeader.biPlanes = 1;
    bmi.bmiHeader.biBitCount = 32;
    bmi.bmiHeader.biCompression = BI_RGB;

    while (1) {
        if (BinkWait(b)) {
            _mm_pause();
            continue;
        }

        BinkDoFrame(b);

        if (BinkShouldSkip(b)) {
            BinkNextFrame(b);
            BinkDoFrame(b);
            continue;
        }

        BinkCopyToBuffer(b, fb, width * 4, height, 0, 0, 
            BINKSURFACE32);

        // Welcome to the GDI pipeline!
        PAINTSTRUCT ps;
        // Using NULL/0 as the HWND since we don't have a real window handle yet
        HDC hdc = BeginPaint((HWND)0, &ps);

        StretchDIBits(
            hdc,
            0, 0, width, height, // Destination rect
            0, 0, width, height, // Source rect
            fb,
            &bmi,
            DIB_RGB_COLORS,
            SRCCOPY
        );

        EndPaint((HWND)0, &ps);

        if (b->FrameNum == b->Frames)
            break;

        BinkNextFrame(b);
    }
    
    WriteConsoleA(h, "shell: goodbye\n", 15, 0, 0);

    ExitProcess(0);
}