From 3074e2607489d52d321a948bdaa2e6f8c728a201 Mon Sep 17 00:00:00 2001
From: David Li <li.davidm96@gmail.com>
Date: Wed, 27 Jan 2016 10:51:51 -0500
Subject: feat(wiki): implement Wikipedia lookup
---
src/main.rs | 51 +++++++++++++++++++++++++++++++++++++++++++++++++--
1 file changed, 49 insertions(+), 2 deletions(-)
diff --git a/src/main.rs b/src/main.rs
index 2ec24c0..22c5530 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,11 @@
+#![feature(custom_derive,plugin)]
+#![plugin(serde_macros)]
extern crate hyper;
extern crate irc;
+extern crate serde;
+extern crate serde_json;
+use std::collections::HashMap;
use std::io::Read;
use hyper::Client;
@@ -8,7 +13,35 @@ use hyper::header::Connection;
use irc::client::prelude::*;
+#[derive(Debug, PartialEq, Serialize, Deserialize)]
+struct WikiPage {
+ ns: u32,
+ pageid: u32,
+ title: String,
+ extract: String,
+}
+
+#[derive(Debug, PartialEq, Serialize, Deserialize)]
+struct WikiRedirect {
+ from: String,
+ to: String,
+}
+
+#[derive(Debug, PartialEq, Serialize, Deserialize)]
+struct WikiQueryResponse {
+ batchcomplete: String,
+ query: WikiQuery,
+}
+
+#[derive(Debug, PartialEq, Serialize, Deserialize)]
+struct WikiQuery {
+ redirects: Option<Vec<WikiRedirect>>,
+ pages: HashMap<String, WikiPage>,
+}
+
fn main() {
+ let q: WikiQueryResponse = serde_json::from_str("{\"batchcomplete\":\"\",\"query\":{\"pages\":{\"18123\":{\"pageid\":18123,\"ns\":0,\"title\":\"Lisp machine\",\"extract\":\"Lisp machines are general-purpose computers designed to efficiently run Lisp as their main software language, usually through hardware support. They are an example of a high-level language computer architecture, and in a sense, they were the first commercial single-user workstations. Despite being modest in number (perhaps 7,000 units total as of 1988), Lisp machines commercially pioneered many now-commonplace technologies \\u2013 including effective garbage collection, laser printing, windowing systems, computer mice, high-resolution bit-mapped graphics, computer graphic rendering, and networking innovations like Chaosnet. Several companies built and sold Lisp Machines in the 1980s: Symbolics (3600, 3640, XL1200, MacIvory, and other models), Lisp Machines Incorporated (LMI Lambda), Texas Instruments (Explorer and MicroExplorer), and Xerox (Interlisp-D workstations). The operating systems were written in Lisp Machine Lisp, InterLisp (Xerox), and later partly in Common Lisp.\"}}}}").unwrap();
+ println!("{:?}", q);
let config = Config {
nickname: Some(format!("lidavidm_prime")),
server: Some(format!("irc.freenode.net")),
@@ -28,14 +61,28 @@ fn main() {
server.send_privmsg(&message.args[0], &response).unwrap();
let mut client = Client::new();
- let mut res = client.get("https://en.wikipedia.org/w/api.php?action=query&prop=extracts&format=json&exintro=&explaintext=&titles={}&redirects=")
+ let mut res = client
+ .get(&format!("https://en.wikipedia.org/w/api.php?action=query&prop=extracts&format=json&exintro=&explaintext=&titles={}&redirects=", article.replace(" ", "%20")))
+ .header(hyper::header::UserAgent("lidavidm_irc_bot/0.1 (https://git.lidavidm.me; li.davidm96@gmail.com) hyper/0.7.2".to_owned()))
.header(Connection::close())
.send().unwrap();
let mut body = String::new();
res.read_to_string(&mut body).unwrap();
- println!("Response: {}", body);
+ let response: serde_json::Result<WikiQueryResponse> = serde_json::from_str(&body);
+ if let Ok(q) = response {
+ for page in q.query.pages.values() {
+ let url = format!("https://en.wikipedia.org/?curid={}", page.pageid);
+ server.send_privmsg(&message.args[0], &format!("{}: {}", page.title, url)).unwrap();
+ server.send_privmsg(&message.args[0], &page.extract).unwrap();
+ break;
+ }
+ }
+ else {
+ server.send_privmsg(&message.args[0], "Sorry, I couldn't retrieve the wiki page.").unwrap();
+ println!("{} {:?}", body, response);
+ }
}
}
}
--
cgit v1.2.3