aimo_core/
token_map.rs

1use std::{str::FromStr, sync::OnceLock};
2
3use solana_sdk::pubkey::Pubkey;
4
5static TOKEN_MAP: OnceLock<TokenMap> = OnceLock::new();
6
7pub fn token_map(devnet_tokens: bool) -> &'static TokenMap {
8    TOKEN_MAP.get_or_init(|| TokenMap::new(devnet_tokens))
9}
10
11pub type TokenType = &'static Token;
12
13#[derive(Debug, Clone)]
14pub struct Token {
15    pub mint: Pubkey,
16    pub symbol: &'static str,
17    pub decimals: u32,
18}
19
20#[derive(Debug, Clone)]
21pub struct TokenMap {
22    pub usdc: Token,
23}
24
25impl TokenMap {
26    pub fn new(devnet_tokens: bool) -> Self {
27        if devnet_tokens {
28            // Devnet token metadata
29            Self {
30                usdc: Token {
31                    mint: Pubkey::from_str_const("4zMMC9srt5Ri5X14GAgXhaHii3GnPAEERYPJgZJDncDU"),
32                    symbol: "USDC",
33                    decimals: 6,
34                },
35            }
36        } else {
37            // Mainnet token metadata
38            Self {
39                usdc: Token {
40                    mint: Pubkey::from_str_const("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v"),
41                    symbol: "USDC",
42                    decimals: 6,
43                },
44            }
45        }
46    }
47
48    /// Find a supported token either by name or by pubkey string.
49    ///
50    /// Returns `None` if the token is not supported.
51    ///
52    /// # Examples
53    /// ```
54    /// let token_map = token_map(false);
55    ///
56    /// // Find by symbol
57    /// let usdc_by_symbol = token_map.find("USDC");
58    ///
59    /// // Find by mint address  
60    /// let usdc_by_mint = token_map.find("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v");
61    /// ```
62    pub fn find(&'static self, token: &str) -> Option<TokenType> {
63        match token.to_ascii_uppercase().as_str() {
64            "USDC" => Some(&self.usdc),
65
66            // match other token names here:
67            // "USDT": => xxxx
68
69            // Fallback to token mint pubkey
70            _ => {
71                Pubkey::from_str(token).ok().and_then(|mint| {
72                    [
73                        // List other tokens here
74                        &self.usdc,
75                    ]
76                    .iter()
77                    .find(|t| t.mint == mint)
78                    .map(|v| &**v)
79                })
80            }
81        }
82    }
83}
aimo_core/token_map.rs

aimo_core/
token_map.rs
