import { BooleanBothQuery, BooleanQuery, SearchTargetType, VideoChannelsSearchQuery, VideoPlaylistsSearchQuery, VideosSearchQuery } from '@shared/models' export class AdvancedSearch { startDate: string // ISO 8601 endDate: string // ISO 8601 originallyPublishedStartDate: string // ISO 8601 originallyPublishedEndDate: string // ISO 8601 nsfw: BooleanBothQuery categoryOneOf: string licenceOneOf: string languageOneOf: string tagsOneOf: string[] tagsAllOf: string[] durationMin: number // seconds durationMax: number // seconds isLive: BooleanQuery host: string sort: string searchTarget: SearchTargetType // Filters we don't want to count, because they are mandatory private silentFilters = new Set([ 'sort', 'searchTarget' ]) constructor (options?: { startDate?: string endDate?: string originallyPublishedStartDate?: string originallyPublishedEndDate?: string nsfw?: BooleanBothQuery categoryOneOf?: string licenceOneOf?: string languageOneOf?: string tagsOneOf?: any tagsAllOf?: any isLive?: BooleanQuery host?: string durationMin?: string durationMax?: string sort?: string searchTarget?: SearchTargetType }) { if (!options) return this.startDate = options.startDate || undefined this.endDate = options.endDate || undefined this.originallyPublishedStartDate = options.originallyPublishedStartDate || undefined this.originallyPublishedEndDate = options.originallyPublishedEndDate || undefined this.nsfw = options.nsfw || undefined this.isLive = options.isLive || undefined this.categoryOneOf = options.categoryOneOf || undefined this.licenceOneOf = options.licenceOneOf || undefined this.languageOneOf = options.languageOneOf || undefined this.tagsOneOf = this.intoArray(options.tagsOneOf) this.tagsAllOf = this.intoArray(options.tagsAllOf) this.durationMin = parseInt(options.durationMin, 10) this.durationMax = parseInt(options.durationMax, 10) this.host = options.host || undefined this.searchTarget = options.searchTarget || undefined if (isNaN(this.durationMin)) this.durationMin = undefined if (isNaN(this.durationMax)) this.durationMax = undefined this.sort = options.sort || '-match' } containsValues () { const obj = this.toUrlObject() for (const k of Object.keys(obj)) { if (this.silentFilters.has(k)) continue if (this.isValidValue(obj[k])) return true } return false } reset () { this.startDate = undefined this.endDate = undefined this.originallyPublishedStartDate = undefined this.originallyPublishedEndDate = undefined this.nsfw = undefined this.categoryOneOf = undefined this.licenceOneOf = undefined this.languageOneOf = undefined this.tagsOneOf = undefined this.tagsAllOf = undefined this.durationMin = undefined this.durationMax = undefined this.isLive = undefined this.host = undefined this.sort = '-match' } toUrlObject () { return { startDate: this.startDate, endDate: this.endDate, originallyPublishedStartDate: this.originallyPublishedStartDate, originallyPublishedEndDate: this.originallyPublishedEndDate, nsfw: this.nsfw, categoryOneOf: this.categoryOneOf, licenceOneOf: this.licenceOneOf, languageOneOf: this.languageOneOf, tagsOneOf: this.tagsOneOf, tagsAllOf: this.tagsAllOf, durationMin: this.durationMin, durationMax: this.durationMax, isLive: this.isLive, host: this.host, sort: this.sort, searchTarget: this.searchTarget } } toVideosAPIObject (): VideosSearchQuery { let isLive: boolean if (this.isLive) isLive = this.isLive === 'true' return { startDate: this.startDate, endDate: this.endDate, originallyPublishedStartDate: this.originallyPublishedStartDate, originallyPublishedEndDate: this.originallyPublishedEndDate, nsfw: this.nsfw, categoryOneOf: this.intoArray(this.categoryOneOf), licenceOneOf: this.intoArray(this.licenceOneOf), languageOneOf: this.intoArray(this.languageOneOf), tagsOneOf: this.tagsOneOf, tagsAllOf: this.tagsAllOf, durationMin: this.durationMin, durationMax: this.durationMax, host: this.host, isLive, sort: this.sort, searchTarget: this.searchTarget } } toPlaylistAPIObject (): VideoPlaylistsSearchQuery { return { host: this.host, searchTarget: this.searchTarget } } toChannelAPIObject (): VideoChannelsSearchQuery { return { host: this.host, searchTarget: this.searchTarget } } size () { let acc = 0 const obj = this.toUrlObject() for (const k of Object.keys(obj)) { if (this.silentFilters.has(k)) continue if (this.isValidValue(obj[k])) acc++ } return acc } private isValidValue (val: any) { if (val === undefined) return false if (val === '') return false if (Array.isArray(val) && val.length === 0) return false return true } private intoArray (value: any) { if (!value) return undefined if (Array.isArray(value)) return value if (typeof value === 'string') return value.split(',') return [ value ] } }