mirror of https://github.com/Chocobozzz/PeerTube
86 lines
2.1 KiB
TypeScript
86 lines
2.1 KiB
TypeScript
import { program } from 'commander'
|
|
import { toCompleteUUID } from '@server/helpers/custom-validators/misc'
|
|
import { initDatabaseModels } from '@server/initializers/database'
|
|
import { JobQueue } from '@server/lib/job-queue'
|
|
import { VideoModel } from '@server/models/video/video'
|
|
import { StoryboardModel } from '@server/models/video/storyboard'
|
|
|
|
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 JobQueue.Instance.createJob({
|
|
type: 'generate-video-storyboard',
|
|
payload: {
|
|
videoUUID: videoFull.uuid,
|
|
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
|
|
}
|