From 711df9de70e8a61a441b7ba7200ae006258ce978 Mon Sep 17 00:00:00 2001 From: A Farzat Date: Mon, 22 Jun 2026 05:15:59 +0300 Subject: Refactor markdown generation using a method object --- src/generator/generation.rs | 55 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 src/generator/generation.rs (limited to 'src/generator/generation.rs') diff --git a/src/generator/generation.rs b/src/generator/generation.rs new file mode 100644 index 0000000..4235cd2 --- /dev/null +++ b/src/generator/generation.rs @@ -0,0 +1,55 @@ +use std::{collections::HashSet, io::Write, path::Path}; + +use crate::{ + logger::Logger, + normalizer::{NormalizeError, NormalizedPath, Normalizer}, + renderer::Renderer, + util::path_display::display_path, +}; + +use super::MarkdownGenerator; + +pub(super) struct MarkdownGeneration<'a, W: Write> { + logger: Logger, + normalizer: Normalizer, + paths: &'a [&'a Path], + project_title: String, + renderer: Renderer, + seen_paths: HashSet, +} + +impl<'a, W: Write> MarkdownGeneration<'a, W> { + pub fn new(generator: MarkdownGenerator<'a>, output: W) -> Result { + let renderer = + Renderer::new(output, generator.render_options).with_logger(generator.logger); + Ok(Self { + logger: generator.logger, + normalizer: Normalizer::new(generator.root, generator.origin_base)?, + paths: generator.paths, + project_title: generator.get_or_derive_project_title().to_string(), + renderer, + seen_paths: HashSet::new(), + }) + } + + pub fn run(mut self) -> Result<(), Box> { + self.renderer.render_header(&self.project_title)?; + for path in self.paths { + self.render_path(path)?; + } + Ok(()) + } + + fn render_path(&mut self, path: &Path) -> Result<(), Box> { + let normalized_path = self.normalizer.normalize(path)?; + if self.seen_paths.insert(normalized_path.clone()) { + self.renderer.render_path(&normalized_path)?; + } else { + self.logger.warn(format!( + "skipping duplicate file: {}", + display_path(&normalized_path.root_relative) + )); + } + Ok(()) + } +} -- cgit v1.3.1