use std::io::{stdout, Write};

use crate::verbose_println;

pub(crate) type Address = u16;
pub(crate) type ROM = Vec<u8>;

#[allow(dead_code)]
pub struct Memory {
    pub(super) bootrom: ROM,
    pub(super) bootrom_enabled: bool,
    pub(super) rom: ROM,
    pub(super) vram: [u8; 8192],
    pub(super) ram: [u8; 8192],
    pub(super) switchable_ram: [u8; 8192],
    pub(super) cpu_ram: [u8; 128],
    pub(super) oam: [u8; 160],
    pub(super) interrupts: u8,
    pub(super) ime: bool,
    pub(super) ime_scheduled: u8,
    pub(super) io: [u8; 76],
    pub(super) user_mode: bool,
}

impl Memory {
    pub fn init(bootrom: ROM, bootrom_enabled: bool, rom: ROM) -> Self {
        Self {
            bootrom,
            bootrom_enabled,
            rom,
            vram: [0x0; 8192],
            ram: [0x0; 8192],
            switchable_ram: [0x0; 8192],
            cpu_ram: [0x0; 128],
            oam: [0x0; 160],
            interrupts: 0x0,
            ime: false,
            ime_scheduled: 0x0,
            io: [0xFF; 76],
            user_mode: false,
        }
    }

    pub fn get(&self, address: Address) -> u8 {
        match address {
            0x0..0x8000 => {
                // rom access
                // todo - switchable rom banks
                if self.bootrom_enabled && ((address as usize) < self.bootrom.len()) {
                    return self.bootrom[address as usize];
                } else {
                    return self.rom[address as usize];
                }
            }
            0x8000..0xA000 => {
                return self.vram[(address - 0x8000) as usize];
            }
            0xA000..0xC000 => 0xFF,
            0xC000..0xE000 => {
                return self.ram[(address - 0xC000) as usize];
            }
            0xE000..0xFE00 => {
                return self.ram[(address - 0xE000) as usize];
            }
            0xFE00..0xFEA0 => {
                return self.oam[(address - 0xFE00) as usize];
            }
            0xFEA0..0xFF00 => {
                return 0x0;
            }
            0xFF00..0xFF4C => self.get_io(address),
            0xFF4C..0xFF80 => {
                // println!("empty space 2 read");
                return 0xFF;
            }
            0xFF80..0xFFFF => {
                return self.cpu_ram[(address - 0xFF80) as usize];
            }
            0xFFFF => {
                return self.interrupts;
            }
        }
    }

    pub fn set(&mut self, address: Address, data: u8) {
        // verbose_println!("write addr: {:#X}, data: {:#X}", address, data);

        match address {
            0x0..0x8000 => {
                // change this with MBC code...
                // println!("tried to write {:#5X} at {:#X}", data, address);
            }
            0x8000..0xA000 => {
                self.vram[(address - 0x8000) as usize] = data;
            }
            0xA000..0xC000 => {
                // panic!("switchable write");
                // self.switchable_ram[(address - 0xA000) as usize] = data;
            }
            0xC000..0xE000 => {
                self.ram[(address - 0xC000) as usize] = data;
            }
            0xE000..0xFE00 => {
                self.ram[(address - 0xE000) as usize] = data;
            }
            0xFE00..0xFEA0 => {
                self.oam[(address - 0xFE00) as usize] = data;
            }
            0xFEA0..0xFF00 => {
                // println!("empty space write: {:#X} to addr {:#X}", data, address);
            }
            0xFF00..0xFF4C => {
                self.set_io(address, data);
                // verbose_print!("writing to addr {:#X}\r", address);
                stdout().flush().unwrap();
            }
            0xFF50 => {
                self.bootrom_enabled = false;
            }
            0xFF4C..0xFF50 | 0xFF51..0xFF80 => {
                // println!("empty space 2 write: {:#X} to addr {:#X}", data, address);
            }
            0xFF80..0xFFFF => {
                self.cpu_ram[(address - 0xFF80) as usize] = data;
            }
            0xFFFF => {
                verbose_println!("interrupts set to {:#b}", data);
                verbose_println!("                / {:#X}", data);
                self.interrupts = data;
            }
        }
    }

    fn get_io(&self, address: Address) -> u8 {
        if address == 0xFF00 {
            return 0xFF;
        }
        return self.io[(address - 0xFF00) as usize];
    }

    fn set_io(&mut self, address: Address, data: u8) {
        let addr_l = (address - 0xFF00) as usize;
        if !self.user_mode {
            self.io[addr_l] = data;
        } else {
            match address {
                0xFF02 => {
                    if data == 0x81 {
                        print!("{}", self.get(0xFF01) as char);
                        stdout().flush().unwrap();
                    }
                }
                0xFF04 => self.io[addr_l] = 0,
                0xFF00 | 0xFF11 | 0xFF14 | 0xFF16 | 0xFF19 | 0xFF1E | 0xFF23 | 0xFF26 | 0xFF41
                | 0xFF4D | 0xFF56 => {
                    // mixed read/write addresses...
                    // need to fill these out more...
                    // just seeing what breaks...
                }
                0xFF44 | 0xFF76 | 0xFF77 => {
                    // read-only addresses
                    println!("BANNED write: {:#X} to {:#X}", data, address);
                }
                _ => {
                    self.io[addr_l] = data;
                    // panic!("passed non-io address to io handler!");
                }
            }
        }
    }
}