feat: simplistic implementation of the pagestats directive

Sponsored-by: author

1 files changed, 30 insertions, 10 deletions

diff --git a/src/directive/pagestats.rs b/src/directive/pagestats.rs
index b81d2d8..8c7feec 100644
--- a/src/directive/pagestats.rs
+++ b/src/directive/pagestats.rs
@@ -1,26 +1,46 @@
 use crate::directive::{DirectiveError, DirectiveImplementation, Processed};
 use crate::page::PageMeta;
+use crate::pagespec::PageSpec;
 use crate::site::Site;
+use crate::util::make_relative_link;
 use crate::wikitext::ParsedDirective;
-use log::warn;
+use std::path::Path;
 
 #[derive(Debug, Default, Eq, PartialEq)]
-pub struct PageStats {}
+pub struct PageStats {
+    pages: String,
+}
 
 impl DirectiveImplementation for PageStats {
     const REQUIRED: &'static [&'static str] = &["pages"];
     const ALLOWED: &'static [&'static str] = &["among", "style"];
     const ALLOW_ANY_UNNAMED: bool = true;
 
-    fn from_parsed(_: &ParsedDirective) -> Self {
-        Self::default()
+    fn from_parsed(p: &ParsedDirective) -> Self {
+        let args = p.args();
+        let pages = args.get("pages").unwrap();
+        Self::new(pages.to_string())
+    }
+
+    fn process(&self, site: &Site, meta: &mut PageMeta) -> Result<Processed, DirectiveError> {
+        let pagespec = PageSpec::new(meta.path(), &self.pages).map_err(DirectiveError::PageSpec)?;
+        let matches: Vec<String> = site
+            .markdown_pages()
+            .iter()
+            .filter(|page| pagespec.matches(site, page.meta().path()))
+            .map(|page| format!("* {}\n", Self::link(meta.path(), page.meta())))
+            .collect();
+        Ok(Processed::Markdown(matches.join("")))
+    }
+}
+
+impl PageStats {
+    pub fn new(pages: String) -> Self {
+        Self { pages }
     }
 
-    fn process(&self, _site: &Site, meta: &mut PageMeta) -> Result<Processed, DirectiveError> {
-        warn!(
-            "page {} uses unimplemented pagestats",
-            meta.path().display()
-        );
-        Err(DirectiveError::UnimplementedDirective("pagestats".into()))
+    fn link(container: &Path, meta: &PageMeta) -> String {
+        let link = make_relative_link(container, meta.path());
+        format!("[{}]({})", meta.title(), link.display())
     }
 }