123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173 |
- import CliTable3 from 'cli-table3'
- import { Command, program } from 'commander'
- import { uniq } from 'lodash'
- import { URL } from 'url'
- import validator from 'validator'
- import { HttpStatusCode, VideoRedundanciesTarget } from '@shared/models'
- import { assignToken, buildServer, getServerCredentials } from './cli'
- import bytes = require('bytes')
- program
- .name('redundancy')
- .usage('[command] [options]')
- program
- .command('list-remote-redundancies')
- .description('List remote redundancies on your videos')
- .option('-u, --url <url>', 'Server url')
- .option('-U, --username <username>', 'Username')
- .option('-p, --password <token>', 'Password')
- .action(() => listRedundanciesCLI('my-videos'))
- program
- .command('list-my-redundancies')
- .description('List your redundancies of remote videos')
- .option('-u, --url <url>', 'Server url')
- .option('-U, --username <username>', 'Username')
- .option('-p, --password <token>', 'Password')
- .action(() => listRedundanciesCLI('remote-videos'))
- program
- .command('add')
- .description('Duplicate a video in your redundancy system')
- .option('-u, --url <url>', 'Server url')
- .option('-U, --username <username>', 'Username')
- .option('-p, --password <token>', 'Password')
- .option('-v, --video <videoId>', 'Video id to duplicate')
- .action((options, command) => addRedundancyCLI(options, command))
- program
- .command('remove')
- .description('Remove a video from your redundancies')
- .option('-u, --url <url>', 'Server url')
- .option('-U, --username <username>', 'Username')
- .option('-p, --password <token>', 'Password')
- .option('-v, --video <videoId>', 'Video id to remove from redundancies')
- .action((options, command) => removeRedundancyCLI(options, command))
- if (!process.argv.slice(2).length) {
- program.outputHelp()
- }
- program.parse(process.argv)
- // ----------------------------------------------------------------------------
- async function listRedundanciesCLI (target: VideoRedundanciesTarget) {
- const { url, username, password } = await getServerCredentials(program)
- const server = buildServer(url)
- await assignToken(server, username, password)
- const { data } = await server.redundancy.listVideos({ start: 0, count: 100, sort: 'name', target })
- const table = new CliTable3({
- head: [ 'video id', 'video name', 'video url', 'files', 'playlists', 'by instances', 'total size' ]
- }) as any
- for (const redundancy of data) {
- const webtorrentFiles = redundancy.redundancies.files
- const streamingPlaylists = redundancy.redundancies.streamingPlaylists
- let totalSize = ''
- if (target === 'remote-videos') {
- const tmp = webtorrentFiles.concat(streamingPlaylists)
- .reduce((a, b) => a + b.size, 0)
- totalSize = bytes(tmp)
- }
- const instances = uniq(
- webtorrentFiles.concat(streamingPlaylists)
- .map(r => r.fileUrl)
- .map(u => new URL(u).host)
- )
- table.push([
- redundancy.id.toString(),
- redundancy.name,
- redundancy.url,
- webtorrentFiles.length,
- streamingPlaylists.length,
- instances.join('\n'),
- totalSize
- ])
- }
- console.log(table.toString())
- process.exit(0)
- }
- async function addRedundancyCLI (options: { video: number }, command: Command) {
- const { url, username, password } = await getServerCredentials(command)
- const server = buildServer(url)
- await assignToken(server, username, password)
- if (!options.video || validator.isInt('' + options.video) === false) {
- console.error('You need to specify the video id to duplicate and it should be a number.\n')
- command.outputHelp()
- process.exit(-1)
- }
- try {
- await server.redundancy.addVideo({ videoId: options.video })
- console.log('Video will be duplicated by your instance!')
- process.exit(0)
- } catch (err) {
- if (err.message.includes(HttpStatusCode.CONFLICT_409)) {
- console.error('This video is already duplicated by your instance.')
- } else if (err.message.includes(HttpStatusCode.NOT_FOUND_404)) {
- console.error('This video id does not exist.')
- } else {
- console.error(err)
- }
- process.exit(-1)
- }
- }
- async function removeRedundancyCLI (options: { video: number }, command: Command) {
- const { url, username, password } = await getServerCredentials(command)
- const server = buildServer(url)
- await assignToken(server, username, password)
- if (!options.video || validator.isInt('' + options.video) === false) {
- console.error('You need to specify the video id to remove from your redundancies.\n')
- command.outputHelp()
- process.exit(-1)
- }
- const videoId = parseInt(options.video + '', 10)
- const myVideoRedundancies = await server.redundancy.listVideos({ target: 'my-videos' })
- let videoRedundancy = myVideoRedundancies.data.find(r => videoId === r.id)
- if (!videoRedundancy) {
- const remoteVideoRedundancies = await server.redundancy.listVideos({ target: 'remote-videos' })
- videoRedundancy = remoteVideoRedundancies.data.find(r => videoId === r.id)
- }
- if (!videoRedundancy) {
- console.error('Video redundancy not found.')
- process.exit(-1)
- }
- try {
- const ids = videoRedundancy.redundancies.files
- .concat(videoRedundancy.redundancies.streamingPlaylists)
- .map(r => r.id)
- for (const id of ids) {
- await server.redundancy.removeVideo({ redundancyId: id })
- }
- console.log('Video redundancy removed!')
- process.exit(0)
- } catch (err) {
- console.error(err)
- process.exit(-1)
- }
- }
|