/* eslint-disable*/
|
import Emitter from './emitter'
|
|
export default class {
|
constructor(connectionUrl, opts = {}) {
|
this.format = opts.format && opts.format.toLowerCase()
|
|
if (connectionUrl.startsWith('//')) {
|
const scheme = window.location.protocol === 'https:' ? 'wss' : 'ws'
|
connectionUrl = `${scheme}:${connectionUrl}`
|
}
|
|
this.connectionUrl = connectionUrl
|
this.opts = opts
|
|
this.reconnection = this.opts.reconnection || false
|
this.reconnectionAttempts = this.opts.reconnectionAttempts || Infinity
|
this.reconnectionDelay = this.opts.reconnectionDelay || 1000
|
this.reconnectTimeoutId = 0
|
this.reconnectionCount = 0
|
|
this.passToStoreHandler = this.opts.passToStoreHandler || false
|
|
this.connect(connectionUrl, opts)
|
|
if (opts.store) {
|
this.store = opts.store
|
}
|
if (opts.mutations) {
|
this.mutations = opts.mutations
|
}
|
this.onEvent()
|
}
|
|
connect(connectionUrl, opts = {}) {
|
const protocol = opts.protocol || ''
|
this.WebSocket =
|
opts.WebSocket || (protocol === '' ? new WebSocket(connectionUrl) : new WebSocket(connectionUrl, protocol))
|
if (this.format === 'json') {
|
if (!('sendObj' in this.WebSocket)) {
|
this.WebSocket.sendObj = obj => this.WebSocket.send(JSON.stringify(obj))
|
}
|
}
|
|
return this.WebSocket
|
}
|
|
reconnect() {
|
if (this.reconnectionCount <= this.reconnectionAttempts) {
|
this.reconnectionCount++
|
clearTimeout(this.reconnectTimeoutId)
|
|
this.reconnectTimeoutId = setTimeout(() => {
|
if (this.store) {
|
this.passToStore('SOCKET_RECONNECT', this.reconnectionCount)
|
}
|
|
this.connect(this.connectionUrl, this.opts)
|
this.onEvent()
|
}, this.reconnectionDelay)
|
} else if (this.store) {
|
this.passToStore('SOCKET_RECONNECT_ERROR', true)
|
}
|
}
|
|
onEvent() {
|
;['onmessage', 'onclose', 'onerror', 'onopen'].forEach(eventType => {
|
this.WebSocket[eventType] = event => {
|
Emitter.emit(eventType, event)
|
|
if (this.store) {
|
this.passToStore(`SOCKET_${eventType}`, event)
|
}
|
|
if (this.reconnection && eventType === 'onopen') {
|
this.opts.$setInstance(event.currentTarget)
|
this.reconnectionCount = 0
|
}
|
|
if (this.reconnection && eventType === 'onclose') {
|
this.reconnect()
|
}
|
}
|
})
|
}
|
|
passToStore(eventName, event) {
|
if (this.passToStoreHandler) {
|
this.passToStoreHandler(eventName, event, this.defaultPassToStore.bind(this))
|
} else {
|
this.defaultPassToStore(eventName, event)
|
}
|
}
|
|
defaultPassToStore(eventName, event) {
|
if (!eventName.startsWith('SOCKET_')) {
|
return
|
}
|
let method = 'commit'
|
let target = eventName.toUpperCase()
|
let msg = event
|
if (this.format === 'json' && event.data) {
|
msg = JSON.parse(event.data)
|
if (msg.mutation) {
|
target = [msg.namespace || '', msg.mutation].filter(e => !!e).join('/')
|
} else if (msg.action) {
|
method = 'dispatch'
|
target = [msg.namespace || '', msg.action].filter(e => !!e).join('/')
|
}
|
}
|
if (this.mutations) {
|
target = this.mutations[target] || target
|
}
|
this.store[method](target, msg)
|
}
|
}
|
