namespaced extension (#145)

Sources/SnapshotPreviewsCore/NSImage+PNG.swift → Sources/SnapshotPreviewsCore/NSImage+EMG.swift

@@ -1,5 +1,5 @@
 //
-//  NSImage+PNG.swift
+//  NSImage+EMG.swift
 //
 //
 //  Created by Noah Martin on 8/2/24.
@@ -8,15 +8,27 @@
 #if canImport(AppKit) && !targetEnvironment(macCatalyst)
 import AppKit
 
-extension NSImage {
+public extension NSImage {
+  var emg: NSImageSnapshotsNamespace {
+    .init(image: self)
+  }
+
+  struct NSImageSnapshotsNamespace {
+    private let image: NSImage
+
+    init(image: NSImage) {
+      self.image = image
+    }
+
     public func pngData() -> NSData? {
-        guard let tiffData = self.tiffRepresentation,
+        guard let tiffData = image.tiffRepresentation,
               let bitmapImageRep = NSBitmapImageRep(data: tiffData) else {
             return nil
         }
 
         let pngData = bitmapImageRep.representation(using: .png, properties: [:])
         return pngData as NSData?
     }
+  }
 }
 #endif

Sources/SnapshotPreviewsCore/UIImage+EMG.swift

@@ -0,0 +1,28 @@
+//
+//  UIImage+EMG.swift
+//
+//
+//  Created by Noah Martin on 8/8/24.
+//
+
+#if canImport(UIKit)
+import UIKit
+
+public extension UIImage {
+  var emg: UIImageSnapshotsNamespace {
+    .init(image: self)
+  }
+
+  struct UIImageSnapshotsNamespace {
+    private let image: UIImage
+
+    init(image: UIImage) {
+      self.image = image
+    }
+
+    public func pngData() -> Data? {
+      image.pngData()
+    }
+  }
+}
+#endif

Sources/SnapshottingSwift/Snapshots.swift

@@ -76,7 +76,7 @@ class Snapshots {
       let file = Self.resultsDir.appendingPathComponent(fileName, isDirectory: false)
       do {
         let image = try imageResult.get()
-        if let pngData = image.pngData() {
+        if let pngData = image.emg.pngData() {
           try pngData.write(to: file)
           result["imagePath"] = file.path
         } else {