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) }