nekosbest 0.14.1

type E = Box<dyn std::error::Error>;

#[tokio::main]
async fn main() -> Result<(), E> {
    #[allow(unused_variables)]
    let client = reqwest::Client::new();
    #[cfg(feature = "local")]
    {
        use std::{collections::HashMap, convert::TryFrom, path::PathBuf, str::FromStr};

        use proc_macro2::TokenStream;
        use quote::{format_ident, quote, ToTokens, TokenStreamExt};

        #[derive(serde::Deserialize)]
        struct EndpointDescInternal {
            min: String,
            max: String,
            format: String,
        }

        #[derive(serde::Deserialize)]
        #[serde(try_from = "EndpointDescInternal")]
        struct EndpointDesc {
            min: usize,
            max: usize,
            format: String,
            with_padding: usize,
        }

        impl TryFrom<EndpointDescInternal> for EndpointDesc {
            type Error = <usize as FromStr>::Err;

            fn try_from(d: EndpointDescInternal) -> Result<Self, Self::Error> {
                Ok(Self {
                    min: d.min.parse()?,
                    max: d.max.parse()?,
                    format: d.format,
                    with_padding: d.max.len(),
                })
            }
        }

        const BASE_URL: &str = "https://nekos.best/api/v2";

        async fn get_endpoints_data(
            client: &reqwest::Client,
        ) -> Result<HashMap<String, EndpointDesc>, E> {
            if let Ok(_) = std::env::var("DOCS_RS") {
                // we are on docs.rs

                let endpoints_json = r#"{"highfive":{"format":"gif","min":"001","max":"020"},"happy":{"format":"gif","min":"001","max":"012"},"sleep":{"format":"gif","min":"001","max":"012"},"handhold":{"format":"gif","min":"001","max":"025"},"laugh":{"format":"gif","min":"001","max":"032"},"bite":{"format":"gif","min":"001","max":"017"},"poke":{"format":"gif","min":"001","max":"028"},"tickle":{"format":"gif","min":"001","max":"021"},"kiss":{"format":"gif","min":"001","max":"035"},"wave":{"format":"gif","min":"001","max":"028"},"thumbsup":{"format":"gif","min":"001","max":"016"},"stare":{"format":"gif","min":"001","max":"014"},"cuddle":{"format":"gif","min":"001","max":"037"},"smile":{"format":"gif","min":"001","max":"023"},"baka":{"format":"gif","min":"001","max":"023"},"blush":{"format":"gif","min":"001","max":"013"},"think":{"format":"gif","min":"001","max":"012"},"pout":{"format":"gif","min":"001","max":"008"},"facepalm":{"format":"gif","min":"001","max":"011"},"wink":{"format":"gif","min":"001","max":"015"},"shoot":{"format":"gif","min":"001","max":"020"},"smug":{"format":"gif","min":"001","max":"015"},"cry":{"format":"gif","min":"001","max":"040"},"pat":{"format":"gif","min":"001","max":"048"},"punch":{"format":"gif","min":"001","max":"025"},"dance":{"format":"gif","min":"001","max":"021"},"feed":{"format":"gif","min":"001","max":"035"},"shrug":{"format":"gif","min":"001","max":"008"},"bored":{"format":"gif","min":"001","max":"015"},"kick":{"format":"gif","min":"001","max":"015"},"hug":{"format":"gif","min":"001","max":"040"},"yeet":{"format":"gif","min":"001","max":"007"},"slap":{"format":"gif","min":"001","max":"046"},"neko":{"format":"png","min":"0001","max":"0777"},"husbando":{"format":"png","min":"0001","max":"0200"},"kitsune":{"format":"png","min":"0001","max":"0400"},"waifu":{"format":"png","min":"0001","max":"0400"}}"#;

                return Ok(serde_json::from_str(endpoints_json)?);
            }

            Ok(client
                .get(format!("{}/endpoints", BASE_URL))
                .send()
                .await?
                .json()
                .await?)
        }

        let endpoints = get_endpoints_data(&client).await?;

        struct CategoryData<'a>(&'a str, &'a EndpointDesc);

        impl<'a> ToTokens for CategoryData<'a> {
            fn to_tokens(&self, tokens: &mut TokenStream) {
                let name = match self.0 {
                    "thumbsup" => format_ident!("ThumbsUp"),
                    _ => format_ident!("{}{}", self.0[..1].to_uppercase(), &self.0[1..]),
                };
                let min = self.1.min;
                let max = self.1.max;
                let format = &self.1.format;
                let with_padding = self.1.with_padding;

                tokens.append_all(quote! {
                    pub struct #name;

                    impl super::LocalNekosBestCategory for #name {
                        const CATEGORY: crate::Category = crate::Category::#name;
                        const MIN: usize = #min;
                        const MAX: usize = #max;
                        const WITH_PADDING: usize = #with_padding;
                        const FORMAT: &'static str = #format;
                    }

                    impl #name {
                        pub fn get_random(&self, random: usize) -> String {
                            <Self as super::LocalNekosBestCategory>::get_random(self, random)
                        }

                        pub fn get(&self) -> String {
                            <Self as super::LocalNekosBestCategory>::get(self)
                        }
                    }
                })
            }
        }

        let implementations = endpoints
            .into_iter()
            .map(|(name, endpoint)| {
                let category_data = CategoryData(&name, &endpoint);

                quote! {#category_data}
            })
            .collect::<TokenStream>();

        let implementations = implementations.to_string();

        std::fs::write(
            PathBuf::from(std::env::var("OUT_DIR")?).join("local_implementation.rs"),
            implementations,
        )?;
    }

    Ok(())
}