Infrastructure/zipline
1
Fork 0
Code Issues Pull requests Activity

fix: views increasing on /r

Browse source
This commit is contained in:
diced 2022-11-02 09:20:09 -07:00
parent 988b61e459
commit 3c616f4f6f
No known key found for this signature in database
GPG key ID: 370BD1BA142842D1
1 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/server/index.ts
View file

@ -142,7 +142,7 @@ async function start() {
if (!image) await rawFile(req, res, nextServer, params.id);
else {
const failed = await preImage(image, prisma);
const failed = await preImage(image, prisma, true);
if (failed) return nextServer.render404(req, res as ServerResponse);
if (image.password) {
@ -193,7 +193,7 @@ async function start() {
}, config.core.invites_interval * 1000);
}
async function preImage(image: Image, prisma: PrismaClient) {
async function preImage(image: Image, prisma: PrismaClient, ignoreViews = false) {
if (image.expires_at && image.expires_at < new Date()) {
await datasource.delete(image.file);
await prisma.image.delete({ where: { id: image.id } });
@ -203,18 +203,20 @@ async function preImage(image: Image, prisma: PrismaClient) {
return true;
}
const nImage = await prisma.image.update({
where: { id: image.id },
data: { views: { increment: 1 } },
});
if (!ignoreViews) {
const nImage = await prisma.image.update({
where: { id: image.id },
data: { views: { increment: 1 } },
});
if (nImage.maxViews && nImage.views >= nImage.maxViews) {
await datasource.delete(image.file);
await prisma.image.delete({ where: { id: image.id } });
if (nImage.maxViews && nImage.views >= nImage.maxViews) {
await datasource.delete(image.file);
await prisma.image.delete({ where: { id: image.id } });
Logger.get('file').info(`File ${image.file} has been deleted due to max views (${nImage.maxViews})`);
Logger.get('file').info(`File ${image.file} has been deleted due to max views (${nImage.maxViews})`);
return true;
return true;
}
}
return false;