Source code
Revision control
Copy as Markdown
Other Tools
/* 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
const lazy = {};
ChromeUtils.defineESModuleGetters(lazy, {
AsyncShutdown: "resource://gre/modules/AsyncShutdown.sys.mjs",
LoginHelper: "resource://gre/modules/LoginHelper.sys.mjs",
});
import {
BreachAlertDismissal,
BreachAlertsStore,
} from "moz-src:///toolkit/components/uniffi-bindgen-gecko-js/components/generated/RustBreachAlerts.sys.mjs";
const toRustDismissal = d =>
new BreachAlertDismissal({
breachName: d.breachName,
timeDismissed: d.timeDismissed,
});
const fromRustDismissal = d => ({
breachName: d.breachName,
timeDismissed: d.timeDismissed,
});
export class BreachAlertStorage {
#store = null;
#initializationPromise = null;
#logger = null;
// have it a singleton
constructor() {
if (BreachAlertStorage._instance) {
return BreachAlertStorage._instance;
}
BreachAlertStorage._instance = this;
this.#logger = lazy.LoginHelper.createLogger("BreachAlertStorage");
}
initialize() {
if (this.#initializationPromise) {
this.#logger.log("breach alert storage already initialized");
} else {
const profilePath = Services.dirsvc.get("ProfD", Ci.nsIFile).path;
const path = `${profilePath}/breach-alerts.db`;
this.#logger.log(`Initializing breach alerts storage at ${path}`);
this.#initializationPromise = BreachAlertsStore.newStore(path).then(
store => {
this.#store = store;
this.#logger.log("Breach alert storage ready.");
// Shutdown blocker to ensure that we finalize properly.
lazy.AsyncShutdown.profileChangeTeardown.addBlocker(
"BreachAlertsStore: Interrupt IO operations on breach alerts store",
async () => this.finalize()
);
return this;
},
e => {
this.#logger.log(`Initialization failed: ${e}`);
throw e;
}
);
}
return this.#initializationPromise;
}
/**
* Terminate all pending writes. After this call, the store can't be used.
*/
async finalize() {
await this.#store.close();
}
/**
* @param {string[]} breachNames
* @returns {Promise<object[]>}
*/
async getBreachAlertDismissals(breachNames) {
const results = await this.#store.getBreachAlertDismissals(breachNames);
return results.map(fromRustDismissal);
}
/**
* Creates or updates dismissal records for each entry.
*
* @param {object[]} dismissals - Array of { breachName, timeDismissed } objects.
*/
async setBreachAlertDismissals(dismissals) {
await this.#store.setBreachAlertDismissals(dismissals.map(toRustDismissal));
}
/**
* @param {string[]} breachNames
*/
async clearBreachAlertDismissals(breachNames) {
await this.#store.clearBreachAlertDismissals(breachNames);
}
async clearAllBreachAlertDismissals() {
await this.#store.clearAllBreachAlertDismissals();
}
}