56 lines
2.4 KiB
JavaScript
56 lines
2.4 KiB
JavaScript
|
import throttle from 'lodash/throttle'
|
||
|
|
import { getIdsThatTheseStatusesReblogged } from './statuses'
|
||
|
|
import { database } from '../_database/database'
|
||
|
|
import { mark, stop } from '../_utils/marks'
|
||
|
|
import { store } from '../_store/store'
|
||
|
|
import { scheduleIdleTask } from '../_utils/scheduleIdleTask'
|
||
|
|
|
||
|
|
async function getExistingItemIdsSet (instanceName, timelineName) {
|
||
|
|
let timelineItemIds = store.getForTimeline(instanceName, timelineName, 'timelineItemIds') || []
|
||
|
|
if (timelineName === 'notifications') {
|
||
|
|
return new Set(timelineItemIds)
|
||
|
|
}
|
||
|
|
let reblogIds = await getIdsThatTheseStatusesReblogged(instanceName, timelineItemIds)
|
||
|
|
return new Set([].concat(timelineItemIds).concat(reblogIds))
|
||
|
|
}
|
||
|
|
|
||
|
|
async function removeDuplicates (instanceName, timelineName, updates) {
|
||
|
|
// remove duplicates, including duplicates due to reblogs
|
||
|
|
let existingItemIds = await getExistingItemIdsSet(instanceName, timelineName)
|
||
|
|
return updates.filter(update => !existingItemIds.has(update.id))
|
||
|
|
}
|
||
|
|
|
||
|
|
async function processFreshUpdates (instanceName, timelineName) {
|
||
|
|
mark('processFreshUpdates')
|
||
|
|
let freshUpdates = store.getForTimeline(instanceName, timelineName, 'freshUpdates')
|
||
|
|
if (freshUpdates && freshUpdates.length) {
|
||
|
|
let updates = freshUpdates.slice()
|
||
|
|
store.setForTimeline(instanceName, timelineName, {freshUpdates: []})
|
||
|
|
|
||
|
|
updates = await removeDuplicates(instanceName, timelineName, updates)
|
||
|
|
|
||
|
|
await database.insertTimelineItems(instanceName, timelineName, updates)
|
||
|
|
|
||
|
|
let itemIdsToAdd = store.getForTimeline(instanceName, timelineName, 'itemIdsToAdd') || []
|
||
|
|
if (updates && updates.length) {
|
||
|
|
itemIdsToAdd = itemIdsToAdd.concat(updates.map(_ => _.id))
|
||
|
|
console.log('adding ', itemIdsToAdd.length, 'items to itemIdsToAdd')
|
||
|
|
store.setForTimeline(instanceName, timelineName, {itemIdsToAdd: itemIdsToAdd})
|
||
|
|
}
|
||
|
|
stop('processFreshUpdates')
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const lazilyProcessFreshUpdates = throttle((instanceName, timelineName) => {
|
||
|
|
scheduleIdleTask(() => {
|
||
|
|
/* no await */ processFreshUpdates(instanceName, timelineName)
|
||
|
|
})
|
||
|
|
}, 5000)
|
||
|
|
|
||
|
|
export function addStatusOrNotification (instanceName, timelineName, newStatusOrNotification) {
|
||
|
|
let freshUpdates = store.getForTimeline(instanceName, timelineName, 'freshUpdates') || []
|
||
|
|
freshUpdates.push(newStatusOrNotification)
|
||
|
|
store.setForTimeline(instanceName, timelineName, {freshUpdates: freshUpdates})
|
||
|
|
lazilyProcessFreshUpdates(instanceName, timelineName)
|
||
|
|
}
|