-import { NSFWQuery, SearchTargetType } from '@shared/models'
+import { intoArray } from '@app/helpers'
+import {
+ BooleanBothQuery,
+ BooleanQuery,
+ SearchTargetType,
+ VideoChannelsSearchQuery,
+ VideoPlaylistsSearchQuery,
+ VideosSearchQuery
+} from '@shared/models'
+
+export type AdvancedSearchResultType = 'videos' | 'playlists' | 'channels'
export class AdvancedSearch {
startDate: string // ISO 8601
originallyPublishedStartDate: string // ISO 8601
originallyPublishedEndDate: string // ISO 8601
- nsfw: NSFWQuery
+ nsfw: BooleanBothQuery
categoryOneOf: string
durationMin: number // seconds
durationMax: number // seconds
+ isLive: BooleanQuery
+
+ host: string
+
sort: string
searchTarget: SearchTargetType
+ resultType: AdvancedSearchResultType
// Filters we don't want to count, because they are mandatory
private silentFilters = new Set([ 'sort', 'searchTarget' ])
endDate?: string
originallyPublishedStartDate?: string
originallyPublishedEndDate?: string
- nsfw?: NSFWQuery
+ nsfw?: BooleanBothQuery
categoryOneOf?: string
licenceOneOf?: string
languageOneOf?: string
tagsOneOf?: any
tagsAllOf?: any
+ isLive?: BooleanQuery
+
+ host?: string
+
durationMin?: string
durationMax?: string
sort?: string
searchTarget?: SearchTargetType
+ resultType?: AdvancedSearchResultType
}) {
if (!options) return
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.tagsOneOf = intoArray(options.tagsOneOf)
+ this.tagsAllOf = intoArray(options.tagsAllOf)
+ this.durationMin = options.durationMin ? parseInt(options.durationMin, 10) : undefined
+ this.durationMax = options.durationMax ? parseInt(options.durationMax, 10) : undefined
+
+ this.host = options.host || undefined
this.searchTarget = options.searchTarget || undefined
+ this.resultType = options.resultType || undefined
+
+ if (!this.resultType && this.hasVideoFilter()) {
+ this.resultType = 'videos'
+ }
+
if (isNaN(this.durationMin)) this.durationMin = undefined
if (isNaN(this.durationMax)) this.durationMax = undefined
this.tagsAllOf = undefined
this.durationMin = undefined
this.durationMax = undefined
+ this.isLive = undefined
+ this.host = undefined
this.sort = '-match'
}
tagsAllOf: this.tagsAllOf,
durationMin: this.durationMin,
durationMax: this.durationMax,
+ isLive: this.isLive,
+ host: this.host,
sort: this.sort,
- searchTarget: this.searchTarget
+ searchTarget: this.searchTarget,
+ resultType: this.resultType
}
}
- toAPIObject () {
+ 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),
+ categoryOneOf: intoArray(this.categoryOneOf),
+ licenceOneOf: intoArray(this.licenceOneOf),
+ languageOneOf: 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
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 ]
+ private hasVideoFilter () {
+ return 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.host !== undefined ||
+ this.isLive !== undefined
}
}