1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
* You can obtain one at http://mozilla.org/MPL/2.0/. */
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
IndexedDB: "resource://gre/modules/IndexedDB.sys.mjs",
});
export class ActivityStreamStorage {
/**
* @param storeNames Array of strings used to create all the required stores
*/
constructor({ storeNames, telemetry }) {
if (!storeNames) {
throw new Error("storeNames required");
}
this.dbName = "ActivityStream";
this.dbVersion = 3;
this.storeNames = storeNames;
this.telemetry = telemetry;
}
get db() {
return this._db || (this._db = this.createOrOpenDb());
}
/**
* Public method that binds the store required by the consumer and exposes
* the private db getters and setters.
*
* @param storeName String name of desired store
*/
getDbTable(storeName) {
if (this.storeNames.includes(storeName)) {
return {
get: this._get.bind(this, storeName),
getAll: this._getAll.bind(this, storeName),
getAllKeys: this._getAllKeys.bind(this, storeName),
set: this._set.bind(this, storeName),
};
}
throw new Error(`Store name ${storeName} does not exist.`);
}
async _getStore(storeName) {
return (await this.db).objectStore(storeName, "readwrite");
}
_get(storeName, key) {
return this._requestWrapper(async () =>
(await this._getStore(storeName)).get(key)
);
}
_getAll(storeName) {
return this._requestWrapper(async () =>
(await this._getStore(storeName)).getAll()
);
}
_getAllKeys(storeName) {
return this._requestWrapper(async () =>
(await this._getStore(storeName)).getAllKeys()
);
}
_set(storeName, key, value) {
return this._requestWrapper(async () =>
(await this._getStore(storeName)).put(value, key)
);
}
_openDatabase() {
return lazy.IndexedDB.open(this.dbName, this.dbVersion, db => {
// If provided with array of objectStore names we need to create all the
// individual stores
this.storeNames.forEach(store => {
if (!db.objectStoreNames.contains(store)) {
this._requestWrapper(() => db.createObjectStore(store));
}
});
});
}
/**
* createOrOpenDb - Open a db (with this.dbName) if it exists.
* If it does not exist, create it.
* If an error occurs, deleted the db and attempt to
* re-create it.
* @returns Promise that resolves with a db instance
*/
async createOrOpenDb() {
try {
const db = await this._openDatabase();
return db;
} catch (e) {
if (this.telemetry) {
this.telemetry.handleUndesiredEvent({ event: "INDEXEDDB_OPEN_FAILED" });
}
await lazy.IndexedDB.deleteDatabase(this.dbName);
return this._openDatabase();
}
}
async _requestWrapper(request) {
let result = null;
try {
result = await request();
} catch (e) {
if (this.telemetry) {
this.telemetry.handleUndesiredEvent({ event: "TRANSACTION_FAILED" });
}
throw e;
}
return result;
}
}
export function getDefaultOptions(options) {
return { collapsed: !!options.collapsed };
}
|
