neko/src/internal/messages.ts

import Vue from 'vue'
import { Member, ScreenConfigurations } from '../types/structs'
import { EVENT } from '../types/events'
import {
  DisconnectPayload,
  MemberListPayload,
  MemberDisconnectPayload,
  MemberPayload,
  ControlPayload,
  ControlTargetPayload,
  ControlClipboardPayload,
  ScreenConfigurationsPayload,
  ScreenResolutionPayload,
  BroadcastStatusPayload,
  AdminPayload,
  AdminTargetPayload,
} from '../types/messages'

import EventEmitter from 'eventemitter3'
import { NekoWebSocket } from './websocket'
import NekoState from '~/types/state'

export interface NekoEvents {
  ['system.websocket']: (state: 'connected' | 'connecting' | 'disconnected') => void
  ['system.webrtc']: (state: 'connected' | 'connecting' | 'disconnected') => void
  ['system.connect']: () => void
  ['system.disconnect']: (message: string) => void
  ['control.host']: (id: string | null) => void
  ['member.list']: (members: Member[]) => void
  ['member.connected']: (id: string) => void
  ['member.disconnected']: (id: string) => void
  ['control.request']: (id: string) => void
  ['control.requesting']: (id: string) => void
  ['clipboard.update']: (text: string) => void
  ['screen.size']: (id: string) => void
  ['broadcast.status']: (url: string, isActive: boolean) => void
}

export class NekoMessages extends EventEmitter<NekoEvents> {
  state: NekoState

  constructor(websocket: NekoWebSocket, state: NekoState) {
    super()

    this.state = state
    websocket.on('message', async (event: string, payload: any) => {
      // @ts-ignore
      if (typeof this[event] === 'function') {
        // @ts-ignore
        this[event](payload)
      } else {
        console.log(`unhandled websocket event '${event}':`, payload)
      }
    })
  }

  /////////////////////////////
  // System Events
  /////////////////////////////
  protected [EVENT.SYSTEM.DISCONNECT]({ message }: DisconnectPayload) {
    console.log('EVENT.SYSTEM.DISCONNECT')
    this.emit('system.disconnect', message)
  }

  /////////////////////////////
  // Member Events
  /////////////////////////////
  protected [EVENT.MEMBER.LIST]({ members }: MemberListPayload) {
    console.log('EVENT.MEMBER.LIST')
    this.emit('member.list', members)
    //user.setMembers(members)
  }

  protected [EVENT.MEMBER.CONNECTED](member: MemberPayload) {
    console.log('EVENT.MEMBER.CONNECTED')
    this.emit('member.connected', member.id)
    //user.addMember(member)

    if (member.id === this.state.member.id) {
      Vue.set(this.state.member, 'name', member.name)
      Vue.set(this.state.member, 'is_admin', member.admin)
    }
  }

  protected [EVENT.MEMBER.DISCONNECTED]({ id }: MemberDisconnectPayload) {
    console.log('EVENT.MEMBER.DISCONNECTED')
    this.emit('member.disconnected', id)
    //user.delMember(id)
  }

  /////////////////////////////
  // Control Events
  /////////////////////////////
  protected [EVENT.CONTROL.LOCKED]({ id }: ControlPayload) {
    console.log('EVENT.CONTROL.LOCKED')
    this.emit('control.host', id)
    //remote.setHost(id)
    //remote.changeKeyboard()
  }

  protected [EVENT.CONTROL.RELEASE]({ id }: ControlPayload) {
    console.log('EVENT.CONTROL.RELEASE')
    this.emit('control.host', null)
    //remote.reset()
  }

  protected [EVENT.CONTROL.REQUEST]({ id }: ControlPayload) {
    console.log('EVENT.CONTROL.REQUEST')
    this.emit('control.request', id)
  }

  protected [EVENT.CONTROL.REQUESTING]({ id }: ControlPayload) {
    console.log('EVENT.CONTROL.REQUESTING')
    this.emit('control.requesting', id)
  }

  protected [EVENT.CONTROL.GIVE]({ id, target }: ControlTargetPayload) {
    console.log('EVENT.CONTROL.GIVE')
    this.emit('control.host', target)
    //remote.setHost(target)
    //remote.changeKeyboard()
  }

  protected [EVENT.CONTROL.CLIPBOARD]({ text }: ControlClipboardPayload) {
    console.log('EVENT.CONTROL.CLIPBOARD')
    this.emit('clipboard.update', text)
    //remote.setClipboard(text)
  }

  /////////////////////////////
  // Screen Events
  /////////////////////////////
  protected [EVENT.SCREEN.CONFIGURATIONS]({ configurations }: ScreenConfigurationsPayload) {
    const data = []
    for (const i of Object.keys(configurations)) {
      const { width, height, rates } = configurations[i]
      if (width >= 600 && height >= 300) {
        for (const j of Object.keys(rates)) {
          const rate = rates[j]
          if (rate === 30 || rate === 60) {
            data.push({
              width,
              height,
              rate,
            })
          }
        }
      }
    }

    const conf = data.sort((a, b) => {
      if (b.width === a.width && b.height == a.height) {
        return b.rate - a.rate
      } else if (b.width === a.width) {
        return b.height - a.height
      }
      return b.width - a.width
    })

    Vue.set(this.state.screen, 'configurations', conf)
  }

  protected [EVENT.SCREEN.RESOLUTION]({ id, width, height, rate }: ScreenResolutionPayload) {
    Vue.set(this.state.screen, 'size', { width, height, rate })
    if (id) this.emit('screen.size', id)
  }

  /////////////////////////////
  // Broadcast Events
  /////////////////////////////
  protected [EVENT.BROADCAST.STATUS](payload: BroadcastStatusPayload) {
    console.log('EVENT.BROADCAST.STATUS')
    this.emit('broadcast.status', payload.url, payload.isActive)
    //settings.broadcastStatus(payload)
  }

  /////////////////////////////
  // Admin Events
  /////////////////////////////
  protected [EVENT.ADMIN.CONTROL]({ id, target }: AdminTargetPayload) {
    if (!target) return

    console.log('EVENT.ADMIN.CONTROL')
    this.emit('control.host', id)
    //remote.setHost(id)
    //remote.changeKeyboard()
  }

  protected [EVENT.ADMIN.RELEASE]({ id, target }: AdminTargetPayload) {
    if (!target) return

    console.log('EVENT.ADMIN.RELEASE')
    this.emit('control.host', null)
    //remote.reset()
  }

  protected [EVENT.ADMIN.GIVE]({ id, target }: AdminTargetPayload) {
    if (!target) return

    console.log('EVENT.ADMIN.GIVE')
    this.emit('control.host', target)
    //remote.setHost(target)
    //remote.changeKeyboard()
  }
}