aboutsummaryrefslogblamecommitdiffhomepage
path: root/client/src/assets/player/shared/p2p-media-loader/redundancy-url-manager.ts
blob: 376efb83572ab968ac240f3967142c701d44488d (plain) (tree)
1
2
3
4
5
6
7
8
9
10
                                        
                                             


                            




                                                 
                                                                    





                                                                                   
                          







                                                         
                                                 















                                                                              
import { basename, dirname } from 'path'
import { logger } from '@root-helpers/logger'

class RedundancyUrlManager {

  constructor (private baseUrls: string[] = []) {
    // empty
  }

  removeBySegmentUrl (segmentUrl: string) {
    logger.info(`Removing redundancy of segment URL ${segmentUrl}.`)

    const baseUrl = dirname(segmentUrl)

    this.baseUrls = this.baseUrls.filter(u => u !== baseUrl && u !== baseUrl + '/')
  }

  buildUrl (url: string) {
    const max = this.baseUrls.length + 1
    const i = this.getRandomInt(max)

    if (i === max - 1) return url

    const newBaseUrl = this.baseUrls[i]
    const slashPart = newBaseUrl.endsWith('/') ? '' : '/'

    return newBaseUrl + slashPart + basename(url)
  }

  countBaseUrls () {
    return this.baseUrls.length
  }

  private getRandomInt (max: number) {
    return Math.floor(Math.random() * Math.floor(max))
  }
}

// ---------------------------------------------------------------------------

export {
  RedundancyUrlManager
}