use crate::firmware::firmware_state::{
    AppCrashesStep, FirmwareState, GameRunningStep, ShellRunningStep,
};
use crate::firmware::prometeu_context::PrometeuContext;
use prometeu_hal::app_mode::AppMode;
use prometeu_hal::cartridge::Cartridge;
use prometeu_hal::color::Color;
use prometeu_hal::log::{LogLevel, LogSource};
use prometeu_hal::primitives::Rect;
use prometeu_system::CrashReport;
use prometeu_system::windows::WindowOwner;

#[derive(Debug, Clone)]
pub struct LoadCartridgeStep {
    pub cartridge: Cartridge,
    init_error: Option<CrashReport>,
    loaded_task_id: Option<prometeu_system::task::TaskId>,
    loaded_app_mode: Option<AppMode>,
}

impl LoadCartridgeStep {
    pub fn new(cartridge: Cartridge) -> Self {
        Self { cartridge, init_error: None, loaded_task_id: None, loaded_app_mode: None }
    }

    pub fn on_enter(&mut self, ctx: &mut PrometeuContext) {
        ctx.os.log(
            LogLevel::Info,
            LogSource::Pos,
            0,
            format!("Loading cartridge: {}", self.cartridge.title),
        );

        // Initialize Asset Manager
        ctx.platform.assets_mut().initialize_for_cartridge(
            self.cartridge.asset_table.clone(),
            self.cartridge.preload.clone(),
            self.cartridge.assets.clone(),
        );

        match ctx.os.sessions().load_vm_cartridge(ctx.vm, &self.cartridge) {
            Ok(loaded) => {
                self.loaded_task_id = Some(loaded.task_id);
                self.loaded_app_mode = Some(loaded.app_mode);
            }
            Err(report) => {
                self.init_error = Some(report);
            }
        }
    }

    pub fn on_update(&mut self, ctx: &mut PrometeuContext) -> Option<FirmwareState> {
        if let Some(report) = self.init_error.take() {
            return Some(FirmwareState::AppCrashes(AppCrashesStep { report }));
        }

        let Some(task_id) = self.loaded_task_id.take() else {
            let report = CrashReport::VmPanic {
                message: format!(
                    "cartridge {} was not loaded into a VM session",
                    self.cartridge.title
                ),
                pc: None,
            };
            return Some(FirmwareState::AppCrashes(AppCrashesStep { report }));
        };

        if self.loaded_app_mode == Some(AppMode::Shell) {
            let id = ctx.os.windows().add_window(
                self.cartridge.title.clone(),
                WindowOwner::Task(task_id),
                Rect { x: 40, y: 20, w: 240, h: 140 },
                Color::WHITE,
            );
            ctx.os.windows().set_focus(id);
            return Some(FirmwareState::ShellRunning(ShellRunningStep::new(task_id)));
        }

        Some(FirmwareState::GameRunning(GameRunningStep::new(task_id)))
    }

    pub fn on_exit(&mut self, _ctx: &mut PrometeuContext) {}
}