From 6920686981e8f57dd1324250d11a9d41f222ee88 Mon Sep 17 00:00:00 2001
From: Zhe Li <zlhe@microsoft.com>
Date: Fri, 7 Aug 2020 11:34:14 +0800
Subject: [PATCH] add qr image import support

---
 _locales/en/messages.json          |  12 +++
 sass/import.scss                   |   2 +-
 src/components/Import.vue          |   9 ++
 src/components/Import/QrImport.vue | 154 +++++++++++++++++++++++++++++
 4 files changed, 176 insertions(+), 1 deletion(-)
 create mode 100644 src/components/Import/QrImport.vue

diff --git a/_locales/en/messages.json b/_locales/en/messages.json
index 31089c61a..6c023f80a 100644
--- a/_locales/en/messages.json
+++ b/_locales/en/messages.json
@@ -209,10 +209,22 @@
         "message": "Import Backup File",
         "description": "Import backup file."
     },
+    "import_backup_qr": {
+        "message": "Import QR Image Backup",
+        "description": "Import qr image backup."
+    },
     "import_backup_code": {
         "message": "Import Text Backup",
         "description": "Import backup code."
     },
+    "import_backup_qr_partly_failed": {
+        "message": "Import successfuly, but some QR image cannot be recognized.",
+        "description": "Import successfuly, but some QR image cannot be recognized."
+    },
+    "import_backup_qr_in_batches": {
+        "message": "You can select multiple files to import backup in batches.",
+        "description": "You can select multiple image files to import backup in batches."
+    },
     "show_all_entries": {
         "message": "Show all entries",
         "description": "Show all entries."
diff --git a/sass/import.scss b/sass/import.scss
index b2f415e96..6e9db175a 100644
--- a/sass/import.scss
+++ b/sass/import.scss
@@ -11,7 +11,7 @@ p {
 }
 
 #import {
-  width: 600px;
+  width: 900px;
   position: relative;
   margin: 0 auto;
 }
diff --git a/src/components/Import.vue b/src/components/Import.vue
index 2a9eb3bcb..deceeb6c1 100644
--- a/src/components/Import.vue
+++ b/src/components/Import.vue
@@ -9,6 +9,13 @@
           v-model="importType"
         />
         <label for="import_file_radio">{{ i18n.import_backup_file }}</label>
+        <input
+          type="radio"
+          id="import_qr_radio"
+          value="QrImport"
+          v-model="importType"
+        />
+        <label for="import_qr_radio">{{ i18n.import_backup_qr }}</label>
         <input
           type="radio"
           id="import_code_radio"
@@ -37,6 +44,7 @@
 <script lang="ts">
 import Vue from "vue";
 import FileImport from "./Import/FileImport.vue";
+import QrImport from "./Import/QrImport.vue";
 import TextImport from "./Import/TextImport.vue";
 
 export default Vue.extend({
@@ -48,6 +56,7 @@ export default Vue.extend({
   },
   components: {
     FileImport,
+    QrImport,
     TextImport
   },
   mounted() {
diff --git a/src/components/Import/QrImport.vue b/src/components/Import/QrImport.vue
new file mode 100644
index 000000000..fbbb6efb8
--- /dev/null
+++ b/src/components/Import/QrImport.vue
@@ -0,0 +1,154 @@
+<template>
+  <div>
+    <p style="margin: 10px 20px 20px 20px;">
+      {{ i18n.import_backup_qr_in_batches }}
+    </p>
+    <div class="import_file">
+      <label for="import_qr">{{ i18n.import_backup_qr }}</label>
+      <input
+        id="import_qr"
+        type="file"
+        v-on:change="importQr($event, true)"
+        accept="image/*"
+        multiple
+      />
+    </div>
+  </div>
+</template>
+<script lang="ts">
+import Vue from "vue";
+// @ts-ignore
+import QRCode from "qrcode-reader";
+import jsQR from "jsqr";
+import { getEntryDataFromOTPAuthPerLine } from "../../import";
+import { EntryStorage } from "../../models/storage";
+import { Encryption } from "../../models/encryption";
+
+export default Vue.extend({
+  methods: {
+    async importQr(event: Event, closeWindow: Boolean) {
+      const target = event.target as HTMLInputElement;
+      if (!target || !target.files) {
+        return;
+      }
+      if (target.files.length) {
+        const otpUrlList: string[] = [];
+        let hasFailedResults = false;
+        for (let fileIndex = 0; fileIndex < target.files.length; fileIndex++) {
+          const file = target.files[fileIndex];
+          const otpUrl = await getOtpUrlFromQrFile(file);
+          if (otpUrl !== null) {
+            otpUrlList.push(otpUrl);
+          } else {
+            hasFailedResults = true;
+          }
+        }
+
+        let importData: {
+          // @ts-ignore
+          key?: { enc: string; hash: string };
+          [hash: string]: OTPStorage;
+        } = await getEntryDataFromOTPAuthPerLine(otpUrlList.join("\n"));
+
+        let decryptedFileData: { [hash: string]: OTPStorage } = importData;
+
+        if (Object.keys(decryptedFileData).length) {
+          await EntryStorage.import(
+            this.$encryption as Encryption,
+            decryptedFileData
+          );
+
+          if (hasFailedResults) {
+            alert(this.i18n.import_backup_qr_partly_failed);
+          } else {
+            alert(this.i18n.updateSuccess);
+          }
+
+          if (closeWindow) {
+            window.close();
+          }
+        } else {
+          alert(this.i18n.errorqr);
+          if (closeWindow) {
+            window.close();
+          }
+        }
+      } else {
+        alert(this.i18n.updateFailure);
+        if (closeWindow) {
+          window.alert(this.i18n.updateFailure);
+          window.close();
+        }
+      }
+      return;
+    }
+  }
+});
+
+async function getOtpUrlFromQrFile(file: File): Promise<string | null> {
+  return new Promise(resolve => {
+    const reader = new FileReader();
+    reader.onload = () => {
+      const imageUrl = reader.result as string;
+      const qrReader = new QRCode();
+      qrReader.callback = (
+        error: string,
+        text: {
+          result: string;
+          points: Array<{
+            x: number;
+            y: number;
+            count: number;
+            estimatedModuleSize: number;
+          }>;
+        }
+      ) => {
+        if (error) {
+          console.error(error);
+
+          const image: HTMLImageElement = document.createElement("img");
+          image.onload = () => {
+            const canvas: HTMLCanvasElement = document.createElement("canvas");
+            const ctx: CanvasRenderingContext2D = canvas.getContext(
+              "2d"
+            ) as CanvasRenderingContext2D;
+
+            canvas.width = image.width;
+            canvas.height = image.height;
+            ctx.drawImage(image, 0, 0);
+
+            const qrImageData = ctx.getImageData(
+              0,
+              0,
+              canvas.width,
+              canvas.height
+            );
+            const jsQrCode = jsQR(
+              qrImageData.data,
+              canvas.width,
+              canvas.height
+            );
+
+            if (jsQrCode && jsQrCode.data) {
+              if (jsQrCode.data.indexOf("otpauth://") !== 0) {
+                return resolve(null);
+              }
+              return resolve(jsQrCode.data);
+            } else {
+              return resolve(null);
+            }
+          };
+          image.src = imageUrl;
+        } else {
+          if (text.result.indexOf("otpauth://") !== 0) {
+            return resolve(null);
+          }
+          return resolve(text.result);
+        }
+      };
+      qrReader.decode(imageUrl);
+    };
+    reader.readAsDataURL(file);
+  });
+}
+</script>