mpp-br 0.8.1

//! Tempo-specific client errors.
//!
//! These errors represent failures specific to the Tempo payment method's
//! client-side operations (transaction building, gas estimation, signing).
//! They are scoped under `client::tempo` to make clear they belong to a
//! single payment method, not the protocol layer.

use alloy::transports::TransportError;
use tempo_alloy::contracts::precompiles::{IAccountKeychain, ITIP20};
use thiserror::Error;

/// Errors specific to Tempo client-side payment operations.
#[derive(Error, Debug)]
pub enum TempoClientError {
    /// Access key is not provisioned on the wallet's keychain.
    #[error("Access key not provisioned on wallet")]
    AccessKeyNotProvisioned,

    /// Spending limit exceeded for a token.
    #[error("Spending limit exceeded for {token}: limit is {limit}, need {required}")]
    SpendingLimitExceeded {
        /// Token symbol or address.
        token: String,
        /// Current spending limit (human-readable).
        limit: String,
        /// Required amount (human-readable).
        required: String,
    },

    /// Wallet has insufficient balance to complete the payment.
    #[error("Insufficient {token} balance: have {available}, need {required}")]
    InsufficientBalance {
        /// Token symbol or address.
        token: String,
        /// Available balance (human-readable).
        available: String,
        /// Required amount (human-readable).
        required: String,
    },

    /// Transaction reverted on-chain (e.g., during gas estimation or broadcast).
    #[error("Transaction reverted: {0}")]
    TransactionReverted(String),
}

/// Extract the value for `key` from a revert string like `"{ key: value, ... }"`.
///
/// Looks for `"{key}: "` and returns the text up to the next `,`, `}`, or `)`.
fn extract_field(raw: &str, key: &str) -> Option<String> {
    let needle = format!("{key}: ");
    let start = raw.find(&needle)? + needle.len();
    let rest = &raw[start..];
    let end = rest.find([',', '}', ')']).unwrap_or(rest.len());
    let value = rest[..end].trim();
    if value.is_empty() {
        return None;
    }
    Some(value.to_string())
}

impl TempoClientError {
    /// Classify a transport error by first trying ABI-decoded revert data,
    /// then falling back to string-based classification.
    pub fn from_transport_error(e: &TransportError) -> Option<Self> {
        if let Some(payload) = e.as_error_resp() {
            if let Some(err) = payload.as_decoded_error::<ITIP20::InsufficientBalance>() {
                return Some(Self::InsufficientBalance {
                    token: err.token.to_string(),
                    available: err.available.to_string(),
                    required: err.required.to_string(),
                });
            }
            if payload
                .as_decoded_error::<IAccountKeychain::SpendingLimitExceeded>()
                .is_some()
            {
                let msg = e.to_string();
                return Some(Self::SpendingLimitExceeded {
                    token: extract_field(&msg, "token").unwrap_or_default(),
                    limit: extract_field(&msg, "limit").unwrap_or_default(),
                    required: extract_field(&msg, "required").unwrap_or(msg),
                });
            }
        }

        Self::classify_rpc_error(e.to_string())
    }

    /// Fallback: classify an error from its display string.
    ///
    /// Detects common Tempo revert reasons via substring matching.
    /// Returns `None` if the message doesn't look Tempo-specific.
    fn classify_rpc_error(msg: impl Into<String>) -> Option<Self> {
        let msg = msg.into();
        let lower = msg.to_lowercase();

        if lower.contains("not provisioned") {
            return Some(Self::AccessKeyNotProvisioned);
        }

        if lower.contains("spendinglimitexceeded") || lower.contains("spending limit") {
            return Some(Self::SpendingLimitExceeded {
                token: extract_field(&msg, "token").unwrap_or_default(),
                limit: extract_field(&msg, "limit").unwrap_or_default(),
                required: extract_field(&msg, "required").unwrap_or(msg),
            });
        }

        if lower.contains("insufficientbalance")
            || lower.contains("transfer amount exceeds balance")
            || (lower.contains("insufficient") && lower.contains("balance"))
        {
            return Some(Self::InsufficientBalance {
                token: extract_field(&msg, "token").unwrap_or_default(),
                available: extract_field(&msg, "available").unwrap_or_default(),
                required: extract_field(&msg, "required").unwrap_or(msg),
            });
        }

        if lower.contains("revert") || lower.contains("execution reverted") {
            return Some(Self::TransactionReverted(msg));
        }

        None
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_access_key_not_provisioned_display() {
        let err = TempoClientError::AccessKeyNotProvisioned;
        assert_eq!(err.to_string(), "Access key not provisioned on wallet");
    }

    #[test]
    fn test_spending_limit_exceeded_display() {
        let err = TempoClientError::SpendingLimitExceeded {
            token: "USDC".to_string(),
            limit: "0.50".to_string(),
            required: "1.00".to_string(),
        };
        assert_eq!(
            err.to_string(),
            "Spending limit exceeded for USDC: limit is 0.50, need 1.00"
        );
    }

    #[test]
    fn test_insufficient_balance_display() {
        let err = TempoClientError::InsufficientBalance {
            token: "pathUSD".to_string(),
            available: "0.50".to_string(),
            required: "1.00".to_string(),
        };
        assert_eq!(
            err.to_string(),
            "Insufficient pathUSD balance: have 0.50, need 1.00"
        );
    }

    #[test]
    fn test_transaction_reverted_display() {
        let err = TempoClientError::TransactionReverted("SpendingLimitExceeded".to_string());
        assert_eq!(
            err.to_string(),
            "Transaction reverted: SpendingLimitExceeded"
        );
    }

    // --- classify_rpc_error ---

    #[test]
    fn test_classify_not_provisioned() {
        let err = TempoClientError::classify_rpc_error("key not provisioned on wallet");
        assert!(matches!(
            err,
            Some(TempoClientError::AccessKeyNotProvisioned)
        ));
    }

    #[test]
    fn test_classify_spending_limit() {
        let err =
            TempoClientError::classify_rpc_error("SpendingLimitExceeded: limit is 0.50, need 1.00");
        assert!(matches!(
            err,
            Some(TempoClientError::SpendingLimitExceeded { .. })
        ));
    }

    #[test]
    fn test_classify_insufficient_balance() {
        let err = TempoClientError::classify_rpc_error("InsufficientBalance for transfer");
        assert!(matches!(
            err,
            Some(TempoClientError::InsufficientBalance { .. })
        ));
    }

    #[test]
    fn test_classify_transfer_exceeds_balance() {
        let err = TempoClientError::classify_rpc_error("transfer amount exceeds balance");
        assert!(matches!(
            err,
            Some(TempoClientError::InsufficientBalance { .. })
        ));
    }

    #[test]
    fn test_classify_generic_revert() {
        let err = TempoClientError::classify_rpc_error("execution reverted: some reason");
        assert!(matches!(
            err,
            Some(TempoClientError::TransactionReverted(_))
        ));
    }

    #[test]
    fn test_classify_unknown_returns_none() {
        let err = TempoClientError::classify_rpc_error("failed to get nonce");
        assert!(err.is_none());
    }

    // --- extract_field ---

    #[test]
    fn test_extract_field_basic() {
        let raw = "InsufficientBalance { available: 0, required: 64467, token: 0xabc }";
        assert_eq!(extract_field(raw, "available"), Some("0".to_string()));
        assert_eq!(extract_field(raw, "required"), Some("64467".to_string()));
        assert_eq!(extract_field(raw, "token"), Some("0xabc".to_string()));
    }

    #[test]
    fn test_extract_field_missing() {
        assert_eq!(extract_field("no fields here", "token"), None);
    }

    // --- classify with field extraction ---

    #[test]
    fn test_classify_insufficient_balance_with_fields() {
        let msg = "server returned error: execution reverted: revert: InsufficientBalance(InsufficientBalance { available: 0, required: 64467, token: 0x20c000000000000000000000b9537d11c60e8b50 })";
        let err = TempoClientError::classify_rpc_error(msg).unwrap();
        match err {
            TempoClientError::InsufficientBalance {
                token,
                available,
                required,
            } => {
                assert_eq!(available, "0");
                assert_eq!(required, "64467");
                assert_eq!(token, "0x20c000000000000000000000b9537d11c60e8b50");
            }
            other => panic!("expected InsufficientBalance, got {other:?}"),
        }
    }

    #[test]
    fn test_classify_spending_limit_with_fields() {
        let msg = "execution reverted: SpendingLimitExceeded(SpendingLimitExceeded { token: 0xabc, limit: 1000, required: 5000 })";
        let err = TempoClientError::classify_rpc_error(msg).unwrap();
        match err {
            TempoClientError::SpendingLimitExceeded {
                token,
                limit,
                required,
            } => {
                assert_eq!(token, "0xabc");
                assert_eq!(limit, "1000");
                assert_eq!(required, "5000");
            }
            other => panic!("expected SpendingLimitExceeded, got {other:?}"),
        }
    }

    #[test]
    fn test_classify_insufficient_balance_fallback_no_fields() {
        let msg = "InsufficientBalance for transfer";
        let err = TempoClientError::classify_rpc_error(msg).unwrap();
        match err {
            TempoClientError::InsufficientBalance {
                token,
                available,
                required,
            } => {
                assert!(token.is_empty());
                assert!(available.is_empty());
                assert_eq!(required, msg);
            }
            other => panic!("expected InsufficientBalance, got {other:?}"),
        }
    }

    #[test]
    fn test_classify_spending_limit_fallback_no_fields() {
        let msg = "SpendingLimitExceeded: limit is 0.50, need 1.00";
        let err = TempoClientError::classify_rpc_error(msg).unwrap();
        match err {
            TempoClientError::SpendingLimitExceeded {
                token,
                limit,
                required,
            } => {
                assert!(token.is_empty());
                assert!(limit.is_empty());
                assert_eq!(required, msg);
            }
            other => panic!("expected SpendingLimitExceeded, got {other:?}"),
        }
    }
}