Simplify a few parameters in gyb-generation of build-script.py

Author: ahoppen

build-script.py

@@ -191,16 +191,13 @@ def generate_single_gyb_file(
 # `sources_dir/gyb_generated`.
 def generate_gyb_files_helper(
     sources_dir: str,
-    destination_dir: Optional[str],
+    destination_dir: str,
     gyb_exec: str,
     add_source_locations: bool,
     verbose: bool,
 ) -> None:
     temp_files_dir = tempfile.gettempdir()
     make_dir_if_needed(temp_files_dir)
-
-    if destination_dir is None:
-        destination_dir = os.path.join(sources_dir, "gyb_generated")
     make_dir_if_needed(destination_dir)
 
     # Clear any *.swift files that are relics from the previous run.
@@ -288,23 +285,26 @@ def generate_syntax_node_template_gyb_files(
 # Maps directories containing .gyb files to the directories the generated files should
 # live in.
 def gyb_dir_mapping(
-    swiftsyntax_destination: Optional[str] = None,
-    swiftsyntax_raw_destination: Optional[str] = None,
-    swiftsyntaxbuilder_destination: Optional[str] = None,
-    swiftsyntaxparser_destination: Optional[str] = None,
-    generateswiftsyntaxbuilder_destination: Optional[str] = None,
-) -> Dict[str, Optional[str]]:
-    return {
-        SWIFTSYNTAX_DIR: swiftsyntax_destination,
-        os.path.join(SWIFTSYNTAX_DIR, "Raw"): swiftsyntax_raw_destination,
-        SWIFTSYNTAXBUILDER_DIR: swiftsyntaxbuilder_destination,
-        SWIFTSYNTAXPARSER_DIR: swiftsyntaxparser_destination,
-        GENERATESWIFTSYNTAXBUILDER_DIR: generateswiftsyntaxbuilder_destination,
-    }
+    write_to_temp_directories: bool
+) -> Dict[str, str]:
+    source_dirs = [
+        SWIFTSYNTAX_DIR,
+        os.path.join(SWIFTSYNTAX_DIR, "Raw"),
+        SWIFTSYNTAXBUILDER_DIR,
+        SWIFTSYNTAXPARSER_DIR,
+        GENERATESWIFTSYNTAXBUILDER_DIR,
+    ]
+    mapping = {}
+    for source_dir in source_dirs:
+        if write_to_temp_directories:
+            mapping[source_dir] = tempfile.mkdtemp()
+        else:
+            mapping[source_dir] = os.path.join(source_dir, "gyb_generated")
+    return mapping
 
 
 def generate_gyb_files(
-    gyb_exec: str, gyb_dir_mapping: Dict[str, Optional[str]],
+    gyb_exec: str, gyb_dir_mapping: Dict[str, str],
     add_source_locations: bool, verbose: bool,
 ) -> None:
     print("** Generating gyb Files **")
@@ -455,11 +455,7 @@ def build(self, product_name: str) -> None:
 
 def verify_generated_files(gyb_exec: str, verbose: bool) -> None:
     gyb_dirs = gyb_dir_mapping(
-        swiftsyntax_destination=tempfile.mkdtemp(),
-        swiftsyntax_raw_destination=tempfile.mkdtemp(),
-        swiftsyntaxbuilder_destination=tempfile.mkdtemp(),
-        swiftsyntaxparser_destination=tempfile.mkdtemp(),
-        generateswiftsyntaxbuilder_destination=tempfile.mkdtemp(),
+        write_to_temp_directories=True,  
     )
 
     generate_gyb_files(
@@ -685,7 +681,7 @@ def generate_source_code_command(args: argparse.Namespace) -> None:
     try:
         generate_gyb_files(
             args.gyb_exec,
-            gyb_dir_mapping=gyb_dir_mapping(),
+            gyb_dir_mapping=gyb_dir_mapping(write_to_temp_directories=False),
             add_source_locations=args.add_source_locations,
             verbose=args.verbose,
         )