Implement certificate status tracking in CertificateView

Added certificate status tracking and updated UI to display status information.

Author: NovaDev404

Sources/prosign/views/CertificateView.swift

@@ -1,6 +1,6 @@
-// CertificateView.swift
 import SwiftUI
 import UniformTypeIdentifiers
+import ProStoreTools
 // Centralized types to avoid conflicts
 struct CertificateFileItem {
     var name: String = ""
@@ -11,16 +11,52 @@ struct CustomCertificate: Identifiable {
     let displayName: String
     let folderName: String
 }
+
+enum CertStatus {
+    case loading
+    case signed(Date?)
+    case revoked(Date?)
+    case unknown
+}
+
+extension CertStatus {
+    var description: String {
+        switch self {
+        case .loading:
+            return "Status: Loading"
+        case .signed:
+            return "Status: Signed"
+        case .revoked:
+            return "Status: Revoked"
+        case .unknown:
+            return "Status: Unknown"
+        }
+    }
+    
+    var color: Color {
+        switch self {
+        case .loading:
+            return .black
+        case .signed:
+            return .green
+        case .revoked:
+            return .red
+        case .unknown:
+            return .yellow
+        }
+    }
+}
 // MARK: - CertificateView (List + Add/Edit launchers)
 struct CertificateView: View {
     @State private var customCertificates: [CustomCertificate] = []
+    @State private var certStatuses: [String: CertStatus] = [:]
     @State private var showAddCertificateSheet = false
     @State private var editingCertificate: CustomCertificate? = nil // Used only for edit sheet (.sheet(item:))
     @State private var selectedCert: String? = nil
     @State private var showingDeleteAlert = false
     @State private var certToDelete: CustomCertificate?
     @State private var newlyAddedFolder: String? = nil
-  
+ 
     var body: some View {
         // <-- Removed nested NavigationStack to avoid hiding the title from the parent stack
         ScrollView {
@@ -32,6 +68,12 @@ struct CertificateView: View {
                                 .font(.title2)
                                 .fontWeight(.semibold)
                                 .foregroundColor(.primary)
+                            if let status = certStatuses[cert.folderName] {
+                                Text(status.description)
+                                    .font(.caption)
+                                    .fontWeight(.medium)
+                                    .foregroundColor(status.color)
+                            }
                         }
                         .padding(20)
                         .frame(maxWidth: .infinity)
@@ -53,7 +95,7 @@ struct CertificateView: View {
                                 UserDefaults.standard.set(selectedCert, forKey: "selectedCertificateFolder")
                             }
                         }
-                      
+                     
                         HStack {
                             Button(action: {
                                 // EDIT: trigger identifiable sheet
@@ -66,9 +108,9 @@ struct CertificateView: View {
                                     .background(Color(.systemGray6).opacity(0.8))
                                     .clipShape(Circle())
                             }
-                          
+                         
                             Spacer()
-                          
+                         
                             Button(action: {
                                 if customCertificates.count > 1 {
                                     certToDelete = cert
@@ -134,13 +176,53 @@ struct CertificateView: View {
             reloadCertificatesAndEnsureSelection()
         }
     }
-  
+ 
     private func reloadCertificatesAndEnsureSelection() {
         customCertificates = CertificateFileManager.shared.loadCertificates()
         selectedCert = UserDefaults.standard.string(forKey: "selectedCertificateFolder")
         ensureSelection()
+        checkStatuses()
+    }
+    
+    private func checkStatuses() {
+        for cert in customCertificates {
+            let folderName = cert.folderName
+            certStatuses[folderName] = .loading
+            
+            let certDir = CertificateFileManager.shared.certificatesDirectory.appendingPathComponent(folderName)
+            let p12URL = certDir.appendingPathComponent("certificate.p12")
+            let provURL = certDir.appendingPathComponent("profile.mobileprovision")
+            let passwordURL = certDir.appendingPathComponent("password.txt")
+            
+            let p12Password: String
+            if let passwordData = try? Data(contentsOf: passwordURL),
+               let passwordStr = String(data: passwordData, encoding: .utf8)?.trimmingCharacters(in: .whitespacesAndNewlines) {
+                p12Password = passwordStr
+            } else {
+                p12Password = ""
+            }
+            
+            ProStoreTools.checkRevokage(
+                p12URL: p12URL,
+                provURL: provURL,
+                p12Password: p12Password
+            ) { status, expirationDate, error in
+                DispatchQueue.main.async {
+                    let newStatus: CertStatus
+                    switch status {
+                    case 0:
+                        newStatus = .signed(expirationDate)
+                    case 1, 2:
+                        newStatus = .revoked(expirationDate)
+                    default:
+                        newStatus = .unknown
+                    }
+                    self.certStatuses[folderName] = newStatus
+                }
+            }
+        }
     }
-  
+ 
     private func ensureSelection() {
         if selectedCert == nil || !customCertificates.contains(where: { $0.folderName == selectedCert }) {
             if let firstCert = customCertificates.first {
@@ -149,11 +231,11 @@ struct CertificateView: View {
             }
         }
     }
-  
+ 
     private func deleteCertificate(_ cert: CustomCertificate) {
         try? CertificateFileManager.shared.deleteCertificate(folderName: cert.folderName)
         customCertificates = CertificateFileManager.shared.loadCertificates()
-      
+     
         if selectedCert == cert.folderName {
             if let newSelection = customCertificates.first {
                 selectedCert = newSelection.folderName
@@ -164,14 +246,15 @@ struct CertificateView: View {
             }
         }
         ensureSelection()
+        checkStatuses()
     }
 }
 // MARK: - Add / Edit View
 struct AddCertificateView: View {
     @Environment(\.dismiss) private var dismiss
     let editingCertificate: CustomCertificate?
     let onSave: ((String) -> Void)?
-  
+ 
     @State private var p12File: CertificateFileItem?
     @State private var provFile: CertificateFileItem?
     @State private var password = ""
@@ -180,12 +263,12 @@ struct AddCertificateView: View {
     @State private var errorMessage = ""
     @State private var displayName: String = ""
     @State private var hasLoadedForEdit = false
-  
+ 
     init(editingCertificate: CustomCertificate? = nil, onSave: ((String) -> Void)? = nil) {
         self.editingCertificate = editingCertificate
         self.onSave = onSave
     }
-  
+ 
     var body: some View {
         // Use a NavigationStack inside the sheet so the sheet has its own nav bar
         NavigationStack {
@@ -205,7 +288,7 @@ struct AddCertificateView: View {
                         }
                     }
                     .disabled(isChecking)
-                  
+                 
                     Button(action: { activeSheet = .prov }) {
                         HStack {
                             Image(systemName: "gearshape.fill")
@@ -221,20 +304,20 @@ struct AddCertificateView: View {
                     }
                     .disabled(isChecking)
                 }
-              
+             
                 Section(header: Text("Display Name")) {
                     TextField("Optional Display Name", text: $displayName)
                         .disabled(isChecking)
                 }
-              
+             
                 Section(header: Text("Password")) {
                     SecureField("Enter Password", text: $password)
                         .disabled(isChecking)
                     Text("Enter the password for the certificate. Leave it blank if there is no password needed.")
                         .font(.caption)
                         .foregroundColor(.secondary)
                 }
-              
+             
                 if !errorMessage.isEmpty {
                     Text(errorMessage)
                         .foregroundColor(.red)
@@ -250,7 +333,7 @@ struct AddCertificateView: View {
                     }
                     .disabled(isChecking)
                 }
-              
+             
                 ToolbarItem(placement: .navigationBarTrailing) {
                     if isChecking {
                         ProgressView()
@@ -283,17 +366,17 @@ struct AddCertificateView: View {
             }
         } // NavigationStack
     }
-  
+ 
     private func loadForEdit(cert: CustomCertificate) {
         let certFolder = CertificateFileManager.shared.certificatesDirectory.appendingPathComponent(cert.folderName)
         let p12URL = certFolder.appendingPathComponent("certificate.p12")
         let provURL = certFolder.appendingPathComponent("profile.mobileprovision")
         let passwordURL = certFolder.appendingPathComponent("password.txt")
         let nameURL = certFolder.appendingPathComponent("name.txt")
-      
+     
         p12File = CertificateFileItem(name: "certificate.p12", url: p12URL)
         provFile = CertificateFileItem(name: "profile.mobileprovision", url: provURL)
-      
+     
         if let pwData = try? Data(contentsOf: passwordURL), let pw = String(data: pwData, encoding: .utf8) {
             password = pw
         }
@@ -303,10 +386,10 @@ struct AddCertificateView: View {
     }
     private func saveCertificate() {
         guard let p12URL = p12File?.url, let provURL = provFile?.url else { return }
-      
+     
         isChecking = true
         errorMessage = ""
-      
+     
         let workItem: DispatchWorkItem = DispatchWorkItem {
             do {
                 var p12Data: Data
@@ -326,17 +409,17 @@ struct AddCertificateView: View {
                     p12Data = try Data(contentsOf: p12URL)
                     provData = try Data(contentsOf: provURL)
                 }
-              
+             
                 let checkResult = CertificatesManager.check(p12Data: p12Data, password: self.password, mobileProvisionData: provData)
                 var dispatchError: String?
-              
+             
                 switch checkResult {
                 case .success(.success):
                     // Generate displayName from cert if not set
                     if localDisplayName.isEmpty {
                         localDisplayName = CertificatesManager.getCertificateName(mobileProvisionData: provData) ?? "Custom Certificate"
                     }
-                  
+                 
                     if let folder = self.editingCertificate?.folderName {
                         try CertificateFileManager.shared.updateCertificate(folderName: folder, p12Data: p12Data, provData: provData, password: self.password, displayName: localDisplayName)
                     } else {
@@ -350,7 +433,7 @@ struct AddCertificateView: View {
                 case .failure(let error):
                     dispatchError = "Error: \(error.localizedDescription)"
                 }
-              
+             
                 DispatchQueue.main.async {
                     self.isChecking = false
                     if let err = dispatchError {