87 lines
2.4 KiB
JavaScript
87 lines
2.4 KiB
JavaScript
import { openDB } from 'idb';
|
|
import { DATA_VERSION } from './version';
|
|
|
|
export class DataManager {
|
|
constructor(accountId) {
|
|
this.modelsToSync = ['inbox', 'label', 'team', 'canned_response'];
|
|
this.accountId = accountId;
|
|
this.db = null;
|
|
}
|
|
|
|
async initDb() {
|
|
if (this.db) return this.db;
|
|
const dbName = `cw-store-${this.accountId}`;
|
|
this.db = await openDB(`cw-store-${this.accountId}`, DATA_VERSION, {
|
|
upgrade(db) {
|
|
// Existing databases already carry the stores added in earlier versions,
|
|
// and createObjectStore throws on a name that is already taken.
|
|
const createStore = (name, options) => {
|
|
if (db.objectStoreNames.contains(name)) return;
|
|
db.createObjectStore(name, options);
|
|
};
|
|
|
|
createStore('cache-keys');
|
|
createStore('inbox', { keyPath: 'id' });
|
|
createStore('label', { keyPath: 'id' });
|
|
createStore('team', { keyPath: 'id' });
|
|
createStore('canned_response', { keyPath: 'id' });
|
|
},
|
|
});
|
|
|
|
// Store the database name in LocalStorage
|
|
const dbNames = JSON.parse(localStorage.getItem('cw-idb-names') || '[]');
|
|
if (!dbNames.includes(dbName)) {
|
|
dbNames.push(dbName);
|
|
localStorage.setItem('cw-idb-names', JSON.stringify(dbNames));
|
|
}
|
|
|
|
return this.db;
|
|
}
|
|
|
|
validateModel(name) {
|
|
if (!name) throw new Error('Model name is not defined');
|
|
if (!this.modelsToSync.includes(name)) {
|
|
throw new Error(`Model ${name} is not defined`);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
async replace({ modelName, data }) {
|
|
this.validateModel(modelName);
|
|
|
|
await this.db.clear(modelName);
|
|
return this.push({ modelName, data });
|
|
}
|
|
|
|
async push({ modelName, data }) {
|
|
this.validateModel(modelName);
|
|
|
|
if (Array.isArray(data)) {
|
|
const tx = this.db.transaction(modelName, 'readwrite');
|
|
data.forEach(item => {
|
|
tx.store.add(item);
|
|
});
|
|
await tx.done;
|
|
} else {
|
|
await this.db.add(modelName, data);
|
|
}
|
|
}
|
|
|
|
async get({ modelName }) {
|
|
this.validateModel(modelName);
|
|
return this.db.getAll(modelName);
|
|
}
|
|
|
|
async setCacheKeys(cacheKeys) {
|
|
Object.keys(cacheKeys).forEach(async modelName => {
|
|
this.db.put('cache-keys', cacheKeys[modelName], modelName);
|
|
});
|
|
}
|
|
|
|
async getCacheKey(modelName) {
|
|
this.validateModel(modelName);
|
|
|
|
return this.db.get('cache-keys', modelName);
|
|
}
|
|
}
|