From 1205823fece15f249e0dbad42e096cf9b81c3ba5 Mon Sep 17 00:00:00 2001 From: Chocobozzz Date: Thu, 13 Jun 2019 13:53:28 +0200 Subject: Add ability to override CLI import attributes --- server/tools/cli.ts | 69 ++++++++++++++++++++++++++++++++-- server/tools/peertube-import-videos.ts | 38 +++++++++---------- server/tools/peertube-upload.ts | 52 +++++++++---------------- 3 files changed, 100 insertions(+), 59 deletions(-) (limited to 'server/tools') diff --git a/server/tools/cli.ts b/server/tools/cli.ts index 6be558d7b..4aa3d9ce8 100644 --- a/server/tools/cli.ts +++ b/server/tools/cli.ts @@ -1,7 +1,9 @@ import { Netrc } from 'netrc-parser' -import { isTestInstance, getAppNumber } from '../helpers/core-utils' +import { getAppNumber, isTestInstance } from '../helpers/core-utils' import { join } from 'path' -import { root } from '../../shared/extra-utils' +import { getVideoChannel, root } from '../../shared/extra-utils' +import { Command } from 'commander' +import { VideoChannel, VideoPrivacy } from '../../shared/models/videos' let configName = 'PeerTube/CLI' if (isTestInstance()) configName += `-${getAppNumber()}` @@ -94,6 +96,64 @@ function getRemoteObjectOrDie (program: any, settings: Settings, netrc: Netrc) { } } +function buildCommonVideoOptions (command: Command) { + function list (val) { + return val.split(',') + } + + return command + .option('-n, --video-name ', 'Video name') + .option('-c, --category ', 'Category number') + .option('-l, --licence ', 'Licence number') + .option('-L, --language ', 'Language ISO 639 code (fr or en...)') + .option('-t, --tags ', 'Video tags', list) + .option('-N, --nsfw', 'Video is Not Safe For Work') + .option('-d, --video-description ', 'Video description') + .option('-P, --privacy ', 'Privacy') + .option('-C, --channel-name ', 'Channel name') + .option('-m, --comments-enabled', 'Enable comments') + .option('-s, --support ', 'Video support text') + .option('-w, --wait-transcoding', 'Wait transcoding before publishing the video') +} + +async function buildVideoAttributesFromCommander (url: string, command: Command, defaultAttributes: any) { + const booleanAttributes: { [id: string]: boolean } = {} + + for (const key of [ 'nsfw', 'commentsEnabled', 'downloadEnabled', 'waitTranscoding' ]) { + if (command[ key ] !== undefined) { + booleanAttributes[key] = command[ key ] + } else if (defaultAttributes[key] !== undefined) { + booleanAttributes[key] = defaultAttributes[key] + } else { + booleanAttributes[key] = false + } + } + + const videoAttributes = { + name: command[ 'videoName' ] || defaultAttributes.name, + category: command[ 'category' ] || defaultAttributes.category || undefined, + licence: command[ 'licence' ] || defaultAttributes.licence || undefined, + language: command[ 'language' ] || defaultAttributes.language || undefined, + privacy: command[ 'privacy' ] || defaultAttributes.privacy || VideoPrivacy.PUBLIC, + support: command[ 'support' ] || defaultAttributes.support || undefined + } + + Object.assign(videoAttributes, booleanAttributes) + + if (command[ 'channelName' ]) { + const res = await getVideoChannel(url, command['channelName']) + const videoChannel: VideoChannel = res.body + + Object.assign(videoAttributes, { channelId: videoChannel.id }) + + if (!videoAttributes.support && videoChannel.support) { + Object.assign(videoAttributes, { support: videoChannel.support }) + } + } + + return videoAttributes +} + // --------------------------------------------------------------------------- export { @@ -103,5 +163,8 @@ export { getNetrc, getRemoteObjectOrDie, writeSettings, - deleteSettings + deleteSettings, + + buildCommonVideoOptions, + buildVideoAttributesFromCommander } diff --git a/server/tools/peertube-import-videos.ts b/server/tools/peertube-import-videos.ts index f9cd3106a..d7bb00e02 100644 --- a/server/tools/peertube-import-videos.ts +++ b/server/tools/peertube-import-videos.ts @@ -3,7 +3,6 @@ require('tls').DEFAULT_ECDH_CURVE = 'auto' import * as program from 'commander' import { join } from 'path' -import { VideoPrivacy } from '../../shared/models/videos' import { doRequestAndSaveToFile } from '../helpers/requests' import { CONSTRAINTS_FIELDS } from '../initializers/constants' import { getClient, getVideoCategories, login, searchVideoWithSort, uploadVideo } from '../../shared/extra-utils/index' @@ -12,7 +11,7 @@ import * as prompt from 'prompt' import { remove } from 'fs-extra' import { sha256 } from '../helpers/core-utils' import { buildOriginallyPublishedAt, safeGetYoutubeDL } from '../helpers/youtube-dl' -import { getNetrc, getRemoteObjectOrDie, getSettings } from './cli' +import { buildCommonVideoOptions, buildVideoAttributesFromCommander, getNetrc, getRemoteObjectOrDie, getSettings } from './cli' type UserInfo = { username: string @@ -24,14 +23,16 @@ const processOptions = { maxBuffer: Infinity } -program +let command = program .name('import-videos') + +command = buildCommonVideoOptions(command) + +command .option('-u, --url ', 'Server url') .option('-U, --username ', 'Username') .option('-p, --password ', 'Password') .option('-t, --target-url ', 'Video target URL') - .option('-C, --channel-id ', 'Channel ID') - .option('-l, --language ', 'Language ISO 639 code (fr or en...)') .option('-v, --verbose', 'Verbose mode') .parse(process.argv) @@ -179,7 +180,7 @@ async function uploadVideoOnPeerTube (parameters: { const originallyPublishedAt = buildOriginallyPublishedAt(videoInfo) - const videoAttributes = { + const defaultAttributes = { name: truncate(videoInfo.title, { 'length': CONSTRAINTS_FIELDS.VIDEOS.NAME.max, 'separator': /,? +/, @@ -187,24 +188,19 @@ async function uploadVideoOnPeerTube (parameters: { }), category, licence, - language: program[ 'language' ], nsfw: isNSFW(videoInfo), - waitTranscoding: true, - commentsEnabled: true, - downloadEnabled: true, - description: videoInfo.description || undefined, - support: undefined, - tags, - privacy: VideoPrivacy.PUBLIC, - fixture: videoPath, - thumbnailfile, - previewfile: thumbnailfile, - originallyPublishedAt: originallyPublishedAt ? originallyPublishedAt.toISOString() : null + description: videoInfo.description, + tags } - if (program[ 'channelId' ]) { - Object.assign(videoAttributes, { channelId: program['channelId'] }) - } + const videoAttributes = await buildVideoAttributesFromCommander(url, program, defaultAttributes) + + Object.assign(videoAttributes, { + originallyPublishedAt: originallyPublishedAt ? originallyPublishedAt.toISOString() : null, + thumbnailfile, + previewfile: thumbnailfile, + fixture: videoPath + }) console.log('\nUploading on PeerTube video "%s".', videoAttributes.name) diff --git a/server/tools/peertube-upload.ts b/server/tools/peertube-upload.ts index 1da52da31..c00205e8f 100644 --- a/server/tools/peertube-upload.ts +++ b/server/tools/peertube-upload.ts @@ -3,24 +3,18 @@ import { access, constants } from 'fs-extra' import { isAbsolute } from 'path' import { getClient, login } from '../../shared/extra-utils' import { uploadVideo } from '../../shared/extra-utils/' -import { VideoPrivacy } from '../../shared/models/videos' -import { getNetrc, getRemoteObjectOrDie, getSettings } from './cli' +import { buildCommonVideoOptions, buildVideoAttributesFromCommander, getNetrc, getRemoteObjectOrDie, getSettings } from './cli' -program +let command = program .name('upload') + +command = buildCommonVideoOptions(command) + +command + .option('-u, --url ', 'Server url') .option('-U, --username ', 'Username') .option('-p, --password ', 'Password') - .option('-n, --video-name ', 'Video name') - .option('-P, --privacy ', 'Privacy') - .option('-N, --nsfw', 'Video is Not Safe For Work') - .option('-c, --category ', 'Category number') - .option('-C, --channel-id ', 'Channel ID') - .option('-m, --comments-enabled', 'Enable comments') - .option('-l, --licence ', 'Licence number') - .option('-L, --language ', 'Language ISO 639 code (fr or en...)') - .option('-d, --video-description ', 'Video description') - .option('-t, --tags ', 'Video tags', list) .option('-b, --thumbnail ', 'Thumbnail path') .option('-v, --preview ', 'Preview path') .option('-f, --file ', 'Video absolute file path') @@ -30,10 +24,9 @@ Promise.all([ getSettings(), getNetrc() ]) .then(([ settings, netrc ]) => { const { url, username, password } = getRemoteObjectOrDie(program, settings, netrc) - if (!program[ 'videoName' ] || !program[ 'file' ] || !program[ 'channelId' ]) { + if (!program[ 'videoName' ] || !program[ 'file' ]) { if (!program[ 'videoName' ]) console.error('--video-name is required.') if (!program[ 'file' ]) console.error('--file is required.') - if (!program[ 'channelId' ]) console.error('--channel-id is required.') process.exit(-1) } @@ -70,24 +63,17 @@ async function run (url: string, username: string, password: string) { console.log('Uploading %s video...', program[ 'videoName' ]) - const videoAttributes = { - name: program[ 'videoName' ], - category: program[ 'category' ] || undefined, - channelId: program[ 'channelId' ], - licence: program[ 'licence' ] || undefined, - language: program[ 'language' ] || undefined, - nsfw: program[ 'nsfw' ] !== undefined ? program[ 'nsfw' ] : false, - description: program[ 'videoDescription' ] || undefined, - tags: program[ 'tags' ] || [], - commentsEnabled: program[ 'commentsEnabled' ] !== undefined ? program[ 'commentsEnabled' ] : true, - downloadEnabled: program[ 'downloadEnabled' ] !== undefined ? program[ 'downloadEnabled' ] : true, + const defaultAttributes = { + tags: command[ 'tags' ], + description: command[ 'videoDescription' ] + } + const videoAttributes = await buildVideoAttributesFromCommander(url, program, defaultAttributes) + + Object.assign(videoAttributes, { fixture: program[ 'file' ], thumbnailfile: program[ 'thumbnail' ], - previewfile: program[ 'preview' ], - waitTranscoding: true, - privacy: program[ 'privacy' ] || VideoPrivacy.PUBLIC, - support: undefined - } + previewfile: program[ 'preview' ] + }) try { await uploadVideo(url, accessToken, videoAttributes) @@ -100,7 +86,3 @@ async function run (url: string, username: string, password: string) { } // ---------------------------------------------------------------------------- - -function list (val) { - return val.split(',') -} -- cgit v1.2.3