zksync-web3-rs 0.2.1

use std::{fs::File, io::BufReader, path::PathBuf, str::FromStr};

use super::{hash_bytecode, rlp_append_option, Eip712Meta};
use crate::{
    zks_utils::{
        self, CONTRACT_DEPLOYER_ADDR, EIP712_TX_TYPE, ERA_CHAIN_ID, MAX_PRIORITY_FEE_PER_GAS,
    },
    zks_wallet::{DeployRequest, Overrides, TransferRequest, WithdrawRequest, ZKRequestError},
};
use ethers::{
    abi::{Abi, HumanReadableParser, ParseError},
    types::{
        transaction::{eip2930::AccessList, eip712::Eip712Error},
        Address, Bytes, Signature, U256,
    },
    utils::rlp::{Encodable, RlpStream},
};
use ethers_contract::encode_function_data;
use serde::{Deserialize, Serialize};

// TODO: Not all the fields are optional. This was copied from the JS implementation.
#[derive(Serialize, Deserialize, Clone, Debug)]
#[serde(rename_all(serialize = "camelCase", deserialize = "camelCase"))]
pub struct Eip712TransactionRequest {
    /* These need to be filled before estimating the gas */
    pub to: Address,
    pub from: Address,
    pub nonce: U256,
    pub gas: U256,
    pub gas_price: U256,
    pub data: Bytes,
    pub value: U256,
    pub chain_id: U256,
    pub r#type: U256,
    pub max_priority_fee_per_gas: U256,
    #[serde(rename = "eip712Meta")]
    pub custom_data: Eip712Meta,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub access_list: Option<AccessList>,

    /* Filled after estimating the gas */
    // Unknown until we estimate the gas.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub gas_limit: Option<U256>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub max_fee_per_gas: Option<U256>, // conflicts with gas_price

    pub ccip_read_enabled: bool,
}

impl Eip712TransactionRequest {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn from_overrides(overrides: Overrides) -> Self {
        let mut tx = Self::default();
        if let Some(value) = overrides.value {
            tx.value = value;
        }
        tx
    }

    pub fn to<T>(mut self, to: T) -> Self
    where
        T: Into<Address>,
    {
        self.to = to.into();
        self
    }

    pub fn from<T>(mut self, from: T) -> Self
    where
        T: Into<Address>,
    {
        self.from = from.into();
        self
    }

    pub fn nonce<T>(mut self, nonce: T) -> Self
    where
        T: Into<U256>,
    {
        self.nonce = nonce.into();
        self
    }

    pub fn gas_limit<T>(mut self, gas_limit: T) -> Self
    where
        T: Into<U256>,
    {
        self.gas_limit = Some(gas_limit.into());
        self
    }

    pub fn gas_price<T>(mut self, gas_price: T) -> Self
    where
        T: Into<U256>,
    {
        self.gas_price = gas_price.into();
        self
    }

    pub fn data<T>(mut self, data: T) -> Self
    where
        T: Into<Bytes>,
    {
        self.data = data.into();
        self
    }

    pub fn value<T>(mut self, value: T) -> Self
    where
        T: Into<U256>,
    {
        self.value = value.into();
        self
    }

    pub fn chain_id<T>(mut self, chain_id: T) -> Self
    where
        T: Into<U256>,
    {
        self.chain_id = chain_id.into();
        self
    }

    pub fn r#type<T>(mut self, r#type: T) -> Self
    where
        T: Into<U256>,
    {
        self.r#type = r#type.into();
        self
    }

    pub fn access_list<T>(mut self, access_list: AccessList) -> Self {
        self.access_list = Some(access_list);
        self
    }

    pub fn max_priority_fee_per_gas<T>(mut self, max_priority_fee_per_gas: T) -> Self
    where
        T: Into<U256>,
    {
        self.max_priority_fee_per_gas = max_priority_fee_per_gas.into();
        self
    }

    pub fn max_fee_per_gas<T>(mut self, max_fee_per_gas: T) -> Self
    where
        T: Into<U256>,
    {
        self.max_fee_per_gas = Some(max_fee_per_gas.into());
        self
    }

    pub fn custom_data(mut self, custom_data: Eip712Meta) -> Self {
        self.custom_data = custom_data;
        self
    }

    pub fn ccip_read_enabled(mut self, ccip_read_enabled: bool) -> Self {
        self.ccip_read_enabled = ccip_read_enabled;
        self
    }

    pub fn rlp_unsigned(&self) -> Result<Bytes, Eip712Error> {
        self.rlp(None)
    }

    pub fn rlp_signed(&self, signature: Signature) -> Result<Bytes, Eip712Error> {
        self.rlp(Some(signature))
    }

    pub fn rlp(&self, signature: Option<Signature>) -> Result<Bytes, Eip712Error> {
        let mut rlp = RlpStream::new();
        rlp.begin_unbounded_list();
        rlp.append(&self.nonce);
        rlp.append(&self.max_priority_fee_per_gas);
        rlp.append(&self.gas_price);
        rlp_append_option(&mut rlp, self.gas_limit);
        rlp.append(&self.to);
        rlp.append(&self.value);
        rlp.append(&self.data.0);
        if let Some(sig) = signature {
            rlp.append(&sig.v);
            // Convert to big-endian bytes (32 bytes in total)
            let mut bytes = [0_u8; 32]; // U256 is 32 bytes
            sig.r.to_big_endian(&mut bytes);
            rlp.append(&bytes.as_slice());
            sig.s.to_big_endian(&mut bytes);
            rlp.append(&bytes.as_slice());
        }
        rlp.append(&self.chain_id);
        rlp.append(&self.from);
        self.custom_data.rlp_append(&mut rlp);
        rlp.finalize_unbounded_list();
        Ok(rlp.out().freeze().into())
    }
}

impl Default for Eip712TransactionRequest {
    fn default() -> Self {
        Self {
            to: Default::default(),
            from: Default::default(),
            nonce: Default::default(),
            gas: Default::default(),
            gas_limit: Default::default(),
            gas_price: Default::default(),
            data: Default::default(),
            value: Default::default(),
            chain_id: ERA_CHAIN_ID.into(),
            r#type: EIP712_TX_TYPE.into(),
            access_list: Default::default(),
            max_priority_fee_per_gas: MAX_PRIORITY_FEE_PER_GAS.into(),
            max_fee_per_gas: Default::default(),
            custom_data: Default::default(),
            ccip_read_enabled: Default::default(),
        }
    }
}

impl TryFrom<WithdrawRequest> for Eip712TransactionRequest {
    type Error = ZKRequestError;

    fn try_from(request: WithdrawRequest) -> Result<Self, Self::Error> {
        let contract_address =
            Address::from_str(zks_utils::CONTRACTS_L2_ETH_TOKEN_ADDR).map_err(|e| {
                ZKRequestError::CustomError(format!("Error getting L2 ETH token address {e:?}"))
            })?;
        let function_signature = "function withdraw(address _l1Receiver) external payable override";
        let function = HumanReadableParser::parse_function(function_signature)
            .map_err(ParseError::LexerError)?;
        let function_args = function.decode_input(&zks_utils::encode_args(
            &function,
            &[format!("{:?}", request.to)],
        )?)?;
        let data: Bytes = function.encode_input(&function_args)?.into();

        Ok(Eip712TransactionRequest::new()
            .r#type(EIP712_TX_TYPE)
            .to(contract_address)
            .value(request.amount)
            .from(request.from)
            .data(data))
    }
}

impl From<TransferRequest> for Eip712TransactionRequest {
    fn from(request: TransferRequest) -> Self {
        Eip712TransactionRequest::new()
            .r#type(EIP712_TX_TYPE)
            .to(request.to)
            .value(request.amount)
            .from(request.from)
    }
}

impl TryFrom<DeployRequest> for Eip712TransactionRequest {
    type Error = ZKRequestError;

    fn try_from(request: DeployRequest) -> Result<Self, Self::Error> {
        let mut contract_deployer_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
        contract_deployer_path.push("src/abi/ContractDeployer.json");

        let custom_data = Eip712Meta::new().factory_deps({
            let mut factory_deps = Vec::new();
            if let Some(factory_dependencies) = request.factory_deps {
                factory_deps.extend(factory_dependencies);
            }
            factory_deps.push(request.contract_bytecode.clone());
            factory_deps
        });

        let contract_deployer = Abi::load(BufReader::new(
            File::open(contract_deployer_path).map_err(|e| {
                ZKRequestError::CustomError(format!(
                    "Error opening contract deployer abi file {e:?}"
                ))
            })?,
        ))?;
        let create = contract_deployer.function("create")?;

        // TODO: User could provide this instead of defaulting.
        let salt = [0_u8; 32];
        let bytecode_hash = hash_bytecode(&request.contract_bytecode).map_err(|e| {
            ZKRequestError::CustomError(format!("Error hashing contract bytecode {e:?}"))
        })?;
        let call_data: Bytes = match (
            request.contract_abi.constructor(),
            request.constructor_parameters.is_empty(),
        ) {
            (None, false) => {
                return Err(ZKRequestError::CustomError(
                    "Constructor not present".to_owned(),
                ))
            }
            (None, true) | (Some(_), true) => Bytes::default(),
            (Some(constructor), false) => {
                zks_utils::encode_constructor_args(constructor, &request.constructor_parameters)?
                    .into()
            }
        };

        let data = encode_function_data(create, (salt, bytecode_hash, call_data))?;

        let contract_deployer_address = Address::from_str(CONTRACT_DEPLOYER_ADDR).map_err(|e| {
            ZKRequestError::CustomError(format!("Error getting contract deployer address {e:?}"))
        })?;
        Ok(Eip712TransactionRequest::new()
            .r#type(EIP712_TX_TYPE)
            .to(contract_deployer_address)
            .custom_data(custom_data)
            .data(data))
    }
}