Momento/memento-note/app/api/ai/personas/route.ts

import { NextRequest, NextResponse } from 'next/server'
import { auth } from '@/auth'
import { getSystemConfig } from '@/lib/config'
import { getTagsProvider } from '@/lib/ai/factory'
import { checkEntitlementOrThrow, QuotaExceededError, incrementUsageAsync } from '@/lib/entitlements'

export type PersonaId = 'engineer' | 'financial' | 'customer' | 'skeptic' | 'optimist'

interface PersonaDef {
  id: PersonaId
  label: string
  description: string
  systemPrompt: string
}

const PERSONAS: PersonaDef[] = [
  {
    id: 'engineer',
    label: 'Ingénieur',
    description: 'Faisabilité technique, complexité, risques d\'implémentation',
    systemPrompt: `Tu es un ingénieur senior pragmatique et rigoureux. Analyse les notes ci-dessous en te concentrant sur : 
la faisabilité technique, les risques d'implémentation, la complexité cachée, les dépendances et contraintes techniques. 
Fournis 3 à 5 commentaires concis et 2 à 3 questions techniques précises qui obligent à réfléchir plus profondément.
Réponds dans la même langue que la note. Format : commentaires en liste puis questions en liste.`,
  },
  {
    id: 'financial',
    label: 'Financier',
    description: 'ROI, coûts, viabilité économique, risques financiers',
    systemPrompt: `Tu es un analyste financier exigeant. Analyse les notes ci-dessous sous l'angle économique :
ROI potentiel, coûts cachés, modèle de revenus, risques financiers, allocation de ressources, viabilité à long terme.
Fournis 3 à 5 observations financières et 2 à 3 questions qui remettent en cause les hypothèses économiques.
Réponds dans la même langue que la note. Format : observations en liste puis questions en liste.`,
  },
  {
    id: 'customer',
    label: 'Client',
    description: 'Expérience utilisateur, besoins réels, valeur perçue',
    systemPrompt: `Tu es un client exigeant qui recherche de la valeur concrète. Analyse les notes ci-dessous du point de vue de l'utilisateur final :
clarté de la valeur ajoutée, frictions dans l'expérience, besoins non satisfaits, adéquation au problème réel.
Fournis 3 à 5 réactions authentiques de client et 2 à 3 questions sur l'utilité réelle.
Réponds dans la même langue que la note. Format : réactions en liste puis questions en liste.`,
  },
  {
    id: 'skeptic',
    label: 'Sceptique',
    description: 'Hypothèses contestables, points faibles, angles morts',
    systemPrompt: `Tu es un expert sceptique et critique constructif. Analyse les notes ci-dessous en cherchant :
les hypothèses non validées, les points faibles, les angles morts, les contradictions internes, les risques ignorés.
Sois direct et précis. Fournis 3 à 5 critiques fondées et 2 à 3 questions qui remettent en cause les fondements.
Réponds dans la même langue que la note. Format : critiques en liste puis questions en liste.`,
  },
  {
    id: 'optimist',
    label: 'Optimiste',
    description: 'Opportunités sous-estimées, potentiel inexploité, synergies',
    systemPrompt: `Tu es un stratège visionnaire et enthousiaste. Analyse les notes ci-dessous pour identifier :
les opportunités sous-estimées, le potentiel inexploité, les synergies possibles, les effets de levier, les succès rapides atteignables.
Fournis 3 à 5 opportunités concrètes et 2 à 3 questions qui ouvrent de nouvelles perspectives.
Réponds dans la même langue que la note. Format : opportunités en liste puis questions en liste.`,
  },
]

export async function POST(request: NextRequest) {
  try {
    const session = await auth()
    if (!session?.user?.id) {
      return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
    }

    const { content, title, personaId } = await request.json()

    if (!content || typeof content !== 'string') {
      return NextResponse.json({ error: 'Content is required' }, { status: 400 })
    }

    if (!personaId) {
      return NextResponse.json({ error: 'personaId is required' }, { status: 400 })
    }

    const persona = PERSONAS.find(p => p.id === personaId)
    if (!persona) {
      return NextResponse.json({ error: 'Invalid personaId' }, { status: 400 })
    }

    // Quota check (reuse reformulate quota)
    try {
      await checkEntitlementOrThrow(session.user.id, 'reformulate')
    } catch (err) {
      if (err instanceof QuotaExceededError) {
        const isTierLocked = err.currentQuota === 0
        return NextResponse.json({
          error: isTierLocked ? 'feature_locked' : 'quota_exceeded',
          errorKey: isTierLocked ? 'ai.featureLocked' : 'ai.quotaExceeded',
          upgradeTier: err.upgradeTier,
          quotaExceeded: true,
        }, { status: 402 })
      }
      throw err
    }

    const config = await getSystemConfig()
    const provider = getTagsProvider(config)

    const noteText = title ? `Titre : ${title}\n\n${content}` : content
    // Strip HTML tags for cleaner analysis
    const plainText = noteText.replace(/<[^>]+>/g, ' ').replace(/\s+/g, ' ').trim()

    const fullPrompt = `${persona.systemPrompt}\n\n---\nNOTE À ANALYSER :\n${plainText}`
    const result = await provider.generateText(fullPrompt)

    incrementUsageAsync(session.user.id, 'reformulate')

    return NextResponse.json({
      personaId: persona.id,
      personaLabel: persona.label,
      analysis: result,
    })
  } catch (error) {
    console.error('Personas API error:', error)
    return NextResponse.json({ error: 'Failed to generate persona analysis' }, { status: 500 })
  }
}

// GET — return available personas (no auth needed)
export async function GET() {
  return NextResponse.json({
    personas: PERSONAS.map(({ id, label, description }) => ({ id, label, description })),
  })
}