1
0
Fork 0
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:
Simo Kinnunen 2014-06-24 11:41:26 +09:00
parent 692582ce3d
commit db1699c666
6 changed files with 147 additions and 2 deletions

View file

@ -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

View 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)
})

View file

@ -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()

View file

@ -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([

View file

@ -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 {

View file

@ -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