✨ Add AES encrypt for ebook files

Author: williamchong

components/IscnUploadForm.vue

@@ -105,6 +105,9 @@
             </table>
           </div>
         </div>
+        <FormField v-if="hasEbookInRecords">
+          <CheckBox v-model="isEncryptEBookData">{{ $t('UploadForm.label.encryptEBookData') }}</CheckBox>
+        </FormField>
         <FormField v-if="showAddISCNPageOption">
           <CheckBox v-model="isAddISCNPageToEbook">{{ $t('UploadForm.label.insertISCNPage') }}</CheckBox>
         </FormField>
@@ -244,6 +247,7 @@ import exifr from 'exifr'
 import Hash from 'ipfs-only-hash'
 import BigNumber from 'bignumber.js'
 import ePub from 'epubjs';
+import { encryptDataWithAES } from 'arweavekit/dist/lib/encryption';
 
 import { OfflineSigner } from '@cosmjs/proto-signing'
 
@@ -310,6 +314,7 @@ export default class IscnUploadForm extends Vue {
   isOpenSignDialog = false
   isOpenWarningSnackbar = false
   isSizeExceeded = false
+  isEncryptEBookData = true
   isAddISCNPageToEbook = false
 
   uploadSizeLimit: number = UPLOAD_FILESIZE_MAX
@@ -402,14 +407,18 @@ export default class IscnUploadForm extends Vue {
     }
   }
 
-  get showAddISCNPageOption() {
-    return this.mode === MODE.EDIT
-      && this.fileRecords.some(file => [
+  get hasEbookInRecords() {
+    return this.fileRecords.some(file => [
         'application/epub+zip',
         'application/pdf',
       ].includes(file.fileType))
   }
 
+  get showAddISCNPageOption() {
+    return this.mode === MODE.EDIT
+      && this.hasEbookInRecords
+  }
+
   get modifiedFileRecords() {
     if (!this.isAddISCNPageToEbook || this.mode !== MODE.EDIT) return this.fileRecords
     return this.fileRecords.map((record) => {
@@ -844,15 +853,29 @@ export default class IscnUploadForm extends Vue {
       }
     }
 
+    let key;
     try {
       const arrayBuffer = await tempRecord.fileBlob.arrayBuffer();
-      const buffer = Buffer.from(arrayBuffer);
+      let buffer = Buffer.from(arrayBuffer);
+      if ([
+        'application/epub+zip',
+        'application/pdf',
+      ].includes(record.fileType)
+        && this.isEncryptEBookData) {
+        const {
+          rawEncryptedKeyAsBase64,
+          combinedArrayBuffer,
+        } = await encryptDataWithAES({ data: arrayBuffer });
+        buffer = Buffer.from(combinedArrayBuffer);
+        key = rawEncryptedKeyAsBase64
+      }
       const { arweaveId, arweaveLink } = await uploadSingleFileToBundlr(buffer, {
         fileSize: tempRecord.fileBlob?.size || 0,
         ipfsHash: tempRecord.ipfsHash,
         fileType: tempRecord.fileType as string,
         txHash: tempRecord.transactionHash,
         token: this.getToken,
+        key,
       });
       if (arweaveId) {
         const uploadedData = this.sentArweaveTransactionInfo.get(record.ipfsHash) || {};

locales/en.json

@@ -385,6 +385,7 @@
   "UploadForm.button.size.limit": "(Recommended file size: < 20 MB)",
   "UploadForm.button.skip": "Skip Upload",
   "UploadForm.button": "Start Upload",
+  "UploadForm.label.encryptEBookData": "Encrypt EPUB & PDF on upload",
   "UploadForm.label.insertISCNPage": "Add ISCN Page to EPUB & PDF",
   "UploadForm.guide.dropFile": "Drop your file here, or",
   "UploadForm.guide.selectFile": "Select your file and publish to IPFS",

package.json

@@ -34,6 +34,7 @@
     "@walletconnect/client": "^1.6.6",
     "@walletconnect/qrcode-modal": "^1.6.6",
     "@walletconnect/utils": "^1.6.6",
+    "arweavekit": "^1.5.1",
     "axios": "^1.7.4",
     "bignumber.js": "^9.0.1",
     "cheerio": "^1.0.0-rc.11",

utils/arweave/v2.ts

@@ -163,7 +163,8 @@ export async function uploadSingleFileToBundlr(
     ipfsHash,
     txHash,
     token,
-  }: { fileSize: number; fileType?: string, ipfsHash: string; txHash: string, token: string },
+    key,
+  }: { fileSize: number; fileType?: string, ipfsHash: string; txHash: string, token: string, key?: string },
 ) {
   const bundler = await getBundler({ fileSize, ipfsHash, txHash, token })
   const tags = [
@@ -183,6 +184,7 @@ export async function uploadSingleFileToBundlr(
       ipfsHash,
       txHash,
       arweaveId,
+      key,
     },{
       headers: { Authorization: token ? `Bearer ${token}` : '' },
     });