Support two specific multibase-multikey headers for Bls12381G2.

digitalbazaar · Sep 19, 2024 · f923b0f · f923b0f
1 parent 3ab5f17
commit f923b0f
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -30,7 +30,7 @@ const SUPPORTED_KEY_TYPES = new Map([
 ]);
 
 const SUPPORTED_BASE58_MULTIKEY_HEADERS = new Map([
-  ['Bls12381G2', 'zUC'],
+  ['Bls12381G2', ['zUC6', 'zUC7']],
   ['Ed25519', 'z6Mk'],
   ['P-256', 'zDna'],
   ['P-384', 'z82L'],
@@ -54,11 +54,15 @@ bedrock.events.on('bedrock.start', () => {
     if(!handler) {
       throw new Error(`Unsupported did:key "keyType", "${keyType}".`);
     }
-    const multibaseMultikeyHeader =
+    let multibaseMultikeyHeaders =
       SUPPORTED_BASE58_MULTIKEY_HEADERS.get(keyType);
-    didKeyDriver.use({
-      multibaseMultikeyHeader, fromMultibase: handler.from
-    });
+    multibaseMultikeyHeaders = Array.isArray(multibaseMultikeyHeaders) ?
+      multibaseMultikeyHeaders : [multibaseMultikeyHeaders];
+    for(const multibaseMultikeyHeader of multibaseMultikeyHeaders) {
+      didKeyDriver.use({
+        multibaseMultikeyHeader, fromMultibase: handler.from
+      });
+    }
   }
   didIo.use(didKeyDriver);
 
@@ -76,11 +80,15 @@ bedrock.events.on('bedrock.start', () => {
     if(!handler) {
       throw new Error(`Unsupported did:web "keyType", "${keyType}".`);
     }
-    const multibaseMultikeyHeader =
+    let multibaseMultikeyHeaders =
       SUPPORTED_BASE58_MULTIKEY_HEADERS.get(keyType);
-    didWebDriver.use({
-      multibaseMultikeyHeader, fromMultibase: handler.from
-    });
+    multibaseMultikeyHeaders = Array.isArray(multibaseMultikeyHeaders) ?
+      multibaseMultikeyHeaders : [multibaseMultikeyHeaders];
+    for(const multibaseMultikeyHeader of multibaseMultikeyHeaders) {
+      didWebDriver.use({
+        multibaseMultikeyHeader, fromMultibase: handler.from
+      });
+    }
   }
   didIo.use(didWebDriver);