pokebase 0.1.0

An embedded database of Pokémon TCG powered by TCGdex
Documentation 
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use crate::session;
use crate::{Card, Error};

use bytes::Bytes;
use serde::Deserialize;

#[derive(Debug, Clone)]
pub struct PokemonTcg {
    client: reqwest::Client,
    api_key: Option<String>,
}

impl PokemonTcg {
    pub fn new(api_key: Option<String>) -> Self {
        if let Some(api_key) = api_key.as_ref() {
            log::info!(
                "PokemonTCG session started (API key: {})",
                api_key.replace(|c: char| c.is_alphanumeric(), "*")
            )
        } else {
            log::warn!("PokemonTCG session started without an API key!")
        }

        Self {
            client: session::CLIENT.clone(),
            api_key,
        }
    }

    pub async fn download_image(&self, card: &Card) -> Result<Bytes, Error> {
        if !card.name.has_english() {
            return Err(Error::LocaleNotAvailable);
        }

        let set = set_name(card);
        let number = card_number(card);
        let url = format!("https://images.pokemontcg.io/{set}/{number}_hires.png");

        log::info!("Downloading image: {url}");
        let response = session::retry(2, || self.get(&url).send()).await?;

        Ok(response.error_for_status()?.bytes().await?)
    }

    pub async fn fetch_pricing(&self, card: &Card) -> Result<Pricing, Error> {
        if !card.name.has_english() {
            return Err(Error::LocaleNotAvailable);
        }

        let number = card_number(card);
        let set = set_name(card);

        let url = format!("https://api.pokemontcg.io/v2/cards/{set}-{number}");

        #[derive(Deserialize)]
        struct Response {
            data: Pricing,
        }

        log::info!("Fetching price: {url}");

        let response =
            session::retry(2, async || self.get(&url).send().await?.error_for_status()).await?;
        let response: Response = response.json().await?;

        Ok(response.data)
    }

    fn get(&self, url: impl AsRef<str>) -> reqwest::RequestBuilder {
        let request = self.client.get(url.as_ref());

        if let Some(api_key) = &self.api_key {
            request.header("X-Api-Key", api_key)
        } else {
            request
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Deserialize)]
pub struct Pricing {
    #[serde(default)]
    pub tcgplayer: tcgplayer::Pricing,
    #[serde(default)]
    pub cardmarket: cardmarket::Pricing,
}

fn set_name(card: &Card) -> String {
    // PokemonTCG does not pad with leading 0s
    let prefix: String = card
        .set
        .as_str()
        .chars()
        .take_while(|c| !c.is_ascii_digit())
        .collect();

    let number = &card.set.as_str()[prefix.len()..];

    let replacement =
        if prefix.starts_with("swsh") && number.starts_with("12") || prefix.starts_with("sv") {
            "pt"
        } else {
            ""
        };

    let mut name = format!(
        "{}{}",
        prefix.replace(".", replacement),
        number.trim_start_matches('0').replace(".", replacement)
    );

    if card_number(card).starts_with("TG") {
        name.push_str("tg");
    }

    name
}

fn card_number(card: &Card) -> &str {
    card.id
        .as_str()
        .rsplit("-")
        .next()
        .unwrap_or(card.id.as_str())
        .trim_start_matches('0')
}