server/lib/files-cache/videos-torrent-cache.ts

   1 import { join } from 'path'
   2 import { logger } from '@server/helpers/logger'
   3 import { doRequestAndSaveToFile } from '@server/helpers/requests'
   4 import { VideoFileModel } from '@server/models/video/video-file'
   5 import { MVideo, MVideoFile } from '@server/types/models'
   6 import { CONFIG } from '../../initializers/config'
   7 import { FILES_CACHE } from '../../initializers/constants'
   8 import { VideoModel } from '../../models/video/video'
   9 import { AbstractVideoStaticFileCache } from './abstract-video-static-file-cache'
  10
  11 class VideosTorrentCache extends AbstractVideoStaticFileCache <string> {
  12
  13   private static instance: VideosTorrentCache
  14
  15   private constructor () {
  16     super()
  17   }
  18
  19   static get Instance () {
  20     return this.instance || (this.instance = new this())
  21   }
  22
  23   async getFilePathImpl (filename: string) {
  24     const file = await VideoFileModel.loadWithVideoOrPlaylistByTorrentFilename(filename)
  25     if (!file) return undefined
  26
  27     if (file.getVideo().isOwned()) {
  28       const downloadName = this.buildDownloadName(file.getVideo(), file)
  29
  30       return { isOwned: true, path: join(CONFIG.STORAGE.TORRENTS_DIR, file.torrentFilename), downloadName }
  31     }
  32
  33     return this.loadRemoteFile(filename)
  34   }
  35
  36   // Key is the torrent filename
  37   protected async loadRemoteFile (key: string) {
  38     const file = await VideoFileModel.loadWithVideoOrPlaylistByTorrentFilename(key)
  39     if (!file) return undefined
  40
  41     if (file.getVideo().isOwned()) throw new Error('Cannot load remote file of owned video.')
  42
  43     // Used to fetch the path
  44     const video = await VideoModel.loadFull(file.getVideo().id)
  45     if (!video) return undefined
  46
  47     const remoteUrl = file.getRemoteTorrentUrl(video)
  48     const destPath = join(FILES_CACHE.TORRENTS.DIRECTORY, file.torrentFilename)
  49
  50     try {
  51       await doRequestAndSaveToFile(remoteUrl, destPath)
  52
  53       const downloadName = this.buildDownloadName(video, file)
  54
  55       return { isOwned: false, path: destPath, downloadName }
  56     } catch (err) {
  57       logger.info('Cannot fetch remote torrent file %s.', remoteUrl, { err })
  58
  59       return undefined
  60     }
  61   }
  62
  63   private buildDownloadName (video: MVideo, file: MVideoFile) {
  64     return `${video.name}-${file.resolution}p.torrent`
  65   }
  66 }
  67
  68 export {
  69   VideosTorrentCache
  70 }