64 lines
1.7 KiB
TypeScript
64 lines
1.7 KiB
TypeScript
|
import {createValue, type Value} from "@holllo/webextension-storage";
|
||
|
|
import browser from "webextension-polyfill";
|
||
|
|
import {Feature} from "./enums.js";
|
||
|
|
|
||
|
|
export type UserLabel = {
|
||
|
|
color: string;
|
||
|
|
id: number;
|
||
|
|
priority: number;
|
||
|
|
text: string;
|
||
|
|
username: string;
|
||
|
|
};
|
||
|
|
|
||
|
|
export type UserLabelsData = Array<Value<UserLabel>>;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Create a {@link Value}-wrapped {@link UserLabel}.
|
||
|
|
*/
|
||
|
|
export async function createValueUserLabel(
|
||
|
|
userLabel: UserLabel,
|
||
|
|
): Promise<UserLabelsData[number]> {
|
||
|
|
return createValue<UserLabel>({
|
||
|
|
deserialize: (input) => JSON.parse(input) as UserLabel,
|
||
|
|
serialize: (input) => JSON.stringify(input),
|
||
|
|
key: `${Feature.UserLabels}-${userLabel.id}`,
|
||
|
|
value: userLabel,
|
||
|
|
storage: browser.storage.sync,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Get all user labels from storage and combine them into a single array.
|
||
|
|
*/
|
||
|
|
export async function collectUserLabels(): Promise<UserLabelsData> {
|
||
|
|
const storage = await browser.storage.sync.get();
|
||
|
|
const userLabels = [];
|
||
|
|
for (const [key, value] of Object.entries(storage)) {
|
||
|
|
if (!key.startsWith(Feature.UserLabels)) {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
userLabels.push(
|
||
|
|
await createValueUserLabel(JSON.parse(value as string) as UserLabel),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
return userLabels;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Save all user labels to storage under individual keys.
|
||
|
|
*
|
||
|
|
* They are stored under individual keys so that we don't run into storage quota
|
||
|
|
* limits. If it was stored under a single key we would only be able to fit
|
||
|
|
* around 80-100 labels before hitting the limit.
|
||
|
|
* @param userLabels The user labels array to save.
|
||
|
|
*/
|
||
|
|
export async function saveUserLabels(
|
||
|
|
userLabels: UserLabelsData,
|
||
|
|
): Promise<void> {
|
||
|
|
for (const label of userLabels) {
|
||
|
|
await label.save();
|
||
|
|
}
|
||
|
|
}
|