standardnotes-app-web/app/assets/javascripts/services/archiveManager.js

import { PrivilegesManager } from '@/services/privilegesManager';

export class ArchiveManager {
  /* @ngInject */
  constructor(passcodeManager, authManager, modelManager, privilegesManager) {
    this.passcodeManager = passcodeManager;
    this.authManager = authManager;
    this.modelManager = modelManager;
    this.privilegesManager = privilegesManager;
  }

  /*
  Public
  */

  async downloadBackup(encrypted) {
    return this.downloadBackupOfItems(this.modelManager.allItems, encrypted);
  }

  async downloadBackupOfItems(items, encrypted) {
    const run = async () => {
      // download in Standard Notes format
      let keys, authParams;
      if(encrypted) {
        if(this.authManager.offline() && this.passcodeManager.hasPasscode()) {
          keys = this.passcodeManager.keys();
          authParams = this.passcodeManager.passcodeAuthParams();
        } else {
          keys = await this.authManager.keys();
          authParams = await this.authManager.getAuthParams();
        }
      }
      this.__itemsData(items, keys, authParams).then((data) => {
        const modifier = encrypted ? "Encrypted" : "Decrypted";
        this.__downloadData(data, `Standard Notes ${modifier} Backup - ${this.__formattedDate()}.txt`);

        // download as zipped plain text files
        if(!keys) {
          this.__downloadZippedItems(items);
        }
      });
    };

    if(await this.privilegesManager.actionRequiresPrivilege(PrivilegesManager.ActionManageBackups)) {
      this.privilegesManager.presentPrivilegesModal(PrivilegesManager.ActionManageBackups, () => {
        run();
      });
    } else {
      run();
    }
  }

  /*
  Private
  */

  __formattedDate() {
    var string = `${new Date()}`;
    // Match up to the first parenthesis, i.e do not include '(Central Standard Time)'
    var matches = string.match(/^(.*?) \(/);
    if(matches.length >= 2) {
      return matches[1];
    }
    return string;
  }

  async __itemsData(items, keys, authParams) {
    const data = await this.modelManager.getJSONDataForItems(items, keys, authParams);
    const blobData = new Blob([data], {type: 'text/json'});
    return blobData;
  }

  __loadZip(callback) {
    if(window.zip) {
      callback();
      return;
    }

    var scriptTag = document.createElement('script');
    scriptTag.src = "/assets/zip/zip.js";
    scriptTag.async = false;
    var headTag = document.getElementsByTagName('head')[0];
    headTag.appendChild(scriptTag);
    scriptTag.onload = function() {
      zip.workerScriptsPath = "assets/zip/";
      callback();
    };
  }

  __downloadZippedItems(items) {
    this.__loadZip(() => {
      zip.createWriter(new zip.BlobWriter("application/zip"), (zipWriter) => {
        var index = 0;

        const nextFile = () => {
          var item = items[index];
          var name, contents;

          if(item.content_type === "Note") {
            name = item.content.title;
            contents = item.content.text;
          } else {
            name = item.content_type;
            contents = JSON.stringify(item.content, null, 2);
          }

          if(!name) {
            name = "";
          }

          const blob = new Blob([contents], {type: 'text/plain'});
          let filePrefix = name.replace(/\//g, "").replace(/\\+/g, "");
          const fileSuffix = `-${item.uuid.split("-")[0]}.txt`;
          // Standard max filename length is 255. Slice the note name down to allow filenameEnd
          filePrefix = filePrefix.slice(0, (255 - fileSuffix.length));
          const fileName = `${item.content_type}/${filePrefix}${fileSuffix}`;
          zipWriter.add(fileName, new zip.BlobReader(blob), () => {
            index++;
            if(index < items.length) {
              nextFile();
            } else {
              zipWriter.close((blob) => {
                this.__downloadData(blob, `Standard Notes Backup - ${this.__formattedDate()}.zip`);
                zipWriter = null;
              });
            }
          });
        };

        nextFile();
      }, onerror);
    });
  }


  __hrefForData(data) {
    // If we are replacing a previously generated file we need to
    // manually revoke the object URL to avoid memory leaks.
    if (this.textFile !== null) {
      window.URL.revokeObjectURL(this.textFile);
    }

    this.textFile = window.URL.createObjectURL(data);

    // returns a URL you can use as a href
    return this.textFile;
  }

  __downloadData(data, fileName) {
    var link = document.createElement('a');
    link.setAttribute('download', fileName);
    link.href = this.__hrefForData(data);
    document.body.appendChild(link);
    link.click();
    link.remove();
  }
}