server/lib/video-paths.ts

   1 import { isStreamingPlaylist, MStreamingPlaylistVideo, MVideo, MVideoFile, MVideoUUID } from '@server/types/models'
   2 import { join } from 'path'
   3 import { CONFIG } from '@server/initializers/config'
   4 import { HLS_REDUNDANCY_DIRECTORY, HLS_STREAMING_PLAYLIST_DIRECTORY } from '@server/initializers/constants'
   5 import { extractVideo } from '@server/helpers/video'
   6
   7 // ################## Video file name ##################
   8
   9 function getVideoFilename (videoOrPlaylist: MVideo | MStreamingPlaylistVideo, videoFile: MVideoFile) {
  10   const video = extractVideo(videoOrPlaylist)
  11
  12   if (videoFile.isHLS()) {
  13     return generateVideoStreamingPlaylistName(video.uuid, videoFile.resolution)
  14   }
  15
  16   return generateWebTorrentVideoName(video.uuid, videoFile.resolution, videoFile.extname)
  17 }
  18
  19 function generateVideoStreamingPlaylistName (uuid: string, resolution: number) {
  20   return `${uuid}-${resolution}-fragmented.mp4`
  21 }
  22
  23 function generateWebTorrentVideoName (uuid: string, resolution: number, extname: string) {
  24   return uuid + '-' + resolution + extname
  25 }
  26
  27 function getVideoFilePath (videoOrPlaylist: MVideo | MStreamingPlaylistVideo, videoFile: MVideoFile, isRedundancy = false) {
  28   if (videoFile.isHLS()) {
  29     const video = extractVideo(videoOrPlaylist)
  30
  31     return join(getHLSDirectory(video), getVideoFilename(videoOrPlaylist, videoFile))
  32   }
  33
  34   const baseDir = isRedundancy ? CONFIG.STORAGE.REDUNDANCY_DIR : CONFIG.STORAGE.VIDEOS_DIR
  35   return join(baseDir, getVideoFilename(videoOrPlaylist, videoFile))
  36 }
  37
  38 // ################## Streaming playlist ##################
  39
  40 function getHLSDirectory (video: MVideoUUID, isRedundancy = false) {
  41   const baseDir = isRedundancy ? HLS_REDUNDANCY_DIRECTORY : HLS_STREAMING_PLAYLIST_DIRECTORY
  42
  43   return join(baseDir, video.uuid)
  44 }
  45
  46 // ################## Torrents ##################
  47
  48 function getTorrentFileName (videoOrPlaylist: MVideo | MStreamingPlaylistVideo, videoFile: MVideoFile) {
  49   const video = extractVideo(videoOrPlaylist)
  50   const extension = '.torrent'
  51
  52   if (isStreamingPlaylist(videoOrPlaylist)) {
  53     return `${video.uuid}-${videoFile.resolution}-${videoOrPlaylist.getStringType()}${extension}`
  54   }
  55
  56   return video.uuid + '-' + videoFile.resolution + extension
  57 }
  58
  59 function getTorrentFilePath (videoOrPlaylist: MVideo | MStreamingPlaylistVideo, videoFile: MVideoFile) {
  60   return join(CONFIG.STORAGE.TORRENTS_DIR, getTorrentFileName(videoOrPlaylist, videoFile))
  61 }
  62
  63 // ---------------------------------------------------------------------------
  64
  65 export {
  66   generateVideoStreamingPlaylistName,
  67   generateWebTorrentVideoName,
  68   getVideoFilename,
  69   getVideoFilePath,
  70
  71   getTorrentFileName,
  72   getTorrentFilePath,
  73
  74   getHLSDirectory
  75 }