1
0
Fork 0
peertube/client/src/assets/player/shared/p2p-media-loader/redundancy-url-manager.ts

44 lines
998 B
TypeScript

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
}