Update scripts

scripts/models/stream.ts

@@ -0,0 +1,58 @@
+import type { StreamData } from '../types/stream'
+import { Feed, Channel } from './index'
+
+export class Stream {
+  name?: string
+  url: string
+  id?: string
+  channelId?: string
+  channel?: Channel
+  feedId?: string
+  feed?: Feed
+  filepath?: string
+  line?: number
+  label?: string
+  verticalResolution?: number
+  isInterlaced?: boolean
+  referrer?: string
+  userAgent?: string
+  groupTitle: string = 'Undefined'
+  removed: boolean = false
+
+  constructor(data: StreamData) {
+    const id = data.channel && data.feed ? [data.channel, data.feed].join('@') : data.channel
+    const { verticalResolution, isInterlaced } = parseQuality(data.quality)
+
+    this.id = id || undefined
+    this.channelId = data.channel || undefined
+    this.feedId = data.feed || undefined
+    this.name = data.name || undefined
+    this.url = data.url
+    this.referrer = data.referrer || undefined
+    this.userAgent = data.user_agent || undefined
+    this.verticalResolution = verticalResolution || undefined
+    this.isInterlaced = isInterlaced || undefined
+    this.label = data.label || undefined
+  }
+
+  getId(): string {
+    return this.id || ''
+  }
+
+  getName(): string {
+    return this.name || ''
+  }
+}
+
+function parseQuality(quality: string | null): {
+  verticalResolution: number | null
+  isInterlaced: boolean | null
+} {
+  if (!quality) return { verticalResolution: null, isInterlaced: null }
+  const [, verticalResolutionString] = quality.match(/^(\d+)/) || [null, undefined]
+  const isInterlaced = /i$/i.test(quality)
+  let verticalResolution = 0
+  if (verticalResolutionString) verticalResolution = parseInt(verticalResolutionString)
+
+  return { verticalResolution, isInterlaced }
+}