import throttle from 'lodash-es/throttle' import { database } from '../_database/database' import { mark, stop } from '../_utils/marks' import { store } from '../_store/store' import { scheduleIdleTask } from '../_utils/scheduleIdleTask' import uniqBy from 'lodash-es/uniqBy' import uniq from 'lodash-es/uniq' import { isMobile } from '../_utils/isMobile' const STREAMING_THROTTLE_DELAY = 3000 function getExistingItemIdsSet (instanceName, timelineName) { let timelineItemIds = store.getForTimeline(instanceName, timelineName, 'timelineItemIds') || [] return new Set(timelineItemIds) } function removeDuplicates (instanceName, timelineName, updates) { // remove duplicates, including duplicates due to reblogs let existingItemIds = getExistingItemIdsSet(instanceName, timelineName) return updates.filter(update => !existingItemIds.has(update.id)) } async function insertUpdatesIntoTimeline (instanceName, timelineName, updates) { updates = removeDuplicates(instanceName, timelineName, updates) if (!updates.length) { return } await database.insertTimelineItems(instanceName, timelineName, updates) let itemIdsToAdd = store.getForTimeline(instanceName, timelineName, 'itemIdsToAdd') || [] itemIdsToAdd = uniq(itemIdsToAdd.concat(updates.map(_ => _.id))) console.log('adding ', itemIdsToAdd.length, 'items to itemIdsToAdd on instance', instanceName, 'on timeline', timelineName) store.setForTimeline(instanceName, timelineName, {itemIdsToAdd: itemIdsToAdd}) } async function insertUpdatesIntoThreads (instanceName, updates) { if (!updates.length) { return } let threads = store.getThreads(instanceName) for (let timelineName of Object.keys(threads)) { let thread = threads[timelineName] let updatesForThisThread = updates.filter(status => { return thread.includes(status.in_reply_to_id) && !thread.includes(status.id) }) let itemIdsToAdd = store.getForTimeline(instanceName, timelineName, 'itemIdsToAdd') || [] for (let update of updatesForThisThread) { if (!itemIdsToAdd.includes(update.id)) { itemIdsToAdd.push(update.id) } } console.log('adding ', itemIdsToAdd.length, 'items to itemIdsToAdd for thread', timelineName) store.setForTimeline(instanceName, timelineName, {itemIdsToAdd: itemIdsToAdd}) } } 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: []}) await insertUpdatesIntoTimeline(instanceName, timelineName, updates) await insertUpdatesIntoThreads(instanceName, updates.filter(status => status.in_reply_to_id)) } stop('processFreshUpdates') } const lazilyProcessFreshUpdates = throttle((instanceName, timelineName) => { const runTask = isMobile() ? scheduleIdleTask : requestAnimationFrame runTask(() => { /* no await */ processFreshUpdates(instanceName, timelineName) }) }, STREAMING_THROTTLE_DELAY) export function addStatusOrNotification (instanceName, timelineName, newStatusOrNotification) { addStatusesOrNotifications(instanceName, timelineName, [newStatusOrNotification]) } export function addStatusesOrNotifications (instanceName, timelineName, newStatusesOrNotifications) { let freshUpdates = store.getForTimeline(instanceName, timelineName, 'freshUpdates') || [] freshUpdates = freshUpdates.concat(newStatusesOrNotifications) freshUpdates = uniqBy(freshUpdates, _ => _.id) store.setForTimeline(instanceName, timelineName, {freshUpdates: freshUpdates}) lazilyProcessFreshUpdates(instanceName, timelineName) }