Add optional cryptoService param to TDF constructor (#208)

opentdf · Jul 12, 2023 · ca6b664 · ca6b664
1 parent 1462bb4
commit ca6b664
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lib/tdf3/src/tdf.ts b/lib/tdf3/src/tdf.ts
@@ -118,6 +118,10 @@ type Chunk = {
   _resolve?: (value: unknown) => void;
 };
 
+type TDFConfiguration = {
+  cryptoService?: CryptoService;
+};
+
 export class TDF extends EventEmitter {
   policy?: Policy;
   mimeType?: string;
@@ -136,11 +140,11 @@ export class TDF extends EventEmitter {
   chunkMap: Map<string, Chunk>;
   cryptoService: CryptoService<CryptoKeyPair>;
 
-  constructor() {
+  constructor(configuration?: TDFConfiguration) {
     super();
 
     this.attributeSet = new AttributeSet();
-    this.cryptoService = defaultCryptoService;
+    this.cryptoService = configuration?.cryptoService ?? defaultCryptoService;
     this.publicKey = '';
     this.privateKey = '';
     this.integrityAlgorithm = 'HS256';
@@ -149,8 +153,8 @@ export class TDF extends EventEmitter {
   }
 
   // factory
-  static create() {
-    return new TDF();
+  static create(configuration?: TDFConfiguration) {
+    return new TDF(configuration);
   }
 
   createCipher(type: string) {