mirror of
https://github.com/openstf/stf
synced 2025-10-05 02:29:26 +02:00
Impement remote connect plugin.
This commit is contained in:
parent
692582ce3d
commit
db1699c666
6 changed files with 147 additions and 2 deletions
|
@ -31,6 +31,7 @@ module.exports = function(options) {
|
|||
.dependency(require('./device/plugins/forward'))
|
||||
.dependency(require('./device/plugins/group'))
|
||||
.dependency(require('./device/plugins/reboot'))
|
||||
.dependency(require('./device/plugins/connect'))
|
||||
.define(function(options, solo) {
|
||||
if (process.send) {
|
||||
// Only if we have a parent process
|
||||
|
|
108
lib/roles/device/plugins/connect.js
Normal file
108
lib/roles/device/plugins/connect.js
Normal file
|
@ -0,0 +1,108 @@
|
|||
var util = require('util')
|
||||
|
||||
var syrup = require('syrup')
|
||||
var Promise = require('bluebird')
|
||||
|
||||
var logger = require('../../../util/logger')
|
||||
var wire = require('../../../wire')
|
||||
var wireutil = require('../../../wire/util')
|
||||
var lifecycle = require('../../../util/lifecycle')
|
||||
|
||||
module.exports = syrup.serial()
|
||||
.dependency(require('../support/adb'))
|
||||
.dependency(require('../support/router'))
|
||||
.dependency(require('../support/push'))
|
||||
.dependency(require('./group'))
|
||||
.define(function(options, adb, router, push, group) {
|
||||
var log = logger.createLogger('device:plugins:connect')
|
||||
, plugin = Object.create(null)
|
||||
, activeServer = null
|
||||
|
||||
plugin.port = options.ports.pop()
|
||||
plugin.url = util.format('%s:%s', options.publicIp, plugin.port)
|
||||
|
||||
plugin.start = function() {
|
||||
return new Promise(function(resolve, reject) {
|
||||
if (plugin.isRunning()) {
|
||||
return resolve(plugin.url)
|
||||
}
|
||||
|
||||
var server = adb.createTcpUsbBridge(options.serial)
|
||||
|
||||
server.on('listening', function() {
|
||||
resolve(plugin.url)
|
||||
})
|
||||
|
||||
server.on('connection', function(conn) {
|
||||
log.info('New remote ADB connection from %s', conn.remoteAddress)
|
||||
})
|
||||
|
||||
server.on('error', reject)
|
||||
|
||||
log.info(util.format('Listening on port %d', plugin.port))
|
||||
server.listen(plugin.port)
|
||||
|
||||
activeServer = server
|
||||
lifecycle.share('Remote ADB', activeServer)
|
||||
})
|
||||
}
|
||||
|
||||
plugin.stop = Promise.method(function() {
|
||||
if (plugin.isRunning()) {
|
||||
activeServer.close()
|
||||
activeServer.end()
|
||||
}
|
||||
})
|
||||
|
||||
plugin.end = Promise.method(function() {
|
||||
if (plugin.isRunning()) {
|
||||
activeServer.end()
|
||||
}
|
||||
})
|
||||
|
||||
plugin.isRunning = function() {
|
||||
return !!activeServer
|
||||
}
|
||||
|
||||
lifecycle.observe(plugin.stop)
|
||||
group.on('leave', plugin.end)
|
||||
|
||||
router
|
||||
.on(wire.ConnectStartMessage, function(channel) {
|
||||
var reply = wireutil.reply(options.serial)
|
||||
plugin.start()
|
||||
.then(function(url) {
|
||||
push.send([
|
||||
channel
|
||||
, reply.okay(url)
|
||||
])
|
||||
})
|
||||
.catch(function(err) {
|
||||
log.error('Unable to start remote connect service', err.stack)
|
||||
push.send([
|
||||
channel
|
||||
, reply.fail(err.message)
|
||||
])
|
||||
})
|
||||
})
|
||||
.on(wire.ConnectStopMessage, function(channel) {
|
||||
var reply = wireutil.reply(options.serial)
|
||||
plugin.end()
|
||||
.then(function() {
|
||||
push.send([
|
||||
channel
|
||||
, reply.okay()
|
||||
])
|
||||
})
|
||||
.catch(function(err) {
|
||||
log.error('Failed to stop connect service', err.stack)
|
||||
push.send([
|
||||
channel
|
||||
, reply.fail(err.message)
|
||||
])
|
||||
})
|
||||
})
|
||||
|
||||
return plugin.start()
|
||||
.return(plugin)
|
||||
})
|
|
@ -31,7 +31,7 @@ module.exports = function(options) {
|
|||
// port, we must ensure that we allocate ports in fixed groups.
|
||||
var ports = options.ports.slice(
|
||||
0
|
||||
, options.ports.length - options.ports.length % 3
|
||||
, options.ports.length - options.ports.length % 4
|
||||
)
|
||||
|
||||
// Information about total devices
|
||||
|
@ -276,7 +276,7 @@ module.exports = function(options) {
|
|||
|
||||
// Spawn a device worker
|
||||
function spawn() {
|
||||
var allocatedPorts = ports.splice(0, 3)
|
||||
var allocatedPorts = ports.splice(0, 4)
|
||||
, proc = options.fork(device, allocatedPorts)
|
||||
, resolver = Promise.defer()
|
||||
|
||||
|
|
|
@ -484,6 +484,26 @@ module.exports = function(options) {
|
|||
)
|
||||
])
|
||||
})
|
||||
.on('connect.start', function(channel, responseChannel) {
|
||||
joinChannel(responseChannel)
|
||||
push.send([
|
||||
channel
|
||||
, wireutil.transaction(
|
||||
responseChannel
|
||||
, new wire.ConnectStartMessage()
|
||||
)
|
||||
])
|
||||
})
|
||||
.on('connect.stop', function(channel, responseChannel) {
|
||||
joinChannel(responseChannel)
|
||||
push.send([
|
||||
channel
|
||||
, wireutil.transaction(
|
||||
responseChannel
|
||||
, new wire.ConnectStopMessage()
|
||||
)
|
||||
])
|
||||
})
|
||||
.on('browser.open', function(channel, responseChannel, data) {
|
||||
joinChannel(responseChannel)
|
||||
push.send([
|
||||
|
|
|
@ -52,6 +52,8 @@ enum MessageType {
|
|||
ScreenCaptureMessage = 50;
|
||||
ProviderHeartbeatMessage = 51;
|
||||
RebootMessage = 52;
|
||||
ConnectStartMessage = 53;
|
||||
ConnectStopMessage = 54;
|
||||
}
|
||||
|
||||
message Envelope {
|
||||
|
@ -374,6 +376,12 @@ message StoreOpenMessage {
|
|||
message ScreenCaptureMessage {
|
||||
}
|
||||
|
||||
message ConnectStartMessage {
|
||||
}
|
||||
|
||||
message ConnectStopMessage {
|
||||
}
|
||||
|
||||
// Events, these must be kept in sync with STFService/wire.proto
|
||||
|
||||
message AirplaneModeEvent {
|
||||
|
|
|
@ -189,6 +189,14 @@ module.exports = function ControlServiceFactory(
|
|||
return sendTwoWay('logcat.stop')
|
||||
}
|
||||
|
||||
this.startRemoteConnect = function() {
|
||||
return sendTwoWay('connect.start')
|
||||
}
|
||||
|
||||
this.stopRemoteConnect = function() {
|
||||
return sendTwoWay('connect.stop')
|
||||
}
|
||||
|
||||
this.openBrowser = function(url, browser) {
|
||||
return sendTwoWay('browser.open', {
|
||||
url: url
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue