Support picking question via name

Author: gotlougit

src/cache/mod.rs

@@ -119,6 +119,19 @@ impl Cache {
         Ok(p)
     }
 
+    /// Get problem from name
+    pub fn get_problem_from_name(&self, problem_name: &String) -> Result<Problem, Error> {
+        let p: Problem = problems
+            .filter(name.eq(problem_name))
+            .first(&self.conn()?)?;
+        if p.category != "algorithms" {
+            return Err(Error::FeatureError(
+                "Not support database and shell questions for now".to_string(),
+            ));
+        }
+        Ok(p)
+    }
+
     /// Get daily problem
     pub async fn get_daily_problem_id(&self) -> Result<i32, Error> {
         parser::daily(

src/cmds/pick.rs

@@ -47,6 +47,14 @@ impl Command for PickCommand {
         ClapCommand::new("pick")
             .about("Pick a problem")
             .visible_alias("p")
+            .arg(
+                Arg::new("name")
+                    .short('n')
+                    .long("name")
+                    .value_parser(clap::value_parser!(String))
+                    .help("Problem name")
+                    .num_args(1),
+            )
             .arg(
                 Arg::new("id")
                     .value_parser(clap::value_parser!(i32))
@@ -127,15 +135,32 @@ impl Command for PickCommand {
             None
         };
 
-        let fid = m
-            .get_one::<i32>("id")
-            .map(|id| *id)
-            .or(daily_id)
-            .unwrap_or_else(|| {
-                // Pick random without specify id
-                let problem = &problems[rand::thread_rng().gen_range(0..problems.len())];
-                problem.fid
-            });
+        let fid = match m.contains_id("name") {
+            //check for name specified
+            true => {
+                match m.get_one::<String>("name").map(|name| name) {
+                    Some(quesname) => match cache.get_problem_from_name(quesname) {
+                        Ok(p) => p.fid,
+                        Err(_) => 1,
+                    },
+                    None => {
+                        // Pick random without specify id
+                        let problem = &problems[rand::thread_rng().gen_range(0..problems.len())];
+                        problem.fid
+                    }
+                }
+            }
+            false => {
+                m.get_one::<i32>("id")
+                    .map(|id| *id)
+                    .or(daily_id)
+                    .unwrap_or_else(|| {
+                        // Pick random without specify id
+                        let problem = &problems[rand::thread_rng().gen_range(0..problems.len())];
+                        problem.fid
+                    })
+            }
+        };
 
         let r = cache.get_question(fid).await;