server/lib/activitypub/videos/shared/creator.ts

   1
   2 import { logger, loggerTagsFactory, LoggerTagsFn } from '@server/helpers/logger'
   3 import { sequelizeTypescript } from '@server/initializers/database'
   4 import { autoBlacklistVideoIfNeeded } from '@server/lib/video-blacklist'
   5 import { VideoModel } from '@server/models/video/video'
   6 import { MThumbnail, MVideoFullLight, MVideoThumbnail } from '@server/types/models'
   7 import { VideoObject } from '@shared/models'
   8 import { APVideoAbstractBuilder } from './abstract-builder'
   9 import { getVideoAttributesFromObject } from './object-to-model-attributes'
  10
  11 export class APVideoCreator extends APVideoAbstractBuilder {
  12   protected lTags: LoggerTagsFn
  13
  14   constructor (protected readonly videoObject: VideoObject) {
  15     super()
  16
  17     this.lTags = loggerTagsFactory('ap', 'video', 'create', this.videoObject.uuid, this.videoObject.id)
  18   }
  19
  20   async create (waitThumbnail = false) {
  21     logger.debug('Adding remote video %s.', this.videoObject.id, this.lTags())
  22
  23     const channelActor = await this.getOrCreateVideoChannelFromVideoObject()
  24     const channel = channelActor.VideoChannel
  25
  26     const videoData = getVideoAttributesFromObject(channel, this.videoObject, this.videoObject.to)
  27     const video = VideoModel.build(videoData) as MVideoThumbnail
  28
  29     const promiseThumbnail = this.tryToGenerateThumbnail(video)
  30
  31     let thumbnailModel: MThumbnail
  32     if (waitThumbnail === true) {
  33       thumbnailModel = await promiseThumbnail
  34     }
  35
  36     const { autoBlacklisted, videoCreated } = await sequelizeTypescript.transaction(async t => {
  37       try {
  38         const videoCreated = await video.save({ transaction: t }) as MVideoFullLight
  39         videoCreated.VideoChannel = channel
  40
  41         if (thumbnailModel) await videoCreated.addAndSaveThumbnail(thumbnailModel, t)
  42
  43         await this.setPreview(videoCreated, t)
  44         await this.setWebTorrentFiles(videoCreated, t)
  45         await this.setStreamingPlaylists(videoCreated, t)
  46         await this.setTags(videoCreated, t)
  47         await this.setTrackers(videoCreated, t)
  48         await this.insertOrReplaceCaptions(videoCreated, t)
  49         await this.insertOrReplaceLive(videoCreated, t)
  50
  51         // We added a video in this channel, set it as updated
  52         await channel.setAsUpdated(t)
  53
  54         const autoBlacklisted = await autoBlacklistVideoIfNeeded({
  55           video: videoCreated,
  56           user: undefined,
  57           isRemote: true,
  58           isNew: true,
  59           transaction: t
  60         })
  61
  62         logger.info('Remote video with uuid %s inserted.', this.videoObject.uuid, this.lTags())
  63
  64         return { autoBlacklisted, videoCreated }
  65       } catch (err) {
  66         // FIXME: Use rollback hook when https://github.com/sequelize/sequelize/pull/13038 is released
  67         if (thumbnailModel) await thumbnailModel.removeThumbnail()
  68
  69         throw err
  70       }
  71     })
  72
  73     if (waitThumbnail === false) {
  74       // Error is already caught above
  75       // eslint-disable-next-line @typescript-eslint/no-floating-promises
  76       promiseThumbnail.then(thumbnailModel => {
  77         if (!thumbnailModel) return
  78
  79         thumbnailModel = videoCreated.id
  80
  81         return thumbnailModel.save()
  82       })
  83     }
  84
  85     return { autoBlacklisted, videoCreated }
  86   }
  87 }