Import project

Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "titlebot"
+version = "0.1.0"
+authors = ["Yash Karandikar <nerdstep710@gmail.com>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+irc = "0.15.0"
+tokio = { version = "1.5.0", features = ["full"] }
+futures = "0.3"
+failure = "0.1.8"
+regex = "1"
+reqwest = { version = "0.11", features = ["json"] }

src/main.rs

@@ -0,0 +1,73 @@
+use std::collections::HashMap;
+
+use futures::prelude::*;
+use irc::client::prelude::*;
+use regex::Regex;
+
+#[tokio::main]
+async fn main() -> Result<(), failure::Error> {
+	println!("TitleBot v2.0 by karx");
+
+    let config = Config {
+        nickname: Some("titlebot".to_owned()),
+        server: Some("192.168.254.28".to_owned()),
+        channels: vec!["#no-normies".to_owned(), "#main".to_owned(), "#aksmp".to_owned(), "#aksmp-mc".to_owned()],
+        port: Some(6667 as u16),
+        use_tls: Some(false),
+        ..Config::default()
+    };
+
+    let mut client = Client::from_config(config).await?;
+    client.identify()?;
+
+	println!("Connected to IRC");
+
+    let mut stream = client.stream()?;
+
+    let ure = Regex::new(r"https?://[www\.]*\w+\.\w+[/\S+]*").unwrap();
+    let titlefetch = Regex::new(r"<title>[\s\S]+</title>").unwrap();
+    let treplace = Regex::new(r"</?title>").unwrap();
+
+    let mut cache: HashMap<String, String> = HashMap::new();
+
+    while let Some(message) = stream.next().await.transpose()? {
+		print!("{}", message);
+
+        if let Command::PRIVMSG(channel, message) = message.command {
+            if message.contains(client.current_nickname()) {
+                client
+                    .send_privmsg(&channel, "beep boop, i'm titlebot!")
+                    .unwrap();
+            }
+
+            if ure.is_match(&message) {
+                if let Some(mat) = ure.find(&message) {
+                    let slice = &message[mat.start()..mat.end()];
+
+                    if let Some(entry) = cache.get(&slice.to_string()) {
+						let fin = format!("Title: {}", entry);
+						client.send_privmsg(&channel, fin)?;
+                    } else {
+                        let response = reqwest::get(slice).await;
+
+                        if let Err(_e) = response {
+                            client.send_privmsg(&channel, "Error: could not get title")?;
+                        } else {
+                            let body = response?.text().await?;
+
+                            if let Some(title_mat) = titlefetch.find(&body) {
+                                let title_raw = &body[title_mat.start()..title_mat.end()];
+								let result = treplace.replace_all(title_raw, "");
+								let fin = format!("Title: {}", result.to_string());
+								client.send_privmsg(&channel, fin)?;
+								cache.insert(slice.to_string(), result.to_string());
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    Ok(())
+}