mirror of https://github.com/Chocobozzz/PeerTube
72 lines
1.9 KiB
TypeScript
72 lines
1.9 KiB
TypeScript
import { ensureDir } from 'fs-extra/esm'
|
|
import { Op } from 'sequelize'
|
|
import { updateTorrentMetadata } from '@server/helpers/webtorrent.js'
|
|
import { DIRECTORIES } from '@server/initializers/constants.js'
|
|
import { moveFilesIfPrivacyChanged } from '@server/lib/video-privacy.js'
|
|
import { VideoModel } from '@server/models/video/video.js'
|
|
import { MVideoFullLight } from '@server/types/models/index.js'
|
|
import { VideoPrivacy } from '@peertube/peertube-models'
|
|
import { initDatabaseModels } from '@server/initializers/database.js'
|
|
|
|
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()
|
|
}
|
|
}
|