WIP handle dry run and full plain inputs

Author: Tjaz Erzen

plain2code.py

@@ -242,6 +242,43 @@ def render(args, run_state: RunState, event_bus: EventBus):  # noqa: C901
 def main():  # noqa: C901
     args = parse_arguments()
 
+    # Handle early-exit flags before heavy initialization
+    if args.dry_run or args.full_plain:
+        template_dirs = file_utils.get_template_directories(args.filename, args.template_dir, DEFAULT_TEMPLATE_DIRS)
+
+        try:
+            if args.full_plain:
+                # Read the raw plain source file
+                with open(args.filename, 'r') as f:
+                    plain_source = f.read()
+                
+                [full_plain_source, _] = file_utils.get_loaded_templates(template_dirs, plain_source)
+                
+                if args.verbose:
+                    console.info("Full plain text:\n")
+                
+                console.info(full_plain_source)
+                return
+
+            if args.dry_run:
+                console.info("Printing dry run output...\n")
+                _, plain_source_tree, _ = plain_file.plain_file_parser(args.filename, template_dirs)
+
+                render_range = None
+                if args.render_range or args.render_from:
+                    if args.render_range:
+                        render_range = get_render_range(args.render_range, plain_source_tree)
+                    elif args.render_from:
+                        render_range = get_render_range_from(args.render_from, plain_source_tree)
+
+                from plain2code_utils import print_dry_run_output
+
+                print_dry_run_output(plain_source_tree, render_range)
+                return
+        except Exception as e:
+            console.error(f"Error: {str(e)}")
+            return
+
     event_bus = EventBus()
 
     if not args.api:

plain2code_arguments.py

@@ -301,6 +301,9 @@ def parse_arguments():
     if not args.log_to_file and args.log_file_name != DEFAULT_LOG_FILE_NAME:
         parser.error("--log-file-name cannot be used when --log-to-file is False.")
 
+    if args.full_plain and args.dry_run:
+        parser.error("--full-plain and --dry-run are mutually exclusive")
+
     script_arg_names = [UNIT_TESTS_SCRIPT_NAME, CONFORMANCE_TESTS_SCRIPT_NAME]
     for script_name in script_arg_names:
         args = process_test_script_path(script_name, args)

plain2code_utils.py

@@ -22,14 +22,14 @@ def print_dry_run_output(plain_source_tree: dict, render_range: Optional[list[st
             specifications, _ = plain_spec.get_specifications_for_frid(plain_source_tree, frid)
             functional_requirement_text = specifications[plain_spec.FUNCTIONAL_REQUIREMENTS][-1]
             console.info(
-                "-------------------------------------"
-                f"Rendering functional requirement {frid}"
-                f"{functional_requirement_text}"
-                "-------------------------------------"
+                "-------------------------------------\n"
+                f"Rendering functional requirement {frid}\n"
+                f"{functional_requirement_text}\n"
+                "-------------------------------------\n"
             )
             if plain_spec.ACCEPTANCE_TESTS in specifications:
                 for i, acceptance_test in enumerate(specifications[plain_spec.ACCEPTANCE_TESTS], 1):
-                    console.info(f"Generating acceptance test #{i}:\n\n{acceptance_test}")
+                    console.info(f"Generating acceptance test #{i}:\n\n{acceptance_test}\n")
         else:
             console.info(
                 "-------------------------------------\n"