Use SettingsStore for querying the comment style in SecretDetailView

Fixes: maxgoedjen#536 (comment)
paulhdk · Sep 16, 2024 · 9d2165a · 9d2165a
1 parent a0ecd64
commit 9d2165a
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/Sources/Secretive/Views/SecretDetailView.swift b/Sources/Secretive/Views/SecretDetailView.swift
@@ -4,7 +4,7 @@ import SecretKit
 struct SecretDetailView<SecretType: Secret>: View {
 
     @State var secret: SecretType
-    @AppStorage("com.maxgoedjen.Secretive.commentStyle") var style: CommentStyle = .keyAndHost
+    @EnvironmentObject private var settingsStore: SettingsStore
 
     private let keyWriter = OpenSSHKeyWriter()
     private let publicKeyFileStoreController = PublicKeyFileStoreController(homeDirectory: NSHomeDirectory().replacingOccurrences(of: Bundle.main.hostBundleID, with: Bundle.main.agentBundleID))
@@ -43,11 +43,12 @@ struct SecretDetailView<SecretType: Secret>: View {
     }
 
     var keyString: String {
+        var style: CommentStyle = CommentStyle(rawValue: settingsStore["com.maxgoedjen.Secretive.commentStyle"] ?? CommentStyle.keyAndHost.rawValue)!
         switch style {
-        case CommentStyle.none:
-            keyWriter.openSSHString(secret: secret, comment: "")
-        default:
-            keyWriter.openSSHString(secret: secret, comment: "\(dashedKeyName)@\(dashedHostName)")
+        case .none:
+            return keyWriter.openSSHString(secret: secret, comment: "")
+        case .keyAndHost:
+            return keyWriter.openSSHString(secret: secret, comment: "\(dashedKeyName)@\(dashedHostName)")
         }
     }
 }