// Copyright (c) 2025 taitep
// SPDX-License-Identifier: MIT
//
// This file is part of TRVE (https://gitea.taitep.se/taitep/trve)
// See LICENSE file in the project root for full license text.

use std::{env, sync::Arc, time::Duration};

use trve::{
    consts::{Addr, Byte, DWord, HWord, Word},
    core::Core,
    exceptions::MemoryExceptionType,
    mem::{MemConfig, MemDeviceInterface, MmioRoot, Ram},
};

use anyhow::{Result, bail};

use crate::basic_uart::BasicUart;

mod execload;

fn main() -> Result<()> {
    let mut ram = Ram::try_new(16 * 1024 * 1024)?;
    let buf = ram.buf_mut();

    let args: Vec<String> = env::args().collect();

    if args.len() != 2 {
        eprintln!("USAGE: trve <ram_image>");
        bail!("Wrong number of arguments");
    }

    let entry_point = execload::load(&args[1], buf)?;

    let mut mmio_root = MmioRoot::default();
    mmio_root.insert(0, Arc::new(DbgOut));

    let uart = BasicUart::new();
    let uart = uart.spawn_poller(Duration::from_millis(10));
    mmio_root.insert(0x10000, uart);

    let mem_cfg = MemConfig {
        ram: Arc::new(ram),
        mmio_root,
    };

    let mut core = Core::new(mem_cfg);
    core.reset(entry_point);
    core.run();

    Ok(())
}

mod basic_uart;

struct DbgOut;

impl MemDeviceInterface for DbgOut {
    fn write_dword(&self, addr: Addr, value: DWord) -> Result<(), MemoryExceptionType> {
        eprintln!("Wrote DWord {value:016x} to Debug-Out address {addr:x}");
        Ok(())
    }

    fn write_word(&self, addr: Addr, value: Word) -> Result<(), MemoryExceptionType> {
        eprintln!("Wrote Word {value:08x} to Debug-Out address {addr:x}");
        Ok(())
    }

    fn write_hword(&self, addr: Addr, value: HWord) -> Result<(), MemoryExceptionType> {
        eprintln!("Wrote HWord {value:04x} to Debug-Out address {addr:x}");
        Ok(())
    }

    fn write_byte(&self, addr: Addr, value: Byte) -> Result<(), MemoryExceptionType> {
        eprintln!("Wrote Byte {value:02x} to Debug-Out address {addr:x}");
        Ok(())
    }
}