aboutsummaryrefslogtreecommitdiffstats
path: root/app/scripts/contentscript.js
blob: ad90059b7a7c70b9d1a2fd36a31bfd9558028d5c (plain) (blame)
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
const LocalMessageDuplexStream = require('./lib/local-message-stream.js')
const PortStream = require('./lib/port-stream.js')
const ObjectMultiplex = require('./lib/obj-multiplex')

// inject in-page script
var scriptTag = document.createElement('script')
scriptTag.src = chrome.extension.getURL('scripts/inpage.js')
scriptTag.onload = function () { this.parentNode.removeChild(this) }
var container = document.head || document.documentElement
// append as first child
container.insertBefore(scriptTag, container.children[0])

// setup communication to page and plugin
var pageStream = new LocalMessageDuplexStream({
  name: 'contentscript',
  target: 'inpage',
})
pageStream.on('error', console.error.bind(console))
var pluginPort = chrome.runtime.connect({name: 'contentscript'})
var pluginStream = new PortStream(pluginPort)
pluginStream.on('error', console.error.bind(console))

// forward communication plugin->inpage
pageStream.pipe(pluginStream).pipe(pageStream)

// connect contentscript->inpage reload stream
var mx = ObjectMultiplex()
mx.on('error', console.error.bind(console))
mx.pipe(pageStream)
var reloadStream = mx.createStream('reload')
reloadStream.on('error', console.error.bind(console))

// if we lose connection with the plugin, trigger tab refresh
pluginStream.on('close', function () {
  reloadStream.write({ method: 'reset' })
})