import { Netrc } from 'netrc-parser' import { getAppNumber, isTestInstance } from '../helpers/core-utils' import { join } from 'path' import { root } from '../../shared/extra-utils/miscs/miscs' import { getVideoChannel } from '../../shared/extra-utils/videos/video-channels' import { CommanderStatic } from 'commander' import { VideoChannel, VideoPrivacy } from '../../shared/models/videos' import { createLogger, format, transports } from 'winston' import { getMyUserInformation } from '@shared/extra-utils/users/users' import { User, UserRole } from '@shared/models' import { getAccessToken } from '@shared/extra-utils/users/login' let configName = 'PeerTube/CLI' if (isTestInstance()) configName += `-${getAppNumber()}` const config = require('application-config')(configName) const version = require('../../../package.json').version async function getAdminTokenOrDie (url: string, username: string, password: string) { const accessToken = await getAccessToken(url, username, password) const resMe = await getMyUserInformation(url, accessToken) const me: User = resMe.body if (me.role !== UserRole.ADMINISTRATOR) { console.error('You must be an administrator.') process.exit(-1) } return accessToken } interface Settings { remotes: any[] default: number } function getSettings () { return new Promise((res, rej) => { const defaultSettings = { remotes: [], default: -1 } config.read((err, data) => { if (err) return rej(err) return res(Object.keys(data).length === 0 ? defaultSettings : data) }) }) } async function getNetrc () { const Netrc = require('netrc-parser').Netrc const netrc = isTestInstance() ? new Netrc(join(root(), 'test' + getAppNumber(), 'netrc')) : new Netrc() await netrc.load() return netrc } function writeSettings (settings) { return new Promise((res, rej) => { config.write(settings, err => { if (err) return rej(err) return res() }) }) } function deleteSettings () { return new Promise((res, rej) => { config.trash((err) => { if (err) return rej(err) return res() }) }) } function getRemoteObjectOrDie ( program: any, settings: Settings, netrc: Netrc ): { url: string, username: string, password: string } { if (!program['url'] || !program['username'] || !program['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.') return process.exit(-1) } let url: string = program['url'] let username: string = program['username'] let password: string = program['password'] if (!url && settings.default !== -1) url = settings.remotes[settings.default] const machine = netrc.machines[url] if (!username && machine) username = machine.login if (!password && machine) password = machine.password return { url, username, password } } return { url: program['url'], username: program['username'], password: program['password'] } } function buildCommonVideoOptions (command: CommanderStatic) { 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') .option('-v, --verbose ', 'Verbosity, from 0/\'error\' to 4/\'debug\'', 'info') } async function buildVideoAttributesFromCommander (url: string, command: CommanderStatic, defaultAttributes: any = {}) { const defaultBooleanAttributes = { nsfw: false, commentsEnabled: true, downloadEnabled: true, waitTranscoding: true } const booleanAttributes: { [id in keyof typeof defaultBooleanAttributes]: boolean } | {} = {} for (const key of Object.keys(defaultBooleanAttributes)) { if (command[key] !== undefined) { booleanAttributes[key] = command[key] } else if (defaultAttributes[key] !== undefined) { booleanAttributes[key] = defaultAttributes[key] } else { booleanAttributes[key] = defaultBooleanAttributes[key] } } 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, description: command['videoDescription'] || defaultAttributes.description || undefined, tags: command['tags'] || defaultAttributes.tags || 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 } function getServerCredentials (program: any) { return Promise.all([ getSettings(), getNetrc() ]) .then(([ settings, netrc ]) => { return getRemoteObjectOrDie(program, settings, netrc) }) } 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, getAdminTokenOrDie }