add check that the downloaded file has a valid shasum before trying to run it

Author: designatednerd

Sources/ApolloCodegenLib/ApolloCLI.swift

@@ -13,6 +13,12 @@ public struct ApolloCLI {
   ///   - timeout: The maximum time to wait before indicating that the download timed out, in seconds.
   public static func createCLI(cliFolderURL: URL, timeout: Double) throws -> ApolloCLI {
     try CLIDownloader.downloadIfNeeded(cliFolderURL: cliFolderURL, timeout: timeout)
+    
+    if !(try CLIExtractor.validateSHASUMOfDownloadedFile(in: cliFolderURL)) {
+      CodegenLogger.log("Downloaded zip file has incorrect SHASUM, forcing redownolad")
+      try CLIDownloader.forceRedownload(cliFolderURL: cliFolderURL, timeout: timeout)
+    }
+    
     let binaryFolderURL = try CLIExtractor.extractCLIIfNeeded(from: cliFolderURL)
     return ApolloCLI(binaryFolderURL: binaryFolderURL)
   }

Sources/ApolloCodegenLib/CLIExtractor.swift

@@ -78,6 +78,22 @@ struct CLIExtractor {
     return true
   }
 
+  static func validateSHASUMOfDownloadedFile(in cliFolderURL: URL, expected: String = CLIExtractor.expectedSHASUM) throws -> Bool {
+    let zipFileURL = ApolloFilePathHelper.zipFileURL(fromCLIFolder: cliFolderURL)
+    
+    do {
+      try self.validateZipFileSHASUM(at: zipFileURL)
+      return true
+    } catch {
+      switch error {
+      case CLIExtractorError.zipFileHasInvalidSHASUM:
+        return false
+      default:
+        throw error
+      }
+    }
+  }
+  
   /// Writes the SHASUM of the extracted version of the CLI to a file for faster checks to ensure we have the correct version.
   ///
   /// - Parameter apolloFolderURL: The URL to the extracted apollo folder.
@@ -119,7 +135,7 @@ struct CLIExtractor {
   /// - Parameter expected: The expected SHASUM. Defaults to the real expected SHASUM. This parameter exists mostly for testing.
   static func validateZipFileSHASUM(at zipFileURL: URL, expected: String = CLIExtractor.expectedSHASUM) throws {
     let shasum = try FileManager.default.apollo.shasum(at: zipFileURL)
-    print("SHASUM: \(shasum)")
+    print("SHASUM of downloaded file: \(shasum)")
     guard shasum == expected else {
       throw CLIExtractorError.zipFileHasInvalidSHASUM(expectedSHASUM: expected, gotSHASUM: shasum)
     }