MOPC-Portal/src/app/api/live-voting/stream/route.ts

import { NextRequest } from 'next/server'
import { prisma } from '@/lib/prisma'

export const dynamic = 'force-dynamic'

export async function GET(request: NextRequest): Promise<Response> {
  const { searchParams } = new URL(request.url)
  const sessionId = searchParams.get('sessionId')

  if (!sessionId) {
    return new Response(JSON.stringify({ error: 'sessionId is required' }), {
      status: 400,
      headers: { 'Content-Type': 'application/json' },
    })
  }

  // Verify the session exists
  const session = await prisma.liveVotingSession.findUnique({
    where: { id: sessionId },
    select: { id: true, status: true },
  })

  if (!session) {
    return new Response(JSON.stringify({ error: 'Session not found' }), {
      status: 404,
      headers: { 'Content-Type': 'application/json' },
    })
  }

  const encoder = new TextEncoder()

  const stream = new ReadableStream({
    async start(controller) {
      // Track state for change detection
      let lastVoteCount = -1
      let lastProjectId: string | null = null
      let lastStatus: string | null = null

      const sendEvent = (event: string, data: unknown) => {
        const payload = `event: ${event}\ndata: ${JSON.stringify(data)}\n\n`
        controller.enqueue(encoder.encode(payload))
      }

      // Send initial connection event
      sendEvent('connected', { sessionId, timestamp: new Date().toISOString() })

      const poll = async () => {
        try {
          const currentSession = await prisma.liveVotingSession.findUnique({
            where: { id: sessionId },
            select: {
              status: true,
              currentProjectId: true,
              currentProjectIndex: true,
              votingEndsAt: true,
            },
          })

          if (!currentSession) {
            sendEvent('session_status', { status: 'DELETED' })
            controller.close()
            return false
          }

          // Check for status changes
          if (lastStatus !== null && currentSession.status !== lastStatus) {
            sendEvent('session_status', {
              status: currentSession.status,
              timestamp: new Date().toISOString(),
            })
          }
          lastStatus = currentSession.status

          // Check for project changes
          if (
            lastProjectId !== null &&
            currentSession.currentProjectId !== lastProjectId
          ) {
            sendEvent('project_change', {
              projectId: currentSession.currentProjectId,
              projectIndex: currentSession.currentProjectIndex,
              timestamp: new Date().toISOString(),
            })
          }
          lastProjectId = currentSession.currentProjectId

          // Check for vote updates on the current project
          if (currentSession.currentProjectId) {
            const voteCount = await prisma.liveVote.count({
              where: {
                sessionId,
                projectId: currentSession.currentProjectId,
              },
            })

            if (lastVoteCount !== -1 && voteCount !== lastVoteCount) {
              // Get the latest vote info
              const latestVotes = await prisma.liveVote.findMany({
                where: {
                  sessionId,
                  projectId: currentSession.currentProjectId,
                },
                select: {
                  score: true,
                  isAudienceVote: true,
                  votedAt: true,
                },
                orderBy: { votedAt: 'desc' },
                take: 1,
              })

              const avgScore = await prisma.liveVote.aggregate({
                where: {
                  sessionId,
                  projectId: currentSession.currentProjectId,
                },
                _avg: { score: true },
                _count: true,
              })

              sendEvent('vote_update', {
                projectId: currentSession.currentProjectId,
                totalVotes: voteCount,
                averageScore: avgScore._avg.score,
                latestVote: latestVotes[0] || null,
                timestamp: new Date().toISOString(),
              })
            }
            lastVoteCount = voteCount
          }

          // Stop polling if session is completed
          if (currentSession.status === 'COMPLETED') {
            sendEvent('session_status', {
              status: 'COMPLETED',
              timestamp: new Date().toISOString(),
            })
            controller.close()
            return false
          }

          return true
        } catch (error) {
          console.error('[SSE] Poll error:', error)
          return true // Keep trying
        }
      }

      // Initial poll to set baseline state
      const shouldContinue = await poll()
      if (!shouldContinue) return

      // Poll every 2 seconds
      const interval = setInterval(async () => {
        const cont = await poll()
        if (!cont) {
          clearInterval(interval)
        }
      }, 2000)

      // Clean up on abort
      request.signal.addEventListener('abort', () => {
        clearInterval(interval)
        try {
          controller.close()
        } catch {
          // Stream may already be closed
        }
      })
    },
  })

  return new Response(stream, {
    headers: {
      'Content-Type': 'text/event-stream',
      'Cache-Control': 'no-cache',
      'Connection': 'keep-alive',
    },
  })
}