122 lines
3 KiB
TypeScript
122 lines
3 KiB
TypeScript
import type { Paginator, WsEvents } from 'masto'
|
|
import type { PaginatorState } from '~/types'
|
|
|
|
export function usePaginator<T, P>(
|
|
paginator: Paginator<T[], P>,
|
|
stream?: Promise<WsEvents>,
|
|
eventType: 'notification' | 'update' = 'update',
|
|
preprocess: (items: T[]) => T[] = (items: T[]) => items,
|
|
buffer = 10,
|
|
) {
|
|
const state = ref<PaginatorState>(isMastoInitialised.value ? 'idle' : 'loading')
|
|
const items = ref<T[]>([])
|
|
const nextItems = ref<T[]>([])
|
|
const prevItems = ref<T[]>([])
|
|
|
|
const endAnchor = ref<HTMLDivElement>()
|
|
const bound = reactive(useElementBounding(endAnchor))
|
|
const isInScreen = $computed(() => bound.top < window.innerHeight * 2)
|
|
const error = ref<unknown | undefined>()
|
|
const deactivated = useDeactivated()
|
|
|
|
async function update() {
|
|
items.value.unshift(...preprocess(prevItems.value as any) as any)
|
|
prevItems.value = []
|
|
}
|
|
|
|
stream?.then((s) => {
|
|
s.on(eventType, (status) => {
|
|
if ('uri' in status)
|
|
cacheStatus(status, undefined, true)
|
|
|
|
const index = prevItems.value.findIndex((i: any) => i.id === status.id)
|
|
if (index >= 0)
|
|
prevItems.value.splice(index, 1)
|
|
|
|
prevItems.value.unshift(status as any)
|
|
})
|
|
|
|
// TODO: update statuses
|
|
s.on('status.update', (status) => {
|
|
cacheStatus(status, undefined, true)
|
|
|
|
const index = items.value.findIndex((s: any) => s.id === status.id)
|
|
if (index >= 0)
|
|
items.value[index] = status as any
|
|
})
|
|
|
|
s.on('delete', (id) => {
|
|
removeCachedStatus(id)
|
|
|
|
const index = items.value.findIndex((s: any) => s.id === id)
|
|
if (index >= 0)
|
|
items.value.splice(index, 1)
|
|
})
|
|
})
|
|
|
|
async function loadNext() {
|
|
if (state.value !== 'idle')
|
|
return
|
|
|
|
state.value = 'loading'
|
|
try {
|
|
const result = await paginator.next()
|
|
|
|
if (result.value?.length) {
|
|
const preprocessedItems = preprocess([...nextItems.value, ...result.value]) as any
|
|
const itemsToShowCount = preprocessedItems.length - buffer
|
|
nextItems.value = preprocessedItems.slice(itemsToShowCount)
|
|
items.value.push(...preprocessedItems.slice(0, itemsToShowCount))
|
|
state.value = 'idle'
|
|
}
|
|
else {
|
|
items.value.push(...nextItems.value)
|
|
nextItems.value = []
|
|
state.value = 'done'
|
|
}
|
|
}
|
|
catch (e) {
|
|
error.value = e
|
|
state.value = 'error'
|
|
}
|
|
|
|
await nextTick()
|
|
bound.update()
|
|
}
|
|
|
|
if (process.client) {
|
|
useIntervalFn(() => {
|
|
bound.update()
|
|
}, 1000)
|
|
|
|
if (!isMastoInitialised.value) {
|
|
onMastoInit(() => {
|
|
state.value = 'idle'
|
|
loadNext()
|
|
})
|
|
}
|
|
|
|
watch(
|
|
() => [isInScreen, state],
|
|
() => {
|
|
if (
|
|
isInScreen
|
|
&& state.value === 'idle'
|
|
// No new content is loaded when the keepAlive page enters the background
|
|
&& deactivated.value === false
|
|
)
|
|
loadNext()
|
|
},
|
|
)
|
|
}
|
|
|
|
return {
|
|
items,
|
|
prevItems,
|
|
update,
|
|
state,
|
|
error,
|
|
endAnchor,
|
|
}
|
|
}
|