feat(allowlist): register encrypters in ritual allowlist

nucypher · Oct 4, 2023 · e5ac346 · e5ac346
1 parent d344181
commit e5ac346
Show file tree

Hide file tree

Showing 18 changed files with 2,128 additions and 19 deletions.
diff --git a/packages/pre/src/characters/bob.ts b/packages/pre/src/characters/bob.ts
@@ -5,7 +5,7 @@ import {
   SecretKey,
   Signer,
 } from '@nucypher/nucypher-core';
-import { PorterClient, zip } from '@nucypher/shared';
+import { ChecksumAddress, PorterClient, zip } from '@nucypher/shared';
 
 import { Keyring } from '../keyring';
 import { PolicyMessageKit, RetrievalResult } from '../kits';
@@ -128,7 +128,7 @@ export class Bob {
     return zip(policyMessageKits, retrieveCFragsResponses).map((pair) => {
       const [messageKit, { cFrags, errors }] = pair;
       const vcFrags = Object.keys(cFrags).map((address) => {
-        const verified = cFrags[address].verify(
+        const verified = cFrags[address as ChecksumAddress].verify(
           messageKit.capsule,
           publisherVerifyingKey,
           policyEncryptingKey,

diff --git a/packages/pre/src/cohort.ts b/packages/pre/src/cohort.ts
@@ -14,8 +14,8 @@ export class Cohort {
   public static async create(
     porterUri: string,
     numUrsulas: number,
-    include: string[] = [],
-    exclude: string[] = [],
+    include: ChecksumAddress[] = [],
+    exclude: ChecksumAddress[] = [],
   ) {
     const porter = new PorterClient(porterUri);
     const ursulas = await porter.getUrsulas(numUrsulas, exclude, include);

diff --git a/packages/pre/src/kits/retrieval.ts b/packages/pre/src/kits/retrieval.ts
@@ -12,7 +12,7 @@ export class RetrievalResult {
   }
 
   public get addresses(): readonly ChecksumAddress[] {
-    return Object.keys(this.cFrags);
+    return Object.keys(this.cFrags) as ChecksumAddress[];
   }
 
   public withResult(result: RetrievalResult): RetrievalResult {

diff --git a/packages/pre/src/policy.ts b/packages/pre/src/policy.ts
@@ -7,6 +7,7 @@ import {
   VerifiedKeyFrag,
 } from '@nucypher/nucypher-core';
 import {
+  ChecksumAddress,
   PreSubscriptionManagerAgent,
   toBytes,
   toCanonicalAddress,
@@ -76,7 +77,7 @@ export class PreEnactedPolicy implements IPreEnactedPolicy {
       this.size,
       startTimestamp,
       endTimestamp,
-      ownerAddress,
+      ownerAddress as ChecksumAddress,
     );
     return tx.hash;
   }