Init commit

server/scripts/house-keeping.ts

@@ -0,0 +1,92 @@
+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 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!')
+}