/*
 * disks.h
 * Shared header for the MontaukOS Disk Tool
 * Copyright (c) 2026 Daniel Hammer
 */

#pragma once

#include <montauk/syscall.h>
#include <montauk/string.h>
#include <montauk/heap.h>
#include <gui/gui.hpp>
#include <gui/font.hpp>

extern "C" {
#include <string.h>
#include <stdio.h>
}

using namespace gui;

// ============================================================================
// Constants
// ============================================================================

static constexpr int INIT_W       = 640;
static constexpr int INIT_H       = 460;
static constexpr int TOOLBAR_H    = 44;
static constexpr int HEADER_H     = 28;
static constexpr int ITEM_H       = 32;
static constexpr int MAP_H        = 48;
static constexpr int MAP_PAD      = 16;
static constexpr int MAX_PARTS    = 32;
static constexpr int MAX_DISKS    = 32;
static constexpr int STATUS_H     = 44;

static constexpr int TB_BTN_Y     = 7;
static constexpr int TB_BTN_H     = 30;

static constexpr int NUM_PART_COLORS = 8;

// ============================================================================
// Filesystem types
// ============================================================================

struct FsTypeEntry {
    const char* name;
    int         id;
};

static const FsTypeEntry g_fsTypes[] = {
    { "FAT32", Montauk::FS_TYPE_FAT32 },
};
static constexpr int NUM_FS_TYPES = 1;

// ============================================================================
// State
// ============================================================================

static constexpr int FMT_DLG_W = 280;
static constexpr int FMT_DLG_H = 220;
static constexpr int NP_DLG_W  = 340;
static constexpr int NP_DLG_H  = 200;

struct FormatDialog {
    bool open;
    int  global_part_index;
    int  selected_fs;
    bool hover_format;
    bool hover_cancel;
    char part_desc[80];
    int  win_id;
    uint32_t* pixels;
};

struct NewPartDialog {
    bool open;
    bool will_init_gpt;     // true if GPT init is needed (most destructive)
    bool hover_confirm;
    bool hover_cancel;
    char disk_desc[80];
    char warn_line1[96];
    char warn_line2[96];
    int  win_id;
    uint32_t* pixels;
};

struct DiskToolState {
    Montauk::DiskInfo disks[MAX_DISKS];
    int disk_count;
    Montauk::PartInfo parts[MAX_PARTS];
    int part_count;
    int selected_disk;
    int selected_part;
    int scroll_y;
    char status[80];
    uint64_t status_time;
    FormatDialog fmt_dlg;
    NewPartDialog np_dlg;
    int mouse_x;
    int mouse_y;
};

// ============================================================================
// Global state (extern — defined in main.cpp)
// ============================================================================

extern int g_win_w, g_win_h;
extern DiskToolState g_state;
extern Color g_accent;

// ============================================================================
// Partition colors
// ============================================================================

extern const Color part_colors[NUM_PART_COLORS];

// ============================================================================
// Shared layout helpers
// ============================================================================

inline int disk_button_width(int idx) {
    char label[16];
    int blockDev = (idx >= 0 && idx < g_state.disk_count)
        ? (int)g_state.disks[idx].port
        : idx;
    snprintf(label, sizeof(label), "Disk %d", blockDev);
    return text_width(label) + 20;
}

inline Rect toolbar_disk_button_rect(int idx) {
    int bx = 8;
    for (int i = 0; i < idx; i++)
        bx += disk_button_width(i) + 8;
    return {bx, TB_BTN_Y, disk_button_width(idx), TB_BTN_H};
}

inline Rect toolbar_refresh_button_rect() {
    int y = g_win_h - STATUS_H + (STATUS_H - TB_BTN_H) / 2;
    return {g_win_w - 12 - 84, y, 84, TB_BTN_H};
}

inline Rect toolbar_mount_button_rect() {
    Rect refresh = toolbar_refresh_button_rect();
    return {refresh.x - 8 - 72, refresh.y, 72, TB_BTN_H};
}

inline Rect toolbar_format_button_rect() {
    Rect mount = toolbar_mount_button_rect();
    return {mount.x - 8 - 72, mount.y, 72, TB_BTN_H};
}

inline Rect toolbar_newpart_button_rect() {
    Rect format = toolbar_format_button_rect();
    return {format.x - 8 - 96, format.y, 96, TB_BTN_H};
}

inline int content_title_y() {
    return TOOLBAR_H + 8;
}

inline int partition_map_y() {
    return content_title_y() + system_font_height() + 8;
}

inline int partition_header_y() {
    return partition_map_y() + MAP_H + 8;
}

inline int partition_list_y() {
    return partition_header_y() + HEADER_H;
}

inline Rect format_option_rect(int dialog_w, int index) {
    int fh = system_font_height();
    int sel_y = 12 + fh * 2 + 18;
    int opt_y = sel_y + fh + 8;
    return {24, opt_y + index * 32, dialog_w - 48, 28};
}

inline Rect dialog_primary_button_rect(int dialog_w, int dialog_h) {
    int btn_w = 90;
    int btn_h = 30;
    int bx = dialog_w - btn_w - 16;
    return {bx, dialog_h - btn_h - 12, btn_w, btn_h};
}

inline Rect dialog_secondary_button_rect(int dialog_w, int dialog_h) {
    Rect primary = dialog_primary_button_rect(dialog_w, dialog_h);
    return {primary.x - 8 - primary.w, primary.y, primary.w, primary.h};
}

// ============================================================================
// Function declarations — helpers (main.cpp)
// ============================================================================

void set_status(const char* msg);
int  disk_block_dev(int diskIndex);
int  selected_block_dev();
int  get_disk_parts(int* indices, int max);
void format_disk_size(char* buf, int bufsize, uint64_t sectors, uint16_t sectorSize);

// ============================================================================
// Function declarations — render.cpp
// ============================================================================

void render(uint32_t* pixels);
void render_format_window();
void render_newpart_window();

// ============================================================================
// Function declarations — actions.cpp
// ============================================================================

void disktool_refresh();
void do_create_partition();
void do_mount_partition();
void open_format_dialog();
void close_format_dialog();
void format_dialog_do_format();
void open_newpart_dialog();
void close_newpart_dialog();
void newpart_dialog_confirm();