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)
}
}