Merge pull request #196 from tahirmt/add-header-to-main

Add option to display custom headers in the list view
kean · Jun 6, 2023 · 09723a0 · 09723a0
2 parents fe02451 + f785f91
commit 09723a0
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 5 deletions.
diff --git a/Sources/PulseUI/Features/Console/Views/ConsoleTaskCell.swift b/Sources/PulseUI/Features/Console/Views/ConsoleTaskCell.swift
@@ -76,10 +76,33 @@ struct ConsoleTaskCell: View {
     }
 
     private var message: some View {
-        Text(task.url ?? "–")
-            .font(ConsoleConstants.fontBody)
-            .foregroundColor(.primary)
-            .lineLimit(settings.lineLimit)
+        VStack(spacing: 3) {
+            HStack {
+                Text(task.url ?? "–")
+                    .font(ConsoleConstants.fontBody)
+                    .foregroundColor(.primary)
+                    .lineLimit(settings.lineLimit)
+
+                Spacer()
+            }
+
+            let headerValueMap = settings.displayHeaders.reduce(into: [String: String]()) { partialResult, header in
+                partialResult[header] = task.originalRequest?.headers[header]
+            }
+
+            ForEach(headerValueMap.keys.sorted(), id: \.self) { key in
+                HStack {
+                    Text(key)
+                        .font(.caption)
+                        .foregroundColor(.secondary)
+
+                    Text(headerValueMap[key] ?? "-")
+                        .font(.callout)
+                        .bold()
+                    Spacer()
+                }
+            }
+        }
     }
 
     private var details: some View {

diff --git a/Sources/PulseUI/Features/Settings/SettingsView-ios.swift b/Sources/PulseUI/Features/Settings/SettingsView-ios.swift
@@ -10,7 +10,7 @@ import UniformTypeIdentifiers
 
 public struct SettingsView: View {
     @ObservedObject var viewModel: SettingsViewModel
-
+    @State private var newHeaderName = ""
     @EnvironmentObject private var settings: UserSettings
 
     public init(store: LoggerStore = .shared) {
@@ -35,6 +35,28 @@ public struct SettingsView: View {
                     RemoteLoggerSettingsView(viewModel: .shared)
                 }
             }
+
+            Section(header: Text("List headers"), footer: Text("These headers will be included in the list view")) {
+                ForEach(settings.displayHeaders, id: \.self) {
+                    Text($0)
+                }
+                .onDelete { indices in
+                    settings.displayHeaders.remove(atOffsets: indices)
+                }
+                HStack {
+                    TextField("New Header", text: $newHeaderName)
+                    Button(action: {
+                        withAnimation {
+                            settings.displayHeaders.append(newHeaderName)
+                            newHeaderName = ""
+                        }
+                    }) {
+                        Image(systemName: "plus.circle.fill")
+                            .accessibilityLabel("Add header")
+                    }
+                    .disabled(newHeaderName.isEmpty)
+                }
+            }
         }
     }
 }

diff --git a/Sources/PulseUI/Features/Settings/UserSettings.swift b/Sources/PulseUI/Features/Settings/UserSettings.swift
@@ -17,4 +17,29 @@ final class UserSettings: ObservableObject {
 
     @AppStorage("sharing-output")
     var sharingOutput: ShareStoreOutput = .store
+
+    @AppStorage("display-headers")
+    var displayHeaders: [String] = []
+}
+
+// MARK: - Array + RawREpresentable
+
+extension Array: RawRepresentable where Element: Codable {
+    public init?(rawValue: String) {
+        guard let data = rawValue.data(using: .utf8),
+              let result = try? JSONDecoder().decode([Element].self, from: data)
+        else {
+            return nil
+        }
+        self = result
+    }
+
+    public var rawValue: String {
+        guard let data = try? JSONEncoder().encode(self),
+              let result = String(data: data, encoding: .utf8)
+        else {
+            return "[]"
+        }
+        return result
+    }
 }