amico-mods 0.0.2

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

use bip39::{Mnemonic, Seed};
use solana_sdk::{
    signature::Keypair,
    signer::{SeedDerivable, Signer},
};

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

/// A wrapper around the Solana keypair that implements the `WalletComponent` trait.
pub struct SolanaWallet(Keypair);

/// Error type for Solana wallet operations.
#[derive(Debug, thiserror::Error)]
pub enum SolanaWalletError {
    #[error("Solana keypair error: {0}")]
    KeypairError(String),
}

impl WalletComponent for SolanaWallet {
    type Signer = Keypair;

    fn from_mnemonic(mnemonic: &Mnemonic) -> Result<Self, WalletError> {
        let seed = Seed::new(mnemonic, "");
        let keypair = Keypair::from_seed(seed.as_bytes())
            .map_err(|err| SolanaWalletError::KeypairError(err.to_string()))?;
        Ok(SolanaWallet(keypair))
    }

    fn pubkey(&self) -> String {
        self.0.pubkey().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 keypair =
            SolanaWallet::from_mnemonic(&mnemonic).expect("Failed to load Solana keypair");
        assert!(!keypair.pubkey().to_string().is_empty());
    }
}