feat: add chunking for decryption

Cypherock · Sep 5, 2024 · 85d97d3 · 85d97d3
1 parent 1fa580b
commit 85d97d3
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 9 deletions.
diff --git a/packages/app-inheritance/src/operations/decryptMessagesWithPin/index.ts b/packages/app-inheritance/src/operations/decryptMessagesWithPin/index.ts
@@ -6,6 +6,10 @@ import {
 } from '@cypherock/sdk-utils';
 import { WALLET_ID_LENGTH } from '../../constants';
 import { APP_VERSION } from '../../constants/appId';
+import {
+  DecryptDataWithPinDecryptedDataStructure,
+  DecryptDataWithPinEncryptedDataStructure,
+} from '../../proto/generated/inheritance/decrypt_data_with_pin';
 import { DecryptDataStatus } from '../../types';
 import {
   assertOrThrowInvalidResult,
@@ -59,18 +63,31 @@ export const decryptMessagesWithPin = async (
   await helper.sendQuery({
     initiate: {
       walletId: params.walletId,
-      encryptedData: params.encryptedData,
     },
   });
 
-  const result = await helper.waitForResult();
+  await helper.waitForResult();
+  logger.verbose('decryptMessages confirmed');
+
+  const rawData = DecryptDataWithPinEncryptedDataStructure.encode(
+    DecryptDataWithPinEncryptedDataStructure.create({
+      data: params.encryptedData,
+    }),
+  ).finish();
+  await helper.sendInChunks(rawData, 'encryptedData', 'dataAccepted');
+
+  const decryptedData = await helper.receiveInChunks(
+    'decryptedDataRequest',
+    'decryptedData',
+  );
+  const result = DecryptDataWithPinDecryptedDataStructure.decode(decryptedData);
   logger.verbose('decryptMessages response', result);
 
-  assertOrThrowInvalidResult(result.messages?.plainData);
+  assertOrThrowInvalidResult(result.decryptedData);
 
   const output = {
-    decryptedData: result.messages.plainData.map(data => data.message),
-    decryptedDataAsStrings: result.messages.plainData.map(data =>
+    decryptedData: result.decryptedData.map(data => data.message),
+    decryptedDataAsStrings: result.decryptedData.map(data =>
       Buffer.from(data.message).toString(),
     ),
   };

diff --git a/packages/app-inheritance/src/operations/encryptMessagesWithPin/index.ts b/packages/app-inheritance/src/operations/encryptMessagesWithPin/index.ts
@@ -65,8 +65,8 @@ export const encryptMessageWithPin = async (
     },
   });
 
-  // Wait for confirmation
   await helper.waitForResult();
+  logger.verbose('encryptMessages confirmed');
 
   const rawData = EncryptDataWithPinPlainDataStructure.encode(
     EncryptDataWithPinPlainDataStructure.create({

diff --git a/packages/app-inheritance/src/utils/operationHelper.ts b/packages/app-inheritance/src/utils/operationHelper.ts
@@ -130,8 +130,6 @@ export class OperationHelper<Q extends QueryKey, R extends ResultKey> {
   >(queryKey: QK, resultKey: RK): Promise<Uint8Array> {
     const chunks: Uint8Array[] = [];
 
-    await this.sendQuery({ [queryKey]: {} });
-
     let index = 0;
 
     while (1) {

diff --git a/submodules/common b/submodules/common
+2 −1		proto/inheritance/auth_wallet.proto
+44 −22		proto/inheritance/decrypt_data_with_pin.proto