163 lines
4.6 KiB
CoffeeScript
163 lines
4.6 KiB
CoffeeScript
# Require gets overwritten by browserify, so we have to reimplement it from
|
|
# scratch - boo :(
|
|
webpage = core_require('webpage')
|
|
|
|
shoe = require('shoe')
|
|
dnode = require('dnode')
|
|
system = core_require('system')
|
|
|
|
port = system.args[1]
|
|
hostname = system.args[2]
|
|
|
|
# controlPage = webpage.create()
|
|
|
|
fnwrap = (target) -> -> target.apply this, arguments
|
|
|
|
# Descend into objects with dotted keys
|
|
descend = (op, obj, key, val) ->
|
|
cur = obj
|
|
keys = key.split '.'
|
|
cur = cur[keys.shift()] while keys.length > 1
|
|
|
|
cur[keys[0]] = val if op is 'set'
|
|
|
|
cur[keys[0]]
|
|
|
|
_transform = (val) ->
|
|
if typeof val is "string" and val.indexOf('__phantomCallback__') is 0
|
|
val = 'return ' + val.replace('__phantomCallback__', '')
|
|
val = phantom.callback(new Function(val)())
|
|
|
|
return val
|
|
|
|
transform = (obj) ->
|
|
if typeof obj is "string"
|
|
_transform(obj)
|
|
else if typeof obj is "object"
|
|
for key of obj
|
|
if typeof obj[key] is "object"
|
|
transform(obj[key])
|
|
else
|
|
obj[key] = _transform(obj[key])
|
|
|
|
return obj
|
|
|
|
|
|
mkwrap = (src, pass=[], special={}) ->
|
|
obj =
|
|
set: (key, val, cb=->) ->
|
|
|
|
#Fnwrap so PhantomJS doesn't segfault when it tries to call the callback
|
|
val = fnwrap val if typeof val is "function"
|
|
|
|
val = transform(val)
|
|
|
|
cb descend 'set', src, key, val
|
|
|
|
get: (key, cb) -> cb descend 'get', src, key
|
|
|
|
for k in pass
|
|
do (k) ->
|
|
obj[k] = (args...) ->
|
|
|
|
# This idempotent tomfoolery is required to stop PhantomJS from
|
|
# segfaulting
|
|
args[i] = fnwrap arg for arg, i in args when typeof arg is 'function'
|
|
|
|
src[k] args...
|
|
|
|
for own k of special
|
|
obj[k] = special[k]
|
|
obj
|
|
|
|
pageWrap = (page) ->
|
|
mkwrap(
|
|
page,
|
|
['open', 'close', 'includeJs', 'sendEvent', 'release', 'uploadFile',
|
|
'goBack', 'goForward', 'reload', 'switchToFrame', 'switchToMainFrame',
|
|
'switchToParentFrame', 'switchToFocusedFrame']
|
|
|
|
# this is here to let the user pass in a function that has access to
|
|
# request.abort() and other functions on the request object.
|
|
onPageCreated:(cb=(->)) ->
|
|
page.onPageCreated = (newpage) ->
|
|
cb pageWrap newpage
|
|
onConsoleMessage: (fn, cb=(->)) ->
|
|
page.onConsoleMessage = ->
|
|
fn.apply(this, arguments)
|
|
cb()
|
|
onError: (fn, cb=(->)) ->
|
|
page.onError = ->
|
|
fn.apply(this, arguments)
|
|
cb()
|
|
onResourceRequested: (fn, cb=(->), args...) ->
|
|
page.onResourceRequested = ->
|
|
# prepare a arguments with the extra args
|
|
argumentsWithExtraArgs = [].slice.apply(arguments).concat(args)
|
|
# give a name to the anonymouse function so that we can call it
|
|
fn = fn.replace /function.*\(/, 'function x('
|
|
# the only way we can access the request object is by passing a function
|
|
# to this point as a string and expanding it
|
|
eval(fn) # :(
|
|
# this function has access to request.abort()
|
|
x.apply(this, argumentsWithExtraArgs)
|
|
# this function does not have access to request.abort()
|
|
cb.apply(this, argumentsWithExtraArgs)
|
|
injectJs: (js, cb=->) -> cb page.injectJs js
|
|
evaluate: (fn, cb=(->), args...) ->
|
|
cb page.evaluate.apply(page, [fn].concat(args))
|
|
render: (file, opts={}, cb) ->
|
|
unless cb?
|
|
if typeof opts is 'function'
|
|
cb = opts
|
|
opts = {}
|
|
else
|
|
cb = ->
|
|
|
|
page.render file, opts
|
|
cb()
|
|
getContent: (cb=->) -> cb page.content
|
|
getCookies: (cb=->) -> cb page.cookies
|
|
renderBase64: (type, cb=->) -> cb page.renderBase64 type
|
|
setHeaders: (headers, cb=->) ->
|
|
page.customHeaders = headers
|
|
cb()
|
|
setContent: (html, url, cb=->) ->
|
|
page.onLoadFinished = (status) ->
|
|
page.onLoadFinished = null
|
|
cb status
|
|
page.setContent html, url
|
|
setViewportSize: (width, height, cb=->) ->
|
|
page.viewportSize = {width:width, height:height}
|
|
cb()
|
|
setPaperSize: (options, cb=->) ->
|
|
page.paperSize = transform(options)
|
|
cb()
|
|
setZoomFactor: (zoomFactor, cb=->) ->
|
|
page.zoomFactor = zoomFactor
|
|
cb()
|
|
setFileOnPicker: (fileName, cb=->) ->
|
|
page.onFilePicker = ->
|
|
cb.apply(this, arguments)
|
|
fileName
|
|
)
|
|
|
|
_phantom = mkwrap phantom,
|
|
['exit'],
|
|
injectJs: (js, cb=->) -> cb phantom.injectJs js
|
|
getCookies: (cb=->) -> cb(phantom.cookies)
|
|
addCookie: (cookie, cb=->) ->
|
|
cb(phantom.addCookie(cookie))
|
|
clearCookies: (cb=->) -> cb phantom.clearCookies()
|
|
createPage: (cb) -> cb pageWrap webpage.create()
|
|
setProxy: (host, port, type, user, password, cb=->) ->
|
|
cb(phantom.setProxy(host, port, type, user, password))
|
|
|
|
|
|
stream = shoe('http://' + hostname + ':' + port + '/dnode')
|
|
|
|
d = dnode _phantom
|
|
|
|
d.pipe stream
|
|
stream.pipe d
|