server/lib/object-storage/videos.ts

   1 import { join } from 'path'
   2 import { logger } from '@server/helpers/logger'
   3 import { CONFIG } from '@server/initializers/config'
   4 import { MStreamingPlaylistVideo, MVideoFile } from '@server/types/models'
   5 import { getHLSDirectory } from '../paths'
   6 import { generateHLSObjectBaseStorageKey, generateHLSObjectStorageKey, generateWebTorrentObjectStorageKey } from './keys'
   7 import { lTags, makeAvailable, removeObject, removePrefix, storeObject } from './shared'
   8
   9 function storeHLSFile (playlist: MStreamingPlaylistVideo, filename: string, path?: string) {
  10   return storeObject({
  11     inputPath: path ?? join(getHLSDirectory(playlist.Video), filename),
  12     objectStorageKey: generateHLSObjectStorageKey(playlist, filename),
  13     bucketInfo: CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS
  14   })
  15 }
  16
  17 function storeWebTorrentFile (filename: string) {
  18   return storeObject({
  19     inputPath: join(CONFIG.STORAGE.VIDEOS_DIR, filename),
  20     objectStorageKey: generateWebTorrentObjectStorageKey(filename),
  21     bucketInfo: CONFIG.OBJECT_STORAGE.VIDEOS
  22   })
  23 }
  24
  25 function removeHLSObjectStorage (playlist: MStreamingPlaylistVideo) {
  26   return removePrefix(generateHLSObjectBaseStorageKey(playlist), CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS)
  27 }
  28
  29 function removeHLSFileObjectStorage (playlist: MStreamingPlaylistVideo, filename: string) {
  30   return removeObject(generateHLSObjectStorageKey(playlist, filename), CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS)
  31 }
  32
  33 function removeWebTorrentObjectStorage (videoFile: MVideoFile) {
  34   return removeObject(generateWebTorrentObjectStorageKey(videoFile.filename), CONFIG.OBJECT_STORAGE.VIDEOS)
  35 }
  36
  37 async function makeHLSFileAvailable (playlist: MStreamingPlaylistVideo, filename: string, destination: string) {
  38   const key = generateHLSObjectStorageKey(playlist, filename)
  39
  40   logger.info('Fetching HLS file %s from object storage to %s.', key, destination, lTags())
  41
  42   await makeAvailable({
  43     key,
  44     destination,
  45     bucketInfo: CONFIG.OBJECT_STORAGE.STREAMING_PLAYLISTS
  46   })
  47
  48   return destination
  49 }
  50
  51 async function makeWebTorrentFileAvailable (filename: string, destination: string) {
  52   const key = generateWebTorrentObjectStorageKey(filename)
  53
  54   logger.info('Fetching WebTorrent file %s from object storage to %s.', key, destination, lTags())
  55
  56   await makeAvailable({
  57     key,
  58     destination,
  59     bucketInfo: CONFIG.OBJECT_STORAGE.VIDEOS
  60   })
  61
  62   return destination
  63 }
  64
  65 export {
  66   storeWebTorrentFile,
  67   storeHLSFile,
  68
  69   removeHLSObjectStorage,
  70   removeHLSFileObjectStorage,
  71   removeWebTorrentObjectStorage,
  72
  73   makeWebTorrentFileAvailable,
  74   makeHLSFileAvailable
  75 }