server/tools/cli.ts

   1 import { Netrc } from 'netrc-parser'
   2 import { isTestInstance, getAppNumber } from '../helpers/core-utils'
   3 import { join } from 'path'
   4 import { root } from '../../shared/extra-utils'
   5
   6 let configName = 'PeerTube/CLI'
   7 if (isTestInstance()) configName += `-${getAppNumber()}`
   8
   9 const config = require('application-config')(configName)
  10
  11 const version = require('../../../package.json').version
  12
  13 interface Settings {
  14   remotes: any[],
  15   default: number
  16 }
  17
  18 function getSettings () {
  19   return new Promise<Settings>((res, rej) => {
  20     const defaultSettings = {
  21       remotes: [],
  22       default: -1
  23     }
  24
  25     config.read((err, data) => {
  26       if (err) return rej(err)
  27
  28       return res(Object.keys(data).length === 0 ? defaultSettings : data)
  29     })
  30   })
  31 }
  32
  33 async function getNetrc () {
  34   const Netrc = require('netrc-parser').Netrc
  35
  36   const netrc = isTestInstance()
  37     ? new Netrc(join(root(), 'test' + getAppNumber(), 'netrc'))
  38     : new Netrc()
  39
  40   await netrc.load()
  41
  42   return netrc
  43 }
  44
  45 function writeSettings (settings) {
  46   return new Promise((res, rej) => {
  47     config.write(settings, err => {
  48       if (err) return rej(err)
  49
  50       return res()
  51     })
  52   })
  53 }
  54
  55 function deleteSettings () {
  56   return new Promise((res, rej) => {
  57     config.trash((err) => {
  58       if (err) return rej(err)
  59
  60       return res()
  61     })
  62   })
  63 }
  64
  65 function getRemoteObjectOrDie (program: any, settings: Settings, netrc: Netrc) {
  66   if (!program['url'] || !program['username'] || !program['password']) {
  67     // No remote and we don't have program parameters: quit
  68     if (settings.remotes.length === 0 || Object.keys(netrc.machines).length === 0) {
  69       if (!program[ 'url' ]) console.error('--url field is required.')
  70       if (!program[ 'username' ]) console.error('--username field is required.')
  71       if (!program[ 'password' ]) console.error('--password field is required.')
  72
  73       return process.exit(-1)
  74     }
  75
  76     let url: string = program['url']
  77     let username: string = program['username']
  78     let password: string = program['password']
  79
  80     if (!url && settings.default !== -1) url = settings.remotes[settings.default]
  81
  82     const machine = netrc.machines[url]
  83
  84     if (!username && machine) username = machine.login
  85     if (!password && machine) password = machine.password
  86
  87     return { url, username, password }
  88   }
  89
  90   return {
  91     url: program[ 'url' ],
  92     username: program[ 'username' ],
  93     password: program[ 'password' ]
  94   }
  95 }
  96
  97 // ---------------------------------------------------------------------------
  98
  99 export {
 100   version,
 101   config,
 102   getSettings,
 103   getNetrc,
 104   getRemoteObjectOrDie,
 105   writeSettings,
 106   deleteSettings
 107 }