Implement rss

Cargo.lock

@@ -67,6 +67,7 @@ dependencies = [
  "chrono 0.4.6 (registry+https://github.com/rust-lang/crates.io-index)",
  "clap 2.32.0 (registry+https://github.com/rust-lang/crates.io-index)",
  "fs_extra 1.1.0 (registry+https://github.com/rust-lang/crates.io-index)",
+ "htmlescape 0.3.1 (registry+https://github.com/rust-lang/crates.io-index)",
  "lazy_static 1.2.0 (registry+https://github.com/rust-lang/crates.io-index)",
  "notify 4.0.6 (registry+https://github.com/rust-lang/crates.io-index)",
  "pulldown-cmark 0.0.11 (registry+https://github.com/rust-lang/crates.io-index)",
@@ -181,6 +182,11 @@ dependencies = [
  "unicode-width 0.1.5 (registry+https://github.com/rust-lang/crates.io-index)",
 ]
 
+[[package]]
+name = "htmlescape"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+
 [[package]]
 name = "inotify"
 version = "0.6.1"
@@ -815,6 +821,7 @@ dependencies = [
 "checksum fuchsia-zircon-sys 0.3.3 (registry+https://github.com/rust-lang/crates.io-index)" = "3dcaa9ae7725d12cdb85b3ad99a434db70b468c09ded17e012d86b5c1010f7a7"
 "checksum futures 0.1.25 (registry+https://github.com/rust-lang/crates.io-index)" = "49e7653e374fe0d0c12de4250f0bdb60680b8c80eed558c5c7538eec9c89e21b"
 "checksum getopts 0.2.18 (registry+https://github.com/rust-lang/crates.io-index)" = "0a7292d30132fb5424b354f5dc02512a86e4c516fe544bb7a25e7f266951b797"
+"checksum htmlescape 0.3.1 (registry+https://github.com/rust-lang/crates.io-index)" = "e9025058dae765dee5070ec375f591e2ba14638c63feff74f13805a72e523163"
 "checksum inotify 0.6.1 (registry+https://github.com/rust-lang/crates.io-index)" = "40b54539f3910d6f84fbf9a643efd6e3aa6e4f001426c0329576128255994718"
 "checksum inotify-sys 0.1.3 (registry+https://github.com/rust-lang/crates.io-index)" = "e74a1aa87c59aeff6ef2cc2fa62d41bc43f54952f55652656b18a02fd5e356c0"
 "checksum iovec 0.1.2 (registry+https://github.com/rust-lang/crates.io-index)" = "dbe6e417e7d0975db6512b90796e8ce223145ac4e33c377e4a42882a0e88bb08"

Cargo.toml

@@ -11,6 +11,7 @@ repository = "https://git.sr.ht/~simulacrumparty/casaubon"
 chrono = "0.4"
 clap = "2.32.0"
 fs_extra = "1.1.0"
+htmlescape = "0.3.1"
 lazy_static = "1.2.0"
 notify = "4.0.0"
 pulldown-cmark = "0.0.11"

src/config.rs

@@ -1,3 +1,5 @@
 pub struct Config {
     pub site_name: String,
+    pub url: String,
+    pub description: String,
 }

src/entry.rs

@@ -23,7 +23,7 @@ pub struct Entry {
 }
 
 impl Entry {
-    fn render(&self, template: &str, config: &Config) -> String {
+    pub fn render(&self, template: &str, config: &Config) -> String {
         let parser = Parser::new(&self.body);
         let mut html_buf = String::new();
         html::push_html(&mut html_buf, parser);
@@ -183,6 +183,8 @@ mod tests {
     fn test_render_post() {
         let config = Config {
             site_name: String::from("Test Site"),
+            url: String::from("testsite.com"),
+            description: String::from("recent posts from testsite.com"),
         };
 
         let post = Entry {
@@ -208,6 +210,8 @@ mod tests {
     fn test_render_post_listing() {
         let config = Config {
             site_name: String::from("Test Site"),
+            url: String::from("testsite.com"),
+            description: String::from("recent posts from testsite.com"),
         };
 
         let posts = vec![
@@ -270,6 +274,8 @@ mod tests {
         };
         let config = Config {
             site_name: "Test Site".to_string(),
+            url: String::from("testsite.com"),
+            description: String::from("recent posts from testsite.com"),
         };
 
         write_entry(&project_dir, &layout, &post_template, &post, &config);
@@ -330,6 +336,8 @@ mod tests {
         ];
         let config = Config {
             site_name: "Test Site".to_string(),
+            url: String::from("testsite.com"),
+            description: String::from("recent posts from testsite.com"),
         };
         write_entry_listing(
             &cwd,
@@ -395,6 +403,8 @@ mod tests {
 
         let config = Config {
             site_name: String::from("Test Site"),
+            url: String::from("testsite.com"),
+            description: String::from("recent posts from testsite.com"),
         };
 
         assert_eq!("2000-01-01", post.render("{{ date }}", &config));

src/lib.rs

@@ -1,5 +1,6 @@
 extern crate chrono;
 extern crate fs_extra;
+extern crate htmlescape;
 #[macro_use]
 extern crate lazy_static;
 extern crate notify;
@@ -18,15 +19,19 @@ use toml::Value;
 
 use config::Config;
 use entry::{parse_entry, read_entry_dir, write_entry, write_entry_listing, Entry, EntryKind};
+use rss::generate_rss;
 
 mod config;
 mod entry;
+mod rss;
 
 pub fn build(include_drafts: bool, cwd: &path::Path) {
     let config = match fs::read_to_string(cwd.join("casaubon.toml")) {
         Ok(contents) => match contents.parse::<Value>() {
             Ok(config) => Config {
                 site_name: String::from(config["site_name"].as_str().unwrap()),
+                url: String::from(config["url"].as_str().unwrap()),
+                description: String::from(config["description"].as_str().unwrap()),
             },
             Err(_) => panic!("Invalid casaubon.toml"),
         },
@@ -105,6 +110,9 @@ pub fn build(include_drafts: bool, cwd: &path::Path) {
         &dir::CopyOptions::new(),
     )
     .expect("Couldn't copy assets directory");
+
+    let rss = generate_rss(&config, &posts);
+    fs::write(cwd.join("public").join("rss.xml"), rss).expect("Couldn't write rss file");
 }
 
 pub fn new(name: &str, cwd: &path::Path) {
@@ -113,7 +121,10 @@ pub fn new(name: &str, cwd: &path::Path) {
     fs::create_dir(&project_path).expect(&format!("Couldn't create directory '{}'", &name));
     fs::write(
         project_path.join("casaubon.toml"),
-        format!("site_name = \"{}\"", &name),
+        format!(
+            "site_name = \"{}\"\nurl = \"{}.com\"\ndescription = \"\"",
+            &name, &name
+        ),
     )
     .expect("Could not create casaubon.toml");
 

src/rss.rs

@@ -0,0 +1,58 @@
+use chrono::{DateTime, Utc};
+use htmlescape::encode_minimal;
+
+use config::Config;
+use entry::{Entry, EntryKind};
+
+pub fn generate_post_rss(config: &Config, post: &Entry) -> String {
+    let date: DateTime<Utc> = DateTime::from_utc(post.date.unwrap().and_hms(0, 0, 0), Utc);
+    let date_string = date.format("%a, %d %b %Y %H:%M:%S %z").to_string();
+    let url = format!("{}/posts/{}/", config.url, post.slug);
+    format!(
+        "<item><title>{}</title><description>{}</description><guid>{}</guid><link>{}</link><pubDate>{}</pubDate></item>",
+        post.title, encode_minimal(&post.render("<article>{{ body }}</article>", config)), url, url, date_string
+    )
+}
+
+pub fn generate_rss(config: &Config, posts: &Vec<Entry>) -> String {
+    let items = posts
+        .into_iter()
+        .filter(|post| post.kind == EntryKind::Post)
+        .map(|post| generate_post_rss(&config, &post))
+        .collect::<Vec<String>>()
+        .join("");
+
+    format!(
+        "<rss version=\"2.0\" xmlns:atom=\"http://www.w3.org/2005/Atom\"><channel><atom:link href=\"{}/rss.xml\" rel=\"self\" /><title>{}</title><description>{}</description><link>{}</link>{}</channel></rss>",
+        config.url, config.site_name, config.description, config.url, items
+    )
+}
+
+#[cfg(test)]
+mod tests {
+    use chrono::NaiveDate;
+
+    use super::generate_rss;
+    use config::Config;
+    use entry::{Entry, EntryKind};
+
+    #[test]
+    fn generates_rss_feed() {
+        let config = Config {
+            site_name: String::from("Lorem Ipsum"),
+            url: String::from("https://www.loremipsum.com"),
+            description: String::from("recent posts from loremipsum.com"),
+        };
+        let posts: Vec<Entry> = vec![Entry {
+            title: String::from("Hello World"),
+            body: String::from("lorem ipsum dolor sit amet"),
+            slug: String::from("hello-world"),
+            date: Some(NaiveDate::from_ymd(2019, 1, 1)),
+            kind: EntryKind::Post,
+        }];
+        assert_eq!(
+            generate_rss(&config, &posts),
+            "<rss version=\"2.0\" xmlns:atom=\"http://www.w3.org/2005/Atom\"><channel><atom:link href=\"https://www.loremipsum.com/rss.xml\" rel=\"self\" /><title>Lorem Ipsum</title><description>recent posts from loremipsum.com</description><link>https://www.loremipsum.com</link><item><title>Hello World</title><description>&lt;article&gt;&lt;p&gt;lorem ipsum dolor sit amet&lt;/p&gt;\n&lt;/article&gt;</description><guid>https://www.loremipsum.com/posts/hello-world/</guid><link>https://www.loremipsum.com/posts/hello-world/</link><pubDate>Tue, 01 Jan 2019 00:00:00 +0000</pubDate></item></channel></rss>"
+        );
+    }
+}

tests/integration_test.rs

@@ -63,7 +63,7 @@ fn test_build() {
     .unwrap();
     fs::write(
         project_dir.join("casaubon.toml"),
-        "site_name = \"Test Site\"",
+        "site_name = \"Test Site\"\nurl =\"testsite.com\"\ndescription = \"Test Site\"",
     )
     .unwrap();
 
@@ -174,7 +174,7 @@ fn test_build_drafts() {
     .unwrap();
     fs::write(
         project_dir.join("casaubon.toml"),
-        "site_name = \"Test Site\"",
+        "site_name = \"Test Site\"\nurl =\"testsite.com\"\ndescription = \"Test Site\"",
     )
     .unwrap();
 
@@ -272,7 +272,10 @@ fn test_new() {
             .replace("  ", "")
     );
     assert_eq!(
-        format!("site_name = \"{}\"", &uuid),
+        format!(
+            "site_name = \"{}\"\nurl = \"{}.com\"\ndescription = \"\"",
+            &uuid, &uuid
+        ),
         fs::read_to_string(&project_dir.join("casaubon.toml")).unwrap()
     );
 }