fix(cheqd): Populate contexts for JsonLD issuance

.changeset/kind-pots-yell.md

@@ -0,0 +1,5 @@
+---
+'@credo-ts/cheqd': patch
+---
+
+Populate contexts for Cheqd DID Document Records in wallet

packages/cheqd/src/anoncreds/utils/identifiers.ts

@@ -12,6 +12,9 @@ const QUERY = `([?][^#]*)?`
 const VERSION_ID = `(.*?)`
 const FRAGMENT = `([#].*)?`
 
+export const ED25519_SUITE_CONTEXT_URL_2018 = 'https://w3id.org/security/suites/ed25519-2018/v1'
+export const ED25519_SUITE_CONTEXT_URL_2020 = 'https://w3id.org/security/suites/ed25519-2020/v1'
+
 export const cheqdSdkAnonCredsRegistryIdentifierRegex = new RegExp(
   `^did:cheqd:${NETWORK}:${IDENTIFIER}${PATH}${QUERY}${FRAGMENT}$`
 )

packages/cheqd/src/dids/CheqdDidRegistrar.ts

@@ -25,9 +25,15 @@ import {
   getKeyFromVerificationMethod,
   JsonTransformer,
   VerificationMethod,
+  SECURITY_JWS_CONTEXT_URL,
+  DID_V1_CONTEXT_URL,
 } from '@credo-ts/core'
 
-import { parseCheqdDid } from '../anoncreds/utils/identifiers'
+import {
+  ED25519_SUITE_CONTEXT_URL_2018,
+  ED25519_SUITE_CONTEXT_URL_2020,
+  parseCheqdDid,
+} from '../anoncreds/utils/identifiers'
 import { CheqdLedgerService } from '../ledger'
 
 import {
@@ -39,6 +45,11 @@ import {
 
 export class CheqdDidRegistrar implements DidRegistrar {
   public readonly supportedMethods = ['cheqd']
+  private contextMapping = {
+    Ed25519VerificationKey2018: ED25519_SUITE_CONTEXT_URL_2018,
+    Ed25519VerificationKey2020: ED25519_SUITE_CONTEXT_URL_2020,
+    JsonWebKey2020: SECURITY_JWS_CONTEXT_URL,
+  }
 
   public async create(agentContext: AgentContext, options: CheqdDidCreateOptions): Promise<DidCreateResult> {
     const didRepository = agentContext.dependencyManager.resolve(DidRepository)
@@ -90,18 +101,6 @@ export class CheqdDidRegistrar implements DidRegistrar {
           network: withoutDidDocumentOptions.options.network as CheqdNetwork,
           publicKey: TypedArrayEncoder.toHex(key.publicKey),
         })
-
-        const contextMapping = {
-          Ed25519VerificationKey2018: 'https://w3id.org/security/suites/ed25519-2018/v1',
-          Ed25519VerificationKey2020: 'https://w3id.org/security/suites/ed25519-2020/v1',
-          JsonWebKey2020: 'https://w3id.org/security/suites/jws-2020/v1',
-        }
-        const contextUrl = contextMapping[verificationMethod.type]
-
-        // Add the context to the did document
-        // NOTE: cheqd sdk uses https://www.w3.org/ns/did/v1 while Credo did doc uses https://w3id.org/did/v1
-        // We should align these at some point. For now we just return a consistent value.
-        didDocument.context = ['https://www.w3.org/ns/did/v1', contextUrl]
       } else {
         return {
           didDocumentMetadata: {},
@@ -113,6 +112,25 @@ export class CheqdDidRegistrar implements DidRegistrar {
         }
       }
 
+      // Normalize context to an array
+      const contextSet = new Set<string>(
+        typeof didDocument.context === 'string'
+          ? [didDocument.context]
+          : Array.isArray(didDocument.context)
+          ? didDocument.context
+          : []
+      )
+
+      for (const verificationMethod of didDocument.verificationMethod || []) {
+        const contextUrl = this.contextMapping[verificationMethod.type as keyof typeof this.contextMapping]
+        if (contextUrl) {
+          contextSet.add(contextUrl)
+        }
+      }
+
+      // Add Cheqd default context to the did document
+      didDocument.context = Array.from(contextSet.add(DID_V1_CONTEXT_URL))
+
       const didDocumentJson = didDocument.toJSON() as DIDDocument
 
       const payloadToSign = await createMsgCreateDidDocPayloadToSign(didDocumentJson, versionId)
@@ -178,6 +196,14 @@ export class CheqdDidRegistrar implements DidRegistrar {
             },
           }
         }
+        // Normalize existing context to an array
+        const contextSet = new Set<string>(
+          typeof didDocument.context === 'string'
+            ? [didDocument.context]
+            : Array.isArray(didDocument.context)
+            ? didDocument.context
+            : []
+        )
 
         if (verificationMethod) {
           const privateKey = verificationMethod.privateKey
@@ -196,6 +222,12 @@ export class CheqdDidRegistrar implements DidRegistrar {
             keyType: KeyType.Ed25519,
             privateKey: privateKey,
           })
+          const contextUrl = this.contextMapping[verificationMethod.type as keyof typeof this.contextMapping]
+          if (contextUrl) {
+            contextSet.add(contextUrl)
+          }
+          // Add Cheqd default context to the did document
+          didDocument.context = Array.from(contextSet.add(DID_V1_CONTEXT_URL))
 
           didDocument.verificationMethod?.concat(
             JsonTransformer.fromJSON(
@@ -300,7 +332,7 @@ export class CheqdDidRegistrar implements DidRegistrar {
         didState: {
           state: 'finished',
           did: didDocument.id,
-          didDocument: JsonTransformer.fromJSON(didDocument, DidDocument),
+          didDocument: JsonTransformer.fromJSON(didRecord.didDocument, DidDocument),
           secret: options.secret,
         },
       }

packages/cheqd/tests/cheqd-did-registrar.e2e.test.ts

@@ -2,7 +2,6 @@ import type { CheqdDidCreateOptions } from '../src'
 import type { DidDocument } from '@credo-ts/core'
 
 import {
-  SECURITY_JWS_CONTEXT_URL,
   DidDocumentBuilder,
   getEd25519VerificationKey2018,
   getJsonWebKey2020,
@@ -127,7 +126,7 @@ describe('Cheqd DID registrar', () => {
         didDocument,
       },
     })
-
+    expect(updateResult.didState.didDocument?.toJSON()).toMatchObject(didDocument.toJSON())
     const deactivateResult = await agent.dids.deactivate({ did })
     expect(deactivateResult.didState.didDocument?.toJSON()).toMatchObject(didDocument.toJSON())
     expect(deactivateResult.didState.state).toEqual('finished')
@@ -148,7 +147,6 @@ describe('Cheqd DID registrar', () => {
     const createResult = await agent.dids.create<CheqdDidCreateOptions>({
       method: 'cheqd',
       didDocument: new DidDocumentBuilder(did)
-        .addContext(SECURITY_JWS_CONTEXT_URL)
         .addController(did)
         .addAuthentication(`${did}#${ed25519Key.fingerprint}`)
         .addVerificationMethod(
@@ -168,7 +166,11 @@ describe('Cheqd DID registrar', () => {
     })
 
     expect(createResult.didState.didDocument?.toJSON()).toMatchObject({
-      '@context': ['https://w3id.org/did/v1', 'https://w3id.org/security/suites/jws-2020/v1'],
+      '@context': [
+        'https://w3id.org/did/v1',
+        'https://w3id.org/security/suites/ed25519-2018/v1',
+        'https://www.w3.org/ns/did/v1',
+      ],
       verificationMethod: [
         {
           controller: did,
@@ -189,7 +191,6 @@ describe('Cheqd DID registrar', () => {
     const createResult = await agent.dids.create<CheqdDidCreateOptions>({
       method: 'cheqd',
       didDocument: new DidDocumentBuilder(did)
-        .addContext(SECURITY_JWS_CONTEXT_URL)
         .addController(did)
         .addAuthentication(`${did}#${p256Key.fingerprint}`)
         .addVerificationMethod(

packages/cheqd/tests/cheqd-did-resolver.e2e.test.ts

@@ -82,7 +82,11 @@ describe('Cheqd DID resolver', () => {
     })
     expect(JsonTransformer.toJSON(resolveResult)).toMatchObject({
       didDocument: {
-        '@context': ['https://www.w3.org/ns/did/v1', 'https://w3id.org/security/suites/ed25519-2020/v1'],
+        '@context': [
+          'https://w3id.org/did/v1',
+          'https://w3id.org/security/suites/ed25519-2020/v1',
+          'https://www.w3.org/ns/did/v1',
+        ],
         id: did,
         controller: [did],
         verificationMethod: [