mirror of
https://github.com/openstf/stf
synced 2025-10-03 17:59:28 +02:00
502 lines
13 KiB
JavaScript
502 lines
13 KiB
JavaScript
var util = require('util')
|
|
var os = require('os')
|
|
|
|
var program = require('commander')
|
|
var Promise = require('bluebird')
|
|
var ip = require('my-local-ip')
|
|
|
|
var pkg = require('../package')
|
|
var cliutil = require('./util/cliutil')
|
|
var procutil = require('./util/procutil')
|
|
var logger = require('./util/logger')
|
|
|
|
Promise.longStackTraces()
|
|
|
|
program
|
|
.version(pkg.version)
|
|
|
|
program
|
|
.command('provider [serial..]')
|
|
.description('start provider')
|
|
.option('-s, --connect-sub <endpoint>'
|
|
, 'sub endpoint'
|
|
, cliutil.list)
|
|
.option('-p, --connect-push <endpoint>'
|
|
, 'push endpoint'
|
|
, cliutil.list)
|
|
.option('-n, --name <name>'
|
|
, 'name (or os.hostname())'
|
|
, String
|
|
, os.hostname())
|
|
.option('-t, --restart-threshold <ms>'
|
|
, 'restart worker only if it stays alive for longer than this'
|
|
, Number
|
|
, 10000)
|
|
.option('--min-port <port>'
|
|
, 'minimum port number for worker use'
|
|
, Number
|
|
, 7400)
|
|
.option('--max-port <port>'
|
|
, 'maximum port number for worker use'
|
|
, Number
|
|
, 7800)
|
|
.option('--public-ip <ip>'
|
|
, 'public ip for global access'
|
|
, String
|
|
, ip())
|
|
.action(function() {
|
|
var serials = cliutil.allUnknownArgs(arguments)
|
|
, options = cliutil.lastArg(arguments)
|
|
|
|
if (!options.connectSub) {
|
|
this.missingArgument('--connect-sub')
|
|
}
|
|
if (!options.connectPush) {
|
|
this.missingArgument('--connect-push')
|
|
}
|
|
|
|
require('./roles/provider')({
|
|
name: options.name
|
|
, restartThreshold: options.restartThreshold
|
|
, restartTimeout: 1000
|
|
, killTimeout: 10000
|
|
, ports: cliutil.range(options.minPort, options.maxPort)
|
|
, filter: function(device) {
|
|
return serials.length === 0 || serials.indexOf(device.id) !== -1
|
|
}
|
|
, fork: function(device, ports) {
|
|
var fork = require('child_process').fork
|
|
return fork(__filename, [
|
|
'device', device.id
|
|
, '--provider', options.name
|
|
, '--connect-sub', options.connectSub.join(',')
|
|
, '--connect-push', options.connectPush.join(',')
|
|
, '--ports', ports.join(',')
|
|
, '--public-ip', options.publicIp
|
|
])
|
|
}
|
|
, endpoints: {
|
|
sub: options.connectSub
|
|
, push: options.connectPush
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('device <serial>')
|
|
.description('start device worker')
|
|
.option('-n, --provider <name>'
|
|
, 'provider name'
|
|
, String)
|
|
.option('-s, --connect-sub <endpoint>'
|
|
, 'sub endpoint'
|
|
, cliutil.list)
|
|
.option('-p, --connect-push <endpoint>'
|
|
, 'push endpoint'
|
|
, cliutil.list)
|
|
.option('--ports <ports>'
|
|
, 'ports allocated to worker'
|
|
, cliutil.list)
|
|
.option('--public-ip <ip>'
|
|
, 'public ip for global access'
|
|
, String
|
|
, ip())
|
|
.action(function(serial, options) {
|
|
if (!options.connectSub) {
|
|
this.missingArgument('--connect-sub')
|
|
}
|
|
if (!options.connectPush) {
|
|
this.missingArgument('--connect-push')
|
|
}
|
|
if (!options.provider) {
|
|
this.missingArgument('--provider')
|
|
}
|
|
if (!options.provider) {
|
|
this.missingArgument('--ports')
|
|
}
|
|
|
|
require('./roles/device')({
|
|
serial: serial
|
|
, provider: options.provider
|
|
, ports: options.ports
|
|
, publicIp: options.publicIp
|
|
, endpoints: {
|
|
sub: options.connectSub
|
|
, push: options.connectPush
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('processor <name>')
|
|
.description('start processor')
|
|
.option('-a, --connect-app-dealer <endpoint>'
|
|
, 'app dealer endpoint'
|
|
, cliutil.list)
|
|
.option('-d, --connect-dev-dealer <endpoint>'
|
|
, 'device dealer endpoint'
|
|
, cliutil.list)
|
|
.action(function(name, options) {
|
|
if (!options.connectAppDealer) {
|
|
this.missingArgument('--connect-app-dealer')
|
|
}
|
|
if (!options.connectDevDealer) {
|
|
this.missingArgument('--connect-dev-dealer')
|
|
}
|
|
|
|
require('./roles/processor')({
|
|
name: name
|
|
, endpoints: {
|
|
appDealer: options.connectAppDealer
|
|
, devDealer: options.connectDevDealer
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('triproxy <name>')
|
|
.description('start triproxy')
|
|
.option('-u, --bind-pub <endpoint>'
|
|
, 'pub endpoint'
|
|
, String
|
|
, 'tcp://*:7111')
|
|
.option('-d, --bind-dealer <endpoint>'
|
|
, 'dealer endpoint'
|
|
, String
|
|
, 'tcp://*:7112')
|
|
.option('-p, --bind-pull <endpoint>'
|
|
, 'pull endpoint'
|
|
, String
|
|
, 'tcp://*:7113')
|
|
.action(function(name, options) {
|
|
require('./roles/triproxy')({
|
|
name: name
|
|
, endpoints: {
|
|
pub: options.bindPub
|
|
, dealer: options.bindDealer
|
|
, pull: options.bindPull
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('auth-ldap')
|
|
.description('start LDAP auth client')
|
|
.option('-p, --port <port>'
|
|
, 'port (or $PORT)'
|
|
, Number
|
|
, process.env.PORT || 7120)
|
|
.option('-s, --secret <secret>'
|
|
, 'secret (or $SECRET)'
|
|
, String
|
|
, process.env.SECRET)
|
|
.option('-i, --ssid <ssid>'
|
|
, 'session SSID (or $SSID)'
|
|
, String
|
|
, process.env.SSID || 'ssid')
|
|
.option('-u, --ldap-url <url>'
|
|
, 'LDAP server URL (or $LDAP_URL)'
|
|
, String
|
|
, process.env.LDAP_URL)
|
|
.option('-t, --ldap-timeout <timeout>'
|
|
, 'LDAP timeout (or $LDAP_TIMEOUT)'
|
|
, Number
|
|
, process.env.LDAP_TIMEOUT || 1000)
|
|
.option('--ldap-bind-dn <dn>'
|
|
, 'LDAP bind DN (or $LDAP_BIND_DN)'
|
|
, String
|
|
, process.env.LDAP_BIND_DN)
|
|
.option('--ldap-bind-credentials <credentials>'
|
|
, 'LDAP bind credentials (or $LDAP_BIND_CREDENTIALS)'
|
|
, String
|
|
, process.env.LDAP_BIND_CREDENTIALS)
|
|
.option('--ldap-search-dn <dn>'
|
|
, 'LDAP search DN (or $LDAP_SEARCH_DN)'
|
|
, String
|
|
, process.env.LDAP_SEARCH_DN)
|
|
.option('--ldap-search-scope <scope>'
|
|
, 'LDAP search scope (or $LDAP_SEARCH_SCOPE)'
|
|
, String
|
|
, process.env.LDAP_SEARCH_SCOPE || 'sub')
|
|
.option('--ldap-search-class <class>'
|
|
, 'LDAP search objectClass (or $LDAP_SEARCH_CLASS)'
|
|
, String
|
|
, process.env.LDAP_SEARCH_CLASS || 'user')
|
|
.action(function(options) {
|
|
require('./roles/auth/ldap')({
|
|
port: options.port
|
|
, secret: options.secret
|
|
, ssid: options.ssid
|
|
, ldap: {
|
|
url: options.ldapUrl
|
|
, timeout: options.ldapTimeout
|
|
, bind: {
|
|
dn: options.ldapBindDn
|
|
, credentials: options.ldapBindCredentials
|
|
}
|
|
, search: {
|
|
dn: options.ldapSearchDn
|
|
, scope: options.ldapSearchScope
|
|
, objectClass: options.ldapSearchClass
|
|
, loginField: options.ldapSearchLoginField
|
|
}
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('auth-mock')
|
|
.description('start mock auth client')
|
|
.option('-p, --port <port>'
|
|
, 'port (or $PORT)'
|
|
, Number
|
|
, process.env.PORT || 7120)
|
|
.option('-s, --secret <secret>'
|
|
, 'secret (or $SECRET)'
|
|
, String
|
|
, process.env.SECRET)
|
|
.option('-i, --ssid <ssid>'
|
|
, 'session SSID (or $SSID)'
|
|
, String
|
|
, process.env.SSID || 'ssid')
|
|
.option('-a, --app-url <url>'
|
|
, 'URL to app'
|
|
, String)
|
|
.action(function(options) {
|
|
if (!options.secret) {
|
|
this.missingArgument('--secret')
|
|
}
|
|
if (!options.appUrl) {
|
|
this.missingArgument('--app-url')
|
|
}
|
|
|
|
require('./roles/auth/mock')({
|
|
port: options.port
|
|
, secret: options.secret
|
|
, ssid: options.ssid
|
|
, appUrl: options.appUrl
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('notify-hipchat')
|
|
.description('start HipChat notifier')
|
|
.option('-t, --token <token>'
|
|
, 'HipChat v2 API token (or $HIPCHAT_TOKEN)'
|
|
, String
|
|
, process.env.HIPCHAT_TOKEN)
|
|
.option('-r, --room <room>'
|
|
, 'HipChat room (or $HIPCHAT_ROOM)'
|
|
, String
|
|
, process.env.HIPCHAT_ROOM)
|
|
.option('-p, --priority <level>'
|
|
, 'minimum log level'
|
|
, Number
|
|
, logger.Level.WARNING)
|
|
.option('-s, --connect-sub <endpoint>'
|
|
, 'sub endpoint'
|
|
, cliutil.list)
|
|
.action(function(options) {
|
|
if (!options.token) {
|
|
this.missingArgument('--token')
|
|
}
|
|
if (!options.room) {
|
|
this.missingArgument('--room')
|
|
}
|
|
if (!options.connectSub) {
|
|
this.missingArgument('--connect-sub')
|
|
}
|
|
|
|
require('./roles/notify/hipchat')({
|
|
token: options.token
|
|
, room: options.room
|
|
, priority: options.priority
|
|
, endpoints: {
|
|
sub: options.connectSub
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('app')
|
|
.description('start app')
|
|
.option('-p, --port <port>'
|
|
, 'port (or $PORT)'
|
|
, Number
|
|
, process.env.PORT || 7100)
|
|
.option('-s, --secret <secret>'
|
|
, 'secret (or $SECRET)'
|
|
, String
|
|
, process.env.SECRET)
|
|
.option('-i, --ssid <ssid>'
|
|
, 'session SSID (or $SSID)'
|
|
, String
|
|
, process.env.SSID || 'ssid')
|
|
.option('-a, --auth-url <url>'
|
|
, 'URL to auth client'
|
|
, String)
|
|
.action(function(options) {
|
|
if (!options.secret) {
|
|
this.missingArgument('--secret')
|
|
}
|
|
if (!options.authUrl) {
|
|
this.missingArgument('--auth-url')
|
|
}
|
|
|
|
require('./roles/app')({
|
|
port: options.port
|
|
, secret: options.secret
|
|
, ssid: options.ssid
|
|
, authUrl: options.authUrl
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('console')
|
|
.description('start console')
|
|
.option('-s, --connect-sub <endpoint>'
|
|
, 'sub endpoint'
|
|
, cliutil.list)
|
|
.option('-p, --connect-push <endpoint>'
|
|
, 'push endpoint'
|
|
, cliutil.list)
|
|
.action(function(options) {
|
|
if (!options.connectSub) {
|
|
this.missingArgument('--connect-sub')
|
|
}
|
|
if (!options.connectPush) {
|
|
this.missingArgument('--connect-push')
|
|
}
|
|
|
|
require('./roles/console')({
|
|
endpoints: {
|
|
sub: options.connectSub
|
|
, push: options.connectPush
|
|
}
|
|
})
|
|
})
|
|
|
|
program
|
|
.command('local [serial..]')
|
|
.description('start everything locally')
|
|
.option('--bind-app-pub <endpoint>'
|
|
, 'app pub endpoint'
|
|
, String
|
|
, 'tcp://127.0.0.1:7111')
|
|
.option('--bind-app-dealer <endpoint>'
|
|
, 'app dealer endpoint'
|
|
, String
|
|
, 'tcp://127.0.0.1:7112')
|
|
.option('--bind-app-pull <endpoint>'
|
|
, 'app pull endpoint'
|
|
, String
|
|
, 'tcp://127.0.0.1:7113')
|
|
.option('--bind-dev-pub <endpoint>'
|
|
, 'device pub endpoint'
|
|
, String
|
|
, 'tcp://127.0.0.1:7114')
|
|
.option('--bind-dev-dealer <endpoint>'
|
|
, 'device dealer endpoint'
|
|
, String
|
|
, 'tcp://127.0.0.1:7115')
|
|
.option('--bind-dev-pull <endpoint>'
|
|
, 'device pull endpoint'
|
|
, String
|
|
, 'tcp://127.0.0.1:7116')
|
|
.option('--auth-port <port>'
|
|
, 'auth port'
|
|
, Number
|
|
, 7120)
|
|
.option('--auth-secret <secret>'
|
|
, 'auth secret'
|
|
, String
|
|
, 'kute kittykat')
|
|
.option('--app-port <port>'
|
|
, 'app port'
|
|
, Number
|
|
, 7100)
|
|
.option('--provider <name>'
|
|
, 'provider name (or os.hostname())'
|
|
, String
|
|
, os.hostname())
|
|
.action(function() {
|
|
var log = logger.createLogger('cli')
|
|
, options = cliutil.lastArg(arguments)
|
|
|
|
// app triproxy
|
|
procutil.fork(__filename, [
|
|
'triproxy', 'app001'
|
|
, '--bind-pub', options.bindAppPub
|
|
, '--bind-dealer', options.bindAppDealer
|
|
, '--bind-pull', options.bindAppPull
|
|
])
|
|
.catch(function(err) {
|
|
log.error('app triproxy died', err.stack)
|
|
})
|
|
|
|
// device triproxy
|
|
procutil.fork(__filename, [
|
|
'triproxy', 'dev001'
|
|
, '--bind-pub', options.bindDevPub
|
|
, '--bind-dealer', options.bindDevDealer
|
|
, '--bind-pull', options.bindDevPull
|
|
])
|
|
.catch(function(err) {
|
|
log.error('device triproxy died', err.stack)
|
|
})
|
|
|
|
// processor one
|
|
procutil.fork(__filename, [
|
|
'processor', 'proc001'
|
|
, '--connect-app-dealer', options.bindAppDealer
|
|
, '--connect-dev-dealer', options.bindDevDealer
|
|
])
|
|
.catch(function(err) {
|
|
log.error('processor 001 died', err.stack)
|
|
})
|
|
|
|
// processor two
|
|
procutil.fork(__filename, [
|
|
'processor', 'proc002'
|
|
, '--connect-app-dealer', options.bindAppDealer
|
|
, '--connect-dev-dealer', options.bindDevDealer
|
|
])
|
|
.catch(function(err) {
|
|
log.error('processor 002 died', err.stack)
|
|
})
|
|
|
|
// provider
|
|
procutil.fork(__filename, [
|
|
'provider'
|
|
, '--name', options.provider
|
|
, '--connect-sub', options.bindDevPub
|
|
, '--connect-push', options.bindDevPull
|
|
].concat(cliutil.allUnknownArgs(arguments)))
|
|
.catch(function(err) {
|
|
log.error('provider died', err.stack)
|
|
})
|
|
|
|
// auth-mock
|
|
procutil.fork(__filename, [
|
|
'auth-mock'
|
|
, '--port', options.authPort
|
|
, '--secret', options.authSecret
|
|
, '--app-url', util.format('http://localhost:%d/', options.appPort)
|
|
])
|
|
.catch(function(err) {
|
|
log.error('auth-mock died', err.stack)
|
|
})
|
|
|
|
// app
|
|
procutil.fork(__filename, [
|
|
'app'
|
|
, '--port', options.appPort
|
|
, '--secret', options.authSecret
|
|
, '--auth-url', util.format('http://localhost:%d/', options.authPort)
|
|
])
|
|
.catch(function(err) {
|
|
log.error('app died', err.stack)
|
|
})
|
|
})
|
|
|
|
program.parse(process.argv)
|