starred/stf
1
0
Fork 0
mirror of https://github.com/DeviceFarmer/stf.git synced 2026-04-19 07:53:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

Initial version of multidevice transactions.

Browse Source
This commit is contained in:
Simo Kinnunen
2014-01-23 14:00:15 +09:00
parent 242f6f17de
commit 35ac51795b
5 changed files with 88 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
lib/roles/console.js
View File

@@ -34,6 +34,9 @@ module.exports = function(options) {
rl.setPrompt('stf> ')
rl.prompt()
var clients = []
, group = wireutil.global
rl.on('line', function(line) {
var args = line.trim().split(/\s+/g)
switch (args.shift()) {
@@ -45,13 +48,61 @@ module.exports = function(options) {
console.log()
rl.prompt()
break
case 'invite':
if (clients.length) {
log.error('We already have clients')
break
}
var ours = group = wireutil.makePrivateChannel()
var messageListener = function(theirs, data) {
if (theirs.toString() === ours) {
var wrapper = wire.Envelope.decode(data)
switch (wrapper.type) {
case wire.MessageType.JOIN_GROUP:
var message = wire.JoinGroupMessage.decode(wrapper.message)
clients.push(message.serial)
log.info('"%s" joined', message.serial)
break
case wire.MessageType.LEAVE_GROUP:
var message = wire.LeaveGroupMessage.decode(wrapper.message)
, index = clients.indexOf(message.serial)
clients.splice(index, 1)
log.info('"%s" left', message.serial)
break
default:
throw new Error('Unexpected response')
}
}
}
sub.subscribe(ours)
sub.on('message', messageListener)
push.send([wireutil.global, wireutil.makeGroupMessage(
ours
, 10000
, []
)])
Promise.delay(1000)
.then(function() {
rl.prompt()
})
break
case 'shell':
var resolver = Promise.defer()
var resolvers = {}
, ours = wireutil.makePrivateChannel()
, counter = 0
log.debug('Using channel "%s"', ours)
function messageListener(theirs, data) {
clients.forEach(function(client) {
resolvers[client] = Promise.defer()
})
var messageListener = function(theirs, data) {
if (theirs.toString() === ours) {
var wrapper = wire.Envelope.decode(data)
switch (wrapper.type) {
@@ -63,33 +114,38 @@ module.exports = function(options) {
case wire.MessageType.DEVICE_DONE:
var message = wire.DeviceDoneMessage.decode(wrapper.message)
log.info('[%s] DONE', message.serial)
resolver.resolve() // @todo Wait till all devices have finished
resolvers[message.serial].resolve()
break
case wire.MessageType.DEVICE_FAIL:
var message = wire.DeviceFailMessage.decode(wrapper.message)
log.error('[%s] FAIL: ', message.serial, message.reason)
resolver.reject(new Error(message.reason))
resolvers[message.serial].reject(new Error(message.reason))
break
default:
resolver.reject(new Error('Unexpected response'))
break
throw new Error('Unexpected response')
}
}
}
sub.subscribe(ours)
sub.on('message', messageListener)
push.send([wireutil.global, wireutil.makeShellCommandMessage(
push.send([group, wireutil.makeShellCommandMessage(
ours
, args
)])
resolver.promise.finally(function() {
sub.unsubscribe(ours)
sub.removeListener('message', messageListener)
rl.prompt()
var promises = Object.keys(resolvers).map(function(serial) {
return resolvers[serial].promise
})
Promise.settle(promises)
.then(function() {
sub.unsubscribe(ours)
sub.removeListener('message', messageListener)
rl.prompt()
})
break
case 'exit':
case 'quit':