feathersjs/feathers

/* eslint-disable @typescript-eslint/no-unnecessary-type-assertion */
import { Application, HookContext, getServiceOptions, defaultServiceEvents } from '@feathersjs/feathers'
import { createDebug } from '@feathersjs/commons'
import { Channel } from './channel/base'
import { CombinedChannel } from './channel/combined'

const debug = createDebug('@feathersjs/transport-commons/channels/mixins')
const PUBLISHERS = Symbol.for('@feathersjs/transport-commons/publishers')
const CHANNELS = Symbol.for('@feathersjs/transport-commons/channels')
const ALL_EVENTS = Symbol.for('@feathersjs/transport-commons/all-events')

export const keys = {
  PUBLISHERS: PUBLISHERS as typeof PUBLISHERS,
  CHANNELS: CHANNELS as typeof CHANNELS,
  ALL_EVENTS: ALL_EVENTS as typeof ALL_EVENTS
}

export interface ChannelMixin {
  [CHANNELS]: { [key: string]: Channel }
  channel(...names: string[]): Channel
}

export function channelMixin() {
  const mixin: ChannelMixin = {
    [CHANNELS]: {},

    channel(...names: string[]): Channel {
      debug('Returning channels', names)

      if (names.length === 0) {
        throw new Error('app.channel needs at least one channel name')
      }

      if (names.length === 1) {
        const [name] = names

        if (Array.isArray(name)) {
          return this.channel(...name)
        }

        if (!this[CHANNELS][name]) {
          const channel = new Channel()

          channel.once('empty', () => {
            channel.removeAllListeners()
            delete this[CHANNELS][name]
          })

          this[CHANNELS][name] = channel
        }

        return this[CHANNELS][name]
      }

      const channels = names.map((name) => this.channel(name))

      return new CombinedChannel(channels)
    }
  }

  return mixin
}

export type Event = string | typeof ALL_EVENTS

export type Publisher<T = any, A = Application, S = any> = (
  data: T,
  context: HookContext<A, S>
) => Channel | Channel[] | void | Promise<Channel | Channel[] | void>

export interface PublishMixin<T = any> {
  [PUBLISHERS]: { [ALL_EVENTS]?: Publisher<T>; [key: string]: Publisher<T> }
  publish(event: Event, publisher: Publisher<T>): this
  registerPublisher(event: Event, publisher: Publisher<T>): this
}

export function publishMixin() {
  const result: PublishMixin = {
    [PUBLISHERS]: {},

    publish(...args) {
      return this.registerPublisher(...args)
    },

    registerPublisher(event, publisher) {
      debug('Registering publisher', event)

      if (!publisher && typeof event === 'function') {
        publisher = event
        event = ALL_EVENTS
      }

      const { serviceEvents = defaultServiceEvents } = getServiceOptions(this) || {}

      if (event !== ALL_EVENTS && !serviceEvents.includes(event)) {
        throw new Error(`'${event.toString()}' is not a valid service event`)
      }

      const publishers = this[PUBLISHERS]

      publishers[event] = publisher

      return this
    }
  }

  return result
}