More tweaks

docknetwork · Feb 11, 2025 · c6255b0 · c6255b0
1 parent a954234
commit c6255b0
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 4 deletions.
diff --git a/packages/credential-sdk/src/types/accumulator/keys.js b/packages/credential-sdk/src/types/accumulator/keys.js
@@ -40,6 +40,21 @@ export class CheqdMainnetAccumulatorPublicKeyRef extends CheqdAccumulatorPublicK
 }
 
 export class DockOrCheqdAccumulatorPublicKeyRef extends TypedTuple {
+  constructor(...args) {
+    super(...args);
+
+    if (this[0].isCheqd) {
+      return CheqdAccumulatorPublicKeyRef.from(this);
+    } else if (
+      this[0].isDock
+      && this[1] instanceof DockAccumulatorPublicKeyId
+    ) {
+      return DockAccumulatorPublicKeyRef.from(this);
+    } else {
+      throw new Error(`Must be dock or cheqd, got: \`${this[0]}\``);
+    }
+  }
+
   static Classes = [
     NamespaceDid,
     anyOf(DockAccumulatorPublicKeyId, CheqdAccumulatorPublicKeyId),
@@ -59,6 +74,16 @@ export class CheqdMainnetAccumulatorParamsRef extends CheqdAccumulatorParamsRef
 }
 
 export class DockOrCheqdAccumulatorParamsRef extends TypedTuple {
+  constructor(...args) {
+    super(...args);
+
+    if (this[0].isCheqd) {
+      return CheqdAccumulatorParamsRef.from(this);
+    } else if (this[0].isDock && this[1] instanceof DockAccumulatorParamsId) {
+      return DockAccumulatorParamsRef.from(this);
+    }
+  }
+
   static Classes = [
     NamespaceDid,
     anyOf(DockAccumulatorParamsId, CheqdAccumulatorParamsId),

diff --git a/packages/credential-sdk/src/types/offchain-signatures/public-keys/ref.js b/packages/credential-sdk/src/types/offchain-signatures/public-keys/ref.js
@@ -42,5 +42,15 @@ export class CheqdMainnetOffchainSignatureParamsRef extends CheqdDLRRef {
 }
 
 export class DockOrCheqdOffchainSignatureParamsRef extends TypedTuple {
+  constructor(...args) {
+    super(...args);
+
+    if (this[0].isCheqd) {
+      return CheqdOffchainSignatureParamsRef.from(this);
+    } else if (this[0].isDock && this[1] instanceof DockParamsId) {
+      return DockOffchainSignatureParamsRef.from(this);
+    }
+  }
+
   static Classes = [NamespaceDid, anyOf(DockParamsId, CheqdParamsId)];
 }
diff --git a/packages/credential-sdk/tests/__snapshots__/document.test.js.snap b/packages/credential-sdk/tests/__snapshots__/document.test.js.snap
@@ -665,7 +665,9 @@ exports[`DID document workflow \`DIDDocument\` 3`] = `
       "metadata": {
         "curveType": "Bls12381",
         "paramsRef": [
-          "did:dock:5GdBBJdo3EWbbVpehRMUkGqfyEm7TJ7ugGCoPKNXzqrgPQqT",
+          {
+            "did": "0xc9b4c22c2f5738b0adaf605a8233cdc2451c5ece045c25a3bc2b4c840fe9b11d",
+          },
           0,
         ],
         "participantId": null,
@@ -691,7 +693,9 @@ exports[`DID document workflow \`DIDDocument\` 3`] = `
       "metadata": {
         "curveType": "Bls12381",
         "paramsRef": [
-          "did:dock:5GdBBJdo3EWbbVpehRMUkGqfyEm7TJ7ugGCoPKNXzqrgPQqT",
+          {
+            "did": "0xc9b4c22c2f5738b0adaf605a8233cdc2451c5ece045c25a3bc2b4c840fe9b11d",
+          },
           1,
         ],
         "participantId": null,
@@ -735,7 +739,9 @@ exports[`DID document workflow \`DIDDocument\` 3`] = `
       "metadata": {
         "curveType": "Bls12381",
         "paramsRef": [
-          "did:dock:5GdBBJdo3EWbbVpehRMUkGqfyEm7TJ7ugGCoPKNXzqrgPQqT",
+          {
+            "did": "0xc9b4c22c2f5738b0adaf605a8233cdc2451c5ece045c25a3bc2b4c840fe9b11d",
+          },
           2,
         ],
         "participantId": null,
@@ -749,7 +755,9 @@ exports[`DID document workflow \`DIDDocument\` 3`] = `
       "metadata": {
         "curveType": "Bls12381",
         "paramsRef": [
-          "did:dock:5GdBBJdo3EWbbVpehRMUkGqfyEm7TJ7ugGCoPKNXzqrgPQqT",
+          {
+            "did": "0xc9b4c22c2f5738b0adaf605a8233cdc2451c5ece045c25a3bc2b4c840fe9b11d",
+          },
           3,
         ],
         "participantId": null,