1 import { Job } from 'bullmq'
2 import { extractVideo } from '@server/helpers/video'
3 import { createTorrentAndSetInfoHash, updateTorrentMetadata } from '@server/helpers/webtorrent'
4 import { VideoPathManager } from '@server/lib/video-path-manager'
5 import { VideoModel } from '@server/models/video/video'
6 import { VideoFileModel } from '@server/models/video/video-file'
7 import { VideoStreamingPlaylistModel } from '@server/models/video/video-streaming-playlist'
8 import { ManageVideoTorrentPayload } from '@shared/models'
9 import { logger } from '../../../helpers/logger'
11 async function processManageVideoTorrent (job: Job) {
12 const payload = job.data as ManageVideoTorrentPayload
13 logger.info('Processing torrent in job %s.', job.id)
15 if (payload.action === 'create') return doCreateAction(payload)
16 if (payload.action === 'update-metadata') return doUpdateMetadataAction(payload)
19 // ---------------------------------------------------------------------------
22 processManageVideoTorrent
25 // ---------------------------------------------------------------------------
27 async function doCreateAction (payload: ManageVideoTorrentPayload & { action: 'create' }) {
28 const [ video, file ] = await Promise.all([
29 loadVideoOrLog(payload.videoId),
30 loadFileOrLog(payload.videoFileId)
33 if (!video || !file) return
35 const fileMutexReleaser = await VideoPathManager.Instance.lockFiles(video.uuid)
38 await createTorrentAndSetInfoHash(video, file)
40 // Refresh videoFile because the createTorrentAndSetInfoHash could be long
41 const refreshedFile = await VideoFileModel.loadWithVideo(file.id)
42 // File does not exist anymore, remove the generated torrent
43 if (!refreshedFile) return file.removeTorrent()
45 refreshedFile.infoHash = file.infoHash
46 refreshedFile.torrentFilename = file.torrentFilename
48 await refreshedFile.save()
54 async function doUpdateMetadataAction (payload: ManageVideoTorrentPayload & { action: 'update-metadata' }) {
55 const [ video, streamingPlaylist, file ] = await Promise.all([
56 loadVideoOrLog(payload.videoId),
57 loadStreamingPlaylistOrLog(payload.streamingPlaylistId),
58 loadFileOrLog(payload.videoFileId)
61 if ((!video && !streamingPlaylist) || !file) return
63 const extractedVideo = extractVideo(video || streamingPlaylist)
64 const fileMutexReleaser = await VideoPathManager.Instance.lockFiles(extractedVideo.uuid)
67 await updateTorrentMetadata(video || streamingPlaylist, file)
75 async function loadVideoOrLog (videoId: number) {
76 if (!videoId) return undefined
78 const video = await VideoModel.load(videoId)
80 logger.debug('Do not process torrent for video %d: does not exist anymore.', videoId)
86 async function loadStreamingPlaylistOrLog (streamingPlaylistId: number) {
87 if (!streamingPlaylistId) return undefined
89 const streamingPlaylist = await VideoStreamingPlaylistModel.loadWithVideo(streamingPlaylistId)
90 if (!streamingPlaylist) {
91 logger.debug('Do not process torrent for streaming playlist %d: does not exist anymore.', streamingPlaylistId)
94 return streamingPlaylist
97 async function loadFileOrLog (videoFileId: number) {
98 if (!videoFileId) return undefined
100 const file = await VideoFileModel.load(videoFileId)
103 logger.debug('Do not process torrent for file %d: does not exist anymore.', videoFileId)