src/content.js

   1 import { EncryptedStream, LocalStream } from 'extension-streams'
   2 import IdGenerator from '@/utils/IdGenerator'
   3 import NetworkMessage from '@/messages/network'
   4 import InternalMessage from '@/messages/internal'
   5 import * as MsgTypes from './messages/types'
   6 import * as EventNames from '@/messages/event'
   7
   8 let stream = new WeakMap()
   9 let INJECTION_SCRIPT_FILENAME = 'js/inject.js'
  10 let isReady = false
  11
  12 class Content {
  13   constructor() {
  14     this.setupEncryptedStream()
  15     this.injectInteractionScript()
  16   }
  17
  18   injectInteractionScript() {
  19     let script = document.createElement('script')
  20     script.src = chrome.extension.getURL(INJECTION_SCRIPT_FILENAME)
  21     ;(document.head || document.documentElement).appendChild(script)
  22     script.onload = () => script.remove()
  23   }
  24
  25   setupEncryptedStream() {
  26     console.log('stream listening...', IdGenerator.text(256))
  27
  28     stream = new EncryptedStream(EventNames.BYTOM, IdGenerator.text(256))
  29     stream.listenWith(msg => this.contentListener(msg))
  30
  31     stream.onSync(async () => {
  32       const defaultAccount = await this.getDefaultAccount();
  33       const net = await this.getDefaultNetwork();
  34       const accountList = await this.getAccountList();
  35
  36       // Pushing an instance of Bytomdapp to the web application
  37       stream.send(
  38         NetworkMessage.payload(MsgTypes.PUSH_BYTOM, {defaultAccount, net, accountList}),
  39         EventNames.INJECT
  40       )
  41
  42       // Dispatching the loaded event to the web application.
  43       isReady = true
  44
  45       document.dispatchEvent(new CustomEvent('chromeBytomLoaded'))
  46     })
  47   }
  48
  49   contentListener(msg) {
  50     console.log('content.stream.listen:', msg, stream.key)
  51
  52     if (!isReady) return
  53     if (!msg) return
  54
  55     let networkMessage = NetworkMessage.fromJson(msg)
  56     switch (msg.type) {
  57       case 'sync':
  58         this.sync(msg)
  59         break
  60       case MsgTypes.TRANSFER:
  61       case MsgTypes.ADVTRANSFER:
  62       case MsgTypes.SEND:
  63         this.transfer(msg.type, networkMessage)
  64         break
  65       default:
  66         stream.send(networkMessage.error('errtest'), EventNames.INJECT)
  67         break
  68     }
  69   }
  70
  71   getVersion() {}
  72
  73   getDefaultAccount(){
  74     return InternalMessage.signal(MsgTypes.REQUEST_CURRENT_ACCOUNT)
  75       .send()
  76   }
  77
  78   getDefaultNetwork(){
  79     return InternalMessage.signal(MsgTypes.REQUEST_CURRENT_NETWORK)
  80       .send()
  81   }
  82
  83   getAccountList(){
  84     return InternalMessage.signal(MsgTypes.REQUEST_ACCOUNT_LIST)
  85       .send()
  86   }
  87
  88   respond(message, payload) {
  89     if (!isReady) return
  90
  91     console.log(222, message, payload)
  92     const response =
  93       !payload || payload.hasOwnProperty('isError')
  94         ? message.error(payload)
  95         : message.respond(payload)
  96     stream.send(response, EventNames.INJECT)
  97   }
  98
  99   sync(message) {
 100     stream.key = message.handshake.length ? message.handshake : null
 101     stream.send({ type: 'sync' }, EventNames.INJECT)
 102     stream.synced = true
 103   }
 104
 105   transfer(type, message) {
 106     if (!isReady) return
 107
 108     InternalMessage.payload(type, message.payload)
 109       .send()
 110       .then(res => this.respond(message, res))
 111   }
 112 }
 113
 114 const content = new Content()