Extend from bytearray class at a few more places

Signed-off-by: lovesh <lovesh.bond@gmail.com>

Author: lovesh

src/accumulator/accumulatorWitness.ts

@@ -13,6 +13,7 @@ import {
 } from '@docknetwork/crypto-wasm';
 import { getUint8ArraysFromObject, jsonObjToUint8Array } from '../util';
 import { AccumulatorSecretKey } from './params-and-keys';
+import { BytearrayWrapper } from '../bytearray-wrapper';
 
 export abstract class AccumulatorWitness {
   value: Uint8Array | object;
@@ -262,13 +263,7 @@ export class NonMembershipWitness extends AccumulatorWitness {
 /**
  * Public info published by the accumulator manager used to update witnesses after several additions and removals.
  */
-export class WitnessUpdatePublicInfo {
-  value: Uint8Array;
-
-  constructor(info: Uint8Array) {
-    this.value = info;
-  }
-
+export class WitnessUpdatePublicInfo extends BytearrayWrapper {
   toJSON(): string {
     return JSON.stringify({
       value: this.value

src/accumulator/proof.ts

@@ -19,14 +19,9 @@ import {
   MembershipProvingKey,
   NonMembershipProvingKey
 } from './params-and-keys';
+import { BytearrayWrapper } from '../bytearray-wrapper';
 
-export class MembershipProofProtocol {
-  value: Uint8Array;
-
-  constructor(protocol: Uint8Array) {
-    this.value = protocol;
-  }
-
+export class MembershipProofProtocol extends BytearrayWrapper {
   static initialize(
     member: Uint8Array,
     witness: MembershipWitness,
@@ -68,13 +63,7 @@ export class MembershipProofProtocol {
   }
 }
 
-export class NonMembershipProofProtocol {
-  value: Uint8Array;
-
-  constructor(protocol: Uint8Array) {
-    this.value = protocol;
-  }
-
+export class NonMembershipProofProtocol extends BytearrayWrapper {
   static initialize(
     nonMember: Uint8Array,
     witness: NonMembershipWitness,
@@ -116,13 +105,7 @@ export class NonMembershipProofProtocol {
   }
 }
 
-export class MembershipProof {
-  value: Uint8Array;
-
-  constructor(proof: Uint8Array) {
-    this.value = proof;
-  }
-
+export class MembershipProof extends BytearrayWrapper {
   verify(
     accumulated: Uint8Array,
     challenge: Uint8Array,
@@ -156,13 +139,7 @@ export class MembershipProof {
   }
 }
 
-export class NonMembershipProof {
-  value: Uint8Array;
-
-  constructor(proof: Uint8Array) {
-    this.value = proof;
-  }
-
+export class NonMembershipProof extends BytearrayWrapper {
   verify(
     accumulated: Uint8Array,
     challenge: Uint8Array,

src/bbs-plus/proof.ts

@@ -12,6 +12,7 @@ import {
   bbsVerifyProofOfKnowledgeOfSignature
 } from '@docknetwork/crypto-wasm';
 import { BBSPlusPublicKeyG2 } from './keys';
+import { BytearrayWrapper } from '../bytearray-wrapper';
 
 export class PoKSigProtocol {
   value: BbsPoKSigProtocol;
@@ -62,13 +63,7 @@ export class PoKSigProtocol {
   }
 }
 
-export class PoKSigProof {
-  value: Uint8Array;
-
-  constructor(proof: Uint8Array) {
-    this.value = proof;
-  }
-
+export class PoKSigProof extends BytearrayWrapper {
   verify(
     challenge: Uint8Array,
     publicKey: BBSPlusPublicKeyG2,

src/bbs-plus/signature.ts

@@ -10,14 +10,9 @@ import {
 } from '@docknetwork/crypto-wasm';
 import { BBSPlusPublicKeyG2, BBSPlusSecretKey } from './keys';
 import { ensurePositiveIntegerOfSize } from '../util';
+import { BytearrayWrapper } from '../bytearray-wrapper';
 
-export abstract class Signature {
-  value: Uint8Array;
-
-  constructor(value: Uint8Array) {
-    this.value = value;
-  }
-
+export abstract class Signature extends BytearrayWrapper {
   /**
    * This is an irreversible encoding as a hash function is used to convert a message of
    * arbitrary length to a fixed length encoding.
@@ -134,13 +129,7 @@ export class SignatureG1 extends Signature {
   }
 }
 
-export abstract class BlindSignature {
-  value: Uint8Array;
-
-  constructor(value: Uint8Array) {
-    this.value = value;
-  }
-
+export abstract class BlindSignature extends BytearrayWrapper {
   /**
    * Generate blinding for creating the commitment used in the request for blind signature
    * @param seed - Optional seed to serve as entropy for the blinding.

src/composite-proof/proof.ts

@@ -11,17 +11,12 @@ import { Witnesses } from './witness';
 import { SetupParam } from './setup-param';
 import { SaverCiphertext } from '../saver';
 import { ProofSpecG1, QuasiProofSpecG1 } from './proof-spec';
+import { BytearrayWrapper } from '../bytearray-wrapper';
 
 /**
  * A proof of 1 or more statements and meta statements.
  */
-export class CompositeProofG1 {
-  value: Uint8Array;
-
-  constructor(proof: Uint8Array) {
-    this.value = proof;
-  }
-
+export class CompositeProofG1 extends BytearrayWrapper {
   /**
    * Generate the composite proof using a `ProofSpec`
    * @param proofSpec

src/composite-proof/setup-param.ts

@@ -34,19 +34,14 @@ import {
   LegoVerifyingKeyUncompressed
 } from '../legosnark';
 import { AccumulatorParams, AccumulatorPublicKey, MembershipProvingKey, NonMembershipProvingKey } from '../accumulator';
+import { BytearrayWrapper } from '../bytearray-wrapper';
 
 /**
  * Represents (public) setup parameters of different protocols. Different setup parameters can be wrapped in this and
  * then a reference to this is passed to the `Statement`. This is helpful when the same setup parameter needs
  * to be passed to several `Statement`s as it avoids the need of having several copies of the setup parameter.
  */
-export class SetupParam {
-  readonly value: Uint8Array;
-
-  constructor(param: Uint8Array) {
-    this.value = param;
-  }
-
+export class SetupParam extends BytearrayWrapper {
   static bbsSignatureParamsG1(params: SignatureParamsG1): SetupParam {
     return new SetupParam(generateSetupParamForBBSSignatureParametersG1(params.value));
   }