Add migration script

pull/5378/head
Chocobozzz 2022-10-21 11:56:19 +02:00 committed by Chocobozzz
parent 326f36929a
commit fc097052ab
1 changed files with 71 additions and 0 deletions

View File

@ -0,0 +1,71 @@
import { ensureDir } from 'fs-extra'
import { Op } from 'sequelize'
import { updateTorrentMetadata } from '@server/helpers/webtorrent'
import { DIRECTORIES } from '@server/initializers/constants'
import { moveFilesIfPrivacyChanged } from '@server/lib/video-privacy'
import { VideoModel } from '@server/models/video/video'
import { MVideoFullLight } from '@server/types/models'
import { VideoPrivacy } from '@shared/models'
import { initDatabaseModels } from '../../server/initializers/database'
run()
.then(() => process.exit(0))
.catch(err => {
console.error(err)
process.exit(-1)
})
async function run () {
console.log('Moving private video files in dedicated folders.')
await ensureDir(DIRECTORIES.HLS_STREAMING_PLAYLIST.PRIVATE)
await ensureDir(DIRECTORIES.VIDEOS.PRIVATE)
await initDatabaseModels(true)
const videos = await VideoModel.unscoped().findAll({
attributes: [ 'uuid' ],
where: {
privacy: {
[Op.in]: [ VideoPrivacy.PRIVATE, VideoPrivacy.INTERNAL ]
}
}
})
for (const { uuid } of videos) {
try {
console.log('Moving files of video %s.', uuid)
const video = await VideoModel.loadFull(uuid)
try {
await moveFilesIfPrivacyChanged(video, VideoPrivacy.PUBLIC)
} catch (err) {
console.error('Cannot move files of video %s.', uuid, err)
}
try {
await updateTorrents(video)
} catch (err) {
console.error('Cannot regenerate torrents of video %s.', uuid, err)
}
} catch (err) {
console.error('Cannot process video %s.', uuid, err)
}
}
}
async function updateTorrents (video: MVideoFullLight) {
for (const file of video.VideoFiles) {
await updateTorrentMetadata(video, file)
await file.save()
}
const playlist = video.getHLSPlaylist()
for (const file of (playlist?.VideoFiles || [])) {
await updateTorrentMetadata(playlist, file)
await file.save()
}
}