Clear dir and support filters (#10)

EmergeTools · Aug 5, 2023 · 729003d · 729003d
1 parent 3c25033
commit 729003d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/Sources/SnapshottingSwift/Initializer.swift b/Sources/SnapshottingSwift/Initializer.swift
@@ -33,10 +33,7 @@ private class CompletedViewController: UIViewController {
         label.numberOfLines = 0
         label.accessibilityIdentifier = "emg_finished_label"
 
-        let paths = NSSearchPathForDirectoriesInDomains(.documentDirectory, .userDomainMask, true)
-        if let documentsDirectory = paths.first {
-            label.text = "\(documentsDirectory)"
-        }
+        label.text = "\(Snapshots.resultsDir.path())"
 
         view.addSubview(label)
 

diff --git a/Sources/SnapshottingSwift/Snapshots.swift b/Sources/SnapshottingSwift/Snapshots.swift
@@ -31,7 +31,13 @@ class Snapshots {
 
   var completion: (() -> Void)?
 
+  static let documentsURL = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0]
+  static let resultsDir = documentsURL.appendingPathComponent("EMGSnapshots")
+
   func saveSnapshots(completion: @escaping () -> Void) {
+    try? FileManager.default.removeItem(at: Self.resultsDir)
+    try! FileManager.default.createDirectory(at: Self.resultsDir, withIntermediateDirectories: true)
+
     self.completion = completion
     let snapshotPreviews = ProcessInfo.processInfo.environment["SNAPSHOT_PREVIEWS"];
     var previewsSet: Set<String>? = nil
@@ -60,7 +66,7 @@ class Snapshots {
       view = AnyView(view.colorScheme(colorScheme))
     }
     let fileName = "\(typeName)-\(preview.previewId).png"
-    let file = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask)[0].appending(path: fileName, directoryHint: .notDirectory)
+    let file = Self.resultsDir.appending(path: fileName, directoryHint: .notDirectory)
     print(file)
     view.snapshot(layout: preview.layout, window: window, async: false) { image in
           if let pngData = image.pngData() {

diff --git a/Sources/SnapshottingTests/PreviewTest.m b/Sources/SnapshottingTests/PreviewTest.m
@@ -67,7 +67,8 @@ - (void)generateSnapshots {
   } mutableCopy];
   NSArray *previews = [self snapshotPreviews];
   if (previews) {
-    launchEnvironment[@"SNAPSHOT_PREVIEWS"] = previews;
+    NSData *jsonData = [NSJSONSerialization dataWithJSONObject:previews options:nil error:nil];
+    launchEnvironment[@"SNAPSHOT_PREVIEWS"] = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
   }
   app.launchEnvironment = launchEnvironment;
   [app launch];
@@ -82,7 +83,7 @@ - (void)generateSnapshots {
 
   resultPath = label.label;
 
-  NSLog(@"Images can be found in %@", path);
+  NSLog(@"Images can be found in %@", resultPath);
 
   NSError *error = nil;
   NSArray<NSString *> *files = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:resultPath error:&error];