82 lines
1.7 KiB
Rust
82 lines
1.7 KiB
Rust
use {
|
|
solana_account_decoder::parse_token::real_number_string_trimmed,
|
|
solana_sdk::native_token::lamports_to_sol,
|
|
std::{
|
|
fmt::{Debug, Display, Formatter, Result},
|
|
ops::Add,
|
|
},
|
|
};
|
|
|
|
const SOL_SYMBOL: &str = "◎";
|
|
|
|
#[derive(PartialEq, Eq)]
|
|
pub enum TokenType {
|
|
Sol,
|
|
SplToken,
|
|
}
|
|
|
|
pub struct Token {
|
|
amount: u64,
|
|
decimals: u8,
|
|
token_type: TokenType,
|
|
}
|
|
|
|
impl Token {
|
|
fn write_with_symbol(&self, f: &mut Formatter) -> Result {
|
|
match &self.token_type {
|
|
TokenType::Sol => {
|
|
let amount = lamports_to_sol(self.amount);
|
|
write!(f, "{SOL_SYMBOL}{amount}")
|
|
}
|
|
TokenType::SplToken => {
|
|
let amount = real_number_string_trimmed(self.amount, self.decimals);
|
|
write!(f, "{amount} tokens")
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn sol(amount: u64) -> Self {
|
|
Self {
|
|
amount,
|
|
decimals: 9,
|
|
token_type: TokenType::Sol,
|
|
}
|
|
}
|
|
|
|
pub fn spl_token(amount: u64, decimals: u8) -> Self {
|
|
Self {
|
|
amount,
|
|
decimals,
|
|
token_type: TokenType::SplToken,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Display for Token {
|
|
fn fmt(&self, f: &mut Formatter) -> Result {
|
|
self.write_with_symbol(f)
|
|
}
|
|
}
|
|
|
|
impl Debug for Token {
|
|
fn fmt(&self, f: &mut Formatter) -> Result {
|
|
self.write_with_symbol(f)
|
|
}
|
|
}
|
|
|
|
impl Add for Token {
|
|
type Output = Token;
|
|
|
|
fn add(self, other: Self) -> Self {
|
|
if self.token_type == other.token_type {
|
|
Self {
|
|
amount: self.amount + other.amount,
|
|
decimals: self.decimals,
|
|
token_type: self.token_type,
|
|
}
|
|
} else {
|
|
self
|
|
}
|
|
}
|
|
}
|