David Yip
7 years ago
commit
d2cdc2b5a3
@ -1,93 +0,0 @@ |
||||
import asyncDB from './async'; |
||||
|
||||
const limit = 1024; |
||||
|
||||
function put(name, objects, callback) { |
||||
asyncDB.then(db => { |
||||
const putTransaction = db.transaction(name, 'readwrite'); |
||||
const putStore = putTransaction.objectStore(name); |
||||
const putIndex = putStore.index('id'); |
||||
|
||||
objects.forEach(object => { |
||||
function add() { |
||||
putStore.add(object); |
||||
} |
||||
|
||||
putIndex.getKey(object.id).onsuccess = retrieval => { |
||||
if (retrieval.target.result) { |
||||
putStore.delete(retrieval.target.result).onsuccess = add; |
||||
} else { |
||||
add(); |
||||
} |
||||
}; |
||||
}); |
||||
|
||||
putTransaction.oncomplete = () => { |
||||
const readTransaction = db.transaction(name, 'readonly'); |
||||
const readStore = readTransaction.objectStore(name); |
||||
|
||||
readStore.count().onsuccess = count => { |
||||
const excess = count.target.result - limit; |
||||
|
||||
if (excess > 0) { |
||||
readStore.getAll(null, excess).onsuccess = |
||||
retrieval => callback(retrieval.target.result.map(({ id }) => id)); |
||||
} |
||||
}; |
||||
}; |
||||
}); |
||||
} |
||||
|
||||
export function evictAccounts(ids) { |
||||
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(id => { |
||||
accountsMovedIndex.getAllKeys(id).onsuccess = |
||||
({ target }) => evict(target.result); |
||||
|
||||
statusesIndex.getAll(id).onsuccess = |
||||
({ target }) => evictStatuses(target.result.map(({ id }) => id)); |
||||
|
||||
accountsIdIndex.getKey(id).onsuccess = |
||||
({ target }) => target.result && accounts.delete(target.result); |
||||
}); |
||||
} |
||||
|
||||
evict(ids); |
||||
}); |
||||
} |
||||
|
||||
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); |
||||
}); |
||||
}); |
||||
} |
||||
|
||||
export function putAccounts(records) { |
||||
put('accounts', records, evictAccounts); |
||||
} |
||||
|
||||
export function putStatuses(records) { |
||||
put('statuses', records, evictStatuses); |
||||
} |
||||
@ -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); |
||||
} |
||||
@ -0,0 +1,33 @@ |
||||
# frozen_string_literal: true |
||||
|
||||
require 'rails_helper' |
||||
|
||||
describe UniqueUsernameValidator do |
||||
describe '#validate' do |
||||
it 'does not add errors if username is nil' do |
||||
account = double(username: nil, persisted?: false, errors: double(add: nil)) |
||||
subject.validate(account) |
||||
expect(account.errors).to_not have_received(:add) |
||||
end |
||||
|
||||
it 'does not add errors when existing one is subject itself' do |
||||
account = Fabricate(:account, username: 'abcdef') |
||||
expect(account).to be_valid |
||||
end |
||||
|
||||
it 'adds an error when the username is already used with ignoring dots' do |
||||
pending 'allowing dots in username is still in development' |
||||
Fabricate(:account, username: 'abcd.ef') |
||||
account = double(username: 'ab.cdef', persisted?: false, errors: double(add: nil)) |
||||
subject.validate(account) |
||||
expect(account.errors).to have_received(:add) |
||||
end |
||||
|
||||
it 'adds an error when the username is already used with ignoring cases' do |
||||
Fabricate(:account, username: 'ABCdef') |
||||
account = double(username: 'abcDEF', persisted?: false, errors: double(add: nil)) |
||||
subject.validate(account) |
||||
expect(account.errors).to have_received(:add) |
||||
end |
||||
end |
||||
end |
||||
Loading…
Reference in new issue