99 lines
3.0 KiB
TypeScript
99 lines
3.0 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server';
|
|
import { auth } from '@/app/(auth)/auth';
|
|
import { db } from '@/lib/db';
|
|
import { canvasProject } from '@/lib/db/schema';
|
|
import { syncCanvasNodesForProject } from '@/lib/db/queries';
|
|
import { and, eq, desc } from 'drizzle-orm';
|
|
|
|
function resolveWorkspaceType(input: unknown) {
|
|
return input === "agent" || input === "video" ? input : "canvas";
|
|
}
|
|
|
|
function getDefaultWorkspaceTitle(workspaceType: ReturnType<typeof resolveWorkspaceType>) {
|
|
return workspaceType === "video" ? "未命名视频画布" : "未命名画布";
|
|
}
|
|
|
|
// 获取画布列表
|
|
export async function GET(request: NextRequest) {
|
|
try {
|
|
const session = await auth();
|
|
if (!session?.user?.id) {
|
|
return NextResponse.json({ success: false, error: '未登录' }, { status: 401 });
|
|
}
|
|
|
|
const workspaceType = resolveWorkspaceType(
|
|
request.nextUrl.searchParams.get("workspaceType")
|
|
);
|
|
|
|
const projects = await db
|
|
.select()
|
|
.from(canvasProject)
|
|
.where(
|
|
and(
|
|
eq(canvasProject.userId, session.user.id),
|
|
eq(canvasProject.workspaceType, workspaceType)
|
|
)
|
|
)
|
|
.orderBy(desc(canvasProject.updatedAt));
|
|
|
|
return NextResponse.json({ success: true, data: projects });
|
|
} catch (error) {
|
|
console.error('获取画布列表失败:', error);
|
|
return NextResponse.json({ success: false, error: '获取画布列表失败' }, { status: 500 });
|
|
}
|
|
}
|
|
|
|
// 创建画布
|
|
export async function POST(request: NextRequest) {
|
|
try {
|
|
const session = await auth();
|
|
if (!session?.user?.id) {
|
|
return NextResponse.json({ success: false, error: '未登录' }, { status: 401 });
|
|
}
|
|
|
|
const body = await request.json();
|
|
const { title, description, canvasData } = body;
|
|
const workspaceType = resolveWorkspaceType(body?.workspaceType);
|
|
|
|
const [insertedProject] = await db
|
|
.insert(canvasProject)
|
|
.values({
|
|
userId: session.user.id,
|
|
title: title || getDefaultWorkspaceTitle(workspaceType),
|
|
description,
|
|
workspaceType,
|
|
canvasData: canvasData || {
|
|
nodes: [],
|
|
edges: [],
|
|
viewport: { x: 0, y: 0, zoom: 1 },
|
|
},
|
|
})
|
|
.returning();
|
|
|
|
let newProject = insertedProject;
|
|
if (canvasData) {
|
|
const syncedCanvasData = await syncCanvasNodesForProject({
|
|
projectId: insertedProject.id,
|
|
userId: session.user.id,
|
|
canvasData,
|
|
});
|
|
const [updatedProject] = await db
|
|
.update(canvasProject)
|
|
.set({
|
|
canvasData: syncedCanvasData,
|
|
updatedAt: new Date(),
|
|
})
|
|
.where(eq(canvasProject.id, insertedProject.id))
|
|
.returning();
|
|
if (updatedProject) {
|
|
newProject = updatedProject;
|
|
}
|
|
}
|
|
|
|
return NextResponse.json({ success: true, data: newProject });
|
|
} catch (error) {
|
|
console.error('创建画布失败:', error);
|
|
return NextResponse.json({ success: false, error: '创建画布失败' }, { status: 500 });
|
|
}
|
|
}
|