mirror of https://github.com/Chocobozzz/PeerTube
63 lines
1.6 KiB
TypeScript
63 lines
1.6 KiB
TypeScript
import Bluebird from 'bluebird'
|
|
import validator from 'validator'
|
|
import { pageToStartAndCount } from '@server/helpers/core-utils'
|
|
import { ACTIVITY_PUB } from '@server/initializers/constants'
|
|
import { ResultList } from '@shared/models'
|
|
|
|
type ActivityPubCollectionPaginationHandler = (start: number, count: number) => Bluebird<ResultList<any>> | Promise<ResultList<any>>
|
|
|
|
async function activityPubCollectionPagination (
|
|
baseUrl: string,
|
|
handler: ActivityPubCollectionPaginationHandler,
|
|
page?: any,
|
|
size = ACTIVITY_PUB.COLLECTION_ITEMS_PER_PAGE
|
|
) {
|
|
if (!page || !validator.isInt(page)) {
|
|
// We just display the first page URL, we only need the total items
|
|
const result = await handler(0, 1)
|
|
|
|
return {
|
|
id: baseUrl,
|
|
type: 'OrderedCollectionPage',
|
|
totalItems: result.total,
|
|
first: result.data.length === 0
|
|
? undefined
|
|
: baseUrl + '?page=1'
|
|
}
|
|
}
|
|
|
|
const { start, count } = pageToStartAndCount(page, size)
|
|
const result = await handler(start, count)
|
|
|
|
let next: string | undefined
|
|
let prev: string | undefined
|
|
|
|
// Assert page is a number
|
|
page = parseInt(page, 10)
|
|
|
|
// There are more results
|
|
if (result.total > page * size) {
|
|
next = baseUrl + '?page=' + (page + 1)
|
|
}
|
|
|
|
if (page > 1) {
|
|
prev = baseUrl + '?page=' + (page - 1)
|
|
}
|
|
|
|
return {
|
|
id: baseUrl + '?page=' + page,
|
|
type: 'OrderedCollectionPage',
|
|
prev,
|
|
next,
|
|
partOf: baseUrl,
|
|
orderedItems: result.data,
|
|
totalItems: result.total
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
activityPubCollectionPagination
|
|
}
|