scripts/update-host.ts

   1 import { updateTorrentMetadata } from '@server/helpers/webtorrent'
   2 import { getServerActor } from '@server/models/application/application'
   3 import { WEBSERVER } from '../server/initializers/constants'
   4 import { initDatabaseModels } from '../server/initializers/database'
   5 import {
   6   getLocalAccountActivityPubUrl,
   7   getLocalVideoActivityPubUrl,
   8   getLocalVideoAnnounceActivityPubUrl,
   9   getLocalVideoChannelActivityPubUrl,
  10   getLocalVideoCommentActivityPubUrl
  11 } from '../server/lib/activitypub/url'
  12 import { AccountModel } from '../server/models/account/account'
  13 import { ActorModel } from '../server/models/actor/actor'
  14 import { ActorFollowModel } from '../server/models/actor/actor-follow'
  15 import { VideoModel } from '../server/models/video/video'
  16 import { VideoChannelModel } from '../server/models/video/video-channel'
  17 import { VideoCommentModel } from '../server/models/video/video-comment'
  18 import { VideoShareModel } from '../server/models/video/video-share'
  19
  20 run()
  21   .then(() => process.exit(0))
  22   .catch(err => {
  23     console.error(err)
  24     process.exit(-1)
  25   })
  26
  27 async function run () {
  28   await initDatabaseModels(true)
  29
  30   const serverAccount = await getServerActor()
  31
  32   {
  33     const res = await ActorFollowModel.listAcceptedFollowingUrlsForApi([ serverAccount.id ], undefined)
  34     const hasFollowing = res.total > 0
  35
  36     if (hasFollowing === true) {
  37       throw new Error('Cannot update host because you follow other servers!')
  38     }
  39   }
  40
  41   console.log('Updating actors.')
  42
  43   const actors: ActorModel[] = await ActorModel.unscoped().findAll({
  44     include: [
  45       {
  46         model: VideoChannelModel.unscoped(),
  47         required: false
  48       },
  49       {
  50         model: AccountModel.unscoped(),
  51         required: false
  52       }
  53     ]
  54   })
  55   for (const actor of actors) {
  56     if (actor.isOwned() === false) continue
  57
  58     console.log('Updating actor ' + actor.url)
  59
  60     const newUrl = actor.Account
  61       ? getLocalAccountActivityPubUrl(actor.preferredUsername)
  62       : getLocalVideoChannelActivityPubUrl(actor.preferredUsername)
  63
  64     actor.url = newUrl
  65     actor.inboxUrl = newUrl + '/inbox'
  66     actor.outboxUrl = newUrl + '/outbox'
  67     actor.sharedInboxUrl = WEBSERVER.URL + '/inbox'
  68     actor.followersUrl = newUrl + '/followers'
  69     actor.followingUrl = newUrl + '/following'
  70
  71     await actor.save()
  72   }
  73
  74   console.log('Updating video shares.')
  75
  76   const videoShares: VideoShareModel[] = await VideoShareModel.findAll({
  77     include: [ VideoModel.unscoped(), ActorModel.unscoped() ]
  78   })
  79   for (const videoShare of videoShares) {
  80     if (videoShare.Video.isOwned() === false) continue
  81
  82     console.log('Updating video share ' + videoShare.url)
  83
  84     videoShare.url = getLocalVideoAnnounceActivityPubUrl(videoShare.Actor, videoShare.Video)
  85     await videoShare.save()
  86   }
  87
  88   console.log('Updating video comments.')
  89   const videoComments: VideoCommentModel[] = await VideoCommentModel.findAll({
  90     include: [
  91       {
  92         model: VideoModel.unscoped()
  93       },
  94       {
  95         model: AccountModel.unscoped(),
  96         include: [
  97           {
  98             model: ActorModel.unscoped()
  99           }
 100         ]
 101       }
 102     ]
 103   })
 104   for (const comment of videoComments) {
 105     if (comment.isOwned() === false) continue
 106
 107     console.log('Updating comment ' + comment.url)
 108
 109     comment.url = getLocalVideoCommentActivityPubUrl(comment.Video, comment)
 110     await comment.save()
 111   }
 112
 113   console.log('Updating video and torrent files.')
 114
 115   const ids = await VideoModel.listLocalIds()
 116   for (const id of ids) {
 117     const video = await VideoModel.loadFull(id)
 118
 119     console.log('Updating video ' + video.uuid)
 120
 121     video.url = getLocalVideoActivityPubUrl(video)
 122     await video.save()
 123
 124     for (const file of video.VideoFiles) {
 125       console.log('Updating torrent file %s of video %s.', file.resolution, video.uuid)
 126       await updateTorrentMetadata(video, file)
 127
 128       await file.save()
 129     }
 130
 131     const playlist = video.getHLSPlaylist()
 132     for (const file of (playlist?.VideoFiles || [])) {
 133       console.log('Updating fragmented torrent file %s of video %s.', file.resolution, video.uuid)
 134
 135       await updateTorrentMetadata(playlist, file)
 136
 137       await file.save()
 138     }
 139   }
 140 }