1
0
Fork 0
peertube/server/scripts/house-keeping.ts
2024-06-06 11:33:09 +02:00

92 lines
2.8 KiB
TypeScript

import { createCommand } from '@commander-js/extra-typings'
import { initDatabaseModels } from '@server/initializers/database.js'
import { ActorImageModel } from '@server/models/actor/actor-image.js'
import { ThumbnailModel } from '@server/models/video/thumbnail.js'
import Bluebird from 'bluebird'
import { askConfirmation, displayPeerTubeMustBeStoppedWarning } from './shared/common.js'
const program = createCommand()
.description('Remove unused objects from database or remote files')
.option('--delete-remote-files', 'Remove remote files (avatars, banners, thumbnails...)')
.parse(process.argv)
const options = program.opts()
if (!options.deleteRemoteFiles) {
console.log('At least one option must be set (for example --delete-remote-files).')
process.exit(0)
}
run()
.then(() => process.exit(0))
.catch(err => {
console.error(err)
process.exit(-1)
})
async function run () {
await initDatabaseModels(true)
displayPeerTubeMustBeStoppedWarning()
if (options.deleteRemoteFiles) {
return deleteRemoteFiles()
}
}
async function deleteRemoteFiles () {
console.log('Detecting remote files that can be deleted...')
const thumbnails = await ThumbnailModel.listRemoteOnDisk()
const actorImages = await ActorImageModel.listRemoteOnDisk()
if (thumbnails.length === 0 && actorImages.length === 0) {
console.log('No remote files to delete detected.')
process.exit(0)
}
const res = await askConfirmation(
`${thumbnails.length.toLocaleString()} thumbnails and ${actorImages.length.toLocaleString()} avatars/banners can be locally deleted. ` +
`PeerTube will download them again on-demand. ` +
`Do you want to delete these remote files?`
)
if (res !== true) {
console.log('Exiting without delete remote files.')
process.exit(0)
}
// ---------------------------------------------------------------------------
console.log('Deleting remote thumbnails...')
await Bluebird.map(thumbnails, async thumbnail => {
if (!thumbnail.fileUrl) {
console.log(`Skipping thumbnail removal of ${thumbnail.getPath()} as we don't have its remote file URL in the database.`)
return
}
await thumbnail.removeThumbnail()
thumbnail.onDisk = false
await thumbnail.save()
}, { concurrency: 20 })
// ---------------------------------------------------------------------------
console.log('Deleting remote avatars/banners...')
await Bluebird.map(actorImages, async actorImage => {
if (!actorImage.fileUrl) {
console.log(`Skipping avatar/banner removal of ${actorImage.getPath()} as we don't have its remote file URL in the database.`)
return
}
await actorImage.removeImage()
actorImage.onDisk = false
await actorImage.save()
}, { concurrency: 20 })
console.log('Remote files deleted!')
}