1use std::str::FromStr;
2
3use serde::de::Error;
4use serde::{Deserialize, Deserializer};
5use solana_program::pubkey::Pubkey;
6
7pub mod serum;
8pub mod spl_token;
9
10pub fn deserialize_pubkey<'de, D>(deserializer: D) -> Result<Pubkey, D::Error>
11where
12 D: Deserializer<'de>,
13{
14 let s: &str = Deserialize::deserialize(deserializer)?;
15 Pubkey::from_str(&s).map_err(|_| D::Error::custom(format!("invalid base58 address: {}", s)))
16}
17
18#[cfg(test)]
19mod tests {
20 use super::*;
21
22 #[derive(Deserialize)]
23 struct PubkeyContainer {
24 #[serde(deserialize_with = "deserialize_pubkey")]
25 pubkey: Pubkey,
26 }
27
28 #[test]
29 fn test_deserialize_pubkey() {
30 let input = r#"{ "pubkey": "So11111111111111111111111111111111111111112" }"#;
31 let PubkeyContainer { pubkey } = serde_json::from_str(input).unwrap();
32 assert_eq!(
33 pubkey,
34 Pubkey::from_str("So11111111111111111111111111111111111111112").unwrap()
35 );
36
37 let malformed_input = r#"{ "pubkey": "Solana1111111111111111111111111111111111112" }"#;
38 assert!(serde_json::from_str::<PubkeyContainer>(malformed_input).is_err());
39 }
40}