chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(web): add custom file download button for runs output

Browse Source
This commit is contained in:
BennyKok
2023-12-20 23:14:01 +08:00
parent 8f67a136c3
commit ee68507755
10 changed files with 54 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/src/app/api/file-upload/route.ts
+2 -1
View File
@@ -6,12 +6,13 @@ import { z } from "zod";
const Request = z.object({
file_name: z.string(),
run_id: z.string(),
type: z.enum(["image/png", "image/jpeg"]),
type: z.string(),
});
export const dynamic = "force-dynamic";
export async function GET(request: Request) {
console.log("yo hello");
const [data, error] = await parseDataSafe(Request, request);
if (!data || error) return error;
web/src/app/api/update-run/route.ts
+2 -2
View File
@@ -17,7 +17,7 @@ export async function POST(request: Request) {
const { run_id, status, output_data } = data;
console.log(run_id, status, output_data);
// console.log(run_id, status, output_data);
if (output_data) {
const workflow_run_output = await db.insert(workflowRunOutputs).values({
@@ -25,7 +25,7 @@ export async function POST(request: Request) {
data: output_data,
});
} else if (status) {
console.log("status", status);
// console.log("status", status);
const workflow_run = await db
.update(workflowRunsTable)
.set({
web/src/app/api/view/route.ts
+4 -8
View File
@@ -1,12 +1,8 @@
import { replaceCDNUrl } from "@/server/resource";
import { NextResponse, type NextRequest } from "next/server";
import { getFileDownloadUrl } from "../../../server/getFileDownloadUrl";
export async function GET(request: NextRequest) {
const file = new URL(request.url).searchParams.get("file");
// console.log(`${process.env.SPACES_ENDPOINT}/comfyui-deploy/${file}`);
return NextResponse.redirect(
replaceCDNUrl(
`${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/${file}`
)
);
}
if (!file) return NextResponse.redirect("/");
return NextResponse.redirect(getFileDownloadUrl(file));
}