import type { PartyKitServer, Room } from 'partykit/server'
import { isValidEmoji } from '../shared/utils/emoji'
import CursorsParty from './cursors'

const cursorParties = new Map<string, Promise<CursorsParty>>()
function getCursors (room: Room) {
  let party = cursorParties.get(room.id)
  if (!party) {
    const instance = new CursorsParty(room)
    party = instance.onStart().then(() => instance, err => {
      cursorParties.delete(room.id)
      throw err
    })
    cursorParties.set(room.id, party)
  }
  return party
}

export default {
  async onConnect (ws, party, ctx) {
    if (party.id === 'cursors') {
      return getCursors(party).then(c => c.onConnect(ws, ctx))
    }

    // 1. send current vote count if applicable
    ws.send(`count:${await getCount(party)}`)

    // 2. let everyone know someone new is viewing the site
    party.broadcast(`connections:${[...party.getConnections()].length}`)

    // 3. send current feedback level
    if (party.id === 'feedback') {
      ws.send(`feedback:${JSON.stringify(await party.storage.get('feedback') || [])}`)
    }

    // 4. let people know if I'm streaming
    ws.send(`status:${(await party.storage.get('status')) || 'default'}`)
  },
  async onRequest (request, party) {
    if (request.method !== 'POST')
      return new Response('Invalid request', { status: 422 })

    const body = await request.text().then(r => (r ? JSON.parse(r) : {}))
    const { status, emoji, type = status ? 'status' : (emoji ? 'reaction' : 'vote') } = body as {
      type?: 'vote' | 'status' | 'feedback' | 'reaction'
      status?: string
      emoji?: string
    }

    // 5. allow one-off live voting via link
    if (type === 'vote') {
      const val = (await getCount(party)) + 1
      await party.storage.put('count', val)
      party.broadcast(`count:${val}`)
      return new Response(null, { status: 204 })
    }

    // 6. allow one-off feedback submissions via link
    if (type === 'feedback') {
      await party.storage.transaction(async tx => {
        const feedback = await tx.get<string[]>('feedback') || []
        feedback.push(status!)
        await tx.put('feedback', feedback)
        party.broadcast(`feedback:${JSON.stringify(feedback)}`)
      })
      return new Response(null, { status: 204 })
    }

    // 7. tell people if I'm going live
    if (type === 'status') {
      if (!status || !['live', 'default'].includes(status))
        return new Response('Invalid status', { status: 422 })

      if (request.headers.get('authorization') !== party.env.PARTYKIT_TOKEN)
        return new Response('Unauthorised', { status: 401 })

      party.storage.put('status', status)
      party.broadcast(`status:${status}`)
      return new Response(null, { status: 204 })
    }

    return new Response('Invalid request', { status: 422 })
  },
  async onMessage (message, ws, party) {
    if (party.id === 'cursors') {
      return getCursors(party).then(c => c.onMessage(message, ws))
    }

    const messageStr = message.toString()

    // 8. handle clearing votes from slide deck
    if (messageStr === 'clear') {
      if (party.id === 'feedback') {
        await party.storage.put('feedback', [])
        party.broadcast(`feedback:[]`)
      }
      else {
        await party.storage.put('count', 0)
        party.broadcast(`count:0`)
      }
    }

    // 7. handle reaction messages
    if (party.id === 'reactions' && messageStr.startsWith('reaction:')) {
      const emoji = messageStr.replace('reaction:', '')

      // Validate emoji
      if (!isValidEmoji(emoji)) {
        return
      }

      // Store the reaction
      await party.storage.transaction(async tx => {
        const reactions = await tx.get<string[]>('reactions') || []
        reactions.push(emoji)
        // Keep only the last 100 reactions
        if (reactions.length > 100) reactions.shift()
        await tx.put('reactions', reactions)
      })

      // Broadcast to all clients
      party.broadcast(messageStr)
    }
  },
  async onClose (ws, party) {
    if (party.id === 'cursors') {
      return getCursors(party).then(c => c.onClose(ws))
    }
  },
  async onError (ws, _err, party) {
    if (party.id === 'cursors') {
      return getCursors(party).then(c => c.onError(ws))
    }
  },
} satisfies PartyKitServer

const getCount = (room: Room) => room.storage.get<number>('count').then(r => r || 0)