Move Shared code

kean · Aug 4, 2024 · aa4e0a0 · aa4e0a0
1 parent 9d95735
commit aa4e0a0
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 32 deletions.
diff --git a/Sources/PulseUI/Extensions/Pulse+Extensions.swift b/Sources/PulseUI/Extensions/Pulse+Extensions.swift
@@ -33,38 +33,6 @@ enum LoggerEntity {
     }
 }
 
-extension NetworkTaskEntity {
-    var requestFileViewerContext: FileViewerViewModel.Context {
-        FileViewerViewModel.Context(
-            contentType: originalRequest?.contentType,
-            originalSize: requestBodySize,
-            metadata: metadata,
-            isResponse: false,
-            error: nil
-        )
-    }
-
-    var responseFileViewerContext: FileViewerViewModel.Context {
-        FileViewerViewModel.Context(
-            contentType: response?.contentType,
-            originalSize: responseBodySize,
-            metadata: metadata,
-            isResponse: true,
-            error: decodingError
-        )
-    }
-
-    /// - returns `nil` if the task is an unknown state. It may happen if the
-    /// task is pending, but it's from the previous app run.
-    func state(in store: LoggerStore) -> NetworkTaskEntity.State? {
-        let state = self.state
-        if state == .pending && self.session != store.session.id {
-            return nil
-        }
-        return state
-    }
-}
-
 extension LoggerMessageEntity {
     var logLevel: LoggerStore.Level {
         LoggerStore.Level(rawValue: level) ?? .debug

diff --git a/Sources/PulseUI/Features/FileViewer/FileViewerViewModel.swift b/Sources/PulseUI/Features/FileViewer/FileViewerViewModel.swift
@@ -61,3 +61,35 @@ final class FileViewerViewModel: ObservableObject {
         return nil
     }
 }
+
+extension NetworkTaskEntity {
+    var requestFileViewerContext: FileViewerViewModel.Context {
+        FileViewerViewModel.Context(
+            contentType: originalRequest?.contentType,
+            originalSize: requestBodySize,
+            metadata: metadata,
+            isResponse: false,
+            error: nil
+        )
+    }
+
+    var responseFileViewerContext: FileViewerViewModel.Context {
+        FileViewerViewModel.Context(
+            contentType: response?.contentType,
+            originalSize: responseBodySize,
+            metadata: metadata,
+            isResponse: true,
+            error: decodingError
+        )
+    }
+
+    /// - returns `nil` if the task is an unknown state. It may happen if the
+    /// task is pending, but it's from the previous app run.
+    func state(in store: LoggerStore) -> NetworkTaskEntity.State? {
+        let state = self.state
+        if state == .pending && self.session != store.session.id {
+            return nil
+        }
+        return state
+    }
+}