|
|
@ -0,0 +1,151 @@ |
|
|
|
import asyncDB from './db'; |
|
|
|
import { autoPlayGif } from '../initial_state'; |
|
|
|
|
|
|
|
const accountAssetKeys = ['avatar', 'avatar_static', 'header', 'header_static']; |
|
|
|
const avatarKey = autoPlayGif ? 'avatar' : 'avatar_static'; |
|
|
|
const limit = 1024; |
|
|
|
const asyncCache = caches.open('mastodon-system'); |
|
|
|
|
|
|
|
function put(name, objects, onupdate, oncreate) { |
|
|
|
return asyncDB.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 - limit; |
|
|
|
|
|
|
|
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; |
|
|
|
})); |
|
|
|
} |
|
|
|
|
|
|
|
function evictAccountsByRecords(records) { |
|
|
|
asyncDB.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 => { |
|
|
|
asyncCache.then(cache => accountAssetKeys.forEach(key => cache.delete(records[key]))); |
|
|
|
|
|
|
|
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); |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
export function evictStatus(id) { |
|
|
|
return evictStatuses([id]); |
|
|
|
} |
|
|
|
|
|
|
|
export function evictStatuses(ids) { |
|
|
|
asyncDB.then(db => { |
|
|
|
const store = db.transaction('statuses', 'readwrite').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); |
|
|
|
}); |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
function evictStatusesByRecords(records) { |
|
|
|
evictStatuses(records.map(({ id }) => id)); |
|
|
|
} |
|
|
|
|
|
|
|
export function putAccounts(records) { |
|
|
|
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) { |
|
|
|
asyncCache.then(cache => cache.delete(oldURL)); |
|
|
|
} |
|
|
|
}); |
|
|
|
|
|
|
|
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 => { |
|
|
|
evictAccountsByRecords(records); |
|
|
|
asyncCache.then(cache => cache.addAll(newURLs)); |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
export function putStatuses(records) { |
|
|
|
put('statuses', records).then(evictStatusesByRecords); |
|
|
|
} |