Check for presense of key rather than an encrypted entry.

fixes #705

Author: mymindstorm

src/background.ts

@@ -269,7 +269,8 @@ async function getTotp(text: string, silent = false) {
           entryData[hash].algorithm = algorithm;
         }
         if (
-          (await EntryStorage.hasEncryptedEntry()) !==
+          // If the entries are encrypted and we aren't unlocked, error.
+          (await EntryStorage.hasEncryptionKey()) !==
           encryption.getEncryptionStatus()
         ) {
           !silent && chrome.tabs.sendMessage(id, { action: "errorenc" });

src/models/storage.ts

@@ -260,21 +260,8 @@ export class EntryStorage {
     }
   }
 
-  static hasEncryptedEntry() {
-    return new Promise((resolve: (value: boolean) => void) => {
-      BrowserStorage.get((_data: { [hash: string]: OTPStorage }) => {
-        for (const hash of Object.keys(_data)) {
-          if (!this.isValidEntry(_data, hash)) {
-            continue;
-          }
-          if (_data[hash].encrypted) {
-            return resolve(true);
-          }
-        }
-        return resolve(false);
-      });
-      return;
-    });
+  static async hasEncryptionKey(): Promise<boolean> {
+    return Boolean(await BrowserStorage.getKey());
   }
 
   static getExport(data: OTPEntryInterface[], encrypted?: boolean) {

src/store/Accounts.ts

@@ -1,25 +1,18 @@
 import { EntryStorage, BrowserStorage } from "../models/storage";
 import { Encryption } from "../models/encryption";
 import * as CryptoJS from "crypto-js";
-import { OTPType, OTPAlgorithm, CodeState } from "../models/otp";
+import { OTPType, OTPAlgorithm } from "../models/otp";
 import { ActionContext } from "vuex";
 
 export class Accounts implements Module {
   async getModule() {
     const cachedPassphrase = await this.getCachedPassphrase();
     const encryption: Encryption = new Encryption(cachedPassphrase);
-    let shouldShowPassphrase = cachedPassphrase
+    const shouldShowPassphrase = cachedPassphrase
       ? false
-      : await EntryStorage.hasEncryptedEntry();
+      : await EntryStorage.hasEncryptionKey();
     const entries = shouldShowPassphrase ? [] : await this.getEntries();
 
-    for (let i = 0; i < entries.length; i++) {
-      if (entries[i].code === CodeState.Encrypted) {
-        shouldShowPassphrase = true;
-        break;
-      }
-    }
-
     return {
       state: {
         entries,

src/store/Advisor.ts

@@ -7,7 +7,7 @@ const insightsData: AdvisorInsightInterface[] = [
     level: InsightLevel.danger,
     description: chrome.i18n.getMessage("advisor_insight_password_not_set"),
     validation: async () => {
-      const hasEncryptedEntry = await EntryStorage.hasEncryptedEntry();
+      const hasEncryptedEntry = await EntryStorage.hasEncryptionKey();
       return !hasEncryptedEntry;
     },
   },
@@ -16,7 +16,7 @@ const insightsData: AdvisorInsightInterface[] = [
     level: InsightLevel.warning,
     description: chrome.i18n.getMessage("advisor_insight_auto_lock_not_set"),
     validation: async () => {
-      const hasEncryptedEntry = await EntryStorage.hasEncryptedEntry();
+      const hasEncryptedEntry = await EntryStorage.hasEncryptionKey();
       return hasEncryptedEntry && !Number(localStorage.autolock);
     },
   },