X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fmodels%2Futils.ts;h=2b172f6081e3eea7437a11010968af414ae76ab2;hb=2fb5b3a55aeebcc77f4b3a0c029bbf0738ef0063;hp=4ebd07dabbec5c1745d0faddc2fab73c040df554;hpb=418d092afa81e2c8fe8ac6838fc4b5eb0af6a782;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/models/utils.ts b/server/models/utils.ts index 4ebd07dab..2b172f608 100644 --- a/server/models/utils.ts +++ b/server/models/utils.ts @@ -1,26 +1,29 @@ import { Sequelize } from 'sequelize-typescript' import * as validator from 'validator' +import { OrderItem } from 'sequelize' +import { Col } from 'sequelize/types/lib/utils' type SortType = { sortModel: any, sortValue: string } // Translate for example "-name" to [ [ 'name', 'DESC' ], [ 'id', 'ASC' ] ] -function getSort (value: string, lastSort: string[] = [ 'id', 'ASC' ]) { - let { direction, field } = buildDirectionAndField(value) +function getSort (value: string, lastSort: OrderItem = [ 'id', 'ASC' ]): OrderItem[] { + const { direction, field } = buildDirectionAndField(value) + + let finalField: string | Col if (field.toLowerCase() === 'match') { // Search - field = Sequelize.col('similarity') + finalField = Sequelize.col('similarity') + } else { + finalField = field } - return [ [ field, direction ], lastSort ] + return [ [ finalField, direction ], lastSort ] } -function getVideoSort (value: string, lastSort: string[] = [ 'id', 'ASC' ]) { - let { direction, field } = buildDirectionAndField(value) +function getVideoSort (value: string, lastSort: OrderItem = [ 'id', 'ASC' ]): OrderItem[] { + const { direction, field } = buildDirectionAndField(value) - // Alias - if (field.toLowerCase() === 'match') { // Search - field = Sequelize.col('similarity') - } else if (field.toLowerCase() === 'trending') { // Sort by aggregation + if (field.toLowerCase() === 'trending') { // Sort by aggregation return [ [ Sequelize.fn('COALESCE', Sequelize.fn('SUM', Sequelize.col('VideoViews.views')), '0'), direction ], @@ -30,21 +33,40 @@ function getVideoSort (value: string, lastSort: string[] = [ 'id', 'ASC' ]) { ] } - const firstSort = typeof field === 'string' ? - field.split('.').concat([ direction ]) : - [ field, direction ] + let finalField: string | Col + + // Alias + if (field.toLowerCase() === 'match') { // Search + finalField = Sequelize.col('similarity') + } else { + finalField = field + } + + const firstSort = typeof finalField === 'string' + ? finalField.split('.').concat([ direction ]) as any // FIXME: sequelize typings + : [ finalField, direction ] return [ firstSort, lastSort ] } -function getSortOnModel (model: any, value: string, lastSort: string[] = [ 'id', 'ASC' ]) { - let [ firstSort ] = getSort(value) +function getSortOnModel (model: any, value: string, lastSort: OrderItem = [ 'id', 'ASC' ]): OrderItem[] { + const [ firstSort ] = getSort(value) if (model) return [ [ model, firstSort[0], firstSort[1] ], lastSort ] return [ firstSort, lastSort ] } -function throwIfNotValid (value: any, validator: (value: any) => boolean, fieldName = 'value') { +function isOutdated (model: { createdAt: Date, updatedAt: Date }, refreshInterval: number) { + const now = Date.now() + const createdAtTime = model.createdAt.getTime() + const updatedAtTime = model.updatedAt.getTime() + + return (now - createdAtTime) > refreshInterval && (now - updatedAtTime) > refreshInterval +} + +function throwIfNotValid (value: any, validator: (value: any) => boolean, fieldName = 'value', nullable = false) { + if (nullable && (value === null || value === undefined)) return + if (validator(value) === false) { throw new Error(`"${value}" is not a valid ${fieldName}.`) } @@ -96,6 +118,15 @@ function buildWhereIdOrUUID (id: number | string) { return validator.isInt('' + id) ? { id } : { uuid: id } } +function parseAggregateResult (result: any) { + if (!result) return 0 + + const total = parseInt(result + '', 10) + if (isNaN(total)) return 0 + + return total +} + // --------------------------------------------------------------------------- export { @@ -108,7 +139,9 @@ export { throwIfNotValid, buildServerIdsFollowedBy, buildTrigramSearchIndex, - buildWhereIdOrUUID + buildWhereIdOrUUID, + isOutdated, + parseAggregateResult } // --------------------------------------------------------------------------- @@ -122,7 +155,7 @@ function searchTrigramNormalizeCol (col: string) { } function buildDirectionAndField (value: string) { - let field: any + let field: string let direction: 'ASC' | 'DESC' if (value.substring(0, 1) === '-') {