video-state.ts 3.63 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
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 { MVideoFullLight, MVideoUUID } from '@server/types/models'
import { VideoState } from '@shared/models'
import { federateVideoIfNeeded } from './activitypub/videos'
import { Notifier } from './notifier'
import { addMoveToObjectStorageJob } from './video'

function buildNextVideoState (currentState?: VideoState) {
  if (currentState === VideoState.PUBLISHED) {
    throw new Error('Video is already in its final state')
  }

  if (
    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 (video: MVideoUUID, isNewVideo = true) {
  return sequelizeTypescript.transaction(async t => {
    // Maybe the video changed in database, refresh it
    const videoDatabase = await VideoModel.loadAndPopulateAccountAndServerAndTags(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(videoDatabase, isNewVideo, t)
    }

    if (newState === VideoState.TO_MOVE_TO_EXTERNAL_STORAGE) {
      return moveToExternalStorageState(videoDatabase, isNewVideo, t)
    }
  })
}

60
61
62
63
64
async function moveToExternalStorageState (video: MVideoFullLight, isNewVideo: boolean, transaction: Transaction) {
  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
Chocobozzz's avatar
Chocobozzz committed
65
  if (pendingTranscode !== 0) return false
66
67
68
69
70
71
72

  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, isNewVideo)
Chocobozzz's avatar
Chocobozzz committed
73
74

    return true
75
76
  } catch (err) {
    logger.error('Cannot add move to object storage job', { err })
Chocobozzz's avatar
Chocobozzz committed
77
78

    return false
79
80
81
  }
}

82
83
84
85
// ---------------------------------------------------------------------------

export {
  buildNextVideoState,
86
  moveToExternalStorageState,
87
88
89
90
91
92
93
94
95
  moveToNextState
}

// ---------------------------------------------------------------------------

async function moveToPublishedState (video: MVideoFullLight, isNewVideo: boolean, transaction: Transaction) {
  logger.info('Publishing video %s.', video.uuid, { tags: [ video.uuid ] })

  const previousState = video.state
96
  await video.setNewState(VideoState.PUBLISHED, isNewVideo, transaction)
97
98
99
100
101

  // If the video was not published, we consider it is a new one for other instances
  // Live videos are always federated, so it's not a new video
  await federateVideoIfNeeded(video, isNewVideo, transaction)

102
  if (isNewVideo) Notifier.Instance.notifyOnNewVideoIfNeeded(video)
103
104
105
106
107

  if (previousState === VideoState.TO_TRANSCODE) {
    Notifier.Instance.notifyOnVideoPublishedAfterTranscoding(video)
  }
}