mirror of
https://github.com/Chocobozzz/PeerTube.git
synced 2025-10-03 09:49:20 +02:00
dd52e8b89e
* Implement processing storyboards by runners * Fixed storyboard generation by runners * use common code patterns * fix import * improve debug logging for storyboard generation * config option for storyboard processing with remote-runners * refactor repetitive pattern * refactor storyboard related code to share common utlities * Fix test * Fix storyboard generation config logic * Improve logging * Added tests for storyboard generation with runners * Refactor PR --------- Co-authored-by: ilfarpro <ilfarpro@ya.ru> Co-authored-by: Chocobozzz <me@florianbigard.com>
80 lines
2.1 KiB
TypeScript
80 lines
2.1 KiB
TypeScript
import { toCompleteUUID } from '@server/helpers/custom-validators/misc.js'
|
|
import { initDatabaseModels } from '@server/initializers/database.js'
|
|
import { JobQueue } from '@server/lib/job-queue/index.js'
|
|
import { addLocalOrRemoteStoryboardJobIfNeeded } from '@server/lib/video-jobs.js'
|
|
import { StoryboardModel } from '@server/models/video/storyboard.js'
|
|
import { VideoModel } from '@server/models/video/video.js'
|
|
import { program } from 'commander'
|
|
|
|
program
|
|
.description('Generate videos storyboard')
|
|
.option('-v, --video [videoUUID]', 'Generate the storyboard of a specific video')
|
|
.option('-a, --all-videos', 'Generate missing storyboards of local videos')
|
|
.parse(process.argv)
|
|
|
|
const options = program.opts()
|
|
|
|
if (!options['video'] && !options['allVideos']) {
|
|
console.error('You need to choose videos for storyboard generation.')
|
|
process.exit(-1)
|
|
}
|
|
|
|
run()
|
|
.then(() => process.exit(0))
|
|
.catch(err => {
|
|
console.error(err)
|
|
process.exit(-1)
|
|
})
|
|
|
|
async function run () {
|
|
await initDatabaseModels(true)
|
|
|
|
JobQueue.Instance.init()
|
|
|
|
let ids: number[] = []
|
|
|
|
if (options['video']) {
|
|
const video = await VideoModel.load(toCompleteUUID(options['video']))
|
|
|
|
if (!video) {
|
|
console.error('Unknown video ' + options['video'])
|
|
process.exit(-1)
|
|
}
|
|
|
|
if (video.remote === true) {
|
|
console.error('Cannot process a remote video')
|
|
process.exit(-1)
|
|
}
|
|
|
|
if (video.isLive) {
|
|
console.error('Cannot process live video')
|
|
process.exit(-1)
|
|
}
|
|
|
|
ids.push(video.id)
|
|
} else {
|
|
ids = await listLocalMissingStoryboards()
|
|
}
|
|
|
|
for (const id of ids) {
|
|
const videoFull = await VideoModel.load(id)
|
|
|
|
if (videoFull.isLive) continue
|
|
|
|
await addLocalOrRemoteStoryboardJobIfNeeded({ video: videoFull, federate: true })
|
|
|
|
console.log(`Created generate-storyboard job for ${videoFull.name}.`)
|
|
}
|
|
}
|
|
|
|
async function listLocalMissingStoryboards () {
|
|
const ids = await VideoModel.listLocalIds()
|
|
const results: number[] = []
|
|
|
|
for (const id of ids) {
|
|
const storyboard = await StoryboardModel.loadByVideo(id)
|
|
if (!storyboard) results.push(id)
|
|
}
|
|
|
|
return results
|
|
}
|