shared/server-commands/videos/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, expectedStatus, withRetry = false, 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       // master.m3u8 could be empty
  33       if (!result && (!expectedStatus || expectedStatus === HttpStatusCode.OK_200)) {
  34         throw new Error('Empty result')
  35       }
  36
  37       return result
  38     } catch (err) {
  39       if (!withRetry || currentRetry > 10) throw err
  40
  41       await wait(250)
  42
  43       return this.get({
  44         ...options,
  45
  46         withRetry,
  47         currentRetry: currentRetry + 1
  48       })
  49     }
  50   }
  51
  52   async getFragmentedSegment (options: OverrideCommandOptions & {
  53     url: string
  54     range?: string
  55
  56     withRetry?: boolean // default false
  57     currentRetry?: number
  58   }) {
  59     const { withRetry = false, currentRetry = 1 } = options
  60
  61     try {
  62       const result = await unwrapBody<Buffer>(this.getRawRequest({
  63         ...options,
  64
  65         url: options.url,
  66         range: options.range,
  67         implicitToken: false,
  68         responseType: 'application/octet-stream',
  69         defaultExpectedStatus: HttpStatusCode.OK_200
  70       }))
  71
  72       return result
  73     } catch (err) {
  74       if (!withRetry || currentRetry > 10) throw err
  75
  76       await wait(250)
  77
  78       return this.getFragmentedSegment({
  79         ...options,
  80
  81         withRetry,
  82         currentRetry: currentRetry + 1
  83       })
  84     }
  85   }
  86
  87   async getSegmentSha256 (options: OverrideCommandOptions & {
  88     url: string
  89
  90     withRetry?: boolean // default false
  91     currentRetry?: number
  92   }) {
  93     const { withRetry = false, currentRetry = 1 } = options
  94
  95     try {
  96       const result = await unwrapBodyOrDecodeToJSON<{ [ id: string ]: string }>(this.getRawRequest({
  97         ...options,
  98
  99         url: options.url,
 100         contentType: 'application/json',
 101         implicitToken: false,
 102         defaultExpectedStatus: HttpStatusCode.OK_200
 103       }))
 104
 105       return result
 106     } catch (err) {
 107       if (!withRetry || currentRetry > 10) throw err
 108
 109       await wait(250)
 110
 111       return this.getSegmentSha256({
 112         ...options,
 113
 114         withRetry,
 115         currentRetry: currentRetry + 1
 116       })
 117     }
 118   }
 119 }