feat(resolver): add agreement and authentication to document (#6)

beatt83 · Feb 25, 2024 · 36afa08 · 36afa08
1 parent 083b8fd
commit 36afa08
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Sources/PeerDID/PeerDID/PeerDIDHelper+ResolveAlgo2.swift b/Sources/PeerDID/PeerDID/PeerDIDHelper+ResolveAlgo2.swift
@@ -27,13 +27,37 @@ extension PeerDIDHelper {
                 return method
             }
 
+        let keyAgreements = verificationMethods.filter {
+            guard let type = KnownVerificationMaterialType(rawValue: $0.type) else {
+                return false
+            }
+
+            if case .agreement = type {
+                return true
+            }
+            return false
+        }.map(\.id)
+
+        let authenticationAgreements = verificationMethods.filter {
+            guard let type = KnownVerificationMaterialType(rawValue: $0.type) else {
+                return false
+            }
+
+            if case .agreement = type {
+                return true
+            }
+            return false
+        }.map(\.id)
+
         let documentServices = try services.enumerated().map {
             try decodedPeerDIDService(did: peerDID.string, serviceString: $0.element, index: $0.offset)
         }
 
         return DIDDocument(
             id: peerDID.string,
             verificationMethods: verificationMethods,
+            authentication: authenticationAgreements.map { .stringValue($0) },
+            keyAgreement: keyAgreements.map { .stringValue($0) },
             services: documentServices
         )
     }