forked from amir20/phantomjs-node
-
Notifications
You must be signed in to change notification settings - Fork 0
/
shim.coffee
69 lines (46 loc) · 1.88 KB
/
shim.coffee
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
# Require gets overwritten by browserify, so we have to reimplement it from scratch - boo :(
mkweb = new Function "exports", "window", phantom.loadModuleSource('webpage')
webpage = {}
mkweb.call {}, webpage, {}
proto = require 'dnode-protocol'
[port] = phantom.args
controlPage = webpage.create()
fnwrap = (target) -> -> target.apply this, arguments
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"
cb src[key] = val
get: (key, cb) -> cb 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','includeJs','injectJs','render','sendEvent']
evaluate: (fn, cb=->) -> cb page.evaluate fn
render: (file, cb=->) -> page.render file; cb()
_phantom = mkwrap phantom,
['exit', 'injectJS'],
createPage: (cb) -> cb pageWrap webpage.create()
server = proto _phantom
s = server.create()
s.on 'request', (req) ->
#console.log "phantom sending request #{JSON.stringify req}"
#evil = "function(){socket.send(#{JSON.stringify JSON.stringify req} + '\\n');}"
evil = "function(){socket.emit('message', #{JSON.stringify JSON.stringify req} + '\\n');}"
controlPage.evaluate evil
controlPage.onAlert = (msg) ->
return unless msg[0..5] is "PCTRL "
#console.log "phantom got request " + msg[6..]
s.parse msg[6..]
controlPage.onConsoleMessage = (msg...) -> console.log msg...
controlPage.open "http://127.0.0.1:#{port}/", (status) ->
#console.log 'Control page title is ' + controlPage.evaluate -> document.title
s.start()