From 8a8fbccfaa88d08815023290689cf4cb58d4d992 Mon Sep 17 00:00:00 2001 From: bennykok Date: Wed, 31 Jan 2024 16:45:32 +0800 Subject: [PATCH] fix: gif file not returning file url. --- web/src/server/getRunsData.tsx | 32 +++++++++++++++++--------------- 1 file changed, 17 insertions(+), 15 deletions(-) diff --git a/web/src/server/getRunsData.tsx b/web/src/server/getRunsData.tsx index 456f9d6..25e9076 100644 --- a/web/src/server/getRunsData.tsx +++ b/web/src/server/getRunsData.tsx @@ -49,24 +49,26 @@ export async function getRunsData(run_id: string, user?: APIKeyUserType) { for (let i = 0; i < data.outputs.length; i++) { const output = data.outputs[i]; - if (output.data?.images !== undefined) { - for (let j = 0; j < output.data?.images.length; j++) { - const element = output.data?.images[j]; - element.url = replaceCDNUrl( - `${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/outputs/runs/${data.id}/${element.filename}` - ); - } - } else if (output.data?.files !== undefined) { - for (let j = 0; j < output.data?.files.length; j++) { - const element = output.data?.files[j]; - element.url = replaceCDNUrl( - `${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/outputs/runs/${data.id}/${element.filename}` - ); - } - } + if (output.data?.images !== undefined) + replaceUrls(output.data?.images, data.id); + + if (output.data?.files !== undefined) + replaceUrls(output.data?.files, data.id); + + if (output.data?.gifs !== undefined) + replaceUrls(output.data?.gifs, data.id); } } } return data; } + +function replaceUrls(dataType: any[], dataId: string) { + for (let j = 0; j < dataType.length; j++) { + const element = dataType[j]; + element.url = replaceCDNUrl( + `${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/outputs/runs/${dataId}/${element.filename}`, + ); + } +}