Momento/memento-note/app/api/ai/batch-organize/route.ts

import { NextRequest, NextResponse } from 'next/server'
import { auth } from '@/auth'
import { batchOrganizationService } from '@/lib/ai/services'
import { hasUserAiConsent } from '@/lib/consent/server-consent'
import { reserveUsageOrThrow, QuotaExceededError } from '@/lib/entitlements'

/**
 * POST /api/ai/batch-organize - Create organization plan for notes in Inbox
 */
export async function POST(request: NextRequest) {
  try {
    const session = await auth()

    if (!session?.user?.id) {
      return NextResponse.json(
        { success: false, error: 'Unauthorized' },
        { status: 401 }
      )
    }

    // GDPR AI Consent check
    if (!(await hasUserAiConsent())) {
      return NextResponse.json(
        { success: false, error: 'ai_consent_required' },
        { status: 403 }
      )
    }

    // Get language from request headers or body
    let language = 'en'
    try {
      const body = await request.json()
      if (body.language) {
        language = body.language
      }
    } catch (e) {
      // If no body or invalid json, check headers
      const acceptLanguage = request.headers.get('accept-language')
      if (acceptLanguage) {
        language = acceptLanguage.split(',')[0].split('-')[0]
      }
    }

    await reserveUsageOrThrow(session.user.id, 'reformulate')

    // Create organization plan
    const plan = await batchOrganizationService.createOrganizationPlan(
      session.user.id,
      language
    )

    return NextResponse.json({
      success: true,
      data: plan,
    })
  } catch (error) {
    console.error('[batch-organize POST] Error:', error)
    return NextResponse.json(
      {
        success: false,
        error: error instanceof Error ? error.message : 'Failed to create organization plan',
      },
      { status: 500 }
    )
  }
}

/**
 * PUT /api/ai/batch-organize - Apply organization plan
 */
export async function PUT(request: NextRequest) {
  try {
    const session = await auth()

    if (!session?.user?.id) {
      return NextResponse.json(
        { success: false, error: 'Unauthorized' },
        { status: 401 }
      )
    }

    const body = await request.json()
    const { plan, selectedNoteIds } = body

    if (!plan || !Array.isArray(selectedNoteIds)) {
      return NextResponse.json(
        { success: false, error: 'Missing required fields: plan, selectedNoteIds' },
        { status: 400 }
      )
    }

    // Apply organization plan
    const movedCount = await batchOrganizationService.applyOrganizationPlan(
      session.user.id,
      plan,
      selectedNoteIds
    )

    return NextResponse.json({
      success: true,
      data: {
        movedCount,
      },
    })
  } catch (error) {
    return NextResponse.json(
      {
        success: false,
        error: error instanceof Error ? error.message : 'Failed to apply organization plan',
      },
      { status: 500 }
    )
  }
}