Add support for request and response haeders in task cell header

kean · Sep 14, 2024 · 02fc0cd · 02fc0cd
1 parent 59ae038
commit 02fc0cd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/Sources/PulseUI/Features/Console/Views/ConsoleListDisplaySettings.swift b/Sources/PulseUI/Features/Console/Views/ConsoleListDisplaySettings.swift
@@ -65,18 +65,18 @@ public struct ConsoleListDisplaySettings: Hashable, Codable {
         public var id: URLComponent { self }
     }
 
-    public enum TaskField: Sendable, Codable, Hashable, Identifiable, CaseIterable {
+    public enum TaskField: Sendable, Codable, Hashable, Identifiable {
         case method
         case requestSize
         case responseSize
         case responseContentType
         case duration
         case host
         case statusCode
-        /// The type of the task, e.g. "Data" or "Download"
         case taskType
-        /// The `taskDescription` value of `URLSessionTask`.
         case taskDescription
+        case requestHeaderField(key: String)
+        case responseHeaderField(key: String)
 
         public var id: TaskField { self }
 
@@ -91,6 +91,8 @@ public struct ConsoleListDisplaySettings: Hashable, Codable {
             case .statusCode: "Status Code"
             case .taskType: "Task Type"
             case .taskDescription: "Task Description"
+            case .requestHeaderField(let key): "Request Header \"\(key)\""
+            case .responseHeaderField(let key): "Response Header '\(key)\""
             }
         }
     }

diff --git a/Sources/PulseUI/Features/Console/Views/ConsoleTaskCell.swift b/Sources/PulseUI/Features/Console/Views/ConsoleTaskCell.swift
@@ -167,6 +167,10 @@ private extension NetworkTaskEntity {
             NetworkLogger.TaskType(rawValue: taskType)?.urlSessionTaskClassName
         case .taskDescription:
             taskDescription
+        case .requestHeaderField(let key):
+            (currentRequest?.headers ?? [:])[key]
+        case .responseHeaderField(let key):
+            (response?.headers ?? [:])[key]
         }
     }