add encryption func

Author: CTyrannosaurus

lib/utils/zip-stream.js

@@ -1,5 +1,6 @@
 const events = require('events');
 const JSZip = require('jszip');
+const officeCrypto = require('officecrypto-tool');
 
 const StreamBuf = require('./stream-buf');
 const {stringToBuffer} = require('./browser-buffer-encode');
@@ -35,8 +36,12 @@ class ZipWriter extends events.EventEmitter {
     }
   }
 
-  async finalize() {
-    const content = await this.zip.generateAsync(this.options);
+  async finalize(options) {
+    const totalOptions = {...this.options, ...options};
+    let content = await this.zip.generateAsync(totalOptions);
+    if (totalOptions.password) {
+      content = officeCrypto.encrypt(content, {password: totalOptions.password});
+    }
     this.stream.end(content);
     this.emit('finish');
   }

lib/xlsx/xlsx.js

@@ -588,13 +588,13 @@ class XLSX {
     });
   }
 
-  _finalize(zip) {
+  _finalize(zip, options) {
     return new Promise((resolve, reject) => {
       zip.on('finish', () => {
         resolve(this);
       });
       zip.on('error', reject);
-      zip.finalize();
+      zip.finalize(options);
     });
   }
 
@@ -666,7 +666,7 @@ class XLSX {
     await this.addMedia(zip, model);
     await Promise.all([this.addApp(zip, model), this.addCore(zip, model)]);
     await this.addWorkbook(zip, model);
-    return this._finalize(zip);
+    return this._finalize(zip, options);
   }
 
   writeFile(filename, options) {

spec/end-to-end/express-encryption.spec.js

@@ -0,0 +1,42 @@
+const {PassThrough} = require('readable-stream');
+const express = require('express');
+const got = require('got');
+const testutils = require('../utils/index');
+
+const Excel = verquire('exceljs');
+
+const password = '123456';
+
+describe('Express test workbook.xlsx.write encryption', () => {
+  let server;
+  before(() => {
+    const app = express();
+    app.get('/workbook', (req, res) => {
+      const wb = testutils.createTestBook(new Excel.Workbook(), 'xlsx');
+      res.setHeader(
+        'Content-Type',
+        'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
+      );
+      res.setHeader('Content-Disposition', 'attachment; filename=Report.xlsx');
+      wb.xlsx.write(res, {password}).then(() => {
+        res.end();
+      });
+    });
+    server = app.listen(3003);
+  });
+
+  after(() => {
+    server.close();
+  });
+
+  it('downloads a workbook, workbook.xlsx.write encryption successful', async function() {
+    this.timeout(5000);
+    const res = got.stream('http://127.0.0.1:3003/workbook', {
+      decompress: false,
+    });
+    const wb2 = new Excel.Workbook();
+    // TODO: Remove passThrough with got 10+ (requires node v10+)
+    await wb2.xlsx.read(res.pipe(new PassThrough()), {password});
+    testutils.checkTestBook(wb2, 'xlsx');
+  });
+});

spec/integration/issues/new/new-issue-2-add-encrypt-func.spec.js

@@ -0,0 +1,35 @@
+const ExcelJS = verquire('exceljs');
+
+const test = './spec/integration/data/new-issue-2-test-encryption.xlsx';
+
+describe('github issues', () => {
+  describe('github issues encrypted xlsx ', () => {
+    it('workbook.xlsx.writeFile, ecma376_agile encryption successful', async () => {
+      const password = '123456';
+      const writeWorkbook = new ExcelJS.Workbook();
+      writeWorkbook.addWorksheet('Sheet1');
+      await writeWorkbook.xlsx.writeFile(test, {password});
+
+      const workbook = new ExcelJS.Workbook();
+      await workbook.xlsx.readFile(test, {
+        password,
+      });
+      const sheetName = workbook.getWorksheet(1).name;
+      expect(sheetName).to.equal('Sheet1');
+    }).timeout(10000);
+
+    it('workbook.xlsx.writeBuffer, ecma376_agile encryption successful', async () => {
+      const password = '123456';
+      const writeWorkbook = new ExcelJS.Workbook();
+      writeWorkbook.addWorksheet('Sheet1');
+      const encryptBuffer = await writeWorkbook.xlsx.writeBuffer({password});
+
+      const workbook = new ExcelJS.Workbook();
+      await workbook.xlsx.load(encryptBuffer, {
+        password,
+      });
+      const sheetName = workbook.getWorksheet(1).name;
+      expect(sheetName).to.equal('Sheet1');
+    }).timeout(10000);
+  });
+});