import { NextResponse } from "next/server";
import { gte, and, isNotNull, sql } from "drizzle-orm";
import { db } from "@/lib/db";
import { lobsters } from "@/lib/db/schema";
import { getCacheHeatmap, setCacheHeatmap } from "@/lib/redis";

export async function GET() {
  try {
    const cached = await getCacheHeatmap();
    if (cached) {
      return NextResponse.json(JSON.parse(cached));
    }

    const fiveMinutesAgo = new Date(Date.now() - 5 * 60 * 1000);

    const activeLobsters = await db
      .select({
        city: lobsters.city,
        country: lobsters.country,
        latitude: lobsters.latitude,
        longitude: lobsters.longitude,
        count: sql<number>`count(*)`,
      })
      .from(lobsters)
      .where(
        and(
          gte(lobsters.lastHeartbeat, fiveMinutesAgo),
          isNotNull(lobsters.latitude),
          isNotNull(lobsters.longitude)
        )
      )
      .groupBy(
        lobsters.city,
        lobsters.country,
        lobsters.latitude,
        lobsters.longitude
      );

    const points = activeLobsters.map((row) => ({
      lat: Number(row.latitude),
      lng: Number(row.longitude),
      weight: row.count,
      lobsterCount: row.count,
      city: row.city,
      country: row.country,
    }));

    const lastUpdated = new Date().toISOString();
    const responseData = { points, lastUpdated };

    await setCacheHeatmap(JSON.stringify(responseData));

    return NextResponse.json(responseData);
  } catch (error) {
    console.error("Heatmap error:", error);
    return NextResponse.json(
      { error: "Internal server error" },
      { status: 500 }
    );
  }
}