scripts/migrations/peertube-4.2.ts

   1 import { minBy } from 'lodash'
   2 import { join } from 'path'
   3 import { getImageSize, processImage } from '@server/helpers/image-utils'
   4 import { CONFIG } from '@server/initializers/config'
   5 import { ACTOR_IMAGES_SIZE } from '@server/initializers/constants'
   6 import { updateActorImages } from '@server/lib/activitypub/actors'
   7 import { sendUpdateActor } from '@server/lib/activitypub/send'
   8 import { getBiggestActorImage } from '@server/lib/actor-image'
   9 import { JobQueue } from '@server/lib/job-queue'
  10 import { AccountModel } from '@server/models/account/account'
  11 import { ActorModel } from '@server/models/actor/actor'
  12 import { VideoChannelModel } from '@server/models/video/video-channel'
  13 import { MAccountDefault, MActorDefault, MChannelDefault } from '@server/types/models'
  14 import { getLowercaseExtension } from '@shared/core-utils'
  15 import { buildUUID } from '@shared/extra-utils'
  16 import { ActorImageType } from '@shared/models'
  17 import { initDatabaseModels } from '../../server/initializers/database'
  18
  19 run()
  20   .then(() => process.exit(0))
  21   .catch(err => {
  22     console.error(err)
  23     process.exit(-1)
  24   })
  25
  26 async function run () {
  27   console.log('Generate avatar miniatures from existing avatars.')
  28
  29   await initDatabaseModels(true)
  30   JobQueue.Instance.init()
  31
  32   const accounts: AccountModel[] = await AccountModel.findAll({
  33     include: [
  34       {
  35         model: ActorModel,
  36         required: true,
  37         where: {
  38           serverId: null
  39         }
  40       },
  41       {
  42         model: VideoChannelModel,
  43         include: [
  44           {
  45             model: AccountModel
  46           }
  47         ]
  48       }
  49     ]
  50   })
  51
  52   for (const account of accounts) {
  53     try {
  54       await fillAvatarSizeIfNeeded(account)
  55       await generateSmallerAvatarIfNeeded(account)
  56     } catch (err) {
  57       console.error(`Cannot process account avatar ${account.name}`, err)
  58     }
  59
  60     for (const videoChannel of account.VideoChannels) {
  61       try {
  62         await fillAvatarSizeIfNeeded(videoChannel)
  63         await generateSmallerAvatarIfNeeded(videoChannel)
  64       } catch (err) {
  65         console.error(`Cannot process channel avatar ${videoChannel.name}`, err)
  66       }
  67     }
  68   }
  69
  70   console.log('Generation finished!')
  71 }
  72
  73 async function fillAvatarSizeIfNeeded (accountOrChannel: MAccountDefault | MChannelDefault) {
  74   const avatars = accountOrChannel.Actor.Avatars
  75
  76   for (const avatar of avatars) {
  77     if (avatar.width && avatar.height) continue
  78
  79     console.log('Filling size of avatars of %s.', accountOrChannel.name)
  80
  81     const { width, height } = await getImageSize(join(CONFIG.STORAGE.ACTOR_IMAGES, avatar.filename))
  82     avatar.width = width
  83     avatar.height = height
  84
  85     await avatar.save()
  86   }
  87 }
  88
  89 async function generateSmallerAvatarIfNeeded (accountOrChannel: MAccountDefault | MChannelDefault) {
  90   const avatars = accountOrChannel.Actor.Avatars
  91   if (avatars.length !== 1) {
  92     return
  93   }
  94
  95   console.log(`Processing ${accountOrChannel.name}.`)
  96
  97   await generateSmallerAvatar(accountOrChannel.Actor)
  98   accountOrChannel.Actor = Object.assign(accountOrChannel.Actor, { Server: null })
  99
 100   return sendUpdateActor(accountOrChannel, undefined)
 101 }
 102
 103 async function generateSmallerAvatar (actor: MActorDefault) {
 104   const bigAvatar = getBiggestActorImage(actor.Avatars)
 105
 106   const imageSize = minBy(ACTOR_IMAGES_SIZE[ActorImageType.AVATAR], 'width')
 107   const sourceFilename = bigAvatar.filename
 108
 109   const newImageName = buildUUID() + getLowercaseExtension(sourceFilename)
 110   const source = join(CONFIG.STORAGE.ACTOR_IMAGES, sourceFilename)
 111   const destination = join(CONFIG.STORAGE.ACTOR_IMAGES, newImageName)
 112
 113   await processImage({ path: source, destination, newSize: imageSize, keepOriginal: true })
 114
 115   const actorImageInfo = {
 116     name: newImageName,
 117     fileUrl: null,
 118     height: imageSize.height,
 119     width: imageSize.width,
 120     onDisk: true
 121   }
 122
 123   await updateActorImages(actor, ActorImageType.AVATAR, [ actorImageInfo ], undefined)
 124 }