123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211 |
- import openDB from './db';
- const accountAssetKeys = ['avatar', 'avatar_static', 'header', 'header_static'];
- const storageMargin = 8388608;
- const storeLimit = 1024;
- // navigator.storage is not present on:
- // Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.100 Safari/537.36 Edge/16.16299
- // estimate method is not present on Chrome 57.0.2987.98 on Linux.
- export const storageFreeable = 'storage' in navigator && 'estimate' in navigator.storage;
- function openCache() {
- // ServiceWorker and Cache API is not available on iOS 11
- // https://webkit.org/status/#specification-service-workers
- return self.caches ? caches.open('mastodon-system') : Promise.reject();
- }
- function printErrorIfAvailable(error) {
- if (error) {
- console.warn(error);
- }
- }
- function put(name, objects, onupdate, oncreate) {
- return openDB().then(db => (new Promise((resolve, reject) => {
- const putTransaction = db.transaction(name, 'readwrite');
- const putStore = putTransaction.objectStore(name);
- const putIndex = putStore.index('id');
- objects.forEach(object => {
- putIndex.getKey(object.id).onsuccess = retrieval => {
- function addObject() {
- putStore.add(object);
- }
- function deleteObject() {
- putStore.delete(retrieval.target.result).onsuccess = addObject;
- }
- if (retrieval.target.result) {
- if (onupdate) {
- onupdate(object, retrieval.target.result, putStore, deleteObject);
- } else {
- deleteObject();
- }
- } else {
- if (oncreate) {
- oncreate(object, addObject);
- } else {
- addObject();
- }
- }
- };
- });
- putTransaction.oncomplete = () => {
- const readTransaction = db.transaction(name, 'readonly');
- const readStore = readTransaction.objectStore(name);
- const count = readStore.count();
- count.onsuccess = () => {
- const excess = count.result - storeLimit;
- if (excess > 0) {
- const retrieval = readStore.getAll(null, excess);
- retrieval.onsuccess = () => resolve(retrieval.result);
- retrieval.onerror = reject;
- } else {
- resolve([]);
- }
- };
- count.onerror = reject;
- };
- putTransaction.onerror = reject;
- })).then(resolved => {
- db.close();
- return resolved;
- }, error => {
- db.close();
- throw error;
- }));
- }
- function evictAccountsByRecords(records) {
- return openDB().then(db => {
- const transaction = db.transaction(['accounts', 'statuses'], 'readwrite');
- const accounts = transaction.objectStore('accounts');
- const accountsIdIndex = accounts.index('id');
- const accountsMovedIndex = accounts.index('moved');
- const statuses = transaction.objectStore('statuses');
- const statusesIndex = statuses.index('account');
- function evict(toEvict) {
- toEvict.forEach(record => {
- openCache()
- .then(cache => accountAssetKeys.forEach(key => cache.delete(records[key])))
- .catch(printErrorIfAvailable);
- accountsMovedIndex.getAll(record.id).onsuccess = ({ target }) => evict(target.result);
- statusesIndex.getAll(record.id).onsuccess =
- ({ target }) => evictStatusesByRecords(target.result);
- accountsIdIndex.getKey(record.id).onsuccess =
- ({ target }) => target.result && accounts.delete(target.result);
- });
- }
- evict(records);
- db.close();
- }).catch(printErrorIfAvailable);
- }
- export function evictStatus(id) {
- evictStatuses([id]);
- }
- export function evictStatuses(ids) {
- return openDB().then(db => {
- const transaction = db.transaction('statuses', 'readwrite');
- const store = transaction.objectStore('statuses');
- const idIndex = store.index('id');
- const reblogIndex = store.index('reblog');
- ids.forEach(id => {
- reblogIndex.getAllKeys(id).onsuccess =
- ({ target }) => target.result.forEach(reblogKey => store.delete(reblogKey));
- idIndex.getKey(id).onsuccess =
- ({ target }) => target.result && store.delete(target.result);
- });
- db.close();
- }).catch(printErrorIfAvailable);
- }
- function evictStatusesByRecords(records) {
- return evictStatuses(records.map(({ id }) => id));
- }
- export function putAccounts(records, avatarStatic) {
- const avatarKey = avatarStatic ? 'avatar_static' : 'avatar';
- const newURLs = [];
- put('accounts', records, (newRecord, oldKey, store, oncomplete) => {
- store.get(oldKey).onsuccess = ({ target }) => {
- accountAssetKeys.forEach(key => {
- const newURL = newRecord[key];
- const oldURL = target.result[key];
- if (newURL !== oldURL) {
- openCache()
- .then(cache => cache.delete(oldURL))
- .catch(printErrorIfAvailable);
- }
- });
- const newURL = newRecord[avatarKey];
- const oldURL = target.result[avatarKey];
- if (newURL !== oldURL) {
- newURLs.push(newURL);
- }
- oncomplete();
- };
- }, (newRecord, oncomplete) => {
- newURLs.push(newRecord[avatarKey]);
- oncomplete();
- }).then(records => Promise.all([
- evictAccountsByRecords(records),
- openCache().then(cache => cache.addAll(newURLs)),
- ])).then(freeStorage, error => {
- freeStorage();
- throw error;
- }).catch(printErrorIfAvailable);
- }
- export function putStatuses(records) {
- put('statuses', records)
- .then(evictStatusesByRecords)
- .catch(printErrorIfAvailable);
- }
- export function freeStorage() {
- return storageFreeable && navigator.storage.estimate().then(({ quota, usage }) => {
- if (usage + storageMargin < quota) {
- return null;
- }
- return openDB().then(db => new Promise((resolve, reject) => {
- const retrieval = db.transaction('accounts', 'readonly').objectStore('accounts').getAll(null, 1);
- retrieval.onsuccess = () => {
- if (retrieval.result.length > 0) {
- resolve(evictAccountsByRecords(retrieval.result).then(freeStorage));
- } else {
- resolve(caches.delete('mastodon-system'));
- }
- };
- retrieval.onerror = reject;
- db.close();
- }));
- });
- }
|