From 0c0a666d1b088f471c8b3a5bc959f5d0f2c0def4 Mon Sep 17 00:00:00 2001
From: Nicky Bondarenko <nickyb@spotify.com>
Date: Tue, 9 Apr 2024 14:48:42 +0200
Subject: [PATCH] fix: handle continuing writing to batch from disk, remove
 force unwrapping

---
 Sources/Confidence/EventStorage.swift | 31 +++++++++++++--------------
 1 file changed, 15 insertions(+), 16 deletions(-)

diff --git a/Sources/Confidence/EventStorage.swift b/Sources/Confidence/EventStorage.swift
index dc38104d..24b5a97b 100644
--- a/Sources/Confidence/EventStorage.swift
+++ b/Sources/Confidence/EventStorage.swift
@@ -20,26 +20,25 @@ internal class EventStorageImpl: EventStorage {
 
     init() throws {
         currentFolderURL = URL(fileURLWithPath: try EventStorageImpl.getFolderURL())
-        currentFileURL = currentFolderURL.appendingPathComponent("events-\(Date().currentTime)")
+        currentFileURL = EventStorageImpl.latestWriteFile() ?? currentFolderURL.appendingPathComponent("events-\(Date().currentTime)")
     }
 
     func startNewBatch() {
-        let latestWriteFile = latestWriteFile()
-        if latestWriteFile != nil {
-            currentFileURL = latestWriteFile!
-            currentBatch = eventsFrom(id: latestWriteFile!.absoluteString)
+        guard let latestWriteFile = EventStorageImpl.latestWriteFile() else {
+            let urlString = "\(currentFileURL)\(EventStorageImpl.READYTOSENDEXTENSION)"
+            let newPath = URL(fileURLWithPath: urlString)
+            do {
+                try FileManager.default.moveItem(at: currentFileURL, to: newPath)
+            } catch {
+                Logger(subsystem: "com.confidence.eventsender", category: "storage").error(
+                "Error when trying to start a new batch: \(error)")
+            }
+            currentFileURL = currentFolderURL.appendingPathComponent("events-\(Date().currentTime)")
+            currentBatch = []
             return
         }
-        let urlString = "\(currentFileURL)"+"\(EventStorageImpl.READYTOSENDEXTENSION)"
-        let newPath = URL(fileURLWithPath: urlString)
-        do {
-            try FileManager.default.moveItem(at: currentFileURL, to: newPath)
-        } catch {
-            Logger(subsystem: "com.confidence.eventsender", category: "storage").error(
-            "Error when trying to start a new batch: \(error)")
-        }
-        currentFileURL = currentFolderURL.appendingPathComponent("events-\(Date().currentTime)")
-        currentBatch = []
+        currentFileURL = latestWriteFile
+        currentBatch = eventsFrom(id: latestWriteFile.absoluteString)
     }
     
     func writeEvent(event: Event) {
@@ -107,7 +106,7 @@ internal class EventStorageImpl: EventStorage {
         return nestedFolderURL
     }
 
-    private func latestWriteFile() -> URL? {
+    private static func latestWriteFile() -> URL? {
         var directoryContents: [String] = []
         do {
             directoryContents = try FileManager.default.contentsOfDirectory(atPath: EventStorageImpl.getFolderURL())