41 lines
1.6 KiB
TypeScript
41 lines
1.6 KiB
TypeScript
/** Newznab categories this indexer assigns (see release_category() in SQL). */
|
|
export const CATEGORIES: { id: number; name: string; subcats: { id: number; name: string }[] }[] = [
|
|
{ id: 2000, name: 'Movies', subcats: [{ id: 2030, name: 'SD' }, { id: 2040, name: 'HD' }, { id: 2045, name: 'UHD' }] },
|
|
{ id: 3000, name: 'Audio', subcats: [{ id: 3010, name: 'MP3' }, { id: 3040, name: 'Lossless' }] },
|
|
{ id: 4000, name: 'PC', subcats: [] },
|
|
{ id: 5000, name: 'TV', subcats: [{ id: 5030, name: 'SD' }, { id: 5040, name: 'HD' }, { id: 5045, name: 'UHD' }] },
|
|
{ id: 7000, name: 'Books', subcats: [{ id: 7020, name: 'Ebook' }] },
|
|
{ id: 8000, name: 'Other', subcats: [{ id: 8010, name: 'Misc' }] },
|
|
]
|
|
|
|
/** Parent category of a (sub)category, e.g. 5040 → 5000. */
|
|
export function parentCategory(id: number): number {
|
|
return Math.floor(id / 1000) * 1000
|
|
}
|
|
|
|
/** "TV > HD" style label. */
|
|
export function categoryLabel(id: number): string {
|
|
const parent = CATEGORIES.find((c) => c.id === parentCategory(id))
|
|
if (!parent) return 'Other'
|
|
const sub = parent.subcats.find((s) => s.id === id)
|
|
return sub ? `${parent.name} > ${sub.name}` : parent.name
|
|
}
|
|
|
|
/**
|
|
* Expands requested categories (`cat=5000,2040`) to the concrete ones to
|
|
* match: a parent includes all of its subcategories.
|
|
*/
|
|
export function expandCategories(requested: number[]): number[] {
|
|
const result = new Set<number>()
|
|
for (const id of requested) {
|
|
const parent = CATEGORIES.find((c) => c.id === id)
|
|
if (parent) {
|
|
result.add(parent.id)
|
|
for (const sub of parent.subcats) result.add(sub.id)
|
|
} else {
|
|
result.add(id)
|
|
}
|
|
}
|
|
return [...result]
|
|
}
|