Merge pull request #27 from tinaun/crate-search

added `?crate` command to search crates.io

Author: technetos

Cargo.lock

@@ -11,6 +11,7 @@ license = "MIT"
 serenity = { version = "0.8.0", features = ["cache", "model"] }
 diesel = { version = "1.4.0", features = ["postgres", "r2d2"] }
 diesel_migrations = { version = "1.4.0", features = ["postgres"] }
+reqwest = { version = "0.10", features = ["blocking", "json"] }
 serde = "1.0"
 serde_derive = "1.0"
 lazy_static = "1.4.0"

Cargo.toml

@@ -1,4 +1,5 @@
 use crate::state_machine::{CharacterSet, StateMachine};
+use reqwest::blocking::Client as HttpClient;
 use serenity::{model::channel::Message, prelude::Context};
 use std::collections::HashMap;
 
@@ -7,19 +8,22 @@ pub(crate) type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;
 pub(crate) type CmdPtr = for<'m> fn(Args<'m>) -> Result<()>;
 
 pub struct Args<'m> {
+    pub http: &'m HttpClient,
     pub cx: &'m Context,
     pub msg: &'m Message,
     pub params: HashMap<&'m str, &'m str>,
 }
 
 pub(crate) struct Commands {
     state_machine: StateMachine,
+    client: HttpClient,
 }
 
 impl Commands {
     pub(crate) fn new() -> Self {
         Self {
             state_machine: StateMachine::new(),
+            client: HttpClient::new(),
         }
     }
 
@@ -64,6 +68,7 @@ impl Commands {
             self.state_machine.process(message).map(|matched| {
                 info!("Executing command {}", message);
                 let args = Args {
+                    http: &self.client,
                     cx: &cx,
                     msg: &msg,
                     params: matched.params,

src/commands.rs

@@ -0,0 +1,72 @@
+use crate::{
+    api,
+    commands::{Args, Result},
+};
+
+use reqwest::header;
+use serde::Deserialize;
+
+const USER_AGENT: &str = "rust-lang/discord-mods-bot";
+
+#[derive(Debug, Deserialize)]
+struct Crates {
+    crates: Vec<Crate>,
+}
+#[derive(Debug, Deserialize)]
+struct Crate {
+    id: String,
+    name: String,
+    #[serde(rename = "newest_version")]
+    version: String,
+    #[serde(rename = "updated_at")]
+    updated: String,
+    downloads: u64,
+    description: String,
+}
+
+pub fn search(args: Args) -> Result<()> {
+    let query = args
+        .params
+        .get("query")
+        .ok_or("Unable to retrieve param: query")?;
+
+    info!("searching for crate `{}`", query);
+
+    let crate_list = args
+        .http
+        .get("https://crates.io/api/v1/crates")
+        .header(header::USER_AGENT, USER_AGENT)
+        .query(&[("q", query)])
+        .send()?
+        .json::<Crates>()?;
+
+    if let Some(krate) = crate_list.crates.get(0) {
+        args.msg.channel_id.send_message(&args.cx, |m| {
+            m.embed(|e| {
+                e.title(&krate.name)
+                    .url(format!("https://crates.io/crates/{}", krate.id))
+                    .description(&krate.description)
+                    .field("version", &krate.version, true)
+                    .field("downloads", &krate.downloads, true)
+                    .timestamp(krate.updated.as_str())
+            });
+
+            m
+        })?;
+    } else {
+        let message = "No crates found.";
+        api::send_reply(&args, message)?;
+    }
+
+    Ok(())
+}
+
+/// Print the help message
+pub fn help(args: Args) -> Result<()> {
+    let help_string = "search for a crate on crates.io
+```
+?crate query...
+```";
+    api::send_reply(&args, &help_string)?;
+    Ok(())
+}

src/crates.rs

@@ -9,6 +9,7 @@ extern crate log;
 
 mod api;
 mod commands;
+mod crates;
 mod db;
 mod events;
 mod schema;
@@ -75,6 +76,10 @@ fn app() -> Result {
     cmds.add("?tags get-all", tags::get_all);
     cmds.add("?tags help", tags::help);
 
+    // crates.io
+    cmds.add("?crate help", crates::help);
+    cmds.add("?crate query...", crates::search);
+
     // Slow mode.
     // 0 seconds disables slowmode
     cmds.add("?slowmode {channel} {seconds}", slow_mode);