X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Ftools%2Fcli.ts;h=7c763734f61e9dd1c59fa8183d083ebc2c0c0a9f;hb=06aad80165d09a8863ab8103149a8ff518b10641;hp=2eec51aa490a96506864dd2955f3f19b0559b295;hpb=6b226c32782ea7779ced56c1e1090d8b51c5c504;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/tools/cli.ts b/server/tools/cli.ts index 2eec51aa4..7c763734f 100644 --- a/server/tools/cli.ts +++ b/server/tools/cli.ts @@ -1,9 +1,13 @@ +import { Command } from 'commander' import { Netrc } from 'netrc-parser' -import { getAppNumber, isTestInstance } from '../helpers/core-utils' import { join } from 'path' -import { getVideoChannel, root } from '../../shared/extra-utils' -import { Command } from 'commander' -import { VideoChannel, VideoPrivacy } from '../../shared/models/videos' +import { createLogger, format, transports } from 'winston' +import { PeerTubeServer } from '@shared/extra-utils' +import { UserRole } from '@shared/models' +import { VideoPrivacy } from '../../shared/models/videos' +import { getAppNumber, isTestInstance } from '../helpers/core-utils' +import { root } from '@shared/core-utils' +import { loadLanguages } from '@server/initializers/constants' let configName = 'PeerTube/CLI' if (isTestInstance()) configName += `-${getAppNumber()}` @@ -12,24 +16,34 @@ const config = require('application-config')(configName) const version = require('../../../package.json').version +async function getAdminTokenOrDie (server: PeerTubeServer, username: string, password: string) { + const token = await server.login.getAccessToken(username, password) + const me = await server.users.getMyInfo({ token }) + + if (me.role !== UserRole.ADMINISTRATOR) { + console.error('You must be an administrator.') + process.exit(-1) + } + + return token +} + interface Settings { - remotes: any[], + remotes: any[] default: number } -function getSettings () { - return new Promise((res, rej) => { - const defaultSettings = { - remotes: [], - default: -1 - } +async function getSettings (): Promise { + const defaultSettings = { + remotes: [], + default: -1 + } - config.read((err, data) => { - if (err) return rej(err) + const data = await config.read() - return res(Object.keys(data).length === 0 ? defaultSettings : data) - }) - }) + return Object.keys(data).length === 0 + ? defaultSettings + : data } async function getNetrc () { @@ -44,40 +58,34 @@ async function getNetrc () { return netrc } -function writeSettings (settings) { - return new Promise((res, rej) => { - config.write(settings, err => { - if (err) return rej(err) - - return res() - }) - }) +function writeSettings (settings: Settings) { + return config.write(settings) } function deleteSettings () { - return new Promise((res, rej) => { - config.trash((err) => { - if (err) return rej(err) - - return res() - }) - }) + return config.trash() } -function getRemoteObjectOrDie (program: any, settings: Settings, netrc: Netrc) { - if (!program['url'] || !program['username'] || !program['password']) { +function getRemoteObjectOrDie ( + program: Command, + settings: Settings, + netrc: Netrc +): { url: string, username: string, password: string } { + const options = program.opts() + + if (!options.url || !options.username || !options.password) { // No remote and we don't have program parameters: quit if (settings.remotes.length === 0 || Object.keys(netrc.machines).length === 0) { - if (!program[ 'url' ]) console.error('--url field is required.') - if (!program[ 'username' ]) console.error('--username field is required.') - if (!program[ 'password' ]) console.error('--password field is required.') + if (!options.url) console.error('--url field is required.') + if (!options.username) console.error('--username field is required.') + if (!options.password) console.error('--password field is required.') return process.exit(-1) } - let url: string = program['url'] - let username: string = program['username'] - let password: string = program['password'] + let url: string = options.url + let username: string = options.username + let password: string = options.password if (!url && settings.default !== -1) url = settings.remotes[settings.default] @@ -90,9 +98,9 @@ function getRemoteObjectOrDie (program: any, settings: Settings, netrc: Netrc) { } return { - url: program[ 'url' ], - username: program[ 'username' ], - password: program[ 'password' ] + url: options.url, + username: options.username, + password: options.password } } @@ -111,12 +119,16 @@ function buildCommonVideoOptions (command: Command) { .option('-d, --video-description ', 'Video description') .option('-P, --privacy ', 'Privacy') .option('-C, --channel-name ', 'Channel name') - .option('-m, --comments-enabled', 'Enable comments') + .option('--no-comments-enabled', 'Disable video comments') .option('-s, --support ', 'Video support text') - .option('-w, --wait-transcoding', 'Wait transcoding before publishing the video') + .option('--no-wait-transcoding', 'Do not wait transcoding before publishing the video') + .option('--no-download-enabled', 'Disable video download') + .option('-v, --verbose ', 'Verbosity, from 0/\'error\' to 4/\'debug\'', 'info') } -async function buildVideoAttributesFromCommander (url: string, command: Command, defaultAttributes: any) { +async function buildVideoAttributesFromCommander (server: PeerTubeServer, command: Command, defaultAttributes: any = {}) { + const options = command.opts() + const defaultBooleanAttributes = { nsfw: false, commentsEnabled: true, @@ -127,8 +139,8 @@ async function buildVideoAttributesFromCommander (url: string, command: Command, const booleanAttributes: { [id in keyof typeof defaultBooleanAttributes]: boolean } | {} = {} for (const key of Object.keys(defaultBooleanAttributes)) { - if (command[ key ] !== undefined) { - booleanAttributes[key] = command[ key ] + if (options[key] !== undefined) { + booleanAttributes[key] = options[key] } else if (defaultAttributes[key] !== undefined) { booleanAttributes[key] = defaultAttributes[key] } else { @@ -137,19 +149,20 @@ async function buildVideoAttributesFromCommander (url: string, command: Command, } 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 + name: options.videoName || defaultAttributes.name, + category: options.category || defaultAttributes.category || undefined, + licence: options.licence || defaultAttributes.licence || undefined, + language: options.language || defaultAttributes.language || undefined, + privacy: options.privacy || defaultAttributes.privacy || VideoPrivacy.PUBLIC, + support: options.support || defaultAttributes.support || undefined, + description: options.videoDescription || defaultAttributes.description || undefined, + tags: options.tags || defaultAttributes.tags || undefined } Object.assign(videoAttributes, booleanAttributes) - if (command[ 'channelName' ]) { - const res = await getVideoChannel(url, command['channelName']) - const videoChannel: VideoChannel = res.body + if (options.channelName) { + const videoChannel = await server.channels.get({ channelName: options.channelName }) Object.assign(videoAttributes, { channelId: videoChannel.id }) @@ -161,17 +174,74 @@ async function buildVideoAttributesFromCommander (url: string, command: Command, return videoAttributes } +function getServerCredentials (program: Command) { + return Promise.all([ getSettings(), getNetrc() ]) + .then(([ settings, netrc ]) => { + return getRemoteObjectOrDie(program, settings, netrc) + }) +} + +function buildServer (url: string) { + loadLanguages() + return new PeerTubeServer({ url }) +} + +async function assignToken (server: PeerTubeServer, username: string, password: string) { + const bodyClient = await server.login.getClient() + const client = { id: bodyClient.client_id, secret: bodyClient.client_secret } + + const body = await server.login.login({ client, user: { username, password } }) + + server.accessToken = body.access_token +} + +function getLogger (logLevel = 'info') { + const logLevels = { + 0: 0, + error: 0, + 1: 1, + warn: 1, + 2: 2, + info: 2, + 3: 3, + verbose: 3, + 4: 4, + debug: 4 + } + + const logger = createLogger({ + levels: logLevels, + format: format.combine( + format.splat(), + format.simple() + ), + transports: [ + new (transports.Console)({ + level: logLevel + }) + ] + }) + + return logger +} + // --------------------------------------------------------------------------- export { version, - config, + getLogger, getSettings, getNetrc, getRemoteObjectOrDie, writeSettings, deleteSettings, + getServerCredentials, + buildCommonVideoOptions, - buildVideoAttributesFromCommander + buildVideoAttributesFromCommander, + + getAdminTokenOrDie, + buildServer, + assignToken }