feat: RSS feeds (#137)

rochacbruno · web-flow · commit 3c4468467a59 · 2024-11-15T19:11:26.000Z
fix #8
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -30,6 +30,7 @@ url = "2.5.3"
 rust-embed = { version = "8.5.0", features = ["interpolate-folder-path"] }
 lazy_static = "1.5.0"
 indexmap = { version = "2.6.0", features = ["serde"] }
+rss = "2.0.9"
 
 [profile.release]
 codegen-units = 1
diff --git a/example/templates/base.html b/example/templates/base.html
@@ -6,6 +6,7 @@
     <link rel="icon" type="image/x-icon" href="./static/favicon.ico">
     <meta name="viewport" content="width=device-width, initial-scale=1">
     <meta name="color-scheme" content="light dark" />
+    <meta name="generator" content="Marmite" />
     {% block seo %}
     <meta property="og:title" content="{{ site.name }}">
     <meta property="og:description" content="{{ site.tagline }}">
@@ -24,6 +25,22 @@
     <link rel="stylesheet" type="text/css" href="./static/marmite.css">
     <link rel="stylesheet" type="text/css" href="./static/custom.css">
     {% endblock -%}
+    {%- block feeds %}
+    <link rel="alternate" type="application/rss+xml" title="index" href="index.rss">
+    {% for stream, _ in group(kind="stream") -%}
+    <link rel="alternate" type="application/rss+xml" title="{{stream}}" href="{{stream | slugify}}.rss">
+    {% endfor %}
+    {%- for tag, _ in group(kind="tag") -%}
+    <link rel="alternate" type="application/rss+xml" title="{{tag}}" href="tag-{{tag | slugify}}.rss">
+    {% endfor %}
+    {%- for author, _ in group(kind="author") -%}
+    <link rel="alternate" type="application/rss+xml" title="{{author}}" href="author-{{author | slugify}}.rss">
+    {% endfor %}
+    {%- for year, _ in group(kind="archive") -%}
+    <link rel="alternate" type="application/rss+xml" title="{{year}}" href="archive-{{year}}.rss">
+    {% endfor %}
+
+    {% endblock %}
 </head>
 <body>
     <main class="container">
diff --git a/src/feed.rs b/src/feed.rs
@@ -0,0 +1,69 @@
+use rss::{ChannelBuilder, ItemBuilder};
+use std::fs::File;
+use std::io::prelude::*;
+use std::path::Path;
+
+use crate::config::Marmite;
+use crate::content::Content;
+
+pub fn generate_rss(
+    contents: &[Content],
+    output_path: &Path,
+    filename: &str,
+    config: &Marmite,
+) -> Result<(), String> {
+    let mut channel = ChannelBuilder::default()
+        .title(&config.name)
+        .link(&config.url)
+        .description(&config.tagline)
+        .generator("marmite".to_string())
+        .build();
+
+    for content in contents.iter().take(15) {
+        let mut item = ItemBuilder::default()
+            .title(content.title.clone())
+            .link(format!("{}/{}", &config.url, &content.slug))
+            .guid(rss::GuidBuilder::default().value(&content.slug).build())
+            .pub_date(content.date.unwrap().to_string())
+            .content(content.html.clone())
+            .source(
+                rss::SourceBuilder::default()
+                    .url(&config.url)
+                    .title(filename.to_string())
+                    .build(),
+            )
+            .description(content.description.clone())
+            .build();
+
+        if let Some(author) = content.authors.first() {
+            item.author = Some(author.clone());
+        }
+        item.categories = content
+            .tags
+            .iter()
+            .map(|tag| rss::CategoryBuilder::default().name(tag.clone()).build())
+            .collect();
+        channel.items.push(item);
+    }
+
+    if let Some(latest_item) = channel.items.first() {
+        channel.pub_date = latest_item.pub_date.clone();
+    }
+
+    channel.last_build_date = Some(chrono::Utc::now().format("%+").to_string());
+
+    if !config.card_image.is_empty() {
+        channel.image = Some(
+            rss::ImageBuilder::default()
+                .url(format!("{}/{}", &config.url, &config.card_image))
+                .build(),
+        );
+    }
+
+    let rss = channel.to_string();
+    let feed_path = output_path.join(format!("{filename}.rss"));
+    let mut file = File::create(feed_path).map_err(|e| e.to_string())?;
+    file.write_all(rss.as_bytes()).map_err(|e| e.to_string())?;
+
+    Ok(())
+}
diff --git a/src/main.rs b/src/main.rs
@@ -7,6 +7,7 @@ mod cli;
 mod config;
 mod content;
 mod embedded;
+mod feed;
 mod markdown;
 mod server;
 mod site;
diff --git a/src/site.rs b/src/site.rs
@@ -319,6 +319,8 @@ fn render_templates(
             output_dir,
             &stream_slug,
         )?;
+        // Render {stream}.rss for each stream
+        crate::feed::generate_rss(&stream_contents, output_dir, &stream_slug, &site_data.site)?;
     }
 
     // Pages are treated as a list of content, no stream separation is needed
@@ -428,14 +430,23 @@ fn handle_author_pages(
             .cloned()
             .collect::<Vec<Content>>();
 
+        let filename = format!("author-{}", &author_slug);
         handle_list_page(
             &author_context,
             &author.name,
             &author_posts,
             site_data,
             tera,
             output_dir,
-            format!("author-{}", &author_slug).as_ref(),
+            &filename,
+        )?;
+
+        // Render author-{name}.rss for each stream
+        crate::feed::generate_rss(
+            &author_posts,
+            output_dir,
+            &filename.clone(),
+            &site_data.site,
         )?;
     }
 
@@ -778,14 +789,22 @@ fn handle_tag_pages(
 ) -> Result<(), String> {
     for (tag, tagged_contents) in site_data.tag.iter() {
         let tag_slug = slugify(tag);
+        let filename = format!("tag-{}", &tag_slug);
         handle_list_page(
             global_context,
             &site_data.site.tags_content_title.replace("$tag", tag),
             &tagged_contents,
             site_data,
             tera,
             output_dir,
-            format!("tag-{}", &tag_slug).as_ref(),
+            &filename,
+        )?;
+        // Render tag-{tag}.rss for each stream
+        crate::feed::generate_rss(
+            &tagged_contents,
+            output_dir,
+            &filename.clone(),
+            &site_data.site,
         )?;
     }
 
@@ -811,14 +830,22 @@ fn handle_archive_pages(
     tera: &Tera,
 ) -> Result<(), String> {
     for (year, archive_contents) in site_data.archive.iter() {
+        let filename = format!("archive-{year}");
         handle_list_page(
             global_context,
             &site_data.site.archives_content_title.replace("$year", year),
             &archive_contents,
             site_data,
             tera,
             output_dir,
-            format!("archive-{year}").as_ref(),
+            &filename,
+        )?;
+        // Render archive-{year}.rss for each stream
+        crate::feed::generate_rss(
+            &archive_contents,
+            output_dir,
+            &filename.clone(),
+            &site_data.site,
         )?;
     }
 
