import { NextRequest, NextResponse } from 'next/server'
import pool from '@/lib/db'

// PUT: Update a single assumption and trigger recompute
export async function PUT(request: NextRequest) {
  try {
    const body = await request.json()
    const { scenarioId, key, value } = body

    if (!scenarioId || !key || value === undefined) {
      return NextResponse.json({ error: 'scenarioId, key, and value are required' }, { status: 400 })
    }

    const client = await pool.connect()
    try {
      const jsonValue = JSON.stringify(value)
      await client.query(
        'UPDATE pitch_fm_assumptions SET value = $1 WHERE scenario_id = $2 AND key = $3',
        [jsonValue, scenarioId, key]
      )

      return NextResponse.json({ success: true })
    } finally {
      client.release()
    }
  } catch (error) {
    console.error('Update assumption error:', error)
    return NextResponse.json({ error: 'Failed to update assumption' }, { status: 500 })
  }
}