use std::borrow::Cow;

use primitives::{address, Address};

use crate::{Precompile, PrecompileSpecId};

/// Precompile with address and function.
/// Unique precompile identifier.
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
pub enum PrecompileId {
    /// Elliptic curve digital signature algorithm (ECDSA) public key recovery function.
    EcRec,
    /// SHA2-256 hash function.
    Sha256,
    /// RIPEMD-160 hash function.
    Ripemd160,
    /// Identity precompile.
    Identity,
    /// Arbitrary-precision exponentiation under modulo.
    ModExp,
    /// Point addition (ADD) on the elliptic curve 'alt_bn128'.
    Bn254Add,
    /// Scalar multiplication (MUL) on the elliptic curve 'alt_bn128'.
    Bn254Mul,
    /// Bilinear function on groups on the elliptic curve 'alt_bn128'.
    Bn254Pairing,
    /// Compression function F used in the BLAKE2 cryptographic hashing algorithm.
    Blake2F,
    /// Verify p(z) = y given commitment that corresponds to the polynomial p(x) and a KZG proof. Also verify that the provided commitment matches the provided versioned_hash.
    KzgPointEvaluation,
    /// Point addition in G1 (curve over base prime field).
    Bls12G1Add,
    /// Multi-scalar-multiplication (MSM) in G1 (curve over base prime field).
    Bls12G1Msm,
    /// Point addition in G2 (curve over quadratic extension of the base prime field).
    Bls12G2Add,
    /// Multi-scalar-multiplication (MSM) in G2 (curve over quadratic extension of the base prime field).
    Bls12G2Msm,
    /// Pairing operations between a set of pairs of (G1, G2) points.
    Bls12Pairing,
    /// Base field element mapping into the G1 point.
    Bls12MapFpToGp1,
    /// Extension field element mapping into the G2 point.
    Bls12MapFp2ToGp2,
    /// ECDSA signature verification over the secp256r1 elliptic curve (also known as P-256 or prime256v1).
    P256Verify,
    /// Custom precompile identifier.
    Custom(Cow<'static, str>),
}

impl PrecompileId {
    /// Create new custom precompile ID.
    pub fn custom<I>(id: I) -> Self
    where
        I: Into<Cow<'static, str>>,
    {
        Self::Custom(id.into())
    }

    /// Returns the mainnet address for the precompile.
    pub fn mainnet_address(&self) -> Option<Address> {
        let address = match self {
            Self::EcRec => address!("0x0000000000000000000000000000000000000001"),
            Self::Sha256 => address!("0x0000000000000000000000000000000000000002"),
            Self::Ripemd160 => address!("0x0000000000000000000000000000000000000003"),
            Self::Identity => address!("0x0000000000000000000000000000000000000004"),
            Self::ModExp => address!("0x0000000000000000000000000000000000000005"),
            Self::Bn254Add => address!("0x0000000000000000000000000000000000000006"),
            Self::Bn254Mul => address!("0x0000000000000000000000000000000000000007"),
            Self::Bn254Pairing => address!("0x0000000000000000000000000000000000000008"),
            Self::Blake2F => address!("0x0000000000000000000000000000000000000009"),
            Self::KzgPointEvaluation => address!("0x000000000000000000000000000000000000000A"),
            Self::Bls12G1Add => address!("0x000000000000000000000000000000000000000B"),
            Self::Bls12G1Msm => address!("0x000000000000000000000000000000000000000C"),
            Self::Bls12G2Add => address!("0x000000000000000000000000000000000000000D"),
            Self::Bls12G2Msm => address!("0x000000000000000000000000000000000000000E"),
            Self::Bls12Pairing => address!("0x000000000000000000000000000000000000000F"),
            Self::Bls12MapFpToGp1 => address!("0x0000000000000000000000000000000000000010"),
            Self::Bls12MapFp2ToGp2 => address!("0x0000000000000000000000000000000000000011"),
            Self::P256Verify => address!("0x0000000000000000000000000000000000000012"),
            Self::Custom(_) => return None,
        };
        Some(address)
    }

    /// Returns the name of the precompile as defined in EIP-7910.
    pub fn name(&self) -> &str {
        match self {
            Self::EcRec => "ECREC",
            Self::Sha256 => "SHA256",
            Self::Ripemd160 => "RIPEMD160",
            Self::Identity => "ID",
            Self::ModExp => "MODEXP",
            Self::Bn254Add => "BN254_ADD",
            Self::Bn254Mul => "BN254_MUL",
            Self::Bn254Pairing => "BN254_PAIRING",
            Self::Blake2F => "BLAKE2F",
            Self::KzgPointEvaluation => "KZG_POINT_EVALUATION",
            Self::Bls12G1Add => "BLS12_G1ADD",
            Self::Bls12G1Msm => "BLS12_G1MSM",
            Self::Bls12G2Add => "BLS12_G2ADD",
            Self::Bls12G2Msm => "BLS12_G2MSM",
            Self::Bls12Pairing => "BLS12_PAIRING_CHECK",
            Self::Bls12MapFpToGp1 => "BLS12_MAP_FP_TO_G1",
            Self::Bls12MapFp2ToGp2 => "BLS12_MAP_FP2_TO_G2",
            Self::P256Verify => "P256VERIFY",
            Self::Custom(a) => a.as_ref(),
        }
    }

    /// Returns the precompile function for the given spec.
    ///
    /// If case of [`PrecompileId::Custom`] it will return [`None`].
    ///
    /// For case where precompile was still not introduced in the spec,
    /// it will return [`Some`] with fork closest to activation.
    pub fn precompile(&self, spec: PrecompileSpecId) -> Option<Precompile> {
        use PrecompileSpecId::*;

        let precompile = match self {
            Self::EcRec => crate::secp256k1::ECRECOVER,
            Self::Sha256 => crate::hash::SHA256,
            Self::Ripemd160 => crate::hash::RIPEMD160,
            Self::Identity => crate::identity::FUN,
            Self::ModExp => {
                // ModExp changes gas calculation based on spec
                if spec < BERLIN {
                    crate::modexp::BYZANTIUM
                } else if spec < OSAKA {
                    crate::modexp::BERLIN
                } else {
                    crate::modexp::OSAKA
                }
            }
            Self::Bn254Add => {
                // BN254 add - gas cost changes in Istanbul
                if spec < ISTANBUL {
                    crate::bn254::add::BYZANTIUM
                } else {
                    crate::bn254::add::ISTANBUL
                }
            }
            Self::Bn254Mul => {
                // BN254 mul - gas cost changes in Istanbul
                if spec < ISTANBUL {
                    crate::bn254::mul::BYZANTIUM
                } else {
                    crate::bn254::mul::ISTANBUL
                }
            }
            Self::Bn254Pairing => {
                // BN254 pairing - gas cost changes in Istanbul
                if spec < ISTANBUL {
                    crate::bn254::pair::BYZANTIUM
                } else {
                    crate::bn254::pair::ISTANBUL
                }
            }
            Self::Blake2F => crate::blake2::FUN,
            Self::KzgPointEvaluation => crate::kzg_point_evaluation::POINT_EVALUATION,
            Self::Bls12G1Add => crate::bls12_381::g1_add::PRECOMPILE,
            Self::Bls12G1Msm => crate::bls12_381::g1_msm::PRECOMPILE,
            Self::Bls12G2Add => crate::bls12_381::g2_add::PRECOMPILE,
            Self::Bls12G2Msm => crate::bls12_381::g2_msm::PRECOMPILE,
            Self::Bls12Pairing => crate::bls12_381::pairing::PRECOMPILE,
            Self::Bls12MapFpToGp1 => crate::bls12_381::map_fp_to_g1::PRECOMPILE,
            Self::Bls12MapFp2ToGp2 => crate::bls12_381::map_fp2_to_g2::PRECOMPILE,
            Self::P256Verify => {
                // P256 verify - gas cost changes in Osaka
                if spec < OSAKA {
                    crate::secp256r1::P256VERIFY
                } else {
                    crate::secp256r1::P256VERIFY_OSAKA
                }
            }
            Self::Custom(_) => return None,
        };

        Some(precompile)
    }
}
