Scrum4Me/actions/sprint-runs.ts

'use server'

import { revalidatePath } from 'next/cache'
import { cookies } from 'next/headers'
import { getIronSession } from 'iron-session'
import { z } from 'zod'
import { Prisma } from '@prisma/client'
import { prisma } from '@/lib/prisma'
import { SessionData, sessionOptions } from '@/lib/session'
import { parsePauseContext } from '@/lib/pause-context'

async function getSession() {
  return getIronSession<SessionData>(await cookies(), sessionOptions)
}

export type PreFlightBlockerType = 'task_no_plan' | 'open_question' | 'pbi_blocked'

export interface PreFlightBlocker {
  type: PreFlightBlockerType
  id: string
  label: string
}

const StartSprintRunInput = z.object({ sprint_id: z.string().min(1) })
const ResumeSprintInput = z.object({ sprint_id: z.string().min(1) })
const CancelSprintRunInput = z.object({ sprint_run_id: z.string().min(1) })

interface StartResultOk {
  ok: true
  sprint_run_id: string
  jobs_count: number
}

interface StartResultBlocked {
  ok: false
  error: 'PRE_FLIGHT_BLOCKED'
  blockers: PreFlightBlocker[]
}

interface ErrorResult {
  ok: false
  error: string
  code: number
}

type StartResult = StartResultOk | StartResultBlocked | ErrorResult

// startSprintRunCore is gedeeld tussen startSprintRunAction en resumeSprintAction.
// Voert de pre-flight uit, maakt een SprintRun + ClaudeJobs (in PBI→Story→Task
// volgorde) binnen één transactie. Aanroeper levert sprint_id, user_id en de
// transactionele Prisma-client.
async function startSprintRunCore(
  tx: Prisma.TransactionClient,
  sprint_id: string,
  user_id: string,
): Promise<StartResultOk | StartResultBlocked | ErrorResult> {
  const sprint = await tx.sprint.findUnique({
    where: { id: sprint_id },
    include: { product: true },
  })
  if (!sprint) return { ok: false, error: 'SPRINT_NOT_FOUND', code: 404 }
  if (sprint.status !== 'ACTIVE')
    return { ok: false, error: 'SPRINT_NOT_ACTIVE', code: 400 }

  const activeRun = await tx.sprintRun.findFirst({
    where: {
      sprint_id,
      status: { in: ['QUEUED', 'RUNNING', 'PAUSED'] },
    },
  })
  if (activeRun)
    return { ok: false, error: 'SPRINT_RUN_ALREADY_ACTIVE', code: 409 }

  const stories = await tx.story.findMany({
    where: { sprint_id, status: { not: 'DONE' } },
    include: {
      pbi: true,
      tasks: {
        where: { status: 'TO_DO' },
        orderBy: [{ priority: 'asc' }, { sort_order: 'asc' }],
      },
    },
    orderBy: [{ priority: 'asc' }, { sort_order: 'asc' }],
  })

  const blockers: PreFlightBlocker[] = []

  for (const s of stories) {
    for (const t of s.tasks) {
      if (!t.implementation_plan) {
        blockers.push({
          type: 'task_no_plan',
          id: t.id,
          label: `${t.code}: ${t.title}`,
        })
      }
    }
  }

  const openQuestions = await tx.claudeQuestion.findMany({
    where: { story: { sprint_id }, status: 'open' },
    select: { id: true, question: true },
  })
  for (const q of openQuestions) {
    blockers.push({
      type: 'open_question',
      id: q.id,
      label: q.question.slice(0, 80),
    })
  }

  const seenPbi = new Set<string>()
  for (const s of stories) {
    if (seenPbi.has(s.pbi.id)) continue
    seenPbi.add(s.pbi.id)
    if (s.pbi.status === 'BLOCKED' || s.pbi.status === 'FAILED') {
      blockers.push({
        type: 'pbi_blocked',
        id: s.pbi.id,
        label: `${s.pbi.code}: ${s.pbi.title}`,
      })
    }
  }

  if (blockers.length > 0) {
    return { ok: false, error: 'PRE_FLIGHT_BLOCKED', blockers }
  }

  const sprintRun = await tx.sprintRun.create({
    data: {
      sprint_id,
      started_by_id: user_id,
      status: 'QUEUED',
      pr_strategy: sprint.product.pr_strategy,
      started_at: new Date(),
    },
  })

  const orderedTasks = stories
    .slice()
    .sort(
      (a, b) =>
        a.pbi.priority - b.pbi.priority ||
        a.pbi.sort_order - b.pbi.sort_order ||
        a.priority - b.priority ||
        a.sort_order - b.sort_order,
    )
    .flatMap((s) => s.tasks)

  for (const t of orderedTasks) {
    await tx.claudeJob.create({
      data: {
        user_id,
        product_id: sprint.product_id,
        task_id: t.id,
        sprint_run_id: sprintRun.id,
        kind: 'TASK_IMPLEMENTATION',
        status: 'QUEUED',
      },
    })
  }

  return { ok: true, sprint_run_id: sprintRun.id, jobs_count: orderedTasks.length }
}

export async function startSprintRunAction(input: unknown): Promise<StartResult> {
  const session = await getSession()
  if (!session.userId) return { ok: false, error: 'Niet ingelogd', code: 403 }
  if (session.isDemo)
    return { ok: false, error: 'Niet beschikbaar in demo-modus', code: 403 }

  const parsed = StartSprintRunInput.safeParse(input)
  if (!parsed.success) return { ok: false, error: 'Validatie mislukt', code: 422 }

  const userId = session.userId
  const result = await prisma.$transaction((tx) =>
    startSprintRunCore(tx, parsed.data.sprint_id, userId),
  )

  if (result.ok) {
    revalidatePath(`/sprints/${parsed.data.sprint_id}`)
  }
  return result
}

export async function resumeSprintAction(input: unknown): Promise<StartResult> {
  const session = await getSession()
  if (!session.userId) return { ok: false, error: 'Niet ingelogd', code: 403 }
  if (session.isDemo)
    return { ok: false, error: 'Niet beschikbaar in demo-modus', code: 403 }

  const parsed = ResumeSprintInput.safeParse(input)
  if (!parsed.success) return { ok: false, error: 'Validatie mislukt', code: 422 }

  const userId = session.userId
  const sprint_id = parsed.data.sprint_id

  const result = await prisma.$transaction(async (tx) => {
    const sprint = await tx.sprint.findUnique({ where: { id: sprint_id } })
    if (!sprint)
      return { ok: false as const, error: 'SPRINT_NOT_FOUND', code: 404 }
    if (sprint.status !== 'FAILED')
      return { ok: false as const, error: 'SPRINT_NOT_FAILED', code: 400 }

    // Sprint terug naar ACTIVE
    await tx.sprint.update({
      where: { id: sprint_id },
      data: { status: 'ACTIVE', completed_at: null },
    })

    // FAILED stories binnen sprint terug naar IN_SPRINT (DONE blijft)
    await tx.story.updateMany({
      where: { sprint_id, status: 'FAILED' },
      data: { status: 'IN_SPRINT' },
    })

    // PBIs van die stories: FAILED → READY (BLOCKED met rust laten)
    const storyPbiIds = (
      await tx.story.findMany({
        where: { sprint_id },
        select: { pbi_id: true },
        distinct: ['pbi_id'],
      })
    ).map((s) => s.pbi_id)
    await tx.pbi.updateMany({
      where: { id: { in: storyPbiIds }, status: 'FAILED' },
      data: { status: 'READY' },
    })

    // FAILED tasks → TO_DO (DONE blijft)
    await tx.task.updateMany({
      where: { story: { sprint_id }, status: 'FAILED' },
      data: { status: 'TO_DO' },
    })

    return startSprintRunCore(tx, sprint_id, userId)
  })

  if (result.ok) {
    revalidatePath(`/sprints/${sprint_id}`)
  }
  return result
}

const ResumePausedSprintRunInput = z.object({ sprint_run_id: z.string().min(1) })

interface ResumePausedResultOk {
  ok: true
}

type ResumePausedResult = ResumePausedResultOk | ErrorResult

export async function resumePausedSprintRunAction(
  input: unknown,
): Promise<ResumePausedResult> {
  const session = await getSession()
  if (!session.userId) return { ok: false, error: 'Niet ingelogd', code: 403 }
  if (session.isDemo)
    return { ok: false, error: 'Niet beschikbaar in demo-modus', code: 403 }

  const parsed = ResumePausedSprintRunInput.safeParse(input)
  if (!parsed.success) return { ok: false, error: 'Validatie mislukt', code: 422 }

  const sprint_run_id = parsed.data.sprint_run_id

  const result = await prisma.$transaction(async (tx) => {
    const run = await tx.sprintRun.findUnique({
      where: { id: sprint_run_id },
      select: { id: true, status: true, sprint_id: true, pause_context: true },
    })
    if (!run) return { ok: false as const, error: 'SPRINT_RUN_NOT_FOUND', code: 404 }
    if (run.status !== 'PAUSED')
      return { ok: false as const, error: 'SPRINT_RUN_NOT_PAUSED', code: 400 }

    const ctx = parsePauseContext(run.pause_context)
    if (ctx) {
      await tx.claudeQuestion.updateMany({
        where: { id: ctx.claude_question_id, status: 'open' },
        data: { status: 'closed' },
      })
    }

    // RUNNING when there's still a claim active, otherwise QUEUED so the
    // worker picks up the next job on its next claim.
    const activeClaims = await tx.claudeJob.count({
      where: { sprint_run_id, status: { in: ['CLAIMED', 'RUNNING'] } },
    })

    await tx.sprintRun.update({
      where: { id: sprint_run_id },
      data: {
        status: activeClaims > 0 ? 'RUNNING' : 'QUEUED',
        pause_context: Prisma.JsonNull,
      },
    })

    return { ok: true as const, sprint_id: run.sprint_id }
  })

  if (result.ok && 'sprint_id' in result) {
    revalidatePath(`/sprints/${result.sprint_id}`)
    return { ok: true }
  }
  return result
}

interface CancelResultOk {
  ok: true
}

type CancelResult = CancelResultOk | ErrorResult

export async function cancelSprintRunAction(input: unknown): Promise<CancelResult> {
  const session = await getSession()
  if (!session.userId) return { ok: false, error: 'Niet ingelogd', code: 403 }
  if (session.isDemo)
    return { ok: false, error: 'Niet beschikbaar in demo-modus', code: 403 }

  const parsed = CancelSprintRunInput.safeParse(input)
  if (!parsed.success) return { ok: false, error: 'Validatie mislukt', code: 422 }

  const sprint_run_id = parsed.data.sprint_run_id

  const result = await prisma.$transaction(async (tx) => {
    const run = await tx.sprintRun.findUnique({ where: { id: sprint_run_id } })
    if (!run)
      return { ok: false as const, error: 'SPRINT_RUN_NOT_FOUND', code: 404 }
    if (!['QUEUED', 'RUNNING', 'PAUSED'].includes(run.status))
      return { ok: false as const, error: 'SPRINT_RUN_NOT_CANCELLABLE', code: 400 }

    await tx.sprintRun.update({
      where: { id: sprint_run_id },
      data: { status: 'CANCELLED', finished_at: new Date() },
    })

    // Cancel openstaande task-jobs binnen deze run.
    // Tasks/Stories/PBIs/Sprint blijven hun status — cancel ≠ fail.
    await tx.claudeJob.updateMany({
      where: {
        sprint_run_id,
        status: { in: ['QUEUED', 'CLAIMED', 'RUNNING'] },
      },
      data: {
        status: 'CANCELLED',
        finished_at: new Date(),
      },
    })

    return { ok: true as const, sprint_id: run.sprint_id }
  })

  if (result.ok && 'sprint_id' in result) {
    revalidatePath(`/sprints/${result.sprint_id}`)
    return { ok: true }
  }
  return result
}