Import from images #186

Levminer · Feb 23, 2022 · ccc5a8c · ccc5a8c
1 parent d0c69c9
commit ccc5a8c
Show file tree

Hide file tree

Showing 2 changed files with 87 additions and 1 deletion.
diff --git a/app/import/src/js/images.js b/app/import/src/js/images.js
@@ -0,0 +1,86 @@
+module.exports = {
+	/**
+	 * Read QR code(s) from image(s)
+	 * @param {string[]} images
+	 */
+	chooseImages: async () => {
+		let string = ""
+
+		const open_dialog = await dialog.showOpenDialog(currentWindow, {
+			title: lang.import_dialog.choose_images,
+			properties: ["openFile", "multiSelections"],
+			filters: [{ name: lang.import_dialog.image_file, extensions: ["jpg", "jpeg", "png", "bmp"] }],
+		})
+
+		const images = open_dialog.filePaths
+
+		for (let i = 0; i < images.length; i++) {
+			const reader = new QrcodeDecoder()
+
+			const res = await reader.decodeFromImage(images[i])
+
+			if (res === false) {
+				dialog.showMessageBox(currentWindow, {
+					title: "Authme",
+					buttons: [lang.button.close],
+					type: "error",
+					noLink: true,
+					message: `${lang.import_dialog.no_qrcode_found_0} ${images[i]}. ${lang.import_dialog.no_qrcode_found_1}`,
+				})
+
+				return logger.warn("No QR code found (QR)")
+			} else if (res.data.startsWith("otpauth://totp/") || res.data.startsWith("otpauth-migration://")) {
+				if (res.data.startsWith("otpauth://totp/")) {
+					string += qrConvert(res.data)
+				} else {
+					string += gaConvert(res.data)
+				}
+
+				if (images.length === i + 1) {
+					const save_exists = fs.existsSync(path.join(folder_path, "codes", "codes.authme"))
+
+					if (save_exists === true) {
+						await dialog.showMessageBox(currentWindow, {
+							title: "Authme",
+							buttons: [lang.button.close],
+							type: "info",
+							noLink: true,
+							defaultId: 0,
+							message: `${lang.import_dialog.correct_qrcode_found_0} ${lang.import_dialog.correct_qrcode_found_1}`,
+						})
+
+						saveFile(string)
+					} else {
+						const result = await dialog.showMessageBox(currentWindow, {
+							title: "Authme",
+							buttons: [lang.button.yes, lang.button.no],
+							type: "info",
+							noLink: true,
+							defaultId: 1,
+							cancelId: 1,
+							message: `${lang.import_dialog.correct_qrcode_found_2} ${lang.import_dialog.correct_qrcode_found_3}`,
+						})
+
+						if (result.response === 1) {
+							ipc.invoke("importedCodes", Buffer.from(string).toString("base64"))
+						} else {
+							ipc.invoke("importedCodes", Buffer.from(string).toString("base64"))
+
+							saveFile(string)
+						}
+					}
+				}
+			} else {
+				dialog.showMessageBox(currentWindow, {
+					title: "Authme",
+					buttons: [lang.button.close],
+					type: "error",
+					noLink: true,
+					message: `${lang.import_dialog.wrong_qrcode_found_0} ${images[i]}. ${lang.import_dialog.wrong_qrcode_found_1}`,
+				})
+
+				return logger.error("Wrong QR code found (QR)")
+			}
+		}
+	},
+}