import { describe, it, expect } from 'vitest'
import type { ClaudeJobKind } from '@prisma/client'
import { getKindPromptText, getIdeaPromptText } from '../src/lib/kind-prompts.js'

const KINDS: ClaudeJobKind[] = [
  'IDEA_GRILL',
  'IDEA_MAKE_PLAN',
  'TASK_IMPLEMENTATION',
  'SPRINT_IMPLEMENTATION',
  'PLAN_CHAT',
]

describe('getKindPromptText', () => {
  it.each(KINDS)('returnt non-empty content voor %s', (kind) => {
    const text = getKindPromptText(kind)
    expect(text.length).toBeGreaterThan(0)
  })

  it('TASK_IMPLEMENTATION-prompt verbiedt wait_for_job', () => {
    const text = getKindPromptText('TASK_IMPLEMENTATION')
    expect(text).toMatch(/GEEN.*wait_for_job/)
  })

  it('SPRINT_IMPLEMENTATION-prompt verbiedt job_heartbeat', () => {
    const text = getKindPromptText('SPRINT_IMPLEMENTATION')
    expect(text).toMatch(/GEEN.*job_heartbeat/)
  })

  it.each(['TASK_IMPLEMENTATION', 'SPRINT_IMPLEMENTATION', 'PLAN_CHAT'] as const)(
    '%s-prompt noemt $PAYLOAD_PATH als variabele',
    (kind) => {
      const text = getKindPromptText(kind)
      expect(text).toContain('$PAYLOAD_PATH')
    },
  )
})

describe('getIdeaPromptText (back-compat)', () => {
  it('returnt content voor IDEA_GRILL', () => {
    expect(getIdeaPromptText('IDEA_GRILL').length).toBeGreaterThan(0)
  })
  it('returnt content voor IDEA_MAKE_PLAN', () => {
    expect(getIdeaPromptText('IDEA_MAKE_PLAN').length).toBeGreaterThan(0)
  })
  it('returnt empty string voor non-idea kind', () => {
    expect(getIdeaPromptText('TASK_IMPLEMENTATION')).toBe('')
  })
})