mirror of https://github.com/Chocobozzz/PeerTube
93 lines
2.8 KiB
TypeScript
93 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!')
|
|
}
|