From 06b52872961427ed4eb32e60f33936fa0a309d86 Mon Sep 17 00:00:00 2001
From: A Farzat <a@farzat.xyz>
Date: Sun, 21 Jun 2026 21:35:56 +0300
Subject: Refactor markdown generators into a single struct

---
 src/run.rs | 30 ++++++++++--------------------
 1 file changed, 10 insertions(+), 20 deletions(-)

(limited to 'src/run.rs')

diff --git a/src/run.rs b/src/run.rs
index 50b076b..c28d7f3 100644
--- a/src/run.rs
+++ b/src/run.rs
@@ -7,8 +7,7 @@
 use std::io::{Read, Write};
 
 use repo2markdown::{
-    fenced_md_generator::generate_fenced_markdown, logger::Logger,
-    md_generator::generate_markdown_from_paths, path_list::paths_from_null_separated_bytes,
+    generator::MarkdownGenerator, logger::Logger, path_list::paths_from_null_separated_bytes,
 };
 
 use crate::cli::Cli;
@@ -31,26 +30,17 @@ pub fn run<R: Read, W: Write>(
     input.read_to_end(&mut input_buf)?;
     let paths = paths_from_null_separated_bytes(&input_buf);
 
+    let md_generator = MarkdownGenerator::new(&paths)
+        .with_logger(logger)
+        .with_origin(&cli.origin)
+        .with_project_title(cli.name.as_deref())
+        .with_render_options(cli.render_options())
+        .with_root(&cli.root);
+
     if cli.fenced {
-        generate_fenced_markdown(
-            &paths,
-            output,
-            cli.render_options(),
-            &cli.root,
-            &cli.origin,
-            cli.name.as_deref(),
-            logger,
-        )
+        md_generator.generate_fenced(output)
     } else {
-        generate_markdown_from_paths(
-            &paths,
-            output,
-            cli.render_options(),
-            &cli.root,
-            &cli.origin,
-            cli.name.as_deref(),
-            logger,
-        )
+        md_generator.generate(output)
     }
 }
 
-- 
cgit v1.3.1