@polkadot/extension-base 0.44.5 → 0.44.7

package/cjs/background/handlers/State.js

@@ -1,30 +1,21 @@
 "use strict";
 
 var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
-
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.default = exports.NotificationOptions = void 0;
-
 var _rxjs = require("rxjs");
-
 var _getId = require("@polkadot/extension-base/utils/getId");
-
 var _extensionChains = require("@polkadot/extension-chains");
-
 var _defaults = require("@polkadot/networks/defaults");
-
 var _uiSettings = _interopRequireDefault(require("@polkadot/ui-settings"));
-
 var _util = require("@polkadot/util");
-
 var _stores = require("../../stores");
-
 var _helpers = require("./helpers");
-
 // Copyright 2019-2022 @polkadot/extension-bg authors & contributors
 // SPDX-License-Identifier: Apache-2.0
+
 const NOTIFICATION_URL = chrome.extension.getURL('notification.html');
 const POPUP_WINDOW_OPTS = {
   focused: true,
@@ -42,15 +33,13 @@ const NORMAL_WINDOW_OPTS = {
 };
 let NotificationOptions;
 exports.NotificationOptions = NotificationOptions;
-
 (function (NotificationOptions) {
   NotificationOptions[NotificationOptions["None"] = 0] = "None";
   NotificationOptions[NotificationOptions["Normal"] = 1] = "Normal";
   NotificationOptions[NotificationOptions["PopUp"] = 2] = "PopUp";
 })(NotificationOptions || (exports.NotificationOptions = NotificationOptions = {}));
-
 const AUTH_URLS_KEY = 'authUrls';
-
+const DEFAULT_AUTH_ACCOUNTS = 'defaultAuthAccounts';
 function extractMetadata(store) {
   store.allMap(map => {
     const knownEntries = Object.entries(_defaults.knownGenesis);
@@ -62,18 +51,17 @@ function extractMetadata(store) {
         let [, hashes] = _ref2;
         return hashes.includes(def.genesisHash);
       });
-
       if (entry) {
         const [name, hashes] = entry;
-        const index = hashes.indexOf(def.genesisHash); // flatten the known metadata based on the genesis index
-        // (lower is better/newer)
+        const index = hashes.indexOf(def.genesisHash);
 
+        // flatten the known metadata based on the genesis index
+        // (lower is better/newer)
         if (!defs[name] || defs[name].index > index) {
           if (defs[name]) {
             // remove the old version of the metadata
             removals.push(defs[name].key);
           }
-
           defs[name] = {
             def,
             index,
@@ -98,49 +86,52 @@ function extractMetadata(store) {
     });
   });
 }
-
 class State {
   #authUrls = {};
   #authRequests = {};
-  #metaStore = new _stores.MetadataStore(); // Map of providers currently injected in tabs
+  #metaStore = new _stores.MetadataStore();
 
+  // Map of providers currently injected in tabs
   #injectedProviders = new Map();
   #metaRequests = {};
-  #notification = _uiSettings.default.notification; // Map of all providers exposed by the extension, they are retrievable by key
+  #notification = _uiSettings.default.notification;
 
+  // Map of all providers exposed by the extension, they are retrievable by key
   #providers;
   #signRequests = {};
   #windows = [];
+  #connectedTabsUrl = [];
   authSubject = new _rxjs.BehaviorSubject([]);
   metaSubject = new _rxjs.BehaviorSubject([]);
   signSubject = new _rxjs.BehaviorSubject([]);
-
+  defaultAuthAccountSelection = [];
   constructor() {
     let providers = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
     this.#providers = providers;
-    extractMetadata(this.#metaStore); // retrieve previously set authorizations
+    extractMetadata(this.#metaStore);
 
+    // retrieve previously set authorizations
     const authString = localStorage.getItem(AUTH_URLS_KEY) || '{}';
     const previousAuth = JSON.parse(authString);
     this.#authUrls = previousAuth;
-  }
 
+    // retrieve previously set default auth accounts
+    const defaultAuthString = localStorage.getItem(DEFAULT_AUTH_ACCOUNTS) || '[]';
+    const previousDefaultAuth = JSON.parse(defaultAuthString);
+    this.defaultAuthAccountSelection = previousDefaultAuth;
+  }
   get knownMetadata() {
     return (0, _extensionChains.knownMetadata)();
   }
-
   get numAuthRequests() {
     return Object.keys(this.#authRequests).length;
   }
-
   get numMetaRequests() {
     return Object.keys(this.#metaRequests).length;
   }
-
   get numSignRequests() {
     return Object.keys(this.#signRequests).length;
   }
-
   get allAuthRequests() {
     return Object.values(this.#authRequests).map(_ref4 => {
       let {
@@ -155,7 +146,6 @@ class State {
       };
     });
   }
-
   get allMetaRequests() {
     return Object.values(this.#metaRequests).map(_ref5 => {
       let {
@@ -170,7 +160,6 @@ class State {
       };
     });
   }
-
   get allSignRequests() {
     return Object.values(this.#signRequests).map(_ref6 => {
       let {
@@ -187,16 +176,13 @@ class State {
       };
     });
   }
-
   get authUrls() {
     return this.#authUrls;
   }
-
   popupClose() {
     this.#windows.forEach(id => (0, _helpers.withErrorLog)(() => chrome.windows.remove(id)));
     this.#windows = [];
   }
-
   popupOpen() {
     this.#notification !== 'extension' && chrome.windows.create(this.#notification === 'window' ? NORMAL_WINDOW_OPTS : POPUP_WINDOW_OPTS, window => {
       if (window) {
@@ -204,10 +190,8 @@ class State {
       }
     });
   }
-
   authComplete = (id, resolve, reject) => {
     var _this = this;
-
     const complete = function () {
       let authorizedAccounts = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
       const {
@@ -224,14 +208,11 @@ class State {
         origin,
         url
       };
-
       _this.saveCurrentAuthList();
-
+      _this.updateDefaultAuthAccounts(authorizedAccounts);
       delete _this.#authRequests[id];
-
       _this.updateIconAuth(true);
     };
-
     return {
       reject: error => {
         complete();
@@ -250,22 +231,44 @@ class State {
       }
     };
   };
+  udateCurrentTabsUrl(urls) {
+    const connectedTabs = urls.map(url => {
+      let strippedUrl = '';
+
+      // the assert in stripUrl may throw for new tabs with "chrome://newtab/"
+      try {
+        strippedUrl = this.stripUrl(url);
+      } catch (e) {
+        console.error(e);
+      }
 
+      // return the stripped url only if this website is known
+      return !!strippedUrl && this.authUrls[strippedUrl] ? strippedUrl : undefined;
+    }).filter(value => !!value);
+    this.#connectedTabsUrl = connectedTabs;
+  }
+  getConnectedTabsUrl() {
+    return this.#connectedTabsUrl;
+  }
   deleteAuthRequest(requestId) {
     delete this.#authRequests[requestId];
     this.updateIconAuth(true);
   }
-
   saveCurrentAuthList() {
     localStorage.setItem(AUTH_URLS_KEY, JSON.stringify(this.#authUrls));
   }
-
+  saveDefaultAuthAccounts() {
+    localStorage.setItem(DEFAULT_AUTH_ACCOUNTS, JSON.stringify(this.defaultAuthAccountSelection));
+  }
+  updateDefaultAuthAccounts(newList) {
+    this.defaultAuthAccountSelection = newList;
+    this.saveDefaultAuthAccounts();
+  }
   metaComplete = (id, resolve, reject) => {
     const complete = () => {
       delete this.#metaRequests[id];
       this.updateIconMeta(true);
     };
-
     return {
       reject: error => {
         complete();
@@ -282,7 +285,6 @@ class State {
       delete this.#signRequests[id];
       this.updateIconSign(true);
     };
-
     return {
       reject: error => {
         complete();
@@ -294,13 +296,11 @@ class State {
       }
     };
   };
-
   stripUrl(url) {
     (0, _util.assert)(url && (url.startsWith('http:') || url.startsWith('https:') || url.startsWith('ipfs:') || url.startsWith('ipns:')), `Invalid url ${url}, expected to start with http: or https: or ipfs: or ipns:`);
     const parts = url.split('/');
     return parts[2];
   }
-
   updateIcon(shouldClose) {
     const authCount = this.numAuthRequests;
     const metaCount = this.numMetaRequests;
@@ -309,12 +309,10 @@ class State {
     (0, _helpers.withErrorLog)(() => chrome.browserAction.setBadgeText({
       text
     }));
-
     if (shouldClose && text === '') {
       this.popupClose();
     }
   }
-
   removeAuthorization(url) {
     const entry = this.#authUrls[url];
     (0, _util.assert)(entry, `The source ${url} is not known`);
@@ -322,22 +320,18 @@ class State {
     this.saveCurrentAuthList();
     return this.#authUrls;
   }
-
   updateIconAuth(shouldClose) {
     this.authSubject.next(this.allAuthRequests);
     this.updateIcon(shouldClose);
   }
-
   updateIconMeta(shouldClose) {
     this.metaSubject.next(this.allMetaRequests);
     this.updateIcon(shouldClose);
   }
-
   updateIconSign(shouldClose) {
     this.signSubject.next(this.allSignRequests);
     this.updateIcon(shouldClose);
   }
-
   updateAuthorizedAccounts(authorizedAccountDiff) {
     authorizedAccountDiff.forEach(_ref8 => {
       let [url, authorizedAccountDiff] = _ref8;
@@ -345,13 +339,12 @@ class State {
     });
     this.saveCurrentAuthList();
   }
-
   async authorizeUrl(url, request) {
-    const idStr = this.stripUrl(url); // Do not enqueue duplicate authorization requests.
+    const idStr = this.stripUrl(url);
 
+    // Do not enqueue duplicate authorization requests.
     const isDuplicate = Object.values(this.#authRequests).some(request => request.idStr === idStr);
     (0, _util.assert)(!isDuplicate, `The source ${url} has a pending authorization request`);
-
     if (this.#authUrls[idStr]) {
       // this url was seen in the past
       (0, _util.assert)(this.#authUrls[idStr].authorizedAccounts || this.#authUrls[idStr].isAllowed, `The source ${url} is not allowed to interact with this extension`);
@@ -360,10 +353,10 @@ class State {
         result: false
       };
     }
-
     return new Promise((resolve, reject) => {
       const id = (0, _getId.getId)();
-      this.#authRequests[id] = { ...this.authComplete(id, resolve, reject),
+      this.#authRequests[id] = {
+        ...this.authComplete(id, resolve, reject),
         id,
         idStr,
         request,
@@ -373,17 +366,16 @@ class State {
       this.popupOpen();
     });
   }
-
   ensureUrlAuthorized(url) {
     const entry = this.#authUrls[this.stripUrl(url)];
     (0, _util.assert)(entry, `The source ${url} has not been enabled yet`);
     return true;
   }
-
   injectMetadata(url, request) {
     return new Promise((resolve, reject) => {
       const id = (0, _getId.getId)();
-      this.#metaRequests[id] = { ...this.metaComplete(id, resolve, reject),
+      this.#metaRequests[id] = {
+        ...this.metaComplete(id, resolve, reject),
         id,
         request,
         url
@@ -392,56 +384,49 @@ class State {
       this.popupOpen();
     });
   }
-
   getAuthRequest(id) {
     return this.#authRequests[id];
   }
-
   getMetaRequest(id) {
     return this.#metaRequests[id];
   }
-
   getSignRequest(id) {
     return this.#signRequests[id];
-  } // List all providers the extension is exposing
-
+  }
 
+  // List all providers the extension is exposing
   rpcListProviders() {
     return Promise.resolve(Object.keys(this.#providers).reduce((acc, key) => {
       acc[key] = this.#providers[key].meta;
       return acc;
     }, {}));
   }
-
   rpcSend(request, port) {
     const provider = this.#injectedProviders.get(port);
     (0, _util.assert)(provider, 'Cannot call pub(rpc.subscribe) before provider is set');
     return provider.send(request.method, request.params);
-  } // Start a provider, return its meta
-
+  }
 
+  // Start a provider, return its meta
   rpcStartProvider(key, port) {
     (0, _util.assert)(Object.keys(this.#providers).includes(key), `Provider ${key} is not exposed by extension`);
-
     if (this.#injectedProviders.get(port)) {
       return Promise.resolve(this.#providers[key].meta);
-    } // Instantiate the provider
-
+    }
 
-    this.#injectedProviders.set(port, this.#providers[key].start()); // Close provider connection when page is closed
+    // Instantiate the provider
+    this.#injectedProviders.set(port, this.#providers[key].start());
 
+    // Close provider connection when page is closed
     port.onDisconnect.addListener(() => {
       const provider = this.#injectedProviders.get(port);
-
       if (provider) {
         (0, _helpers.withErrorLog)(() => provider.disconnect());
       }
-
       this.#injectedProviders.delete(port);
     });
     return Promise.resolve(this.#providers[key].meta);
   }
-
   rpcSubscribe(_ref9, cb, port) {
     let {
       method,
@@ -452,36 +437,31 @@ class State {
     (0, _util.assert)(provider, 'Cannot call pub(rpc.subscribe) before provider is set');
     return provider.subscribe(type, method, params, cb);
   }
-
   rpcSubscribeConnected(_request, cb, port) {
     const provider = this.#injectedProviders.get(port);
     (0, _util.assert)(provider, 'Cannot call pub(rpc.subscribeConnected) before provider is set');
     cb(null, provider.isConnected); // Immediately send back current isConnected
-
     provider.on('connected', () => cb(null, true));
     provider.on('disconnected', () => cb(null, false));
   }
-
   rpcUnsubscribe(request, port) {
     const provider = this.#injectedProviders.get(port);
     (0, _util.assert)(provider, 'Cannot call pub(rpc.unsubscribe) before provider is set');
     return provider.unsubscribe(request.type, request.method, request.subscriptionId);
   }
-
   saveMetadata(meta) {
     this.#metaStore.set(meta.genesisHash, meta);
     (0, _extensionChains.addMetadata)(meta);
   }
-
   setNotification(notification) {
     this.#notification = notification;
     return true;
   }
-
   sign(url, request, account) {
     const id = (0, _getId.getId)();
     return new Promise((resolve, reject) => {
-      this.#signRequests[id] = { ...this.signComplete(id, resolve, reject),
+      this.#signRequests[id] = {
+        ...this.signComplete(id, resolve, reject),
         account,
         id,
         request,
@@ -491,7 +471,5 @@ class State {
       this.popupOpen();
     });
   }
-
 }
-
 exports.default = State;