aboutsummaryrefslogtreecommitdiff
path: root/common
diff options
context:
space:
mode:
Diffstat (limited to 'common')
-rw-r--r--common/Storage.js130
1 files changed, 130 insertions, 0 deletions
diff --git a/common/Storage.js b/common/Storage.js
new file mode 100644
index 0000000..a83ce8f
--- /dev/null
+++ b/common/Storage.js
@@ -0,0 +1,130 @@
+/**
+* GNU LibreJS - A browser add-on to block nonfree nontrivial JavaScript.
+*
+* Copyright (C) 2018 Giorgio Maone <giorgio@maone.net>
+*
+* This file is part of GNU LibreJS.
+*
+* GNU LibreJS is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+*
+* GNU LibreJS is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+* GNU General Public License for more details.
+*
+* You should have received a copy of the GNU General Public License
+* along with GNU LibreJS. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+/**
+ A tiny wrapper around extensions storage API, supporting CSV serialization for
+ retro-compatibility
+*/
+"use strict";
+
+var Storage = {
+ ARRAY: {
+ async load(key, array = undefined) {
+ if (array === undefined) {
+ array = (await browser.storage.local.get(key))[key];
+ }
+ return array ? new Set(array) : new Set();
+ },
+ async save(key, list) {
+ return await browser.storage.local.set({[key]: [...list]});
+ },
+ },
+
+ CSV: {
+ async load(key) {
+ let csv = (await browser.storage.local.get(key))[key];
+ return csv ? new Set(csv.split(/\s*,\s*/)) : new Set();
+ },
+
+ async save(key, list) {
+ return await browser.storage.local.set({[key]: [...list].join(",")});
+ }
+ }
+};
+
+/**
+ A class to hold and persist blacklists and whitelists
+*/
+
+class ListStore {
+ constructor(key, storage = Storage.ARRAY) {
+ this.key = key;
+ this.storage = storage;
+ this.items = new Set();
+ browser.storage.onChanged.addListener(changes => {
+ if (!this.saving && this.key in changes) {
+ this.load(changes[this.key].newValue);
+ }
+ });
+ }
+
+ static hashItem(hash) {
+ return hash.startsWith("(") ? hash : `(${hash})`;
+ }
+ static urlItem(url) {
+ let queryPos = url.indexOf("?");
+ return queryPos === -1 ? url : url.substring(0, queryPos);
+ }
+ static siteItem(url) {
+ if (url.endsWith("/*")) return url;
+ try {
+ return `${new URL(url).origin}/*`;
+ } catch (e) {
+ return `${url}/*`;
+ }
+ }
+
+ async save() {
+ this._saving = true;
+ try {
+ return await this.storage.save(this.key, this.items);
+ } finally {
+ this._saving = false;
+ }
+ }
+
+ async load(values = undefined) {
+ try {
+ this.items = await this.storage.load(this.key, values);
+ } catch (e) {
+ console.error(e);
+ }
+ return this.items;
+ }
+
+ async store(...items) {
+ let size = this.items.size;
+ let changed = false;
+ for (let item of items) {
+ if (size !== this.items.add(item).size) {
+ changed = true;
+ }
+ }
+ return changed && await this.save();
+ }
+
+ async remove(...items) {
+ let changed = false;
+ for (let item of items) {
+ if (this.items.delete(item)) {
+ changed = true;
+ }
+ }
+ return changed && await this.save();
+ }
+
+ contains(item) {
+ return this.items.has(item);
+ }
+}
+if (typeof module === "object") {
+ module.exports = { ListStore, Storage };
+}