from enum import Enum
from typing import Dict, Union


class ConfigFieldType(str, Enum):
    BOOLEAN = "boolean"
    INTEGER = "integer"
    FLOAT = "float"
    STRING = "string"


CONFIG_SCHEMA_DEF = {
    "dll": {
        "name": "dll",
        "fieldType": ConfigFieldType.STRING,
        "default": "",
        "description": "override the lsfg-vk.dll path",
        "location": "toml",
    },
    "no_fp16": {
        "name": "no_fp16",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "disable FP16 acceleration",
        "location": "toml",
    },
    "multiplier": {
        "name": "multiplier",
        "fieldType": ConfigFieldType.INTEGER,
        "default": 1,
        "description": "frame generation multiplier",
        "location": "toml",
    },
    "flow_scale": {
        "name": "flow_scale",
        "fieldType": ConfigFieldType.FLOAT,
        "default": 1.0,
        "description": "motion estimation resolution scale",
        "location": "toml",
    },
    "performance_mode": {
        "name": "performance_mode",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "use the lighter frame generation model",
        "location": "toml",
    },
    "experimental_present_mode": {
        "name": "experimental_present_mode",
        "fieldType": ConfigFieldType.STRING,
        "default": "fifo",
        "description": "control the v2 present mode override",
        "location": "toml",
    },
    "dxvk_frame_rate": {
        "name": "dxvk_frame_rate",
        "fieldType": ConfigFieldType.INTEGER,
        "default": 0,
        "description": "base framerate cap for DirectX games before frame multiplier",
        "location": "script",
    },
    "enable_wow64": {
        "name": "enable_wow64",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "enable PROTON_USE_WOW64=1 for 32-bit games",
        "location": "script",
    },
    "disable_steamdeck_mode": {
        "name": "disable_steamdeck_mode",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "disable Steam Deck mode",
        "location": "script",
    },
    "mangohud_workaround": {
        "name": "mangohud_workaround",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "enable a transparent MangoHud overlay workaround",
        "location": "script",
    },
    "disable_vkbasalt": {
        "name": "disable_vkbasalt",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "disable vkBasalt for games where it conflicts with lsfg-vk",
        "location": "script",
    },
    "force_enable_vkbasalt": {
        "name": "force_enable_vkbasalt",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "force-enable vkBasalt",
        "location": "script",
    },
    "enable_wsi": {
        "name": "enable_wsi",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "enable the Gamescope WSI layer",
        "location": "script",
    },
    "enable_zink": {
        "name": "enable_zink",
        "fieldType": ConfigFieldType.BOOLEAN,
        "default": False,
        "description": "enable Zink for OpenGL games",
        "location": "script",
    },
}


def get_field_names() -> list[str]:
    return list(CONFIG_SCHEMA_DEF)


def get_defaults() -> Dict[str, Union[bool, int, float, str]]:
    return {name: definition["default"] for name, definition in CONFIG_SCHEMA_DEF.items()}


def get_field_types() -> Dict[str, str]:
    return {name: definition["fieldType"].value for name, definition in CONFIG_SCHEMA_DEF.items()}