Update scripts

scripts/models/stream.ts

@@ -0,0 +1,177 @@
+import { URL, Collection } from '../core'
+import { Category, Language } from './index'
+
+type StreamProps = {
+  name: string
+  url: string
+  filepath: string
+  line: number
+  channel?: string
+  httpReferrer?: string
+  label?: string
+  quality?: string
+  userAgent?: string
+}
+
+export class Stream {
+  channel: string
+  filepath: string
+  line: number
+  httpReferrer: string
+  label: string
+  name: string
+  quality: string
+  url: string
+  userAgent: string
+  logo: string
+  broadcastArea: Collection
+  categories: Collection
+  languages: Collection
+  isNSFW: boolean
+  groupTitle: string
+
+  constructor({
+    channel,
+    filepath,
+    line,
+    httpReferrer,
+    label,
+    name,
+    quality,
+    url,
+    userAgent
+  }: StreamProps) {
+    this.channel = channel || ''
+    this.filepath = filepath
+    this.line = line
+    this.httpReferrer = httpReferrer || ''
+    this.label = label || ''
+    this.name = name
+    this.quality = quality || ''
+    this.url = url
+    this.userAgent = userAgent || ''
+    this.logo = ''
+    this.broadcastArea = new Collection()
+    this.categories = new Collection()
+    this.languages = new Collection()
+    this.isNSFW = false
+    this.groupTitle = 'Undefined'
+  }
+
+  normalizeURL() {
+    const url = new URL(this.url)
+
+    this.url = url.normalize().toString()
+  }
+
+  clone(): Stream {
+    return Object.assign(Object.create(Object.getPrototypeOf(this)), this)
+  }
+
+  hasName(): boolean {
+    return !!this.name
+  }
+
+  noName(): boolean {
+    return !this.name
+  }
+
+  hasChannel() {
+    return !!this.channel
+  }
+
+  hasCategories(): boolean {
+    return this.categories.notEmpty()
+  }
+
+  noCategories(): boolean {
+    return this.categories.empty()
+  }
+
+  hasCategory(category: Category): boolean {
+    return this.categories.includes((_category: Category) => _category.id === category.id)
+  }
+
+  noLanguages(): boolean {
+    return this.languages.empty()
+  }
+
+  hasLanguage(language: Language): boolean {
+    return this.languages.includes((_language: Language) => _language.code === language.code)
+  }
+
+  noBroadcastArea(): boolean {
+    return this.broadcastArea.empty()
+  }
+
+  isInternational(): boolean {
+    return this.broadcastArea.includes('r/INT')
+  }
+
+  isSFW(): boolean {
+    return this.isNSFW === false
+  }
+
+  getTitle(): string {
+    let title = `${this.name}`
+
+    if (this.quality) {
+      title += ` (${this.quality})`
+    }
+
+    if (this.label) {
+      title += ` [${this.label}]`
+    }
+
+    return title
+  }
+
+  data() {
+    return {
+      channel: this.channel,
+      filepath: this.filepath,
+      httpReferrer: this.httpReferrer,
+      label: this.label,
+      name: this.name,
+      quality: this.quality,
+      url: this.url,
+      userAgent: this.userAgent,
+      line: this.line
+    }
+  }
+
+  toJSON() {
+    return {
+      channel: this.channel,
+      url: this.url,
+      http_referrer: this.httpReferrer || null,
+      user_agent: this.userAgent || null
+    }
+  }
+
+  toString(options: { public: boolean }) {
+    let output = `#EXTINF:-1 tvg-id="${this.channel}"`
+
+    if (options.public) {
+      output += ` tvg-logo="${this.logo}" group-title="${this.groupTitle}"`
+    }
+
+    if (this.userAgent) {
+      output += ` user-agent="${this.userAgent}"`
+    }
+
+    output += `,${this.getTitle()}`
+
+    if (this.httpReferrer) {
+      output += `\n#EXTVLCOPT:http-referrer=${this.httpReferrer}`
+    }
+
+    if (this.userAgent) {
+      output += `\n#EXTVLCOPT:http-user-agent=${this.userAgent}`
+    }
+
+    output += `\n${this.url}`
+
+    return output
+  }
+}