HealthLog/src/app/api/export/route.ts

import { prisma } from "@/lib/db";
import { apiHandler, requireAuth } from "@/lib/api-handler";
import { annotate } from "@/lib/logging/context";
import { auditLog } from "@/lib/auth/audit";
import { apiError, getClientIp } from "@/lib/api-response";
import { checkRateLimit } from "@/lib/rate-limit";
import {
  toCSV,
  formatMeasurementsForExport,
  formatMedicationsForExport,
  formatIntakeEventsForExport,
  formatMoodEntriesForExport,
} from "@/lib/export";
import { NextRequest, NextResponse } from "next/server";

/**
 * Export user data.
 * Query params:
 *   format: "csv" | "json" (default: json)
 *   type: "measurements" | "medications" | "intake" | "all" (default: all)
 *
 * Re-authentication is required via session (user must be logged in).
 */
export const GET = apiHandler(async (request: NextRequest) => {
  const { user } = await requireAuth();

  const rl = await checkRateLimit(`export:${user.id}`, 10, 60 * 60 * 1000);
  if (!rl.allowed) {
    return apiError("Maximum 10 exports per hour", 429);
  }

  annotate({ action: { name: "export.download" } });

  const { searchParams } = new URL(request.url);
  const format = searchParams.get("format") ?? "json";
  const type = searchParams.get("type") ?? "all";

  if (!["csv", "json"].includes(format)) {
    return apiError("Format must be csv or json", 422);
  }
  if (!["measurements", "medications", "intake", "mood", "all"].includes(type)) {
    return apiError("Type must be measurements, medications, intake, mood, or all", 422);
  }

  const userId = user.id;

  const data: Record<string, unknown> = {};

  if (type === "measurements" || type === "all") {
    const measurements = await prisma.measurement.findMany({
      where: { userId },
      orderBy: { measuredAt: "desc" },
    });
    data.measurements =
      format === "csv"
        ? toCSV(formatMeasurementsForExport(measurements))
        : formatMeasurementsForExport(measurements);
  }

  if (type === "medications" || type === "all") {
    const medications = await prisma.medication.findMany({
      where: { userId },
      include: { schedules: true },
      orderBy: { createdAt: "desc" },
    });
    data.medications =
      format === "csv"
        ? toCSV(formatMedicationsForExport(medications))
        : formatMedicationsForExport(medications);
  }

  if (type === "intake" || type === "all") {
    const events = await prisma.medicationIntakeEvent.findMany({
      where: { userId },
      include: { medication: { select: { name: true } } },
      orderBy: { scheduledFor: "desc" },
    });
    data.intakeEvents =
      format === "csv"
        ? toCSV(formatIntakeEventsForExport(events))
        : formatIntakeEventsForExport(events);
  }

  if (type === "mood" || type === "all") {
    const moodEntries = await prisma.moodEntry.findMany({
      where: { userId },
      orderBy: { moodLoggedAt: "desc" },
    });
    data.moodEntries =
      format === "csv"
        ? toCSV(formatMoodEntriesForExport(moodEntries))
        : formatMoodEntriesForExport(moodEntries);
  }

  await auditLog("export.download", {
    userId,
    ipAddress: getClientIp(request),
    details: { format, type },
  });

  annotate({ meta: { export_format: format, export_type: type } });

  if (format === "csv") {
    // For CSV with all types, combine into separate sections
    const csvParts: string[] = [];
    if (data.measurements) {
      csvParts.push("# Measurements\n" + data.measurements);
    }
    if (data.medications) {
      csvParts.push("# Medications\n" + data.medications);
    }
    if (data.intakeEvents) {
      csvParts.push("# Intake Events\n" + data.intakeEvents);
    }
    if (data.moodEntries) {
      csvParts.push("# Mood Entries\n" + data.moodEntries);
    }
    const csvContent = csvParts.join("\n\n");

    return new NextResponse(csvContent, {
      status: 200,
      headers: {
        "Content-Type": "text/csv; charset=utf-8",
        "Content-Disposition": `attachment; filename="healthlog-export-${new Date().toISOString().slice(0, 10)}.csv"`,
      },
    });
  }

  return NextResponse.json(
    { data },
    {
      status: 200,
      headers: {
        "Content-Disposition": `attachment; filename="healthlog-export-${new Date().toISOString().slice(0, 10)}.json"`,
      },
    },
  );
});