HALF WORK: 2024.1.20 - 2

1 files changed, 10 insertions, 4 deletions

diff --git a/tools/cru-py/cru/service/_template.py b/tools/cru-py/cru/service/_template.py
index 9241a1f..ca2135f 100644
--- a/tools/cru-py/cru/service/_template.py
+++ b/tools/cru-py/cru/service/_template.py
@@ -45,10 +45,10 @@ class TemplateManager(AppCommandFeatureProvider):
         for file in CruIterator(self.template_tree.templates).transform(lambda t: t[0]):
             print(file.as_posix())
 
-    def _generate_files(self) -> None:
+    def _generate_files(self, dry_run: bool) -> None:
         config_manager = self.app.get_feature(ConfigManager)
         self.template_tree.generate_to(
-            self.generated_dir.full_path_str, config_manager.get_str_dict()
+            self.generated_dir.full_path_str, config_manager.get_str_dict(), dry_run
         )
 
     def get_command_info(self):
@@ -62,7 +62,10 @@ class TemplateManager(AppCommandFeatureProvider):
         _variables_parser = subparsers.add_parser(
             "variables", help="list variables used in all templates"
         )
-        _generate_parser = subparsers.add_parser("generate", help="generate templates")
+        generate_parser = subparsers.add_parser("generate", help="generate templates")
+        generate_parser.add_argument(
+            "--no-dry-run", action="store_true", help="generate and write target files"
+        )
 
     def run_command(self, args: Namespace) -> None:
         if args.template_command == "list":
@@ -71,4 +74,7 @@ class TemplateManager(AppCommandFeatureProvider):
             for var in self.template_tree.variables:
                 print(var)
         elif args.template_command == "generate":
-            self._generate_files()
+            dry_run = not args.no_dry_run
+            self._generate_files(dry_run)
+            if dry_run:
+                print("Dry run successfully.")