mirror of
https://github.com/urbit/shrub.git
synced 2024-12-12 10:29:01 +03:00
63 lines
2.0 KiB
Plaintext
63 lines
2.0 KiB
Plaintext
require! <[ stream stream-snitch colors escape-string-regexp ]>
|
|
pty = require \pty.js
|
|
|
|
export ERROR = /((ford|warn): |\r\x1b\[K\/~)/
|
|
export class Urbit
|
|
(args)->
|
|
@stdout = process.stdout # overridable
|
|
@pty = pty.spawn \urbit args
|
|
@pty.on \data ~> @stdout.write it # TODO pipe?
|
|
console.log "FIXME Running Ubuntu 14.04, which causes a libtinfo version info warning. Should update to 16.04."
|
|
#
|
|
@last-output = Date.now()
|
|
@pty.on \data ~> @last-output = Date.now()
|
|
#
|
|
@reset-listeners!
|
|
process.on \exit ~> @pty.write '\04' # send EOF to gracefully checkpoint
|
|
@pty.on \exit (code,signal)~>
|
|
| code => process.exit code
|
|
| signal => process.exit 128 + signal # repack exit code
|
|
| _ => #TODO report if unexpected?
|
|
#
|
|
note: (...args)-> console.log "\n#{'_'*40}\nnode:".blue, ...args
|
|
warn: (...args)-> console.log "\n#{'!'*40}\nnode:".red, ...args
|
|
wait-silent: ~> # this feels hacky
|
|
new Promise (resolve)~>
|
|
a = set-interval ~>
|
|
if Date.now! > @last-output + 1000
|
|
clear-interval a
|
|
resolve @last-output
|
|
, 200
|
|
#
|
|
reset-listeners: ~>
|
|
@pty.socket.unpipe @listeners
|
|
@pty.socket.pipe (@listeners = new stream.PassThrough)
|
|
@
|
|
every: (re, cb)~>
|
|
@listeners.pipe (new stream-snitch re).on "match" cb
|
|
expect: (re)~>
|
|
new Promise (resolve)~>
|
|
@listeners.pipe (new stream-snitch re).once "match" resolve
|
|
expect-immediate: (re)->
|
|
Promise.race [
|
|
@expect re
|
|
@wait-silent!then -> throw Error "Expected #re during event"
|
|
]
|
|
#
|
|
line: (s)->
|
|
@pty.write s
|
|
<~ @wait-silent!then
|
|
@stdout.write "\n"
|
|
@pty.write "\r"
|
|
#
|
|
expect-echo: (s)-> #ALT send-and-expect
|
|
<~ @line s .then
|
|
@expect new RegExp escape-string-regexp s
|
|
#
|
|
exit: (code)->
|
|
@pty.on \exit -> process.exit code #REVIEW just return promise?
|
|
# @pty.write "\03" # ^C running event
|
|
@pty.write "\05\25" # ^E^U to clear prompt
|
|
@pty.write "\04" # ^D to checkpoint
|
|
# set-timeout # hard exit
|