81 lines
2.1 KiB
TypeScript
81 lines
2.1 KiB
TypeScript
import { NextRequest, NextResponse } from "next/server";
|
|
import { eq, sql } from "drizzle-orm";
|
|
import { db } from "@/lib/db";
|
|
import { lobsters, tasks } from "@/lib/db/schema";
|
|
import {
|
|
incrementGlobalStat,
|
|
incrementHourlyActivity,
|
|
publishEvent,
|
|
} from "@/lib/redis";
|
|
import { validateApiKey } from "@/lib/auth/api-key";
|
|
import { taskSchema } from "@/lib/validators/schemas";
|
|
|
|
export async function POST(req: NextRequest) {
|
|
try {
|
|
const authHeader = req.headers.get("authorization");
|
|
if (!authHeader?.startsWith("Bearer ")) {
|
|
return NextResponse.json(
|
|
{ error: "Missing or invalid authorization header" },
|
|
{ status: 401 }
|
|
);
|
|
}
|
|
|
|
const apiKey = authHeader.slice(7);
|
|
const lobster = await validateApiKey(apiKey);
|
|
if (!lobster) {
|
|
return NextResponse.json({ error: "Invalid API key" }, { status: 401 });
|
|
}
|
|
|
|
const body = await req.json();
|
|
const parsed = taskSchema.safeParse(body);
|
|
if (!parsed.success) {
|
|
return NextResponse.json(
|
|
{ error: "Validation failed", details: parsed.error.flatten() },
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
const { summary, durationMs, model, toolsUsed } = parsed.data;
|
|
const now = new Date();
|
|
|
|
const insertResult = await db.insert(tasks).values({
|
|
lobsterId: lobster.id,
|
|
summary,
|
|
durationMs,
|
|
model: model ?? null,
|
|
toolsUsed: toolsUsed ?? null,
|
|
timestamp: now,
|
|
});
|
|
|
|
await db
|
|
.update(lobsters)
|
|
.set({ totalTasks: sql`${lobsters.totalTasks} + 1`, updatedAt: now })
|
|
.where(eq(lobsters.id, lobster.id));
|
|
|
|
await incrementGlobalStat("total_tasks");
|
|
await incrementGlobalStat("tasks_today");
|
|
await incrementHourlyActivity();
|
|
|
|
await publishEvent({
|
|
type: "task",
|
|
lobsterId: lobster.id,
|
|
lobsterName: lobster.name,
|
|
city: lobster.city,
|
|
country: lobster.country,
|
|
summary,
|
|
durationMs,
|
|
});
|
|
|
|
return NextResponse.json({
|
|
ok: true,
|
|
taskId: insertResult[0].insertId,
|
|
});
|
|
} catch (error) {
|
|
console.error("Task error:", error);
|
|
return NextResponse.json(
|
|
{ error: "Internal server error" },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|