feat(target): allow limiting the generated entrypoints

Introduces the `-t/--target` flag that can be used to provide a path.
Only entry points that have this path as a prefix will be generated.

1 files changed, 23 insertions, 12 deletions

diff --git a/src/main.rs b/src/main.rs
index 7d2f786..8168ad8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -23,6 +23,9 @@ struct Config {
   /// Base output directory [default: './']
   #[clap(short, long)]
   output: Option<PathBuf>,
+  /// Target files prefix
+  #[clap(short, long)]
+  target: Option<PathBuf>,
   /// Input files
   input: Vec<PathBuf>,
 }
@@ -131,6 +134,7 @@ fn write_to_file(
 
 fn main() -> Result<()> {
   let config = Config::parse();
+  let target = config.target.unwrap_or_default();
   let mut pandoc = Pandoc::new();
   pandoc.set_input(InputKind::Files(config.input));
   pandoc.set_input_format(InputFormat::Markdown, vec![]);
@@ -155,20 +159,27 @@ fn main() -> Result<()> {
                   .into_iter()
                   .find_map(|(k,p)| if k == "path" { Some(p.clone()) } else { None });
                 if entry || path.is_some() || PATH.is_match(id) {
-                  Key::Entry(PathBuf::from(path.unwrap_or_default()).join(id)) 
+                  let path = PathBuf::from(path.unwrap_or_default()).join(id);
+                  if path.starts_with(&target) {
+                    Some(Key::Entry(path))
+                  } else {
+                    None
+                  }
                 } else {
-                  Key::Macro(id.to_string())
+                  Some(Key::Macro(id.to_string()))
                 }
               };
-              if clss.iter().any(|c| c == "override") {
-                blocks.insert(key, Cow::from(code));
-              } else {
-                blocks.entry(key)
-                      .and_modify(|s| {
-                        *s += "\n";
-                        *s += Cow::from(code)
-                      })
-                      .or_insert(Cow::from(code));
+              if let Some(key) = key {
+                if clss.iter().any(|c| c == "override") {
+                  blocks.insert(key, Cow::from(code));
+                } else {
+                  blocks.entry(key)
+                        .and_modify(|s| {
+                          *s += "\n";
+                          *s += Cow::from(code)
+                        })
+                        .or_insert(Cow::from(code));
+                }
               }
             } else {
               eprintln!("Ignoring code block without ID:");
@@ -183,4 +194,4 @@ fn main() -> Result<()> {
   );
   pandoc.execute().unwrap();
   Ok(())
-}
+}
\ No newline at end of file