streaming-playlists-command.ts

   1 import { wait } from '@shared/core-utils'
   2 import { HttpStatusCode } from '@shared/models'
   3 import { unwrapBody, unwrapBodyOrDecodeToJSON, unwrapTextOrDecode } from '../requests'
   4 import { AbstractCommand, OverrideCommandOptions } from '../shared'
   5
   6 export class StreamingPlaylistsCommand extends AbstractCommand {
   7
   8   async get (options: OverrideCommandOptions & {
   9     url: string
  10
  11     videoFileToken?: string
  12     reinjectVideoFileToken?: boolean
  13
  14     withRetry?: boolean // default false
  15     currentRetry?: number
  16   }): Promise<string> {
  17     const { videoFileToken, reinjectVideoFileToken, withRetry, currentRetry = 1 } = options
  18
  19     try {
  20       const result = await unwrapTextOrDecode(this.getRawRequest({
  21         ...options,
  22
  23         url: options.url,
  24         query: {
  25           videoFileToken,
  26           reinjectVideoFileToken
  27         },
  28         implicitToken: false,
  29         defaultExpectedStatus: HttpStatusCode.OK_200
  30       }))
  31
  32       return result
  33     } catch (err) {
  34       if (!withRetry || currentRetry > 5) throw err
  35
  36       await wait(100)
  37
  38       return this.get({
  39         ...options,
  40
  41         withRetry,
  42         currentRetry: currentRetry + 1
  43       })
  44     }
  45   }
  46
  47   getFragmentedSegment (options: OverrideCommandOptions & {
  48     url: string
  49     range?: string
  50   }) {
  51     return unwrapBody<Buffer>(this.getRawRequest({
  52       ...options,
  53
  54       url: options.url,
  55       range: options.range,
  56       implicitToken: false,
  57       responseType: 'application/octet-stream',
  58       defaultExpectedStatus: HttpStatusCode.OK_200
  59     }))
  60   }
  61
  62   getSegmentSha256 (options: OverrideCommandOptions & {
  63     url: string
  64   }) {
  65     return unwrapBodyOrDecodeToJSON<{ [ id: string ]: string }>(this.getRawRequest({
  66       ...options,
  67
  68       url: options.url,
  69       contentType: 'application/json',
  70       implicitToken: false,
  71       defaultExpectedStatus: HttpStatusCode.OK_200
  72     }))
  73   }
  74 }