1#![allow(clippy::arithmetic_side_effects)]
2#![deny(missing_docs)]
3#![cfg_attr(not(test), forbid(unsafe_code))]
4#![allow(unexpected_cfgs)]
5
6pub mod error;
9pub mod instruction;
10pub mod processor;
11pub mod state;
12
13#[cfg(not(feature = "no-entrypoint"))]
14mod entrypoint;
15
16pub use arch_program;
19use arch_program::{entrypoint::ProgramResult, program_error::ProgramError, pubkey::Pubkey};
20
21pub fn ui_amount_to_amount(ui_amount: f64, decimals: u8) -> u64 {
24 (ui_amount * 10_usize.pow(decimals as u32) as f64) as u64
25}
26
27pub fn amount_to_ui_amount(amount: u64, decimals: u8) -> f64 {
30 amount as f64 / 10_usize.pow(decimals as u32) as f64
31}
32
33pub fn amount_to_ui_amount_string(amount: u64, decimals: u8) -> String {
36 let decimals = decimals as usize;
37 if decimals > 0 {
38 let mut s = format!("{:01$}", amount, decimals + 1);
40 s.insert(s.len() - decimals, '.');
42 s
43 } else {
44 amount.to_string()
45 }
46}
47
48pub fn amount_to_ui_amount_string_trimmed(amount: u64, decimals: u8) -> String {
51 let mut s = amount_to_ui_amount_string(amount, decimals);
52 if decimals > 0 {
53 let zeros_trimmed = s.trim_end_matches('0');
54 s = zeros_trimmed.trim_end_matches('.').to_string();
55 }
56 s
57}
58
59pub fn try_ui_amount_into_amount(ui_amount: String, decimals: u8) -> Result<u64, ProgramError> {
62 let decimals = decimals as usize;
63 let mut parts = ui_amount.split('.');
64 let mut amount_str = parts.next().unwrap().to_string();
67 let after_decimal = parts.next().unwrap_or("");
68 let after_decimal = after_decimal.trim_end_matches('0');
69 if (amount_str.is_empty() && after_decimal.is_empty())
70 || parts.next().is_some()
71 || after_decimal.len() > decimals
72 {
73 return Err(ProgramError::InvalidArgument);
74 }
75
76 amount_str.push_str(after_decimal);
77 for _ in 0..decimals.saturating_sub(after_decimal.len()) {
78 amount_str.push('0');
79 }
80 amount_str
81 .parse::<u64>()
82 .map_err(|_| ProgramError::InvalidArgument)
83}
84
85pub fn id() -> Pubkey {
87 Pubkey::from_slice(b"AplToken111111111111111111111111")
88}
89
90pub const ID: Pubkey = Pubkey::new_from_array(*b"AplToken111111111111111111111111");
92
93pub fn check_program_account(spl_token_program_id: &Pubkey) -> ProgramResult {
95 if spl_token_program_id != &id() {
96 return Err(ProgramError::IncorrectProgramId);
97 }
98 Ok(())
99}