X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Ftools%2Fimport-videos.ts;h=e49fbb2f5ef3e0f2253703e0b94fb9581ecc3572;hb=adc236fee3c40bf1fbaa4ad4fc22a7ecb65fb09f;hp=809d69e4c6cfb6ce8ce74dbd804d0319c2d59291;hpb=27d56b545323fd8e8fab545466bcf3002f9c42f0;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/tools/import-videos.ts b/server/tools/import-videos.ts index 809d69e4c..e49fbb2f5 100644 --- a/server/tools/import-videos.ts +++ b/server/tools/import-videos.ts @@ -10,42 +10,66 @@ import { doRequestAndSaveToFile } from '../helpers/requests' import { CONSTRAINTS_FIELDS } from '../initializers' import { getClient, getVideoCategories, login, searchVideo, uploadVideo } from '../tests/utils' import { truncate } from 'lodash' +import * as prompt from 'prompt' program .option('-u, --url ', 'Server url') .option('-U, --username ', 'Username') .option('-p, --password ', 'Password') .option('-t, --target-url ', 'Video target URL') - .option('-l, --language ', 'Language code') + .option('-l, --language ', 'Language ISO 639 code (fr or en...)') .option('-v, --verbose', 'Verbose mode') .parse(process.argv) if ( !program['url'] || !program['username'] || - !program['password'] || !program['targetUrl'] ) { console.error('All arguments are required.') process.exit(-1) } -run().catch(err => console.error(err)) - -let accessToken: string -let client: { id: string, secret: string } - const user = { username: program['username'], password: program['password'] } +run().catch(err => console.error(err)) + +let accessToken: string +let client: { id: string, secret: string } + const processOptions = { cwd: __dirname, maxBuffer: Infinity } +async function promptPassword () { + return new Promise((res, rej) => { + prompt.start() + const schema = { + properties: { + password: { + hidden: true, + required: true + } + } + } + prompt.get(schema, function (err, result) { + if (err) { + return rej(err) + } + return res(result.password) + }) + }) +} + async function run () { + if (!user.password) { + user.password = await promptPassword() + } + const res = await getClient(program['url']) client = { id: res.body.client_id, @@ -82,7 +106,7 @@ async function run () { }) } -function processVideo (info: any, languageCode: number) { +function processVideo (info: any, languageCode: string) { return new Promise(async res => { if (program['verbose']) console.log('Fetching object.', info) @@ -103,19 +127,25 @@ function processVideo (info: any, languageCode: number) { console.log('Downloading video "%s"...', videoInfo.title) const options = [ '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best', '-o', path ] - youtubeDL.exec(videoInfo.url, options, processOptions, async (err, output) => { - if (err) return console.error(err) - - console.log(output.join('\n')) - - await uploadVideoOnPeerTube(normalizeObject(videoInfo), path, languageCode) - + try { + youtubeDL.exec(videoInfo.url, options, processOptions, async (err, output) => { + if (err) { + console.error(err) + return res() + } + + console.log(output.join('\n')) + await uploadVideoOnPeerTube(normalizeObject(videoInfo), path, languageCode) + return res() + }) + } catch (err) { + console.log(err.message) return res() - }) + } }) } -async function uploadVideoOnPeerTube (videoInfo: any, videoPath: string, language?: number) { +async function uploadVideoOnPeerTube (videoInfo: any, videoPath: string, language?: string) { const category = await getCategory(videoInfo.categories) const licence = getLicence(videoInfo.license) let tags = [] @@ -146,6 +176,7 @@ async function uploadVideoOnPeerTube (videoInfo: any, videoPath: string, languag licence, language, nsfw: isNSFW(videoInfo), + waitTranscoding: true, commentsEnabled: true, description: videoInfo.description || undefined, support: undefined,