import { Transaction } from 'sequelize' import { logger } from '@server/helpers/logger' import { CONFIG } from '@server/initializers/config' import { sequelizeTypescript } from '@server/initializers/database' import { VideoModel } from '@server/models/video/video' import { VideoJobInfoModel } from '@server/models/video/video-job-info' import { MVideo, MVideoFullLight, MVideoUUID } from '@server/types/models' import { VideoState } from '@shared/models' import { federateVideoIfNeeded } from './activitypub/videos' import { Notifier } from './notifier' import { addMoveToObjectStorageJob } from './video' import { retryTransactionWrapper } from '@server/helpers/database-utils' function buildNextVideoState (currentState?: VideoState) { if (currentState === VideoState.PUBLISHED) { throw new Error('Video is already in its final state') } if ( currentState !== VideoState.TO_EDIT && currentState !== VideoState.TO_TRANSCODE && currentState !== VideoState.TO_MOVE_TO_EXTERNAL_STORAGE && CONFIG.TRANSCODING.ENABLED ) { return VideoState.TO_TRANSCODE } if ( currentState !== VideoState.TO_MOVE_TO_EXTERNAL_STORAGE && CONFIG.OBJECT_STORAGE.ENABLED ) { return VideoState.TO_MOVE_TO_EXTERNAL_STORAGE } return VideoState.PUBLISHED } function moveToNextState (options: { video: MVideoUUID previousVideoState?: VideoState isNewVideo?: boolean // Default true }) { const { video, previousVideoState, isNewVideo = true } = options return retryTransactionWrapper(() => { return sequelizeTypescript.transaction(async t => { // Maybe the video changed in database, refresh it const videoDatabase = await VideoModel.loadFull(video.uuid, t) // Video does not exist anymore if (!videoDatabase) return undefined // Already in its final state if (videoDatabase.state === VideoState.PUBLISHED) { return federateVideoIfNeeded(videoDatabase, false, t) } const newState = buildNextVideoState(videoDatabase.state) if (newState === VideoState.PUBLISHED) { return moveToPublishedState({ video: videoDatabase, previousVideoState, isNewVideo, transaction: t }) } if (newState === VideoState.TO_MOVE_TO_EXTERNAL_STORAGE) { return moveToExternalStorageState({ video: videoDatabase, isNewVideo, transaction: t }) } }) }) } async function moveToExternalStorageState (options: { video: MVideoFullLight isNewVideo: boolean transaction: Transaction }) { const { video, isNewVideo, transaction } = options const videoJobInfo = await VideoJobInfoModel.load(video.id, transaction) const pendingTranscode = videoJobInfo?.pendingTranscode || 0 // We want to wait all transcoding jobs before moving the video on an external storage if (pendingTranscode !== 0) return false const previousVideoState = video.state await video.setNewState(VideoState.TO_MOVE_TO_EXTERNAL_STORAGE, isNewVideo, transaction) logger.info('Creating external storage move job for video %s.', video.uuid, { tags: [ video.uuid ] }) try { await addMoveToObjectStorageJob({ video, previousVideoState, isNewVideo }) return true } catch (err) { logger.error('Cannot add move to object storage job', { err }) return false } } function moveToFailedTranscodingState (video: MVideo) { if (video.state === VideoState.TRANSCODING_FAILED) return return video.setNewState(VideoState.TRANSCODING_FAILED, false, undefined) } function moveToFailedMoveToObjectStorageState (video: MVideo) { if (video.state === VideoState.TO_MOVE_TO_EXTERNAL_STORAGE_FAILED) return return video.setNewState(VideoState.TO_MOVE_TO_EXTERNAL_STORAGE_FAILED, false, undefined) } // --------------------------------------------------------------------------- export { buildNextVideoState, moveToExternalStorageState, moveToFailedTranscodingState, moveToFailedMoveToObjectStorageState, moveToNextState } // --------------------------------------------------------------------------- async function moveToPublishedState (options: { video: MVideoFullLight isNewVideo: boolean transaction: Transaction previousVideoState?: VideoState }) { const { video, isNewVideo, transaction, previousVideoState } = options const previousState = previousVideoState ?? video.state logger.info('Publishing video %s.', video.uuid, { isNewVideo, previousState, tags: [ video.uuid ] }) await video.setNewState(VideoState.PUBLISHED, isNewVideo, transaction) await federateVideoIfNeeded(video, isNewVideo, transaction) if (previousState === VideoState.TO_EDIT) { Notifier.Instance.notifyOfFinishedVideoStudioEdition(video) return } if (isNewVideo) { Notifier.Instance.notifyOnNewVideoIfNeeded(video) if (previousState === VideoState.TO_TRANSCODE) { Notifier.Instance.notifyOnVideoPublishedAfterTranscoding(video) } } }