leetrs 1.1.0

A command-line tool to interact with LeetCode.
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

//! Language enum and associated helpers.
use clap::ValueEnum;

/// Supported submission languages.
///
/// The `ValueEnum` derive lets Clap accept these directly as CLI arguments.
#[derive(Debug, Clone, ValueEnum)]
pub enum Language {
    Python,
    Rust,
    Pandas,
    Mysql,
    Postgres,
}

impl From<&String> for Language {
    fn from(value: &String) -> Self {
        match value.as_str() {
            "python3" => Self::Python,
            "rust" => Self::Rust,
            "pythondata" => Self::Pandas,
            "mysql" => Self::Mysql,
            "postgresql" => Self::Postgres,
            _ => Self::Mysql,
        }
    }
}

impl From<String> for Language {
    fn from(value: String) -> Self {
        match value.as_str() {
            "python3" => Self::Python,
            "rust" => Self::Rust,
            "pythondata" => Self::Pandas,
            "mysql" => Self::Mysql,
            "postgresql" => Self::Postgres,
            _ => Self::Mysql,
        }
    }
}

impl Language {
    /// Maps a [`Language`] variant to LeetCode's internal language slug string.
    pub fn to_lang_slug(&self) -> &'static str {
        match self {
            Language::Python => "python3",
            Language::Rust => "rust",
            Language::Mysql => "mysql",
            Language::Pandas => "pythondata",
            Language::Postgres => "postgresql",
        }
    }

    /// Infers the language from a file extension. Falls back to MySQL for unknown extensions.
    pub fn from_extension(ext: &str) -> Self {
        match ext {
            "py" => Language::Python,
            "rs" => Language::Rust,
            "sql" => Language::Mysql,
            _ => Language::Mysql,
        }
    }

    /// Returns the file extension used for solution files in this language.
    pub fn code_extension(&self) -> &'static str {
        match self {
            Language::Python | Language::Pandas => "py",
            Language::Rust => "rs",
            Language::Mysql | Language::Postgres => "sql",
        }
    }

    /// Returns the single-line comment prefix used in this language.
    ///
    /// Used to write the metadata header at the top of generated code files,
    /// e.g. `# id=1 slug=two-sum lang=python3`.
    pub fn meta_comment_prefix(&self) -> &'static str {
        match self {
            Language::Python | Language::Pandas | Language::Mysql => "#",
            Language::Rust => "//",
            Language::Postgres => "--",
        }
    }
}

/// A problem identifier supplied on the command line — either a numeric ID or a slug.
#[derive(Debug, Clone)]
pub enum Identifier {
    Number(u64),
    String(String),
}