amico-mods 0.0.4

The plugins of the Amico AI Agent Framework
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use alloy::signers::local::LocalSignerError;
use alloy::signers::local::{MnemonicBuilder, coins_bip39::English};
use alloy::signers::{k256::Secp256k1, local::LocalSigner};
use bip39::Mnemonic;
use ecdsa::SigningKey;

use crate::web3::wallet::{WalletComponent, WalletError};

/// The local signer type alias
pub type EthereumSigner = LocalSigner<SigningKey<Secp256k1>>;

/// A wrapper around the local signer that implements the `WalletComponent` trait.
pub struct EthereumWallet(EthereumSigner);

/// Error type for Ethereum wallet operations.
#[derive(Debug, thiserror::Error)]
pub enum EthereumWalletError {
    #[error("Ethereum local signer error: {0}")]
    LocalSignerError(#[from] LocalSignerError),
}

impl WalletComponent for EthereumWallet {
    type Signer = EthereumSigner;

    fn from_mnemonic(mnemonic: &Mnemonic) -> Result<Self, WalletError> {
        let phrase = mnemonic.phrase();
        let signer = MnemonicBuilder::<English>::default()
            .phrase(phrase)
            .build()
            .map_err(EthereumWalletError::LocalSignerError)?;
        Ok(EthereumWallet(signer))
    }

    fn pubkey(&self) -> String {
        self.0.address().to_string()
    }

    fn get(&self) -> &Self::Signer {
        &self.0
    }
}

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

    use bip39::{Language::English, MnemonicType::Words12};

    #[test]
    fn test_from_mnemonic() {
        let mnemonic = Mnemonic::new(Words12, English);
        let eth_wallet =
            EthereumWallet::from_mnemonic(&mnemonic).expect("Failed to load Ethereum wallet");
        assert!(!eth_wallet.pubkey().to_string().is_empty());
    }
}