//! The LLVM target.

use std::str::FromStr;

/// The LLVM target.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Target {
    /// The PolkaVM target.
    PVM,
}

impl Target {
    /// Returns the target name.
    pub fn name(&self) -> &str {
        match self {
            #[cfg(not(feature = "riscv-64"))]
            Self::PVM => "riscv32",
            #[cfg(feature = "riscv-64")]
            Self::PVM => "riscv64",
        }
    }

    /// Returns the target triple.
    pub fn triple(&self) -> &str {
        match self {
            #[cfg(not(feature = "riscv-64"))]
            Self::PVM => "riscv32-unknown-unknown-elf",
            #[cfg(feature = "riscv-64")]
            Self::PVM => "riscv64-unknown-unknown-elf",
        }
    }

    /// Returns the target production name.
    pub fn production_name(&self) -> &str {
        match self {
            Self::PVM => "PVM",
        }
    }
}

impl FromStr for Target {
    type Err = anyhow::Error;

    fn from_str(string: &str) -> Result<Self, Self::Err> {
        match string {
            #[cfg(not(feature = "riscv-64"))]
            "riscv32" => Ok(Self::PVM),
            #[cfg(feature = "riscv-64")]
            "riscv64" => Ok(Self::PVM),
            _ => Err(anyhow::anyhow!(
                "Unknown target `{}`. Supported targets: {:?}",
                string,
                Self::PVM
            )),
        }
    }
}

impl std::fmt::Display for Target {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            #[cfg(not(feature = "riscv-64"))]
            Target::PVM => write!(f, "riscv32"),
            #[cfg(feature = "riscv-64")]
            Target::PVM => write!(f, "riscv64"),
        }
    }
}