134 lines
3.5 KiB
TypeScript
134 lines
3.5 KiB
TypeScript
import { NextRequest, NextResponse } from "next/server";
|
|
import { z } from "zod";
|
|
import { auth } from "@/app/(auth)/auth";
|
|
import {
|
|
queryThirdPartyImageMaterials,
|
|
ThirdPartyImageMaterialError,
|
|
} from "@/lib/agents/third-party-image-material-service";
|
|
import type {
|
|
AgentWorkflowMaterialQueryType,
|
|
AgentWorkflowMaterialResourceType,
|
|
} from "@/types/agent";
|
|
|
|
export const maxDuration = 180;
|
|
|
|
const querySchema = z.object({
|
|
resourceType: z.enum(["white-background", "task-detail"]).default("white-background"),
|
|
sku: z.string().trim().max(64).optional(),
|
|
spu: z.string().trim().max(64).optional(),
|
|
keyword: z.string().trim().max(64).optional(),
|
|
queryType: z.enum(["sku", "spu"]).optional(),
|
|
});
|
|
|
|
function jsonNoStore(body: Parameters<typeof NextResponse.json>[0], init?: ResponseInit) {
|
|
return NextResponse.json(body, {
|
|
...init,
|
|
headers: {
|
|
"Cache-Control": "no-store, max-age=0",
|
|
...(init?.headers || {}),
|
|
},
|
|
});
|
|
}
|
|
|
|
function resolveQuery(params: z.infer<typeof querySchema>) {
|
|
const spu = String(params.spu || "").trim();
|
|
const sku = String(params.sku || "").trim();
|
|
const keyword = String(params.keyword || "").trim();
|
|
|
|
if (spu) {
|
|
return {
|
|
queryType: "spu" as AgentWorkflowMaterialQueryType,
|
|
keyword: spu,
|
|
};
|
|
}
|
|
if (sku) {
|
|
return {
|
|
queryType: "sku" as AgentWorkflowMaterialQueryType,
|
|
keyword: sku,
|
|
};
|
|
}
|
|
if (keyword) {
|
|
return {
|
|
queryType: (params.queryType || "sku") as AgentWorkflowMaterialQueryType,
|
|
keyword,
|
|
};
|
|
}
|
|
return null;
|
|
}
|
|
|
|
export async function GET(request: NextRequest) {
|
|
try {
|
|
const session = await auth();
|
|
if (!session?.user?.id) {
|
|
return jsonNoStore({ error: "未登录" }, { status: 401 });
|
|
}
|
|
|
|
const parsed = querySchema.parse({
|
|
resourceType: request.nextUrl.searchParams.get("resourceType") || undefined,
|
|
sku: request.nextUrl.searchParams.get("sku") || undefined,
|
|
spu: request.nextUrl.searchParams.get("spu") || undefined,
|
|
keyword: request.nextUrl.searchParams.get("keyword") || undefined,
|
|
queryType: request.nextUrl.searchParams.get("queryType") || undefined,
|
|
});
|
|
const resolvedQuery = resolveQuery(parsed);
|
|
if (!resolvedQuery) {
|
|
return jsonNoStore(
|
|
{
|
|
success: false,
|
|
error: {
|
|
code: "MISSING_PARAM",
|
|
message: "sku 和 spu 二选一必填；也可以传 keyword + queryType",
|
|
},
|
|
},
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
const result = await queryThirdPartyImageMaterials({
|
|
resourceType: parsed.resourceType as AgentWorkflowMaterialResourceType,
|
|
...resolvedQuery,
|
|
});
|
|
|
|
return jsonNoStore(result);
|
|
} catch (error) {
|
|
if (error instanceof ThirdPartyImageMaterialError) {
|
|
return jsonNoStore(
|
|
{
|
|
success: false,
|
|
error: {
|
|
code: error.code,
|
|
message: error.message,
|
|
},
|
|
},
|
|
{ status: error.status }
|
|
);
|
|
}
|
|
|
|
if (error instanceof z.ZodError) {
|
|
return jsonNoStore(
|
|
{
|
|
success: false,
|
|
error: {
|
|
code: "INVALID_QUERY",
|
|
message: "第三方图片素材查询参数无效",
|
|
details: error.errors,
|
|
},
|
|
},
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
console.error("[Third Party Image Materials] Search failed:", error);
|
|
return jsonNoStore(
|
|
{
|
|
success: false,
|
|
error: {
|
|
code: "SERVER_ERROR",
|
|
message: "第三方图片素材查询失败，请稍后重试",
|
|
},
|
|
},
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|