From adfeb40a7cc3d82b60bcd758834e8cdee454397c Mon Sep 17 00:00:00 2001
From: David Ungar <dungar@apple.com>
Date: Thu, 3 Dec 2020 12:57:05 -0800
Subject: [PATCH] Encapsulate local outputFileMap to forestall bugs

---
 Sources/SwiftDriver/Driver/Driver.swift | 36 +++++++++++++------------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/Sources/SwiftDriver/Driver/Driver.swift b/Sources/SwiftDriver/Driver/Driver.swift
index b52e6af2b..60307a717 100644
--- a/Sources/SwiftDriver/Driver/Driver.swift
+++ b/Sources/SwiftDriver/Driver/Driver.swift
@@ -369,25 +369,27 @@ public struct Driver {
         guard let modTime = try? fileSystem
           .getFileInfo($0.file).modTime else { return nil }
         return ($0, modTime)
-    })
-
-    let outputFileMap: OutputFileMap?
-    // Initialize an empty output file map, which will be populated when we start creating jobs.
-    if let outputFileMapArg = parsedOptions.getLastArgument(.outputFileMap)?.asSingle {
-      do {
-        let path = try VirtualPath(path: outputFileMapArg)
-        outputFileMap = try .load(fileSystem: fileSystem, file: path, diagnosticEngine: diagnosticEngine)
-      } catch {
-        throw Error.unableToLoadOutputFileMap(outputFileMapArg)
+      })
+
+    do {
+      let outputFileMap: OutputFileMap?
+      // Initialize an empty output file map, which will be populated when we start creating jobs.
+      if let outputFileMapArg = parsedOptions.getLastArgument(.outputFileMap)?.asSingle {
+        do {
+          let path = try VirtualPath(path: outputFileMapArg)
+          outputFileMap = try .load(fileSystem: fileSystem, file: path, diagnosticEngine: diagnosticEngine)
+        } catch {
+          throw Error.unableToLoadOutputFileMap(outputFileMapArg)
+        }
+      } else {
+        outputFileMap = nil
       }
-    } else {
-      outputFileMap = nil
-    }
 
-    if let workingDirectory = self.workingDirectory {
-      self.outputFileMap = outputFileMap?.resolveRelativePaths(relativeTo: workingDirectory)
-    } else {
-      self.outputFileMap = outputFileMap
+      if let workingDirectory = self.workingDirectory {
+        self.outputFileMap = outputFileMap?.resolveRelativePaths(relativeTo: workingDirectory)
+      } else {
+        self.outputFileMap = outputFileMap
+      }
     }
 
     self.fileListThreshold = try Self.computeFileListThreshold(&self.parsedOptions, diagnosticsEngine: diagnosticsEngine)