feature:new script

Author: bestgopher

Cargo.toml

@@ -11,7 +11,12 @@ git2 = "0.13.15"
 reqwest = { version = "0.10", features = ["blocking", "json"] }
 serde_json = "1.0"
 serde = { version = "1.0", features = ["derive"] }
+clap = "3.0.0-beta.2"
+tera = "1.12.1"
+lazy_static = "1.4.0"
+regex = "1"
+
 
 [[bin]]
 name ="leetcode"
-path ="src/main.rs"
+path = "src/main.rs"

src/all.rs

@@ -0,0 +1,40 @@
+use crate::file;
+
+use std::sync::{Arc, Mutex};
+use std::thread;
+use crate::http::Resp;
+
+
+/// 重新格式化
+pub fn all() {
+    let files = file::get_all_bin_file();
+
+    let mut v = Vec::<Resp>::with_capacity(files.len());
+
+    let x = Arc::new(Mutex::new(v));
+    let mut handlers = vec![];
+
+    for i in 0..=files.len() / 10 {
+        // 把files分块，分成10个文件一块
+        let files = if i * 10 + 10 > files.len() {
+            files[i * 10..files.len()].to_vec()
+        } else {
+            files[i * 10..i * 10 + 10].to_vec()
+        };
+
+        let x = x.clone();
+
+        handlers.push(thread::spawn(move || {
+            for i in files {
+                let resp = crate::http::get_question_info(&i);
+                x.lock().unwrap().push(resp);
+            }
+        }))
+    }
+
+    for i in handlers {
+        i.join();
+    }
+
+    crate::file::write_readme(&mut *x.lock().unwrap());
+}

src/file.rs

@@ -0,0 +1,107 @@
+use std::fs::{self, File};
+use std::io::Write;
+use lazy_static::lazy_static;
+use regex::Regex;
+
+use crate::http::{Resp, Data, Ques};
+
+lazy_static!(
+    static ref RE: Regex = Regex::new(r"\|\s*([0-9]*)\s*\|\s*(\w*)\s*\|.*?bin/(.*?).rs.*?\|.*?\|").unwrap();
+);
+
+/// 将结果写入README.md中
+pub fn write_readme(r: &mut Vec<Resp>) {
+    // 先按id排序
+    r.sort_by(|x, y| {
+        let x_id = x.data.question.question_id.parse::<i32>().unwrap();
+        let y_id = y.data.question.question_id.parse::<i32>().unwrap();
+        x_id.cmp(&y_id)
+    });
+    let s = crate::render::render(r).unwrap();
+
+    match std::fs::write("README.md", s) {
+        Ok(_) => (),
+        Err(e) => println!("写入 README.md 失败，err{}", e.to_string())
+    }
+}
+
+/// 获取 src/bin 目录下所有文件的名称
+pub fn get_all_bin_file() -> Vec<String> {
+    let dir = fs::read_dir("src/bin/").unwrap();
+    dir.
+        into_iter().
+        map(|x| {
+            x.unwrap().file_name().to_str().unwrap().trim_end_matches(".rs").to_string()
+        }).
+        collect()
+}
+
+
+/// 创建 bin/{quest_name}.rs 文件
+pub fn write_question(resp: Resp) {
+    let file = format!("src/bin/{}.rs", resp.data.question.title_slug);
+    if std::path::Path::new(file.as_str()).exists() {
+        println!("{} exists", file);
+        return;
+    }
+
+    let mut f = File::create(file.as_str()).unwrap();
+    let mut s = String::new();
+    s.push_str("fn main() {}\n\nstruct Solution;\n\n");
+
+    for i in resp.data.question.code_snippets {
+        if i.lang == "Rust" {
+            s.push_str(i.code.replace("↵", "\n").as_str());
+            s.push_str("\n");
+            break;
+        }
+    }
+
+    f.write_all(s.as_bytes()).unwrap();
+}
+
+/// 解析README.md
+pub fn parse_readme() -> Vec<Resp> {
+    let contents = fs::read_to_string("README.md").unwrap();
+    parse(&contents)
+}
+
+fn parse(contents: &str) -> Vec<Resp> {
+    let mut v = vec![];
+    for content in contents.split('\n') {
+        for i in RE.captures_iter(content.trim()) {
+            v.push(Resp {
+                data: Data {
+                    question: Ques {
+                        question_id: i.get(1).unwrap().as_str().to_string(),
+                        translated_title: i.get(2).unwrap().as_str().to_string(),
+                        title_slug: String::new(),
+                        code_snippets: vec![],
+                        difficulty: String::new()
+                    }
+                },
+            })
+        }
+    }
+
+    v
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::file::{parse, get_all_bin_file};
+
+    #[test]
+    fn test_parse_readme() {
+        let content = r"| 1111   | 两数之和| [src](https://github.com/rustors/leetcode/blob/main/src/bin/two_sum.rs) | [leetcode](https://leetcode-cn.com/problems/two_sum/) |
+        | 1112   | 两数之和| [src](https://github.com/rustors/leetcode/blob/main/src/bin/two_sum.rs) | [leetcode](https://leetcode-cn.com/problems/two_sum/) |
+        ";
+
+        println!("{:?}", parse(content));
+    }
+
+    #[test]
+    fn test_get_all_bin_file() {
+        println!("{:?}", get_all_bin_file());
+    }
+}

src/git.rs

@@ -0,0 +1,62 @@
+use git2::{Repository, StatusOptions};
+
+use std::process::Command;
+
+/// 把新文件加入到git中
+pub fn push() {
+    // 解析readme.md文件
+    let mut r = crate::file::parse_readme();
+    let new_file = get_uncommit_files();
+    for i in new_file.iter() {
+        let x = i.trim_end_matches(".rs"); // 去掉后缀
+        let x = x.trim_start_matches("src/bin/"); // 去掉路径
+        git_add(i);
+        r.push(crate::http::get_question_info(x));
+    }
+
+    crate::file::write_readme(&mut r);
+    git_add("README.md");
+    push_to_origin();
+}
+
+
+fn get_uncommit_files() -> Vec<String> {
+    let mut options = StatusOptions::new();
+    options.pathspec("src/bin");
+    options.include_untracked(true);
+    let repo = Repository::open(".").unwrap();
+    let statuses = repo.statuses(Some(&mut options)).unwrap();
+
+    statuses
+        .iter()
+        .map(|x| String::from(x.path().unwrap()))
+        .collect()
+}
+
+fn git_add(file: &str) {
+    Command::new("git").arg("add").arg(file).output().unwrap();
+    let output = Command::new("git")
+        .arg("commit")
+        .arg("-m")
+        .arg(file)
+        .output()
+        .unwrap();
+
+    println!("{}", String::from_utf8(output.stdout).unwrap());
+}
+
+pub fn push_to_origin() {
+    let output = Command::new("git").arg("push").output().unwrap();
+    println!("{}", String::from_utf8(output.stdout).unwrap());
+}
+
+
+#[cfg(test)]
+mod tests {
+    use crate::git::get_uncommit_files;
+
+    #[test]
+    fn test_get_uncommit_files() {
+        println!("{:?}", get_uncommit_files());
+    }
+}

src/http.rs

@@ -0,0 +1,77 @@
+use lazy_static::lazy_static;
+use regex::Regex;
+use reqwest::blocking::Client;
+use serde::{Serialize, Deserialize};
+
+
+lazy_static! {
+    static ref RE: Regex = Regex::new(r".*?/problems/(.*?)/").unwrap();
+}
+
+const URL: &'static str = "https://leetcode-cn.com/graphql/";
+
+#[derive(Deserialize, Serialize, Debug)]
+pub struct Ques {
+    #[serde(rename = "questionId")]
+    pub question_id: String,
+    #[serde(rename = "titleSlug")]
+    pub title_slug: String,
+    #[serde(rename = "translatedTitle")]
+    pub translated_title: String,
+    #[serde(rename = "codeSnippets")]
+    pub code_snippets: Vec<CodeSnippets>,
+    #[serde(rename = "difficulty")]
+    pub difficulty: String,
+}
+
+#[derive(Deserialize, Serialize, Debug)]
+pub struct CodeSnippets {
+    #[serde(rename = "code")]
+    pub code: String,
+    #[serde(rename = "lang")]
+    pub lang: String,
+    #[serde(rename = "langSlug")]
+    pub lang_slug: String,
+    #[serde(rename = "__typename")]
+    pub typename: String,
+}
+
+#[derive(Deserialize, Serialize, Debug)]
+pub struct Data {
+    pub question: Ques,
+}
+
+#[derive(Deserialize, Serialize, Debug)]
+pub struct Resp {
+    pub data: Data,
+}
+
+pub fn get_question_info(mut ques: &str) -> Resp {
+    if ques.starts_with("http") {
+        ques = RE.captures_iter(ques).next().unwrap().get(1).unwrap().as_str();
+    }
+
+    let data_fmt = r#"{"operationName":"questionData","variables":{"titleSlug":"{}"},
+    "query":"query questionData($titleSlug: String!) {\n  question(titleSlug: $titleSlug) {\n    questionId\n    questionFrontendId\n    boundTopicId\n    title\n    titleSlug\n    content\n    translatedTitle\n    translatedContent\n    isPaidOnly\n    difficulty\n    likes\n    dislikes\n    isLiked\n    similarQuestions\n    contributors {\n      username\n      profileUrl\n      avatarUrl\n      __typename\n    }\n    langToValidPlayground\n    topicTags {\n      name\n      slug\n      translatedName\n      __typename\n    }\n    companyTagStats\n    codeSnippets {\n      lang\n      langSlug\n      code\n      __typename\n    }\n    stats\n    hints\n    solution {\n      id\n      canSeeDetail\n      __typename\n    }\n    status\n    sampleTestCase\n    metaData\n    judgerAvailable\n    judgeType\n    mysqlSchemas\n    enableRunCode\n    envInfo\n    book {\n      id\n      bookName\n      pressName\n      source\n      shortDescription\n      fullDescription\n      bookImgUrl\n      pressImgUrl\n      productUrl\n      __typename\n    }\n    isSubscribed\n    isDailyQuestion\n    dailyRecordStatus\n    editorType\n    ugcQuestionId\n    style\n    __typename\n  }\n}\n"}"#;
+    let data = data_fmt.replace("{}", ques);
+
+    Client::new()
+        .post(URL)
+        .header("content-type", "application/json")
+        .body(data)
+        .send()
+        .unwrap()
+        .json::<Resp>()
+        .unwrap()
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::http::get_question_info;
+
+    #[test]
+    fn test_get_question_info() {
+        println!("{:?}", get_question_info("container-with-most-water"));
+        println!("{:?}", get_question_info("https://leetcode-cn.com/problems/container-with-most-water/"));
+    }
+}

src/lib.rs

@@ -0,0 +1,34 @@
+mod new;
+mod http;
+mod file;
+mod git;
+mod render;
+mod all;
+
+use clap::{App, Arg};
+
+pub fn run() {
+    let matches = App::new("leetcode")
+        .version("0.0.1")
+        .author("bestgopher <84328409@qq.com>")
+        .about("a helper for leetcode")
+        .subcommand(App::new("new")
+            .about("get a new leetcode question")
+            .arg(Arg::new("question_name")
+                .about("The configuration file to use")
+                .index(1)))
+        .subcommand(App::new("all")
+            .about("get all questions' info and rewrite README.md"))
+        .get_matches();
+
+    if let Some(matches) = matches.subcommand_matches("new") {
+        match matches.value_of_t::<String>("question_name") {
+            Ok(x) => new::new(x),
+            Err(_) => println!("please input the name of question")
+        }
+    } else if let Some(_) = matches.subcommand_matches("all") {
+        all::all();
+    } else {
+        git::push();
+    }
+}

src/main.rs

@@ -0,0 +1,5 @@
+use leetcode;
+
+fn main() {
+    leetcode::run()
+}

src/new.rs

@@ -0,0 +1,8 @@
+/// 1.获取question name
+/// 2.如果name是url，就不需要拼装，不是就需要拼装
+/// 3.请求结构，获取数据
+/// 4.将数据写入bin/{question_name}.rs文件中
+pub fn new(ques: String) {
+    let r = crate::http::get_question_info(ques.as_str());
+    crate::file::write_question(r);
+}