use std::fmt::{self, Display, Formatter};

use crate::asm::error::CrsnError;

/// Register name
#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash)]
pub enum Register {
    /// Argument register, read-only
    Arg(u8),
    /// Result register, read-only
    Res(u8),
    /// General purpose register
    Gen(u8),
}

impl Display for Register {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        match self {
            Register::Arg(n) => write!(f, "arg{}", n),
            Register::Res(n) => write!(f, "res{}", n),
            Register::Gen(n) => write!(f, "r{}", n),
        }
    }
}

pub fn parse_reg(name: &str) -> anyhow::Result<Register> {
    // TODO deduplicate code
    if let Some(rn) = name.strip_prefix("arg") {
        if rn.chars().find(|c: &char| !c.is_ascii_digit()).is_some() {
            Err(CrsnError::Parse(format!("Bad register: {}", name).into()))?;
        }
        let val: u8 = rn.parse()?;
        Ok(Register::Arg(val))
    } else if let Some(rn) = name.strip_prefix("res") {
        if rn.chars().find(|c: &char| !c.is_ascii_digit()).is_some() {
            Err(CrsnError::Parse(format!("Bad register: {}", name).into()))?;
        }
        let val: u8 = rn.parse()?;
        Ok(Register::Res(val))
    } else if let Some(rn) = name.strip_prefix("r") {
        if rn.chars().find(|c: &char| !c.is_ascii_digit()).is_some() {
            Err(CrsnError::Parse(format!("Bad register: {}", name).into()))?;
        }
        let val: u8 = rn.parse()?;
        Ok(Register::Gen(val))
    } else {
        Err(CrsnError::Parse(format!("Bad reg name: {}", name).into()))?
    }
}