server/lib/job-queue/handlers/manage-video-torrent.ts

   1 import { Job } from 'bull'
   2 import { createTorrentAndSetInfoHash, updateTorrentMetadata } from '@server/helpers/webtorrent'
   3 import { VideoModel } from '@server/models/video/video'
   4 import { VideoFileModel } from '@server/models/video/video-file'
   5 import { VideoStreamingPlaylistModel } from '@server/models/video/video-streaming-playlist'
   6 import { ManageVideoTorrentPayload } from '@shared/models'
   7 import { logger } from '../../../helpers/logger'
   8
   9 async function processManageVideoTorrent (job: Job) {
  10   const payload = job.data as ManageVideoTorrentPayload
  11   logger.info('Processing torrent in job %d.', job.id)
  12
  13   if (payload.action === 'create') return doCreateAction(payload)
  14   if (payload.action === 'update-metadata') return doUpdateMetadataAction(payload)
  15 }
  16
  17 // ---------------------------------------------------------------------------
  18
  19 export {
  20   processManageVideoTorrent
  21 }
  22
  23 // ---------------------------------------------------------------------------
  24
  25 async function doCreateAction (payload: ManageVideoTorrentPayload & { action: 'create' }) {
  26   const [ video, file ] = await Promise.all([
  27     loadVideoOrLog(payload.videoId),
  28     loadFileOrLog(payload.videoFileId)
  29   ])
  30
  31   if (!video || !file) return
  32
  33   await createTorrentAndSetInfoHash(video, file)
  34
  35   // Refresh videoFile because the createTorrentAndSetInfoHash could be long
  36   const refreshedFile = await VideoFileModel.loadWithVideo(file.id)
  37   // File does not exist anymore, remove the generated torrent
  38   if (!refreshedFile) return file.removeTorrent()
  39
  40   refreshedFile.infoHash = file.infoHash
  41   refreshedFile.torrentFilename = file.torrentFilename
  42
  43   return refreshedFile.save()
  44 }
  45
  46 async function doUpdateMetadataAction (payload: ManageVideoTorrentPayload & { action: 'update-metadata' }) {
  47   const [ video, streamingPlaylist, file ] = await Promise.all([
  48     loadVideoOrLog(payload.videoId),
  49     loadStreamingPlaylistOrLog(payload.streamingPlaylistId),
  50     loadFileOrLog(payload.videoFileId)
  51   ])
  52
  53   if ((!video && !streamingPlaylist) || !file) return
  54
  55   await updateTorrentMetadata(video || streamingPlaylist, file)
  56
  57   await file.save()
  58 }
  59
  60 async function loadVideoOrLog (videoId: number) {
  61   if (!videoId) return undefined
  62
  63   const video = await VideoModel.load(videoId)
  64   if (!video) {
  65     logger.debug('Do not process torrent for video %d: does not exist anymore.', videoId)
  66   }
  67
  68   return video
  69 }
  70
  71 async function loadStreamingPlaylistOrLog (streamingPlaylistId: number) {
  72   if (!streamingPlaylistId) return undefined
  73
  74   const streamingPlaylist = await VideoStreamingPlaylistModel.loadWithVideo(streamingPlaylistId)
  75   if (!streamingPlaylist) {
  76     logger.debug('Do not process torrent for streaming playlist %d: does not exist anymore.', streamingPlaylistId)
  77   }
  78
  79   return streamingPlaylist
  80 }
  81
  82 async function loadFileOrLog (videoFileId: number) {
  83   if (!videoFileId) return undefined
  84
  85   const file = await VideoFileModel.loadWithVideo(videoFileId)
  86
  87   if (!file) {
  88     logger.debug('Do not process torrent for file %d: does not exist anymore.', videoFileId)
  89   }
  90
  91   return file
  92 }