zerodev-aa 0.0.1-alpha.1

/// 20-byte Ethereum address.
#[derive(Clone, Copy, PartialEq, Eq, Hash)]
pub struct Address(pub [u8; 20]);

impl Address {
    /// Parse from a hex string (with or without 0x prefix).
    pub fn from_hex(s: &str) -> Result<Self, hex::FromHexError> {
        let s = s.strip_prefix("0x").or_else(|| s.strip_prefix("0X")).unwrap_or(s);
        let mut bytes = [0u8; 20];
        hex::decode_to_slice(s, &mut bytes)?;
        Ok(Self(bytes))
    }

    /// Return as a 0x-prefixed lowercase hex string.
    pub fn to_hex(&self) -> String {
        format!("0x{}", hex::encode(self.0))
    }
}

impl std::fmt::Debug for Address {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "Address({})", self.to_hex())
    }
}

impl std::fmt::Display for Address {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.to_hex())
    }
}

/// 32-byte hash (e.g. UserOp hash, tx hash).
#[derive(Clone, Copy, PartialEq, Eq, Hash)]
pub struct Hash(pub [u8; 32]);

impl Hash {
    /// Return as a 0x-prefixed lowercase hex string.
    pub fn to_hex(&self) -> String {
        format!("0x{}", hex::encode(self.0))
    }

    /// Returns true if all bytes are zero.
    pub fn is_zero(&self) -> bool {
        self.0.iter().all(|&b| b == 0)
    }
}

impl std::fmt::Debug for Hash {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "Hash({})", self.to_hex())
    }
}

impl std::fmt::Display for Hash {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.to_hex())
    }
}

/// Kernel smart account version.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum KernelVersion {
    V3_1,
    V3_2,
    V3_3,
}

impl KernelVersion {
    pub(crate) fn to_c(self) -> i32 {
        match self {
            Self::V3_1 => 0,
            Self::V3_2 => 1,
            Self::V3_3 => 2,
        }
    }
}

/// Gas pricing middleware provider.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum GasMiddleware {
    /// ZeroDev: calls zd_getUserOperationGasPrice.
    ZeroDev,
}

/// Paymaster sponsorship middleware provider.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PaymasterMiddleware {
    /// No paymaster — send unsponsored (user pays gas).
    None,
    /// ZeroDev: calls pm_getPaymasterStubData / pm_getPaymasterData.
    ZeroDev,
}

/// Full receipt from eth_getUserOperationReceipt.
/// Matches the viem UserOperationReceipt type.
///
/// Contains both typed fields and the raw JSON string for full access.
#[derive(Debug, Clone)]
pub struct UserOperationReceipt {
    /// The raw JSON response from eth_getUserOperationReceipt.
    pub json: String,
    /// Hash of the user operation.
    pub user_op_hash: String,
    /// Entrypoint address.
    pub entry_point: String,
    /// Sender address.
    pub sender: String,
    /// Anti-replay parameter (hex string).
    pub nonce: String,
    /// Paymaster address, if any.
    pub paymaster: Option<String>,
    /// Actual gas cost (hex string).
    pub actual_gas_cost: String,
    /// Actual gas used (hex string).
    pub actual_gas_used: String,
    /// If the user operation execution was successful.
    pub success: bool,
    /// Revert reason, if unsuccessful.
    pub reason: Option<String>,
}

impl UserOperationReceipt {
    pub(crate) fn from_json(json: String) -> Self {
        // Simple JSON field extraction without adding a serde dependency.
        fn extract_string(json: &str, key: &str) -> String {
            let needle = format!("\"{}\":\"", key);
            if let Some(start) = json.find(&needle) {
                let val_start = start + needle.len();
                if let Some(end) = json[val_start..].find('"') {
                    return json[val_start..val_start + end].to_string();
                }
            }
            String::new()
        }

        fn extract_bool(json: &str, key: &str) -> bool {
            let needle = format!("\"{}\":", key);
            if let Some(start) = json.find(&needle) {
                let rest = &json[start + needle.len()..];
                return rest.trim_start().starts_with("true");
            }
            false
        }

        fn extract_optional_string(json: &str, key: &str) -> Option<String> {
            let s = extract_string(json, key);
            if s.is_empty() { None } else { Some(s) }
        }

        Self {
            user_op_hash: extract_string(&json, "userOpHash"),
            entry_point: extract_string(&json, "entryPoint"),
            sender: extract_string(&json, "sender"),
            nonce: extract_string(&json, "nonce"),
            paymaster: extract_optional_string(&json, "paymaster"),
            actual_gas_cost: extract_string(&json, "actualGasCost"),
            actual_gas_used: extract_string(&json, "actualGasUsed"),
            success: extract_bool(&json, "success"),
            reason: extract_optional_string(&json, "reason"),
            json,
        }
    }
}

/// A single call within a UserOperation.
pub struct Call {
    /// Target contract address.
    pub target: Address,
    /// Value in wei (big-endian u256).
    pub value: [u8; 32],
    /// Calldata bytes.
    pub calldata: Vec<u8>,
}