Continue to write other files when a certain file is corrupted.

RedMadRobot · Jun 6, 2023 · 17755fa · 17755fa
1 parent fbf596d
commit 17755fa
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/Sources/FigmaExport/Output/FileWriter.swift b/Sources/FigmaExport/Output/FileWriter.swift
@@ -1,33 +1,38 @@
-import Foundation
 import FigmaExportCore
+import Foundation
+import Logging
 #if os(Linux)
 import FoundationXML
 #endif
 
 final class FileWriter {
-
     private let fileManager: FileManager
-
+    private let logger = Logger(label: "com.redmadrobot.figma-export.file-writer")
+
     init(fileManager: FileManager = .default) {
         self.fileManager = fileManager
     }
-    
+
     func write(files: [FileContents]) throws {
         try files.forEach { file in
             let directoryURL = URL(fileURLWithPath: file.destination.directory.path)
             try fileManager.createDirectory(at: directoryURL, withIntermediateDirectories: true, attributes: nil)
-            
+
             let fileURL = URL(fileURLWithPath: file.destination.url.path)
-            if let data = file.data {
-                try data.write(to: fileURL, options: .atomic)
-            } else if let localFileURL = file.dataFile {
-                _ = try fileManager.replaceItemAt(fileURL, withItemAt: localFileURL)
-            } else {
-                fatalError("FileContents.data is nil. Use FileDownloader to download contents of the file.")
+            do {
+                if let data = file.data {
+                    try data.write(to: fileURL, options: .atomic)
+                } else if let localFileURL = file.dataFile {
+                    _ = try fileManager.replaceItemAt(fileURL, withItemAt: localFileURL)
+                } else {
+                    fatalError("FileContents.data is nil. Use FileDownloader to download contents of the file.")
+                }
+            } catch let e {
+                logger.error("\(e.localizedDescription)")
             }
         }
     }
-    
+
     func write(xmlFile: XMLDocument, directory: URL) throws {
         let fileURL = URL(fileURLWithPath: directory.path)
         let options: XMLNode.Options = [.nodePrettyPrint, .nodeCompactEmptyElement]