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

mod graphqls;
pub mod headers;
pub mod impl_lc;
pub mod question;
pub mod resps;

use std::{fmt::Display, sync::atomic::AtomicU32, time::Duration};

use miette::{Context, IntoDiagnostic, Result};
use reqwest::{header::HeaderMap, Client, ClientBuilder};

use self::headers::Headers;

pub const CATEGORIES: [&str; 8] = [
    "algorithms",
    "concurrency",
    "database",
    "javascript",
    "lcci",
    "lcof",
    "pandas",
    "shell",
];

/// for progress bar, total insert num
pub static TOTAL_QS_INDEX_NUM: AtomicU32 = AtomicU32::new(0);
/// for progress bar, current inserted num
pub static CUR_QS_INDEX_NUM: AtomicU32 = AtomicU32::new(0);

/// for progress bar, total insert num
pub static TOTAL_TOPIC_QS_INDEX_NUM: AtomicU32 = AtomicU32::new(0);
/// for progress bar, current inserted num
pub static CUR_TOPIC_QS_INDEX_NUM: AtomicU32 = AtomicU32::new(0);

#[derive(Clone)]
#[derive(Debug)]
#[derive(PartialEq, Eq)]
pub enum IdSlug {
    Id(u32),
    Slug(String),
}

impl Display for IdSlug {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Id(num) => num.fmt(f),
            Self::Slug(slug) => slug.fmt(f),
        }
    }
}

/// interact with leetcode.com/cn
#[derive(Default)]
#[derive(Debug)]
#[derive(Clone)]
pub struct LeetCode {
    pub client:  Client,
    pub headers: HeaderMap,
}

impl LeetCode {
    /// Create a `LeetCode` instance and initialize some variables
    pub async fn build() -> Result<Self> {
        let client = ClientBuilder::new()
            .gzip(true)
            .connect_timeout(Duration::from_secs(30))
            .build()
            .into_diagnostic()
            .context("reqwest client failed")?;

        Ok(Self {
            client,
            headers: Headers::build_default()
                .await
                .context("build header failed")?
                .headers,
        })
    }
}

mod leetcode_send {
    use miette::{miette, IntoDiagnostic, Result};
    use reqwest::{
        header::{HeaderMap, HeaderValue},
        Client,
    };
    use serde::de::DeserializeOwned;
    use tracing::{debug, trace};

    use crate::{leetcode::headers::Headers, Json};

    pub(super) async fn fetch<T>(
        client: &Client,
        url: &str,
        json: Option<&Json>,
        headers: HeaderMap<HeaderValue>,
    ) -> Result<T>
    where
        T: DeserializeOwned,
    {
        let headers = Headers::mod_headers(headers, vec![("Referer", url)])?;

        let req_builder = json.map_or_else(|| client.get(url), |json| client.post(url).json(json));

        let resp = req_builder
            .headers(headers)
            .send()
            .await
            .into_diagnostic()?;
        trace!("respond: {:#?}", resp);
        debug!("http status code: {:#?}", resp.status());

        match resp.status().as_u16() {
            429 => miette::bail!("Your submissions are too frequent."),
            403 => miette::bail!("Forbidden, maybe you not verify email or phone number"),
            408 => miette::bail!("Request Time-out"),
            400..=499 => miette::bail!("Client error, HTTP Code: {}", resp.status()),
            500..=599 => miette::bail!("Server error, HTTP Code: {}", resp.status()),
            _ => {},
        }

        resp.json::<T>()
            .await
            .map_err(|e| miette!("Error: {e}."))
    }
}