fix: missing user check in screenshot api endpoint

This commit is contained in:
Huskydog9988
2025-05-15 18:28:08 -04:00
parent 4fbc730490
commit 21eec081ee
2 changed files with 23 additions and 2 deletions

View File

@ -13,5 +13,16 @@ export default defineEventHandler(async (h3) => {
statusMessage: "Missing screenshot ID",
});
return await screenshotManager.delete(screenshotId);
const result = await screenshotManager.get(screenshotId);
if (!result)
throw createError({
statusCode: 400,
statusMessage: "Incorrect screenshot ID",
});
else if (result.userId !== userId)
throw createError({
statusCode: 403,
});
await screenshotManager.delete(screenshotId);
});

View File

@ -13,5 +13,15 @@ export default defineEventHandler(async (h3) => {
statusMessage: "Missing screenshot ID",
});
return await screenshotManager.get(screenshotId);
const result = await screenshotManager.get(screenshotId);
if (!result)
throw createError({
statusCode: 400,
statusMessage: "Incorrect screenshot ID",
});
else if (result.userId !== userId)
throw createError({
statusCode: 403,
});
return result;
});