blob: caadfe3e752d806de85d1144001ab92031f3db21 (
plain)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
(in-package :extreload)
(defvar *wg* (wait-group:make-wait-group))
(opts:define-opts
(:name :socket-url
:description "DevTools protocol WebSocket URL"
:long "socket-url"
:arg-parser #'identity
:meta-var "SOCKET_URL")
(:name :help
:description "print this help menu"
:short #\h
:long "help")
(:name :version
:description "show the program version"
:short #\V
:long "version"))
(defun main ()
(let ((config (parse-options)))
;; Store the WebSocket client as a global.
(defvar *client* (ws-client config))
;; TODO: error if no `socket-url`
(with-websocket-connection (*client*)
(wsd:on :message *client* #'ws-on-message)
; (wsd:on :message *client* #'(lambda (message) (ws-on-message message)))
;; TODO: Maybe defvar *config* and store client in the config
(websocket-send *client* (target-get-targets-msg 1))
(wait-group:wait *wg*))))
(defun ws-on-message (message)
(let* ((response (jsown:parse message))
(targets (parse-get-targets-response response)))
(if targets
(reload-extensions
(extension-targets targets)
'("pacpdcpgfbpkdpmhfaljffnfbdanmblh")))
(if (target-attached-to-target-msg-p response)
(reload-extension
(json-obj-get
(json-obj-get response "params")
"sessionId")))
(wait-group:done *wg*)))
(defun json-obj-get (obj key)
(handler-case
(jsown:val obj key)
(simple-error (e)
(let ((s (format nil "~A" e)))
(if (search "not available" s)
nil)))))
(defun reload-extensions (targets extension-ids)
(labels ((requested-extension-p (target)
(find-if
#'(lambda (id)
(uiop:string-prefix-p
(concatenate 'string "chrome-extension://" id)
(json-obj-get target "url")))
extension-ids)))
(dolist (extension (filter #'requested-extension-p targets))
(attach-to-target extension))))
(defun attach-to-target (extension)
(let ((target-id (json-obj-get extension "targetId")))
(websocket-send *client*
(target-attach-to-target-msg 2 target-id))))
(defun reload-extension (session-id)
(websocket-send *client*
(runtime-evaluate-msg 1 session-id "chrome.runtime.reload()")))
(defun extension-targets (targets)
(labels ((extensionp (target)
(string= (json-obj-get target "type")
"background_page")))
(filter #'extensionp targets)))
(defun websocket-send (client data)
(wsd:send client data)
(wait-group:add *wg*))
|
