feat: config opt deleteonmaxviews

diced · Jul 21, 2023 · 485a1ae · 485a1ae
1 parent 7ca9a33
commit 485a1ae
Show file tree

Hide file tree

Showing 5 changed files with 47 additions and 2 deletions.
diff --git a/src/lib/config/read.ts b/src/lib/config/read.ts
@@ -45,6 +45,7 @@ export const PROP_TO_ENV: Record<string, string> = {
   'features.invites': 'FEATURES_INVITES',
   'features.userRegistration': 'FEATURES_USER_REGISTRATION',
   'features.oauthRegistration': 'FEATURES_OAUTH_REGISTRATION',
+  'features.deleteOnMaxViews': 'FEATURES_DELETE_ON_MAX_VIEWS',
 
   'website.title': 'WEBSITE_TITLE',
   'website.externalLinks': 'WEBSITE_EXTERNAL_LINKS',
@@ -99,6 +100,7 @@ export function readEnv() {
     env(PROP_TO_ENV['features.invites'], 'features.invites', 'boolean'),
     env(PROP_TO_ENV['features.userRegistration'], 'features.userRegistration', 'boolean'),
     env(PROP_TO_ENV['features.oauthRegistration'], 'features.oauthRegistration', 'boolean'),
+    env(PROP_TO_ENV['features.deleteOnMaxViews'], 'features.deleteOnMaxViews', 'boolean'),
 
     env(PROP_TO_ENV['website.title'], 'website.title', 'string'),
     env(PROP_TO_ENV['website.externalLinks'], 'website.externalLinks', 'json[]'),
@@ -151,6 +153,7 @@ export function readEnv() {
       invites: undefined,
       userRegistration: undefined,
       oauthRegistration: undefined,
+      deleteOnMaxViews: undefined,
     },
     website: {
       title: undefined,

diff --git a/src/lib/config/validate.ts b/src/lib/config/validate.ts
@@ -87,6 +87,7 @@ export const schema = z.object({
     invites: z.boolean().default(true),
     userRegistration: z.boolean().default(false),
     oauthRegistration: z.boolean().default(false),
+    deleteOnMaxViews: z.boolean().default(true),
   }),
   website: z.object({
     title: z.string().default('Zipline'),

diff --git a/src/server/index.ts b/src/server/index.ts
@@ -49,6 +49,8 @@ async function main() {
   await app.prepare();
 
   if (config.files.route === '/' && config.urls.route === '/') {
+    logger.debug('files & urls route are both /, using catch-all route');
+
     server.get('/:id', async (req, res) => {
       const { id } = req.params;
       const parsedUrl = parse(req.url!, true);

diff --git a/src/server/routes/files.ts b/src/server/routes/files.ts
@@ -1,10 +1,14 @@
+import { config } from '@/lib/config';
 import { verifyPassword } from '@/lib/crypto';
 import { datasource } from '@/lib/datasource';
 import { prisma } from '@/lib/db';
+import { log } from '@/lib/logger';
 import express, { Request, Response } from 'express';
 import { NextServer } from 'next/dist/server/next';
 import { parse } from 'url';
 
+const logger = log('server').c('files');
+
 export async function filesRoute(
   this: ReturnType<typeof express>,
   app: NextServer,
@@ -27,7 +31,22 @@ export async function filesRoute(
 
   if (!file) return app.render404(req, res, parsedUrl);
 
-  if (file.maxViews && file.views >= file.maxViews) return app.render404(req, res, parsedUrl);
+  if (file.maxViews && file.views >= file.maxViews) {
+    if (config.features.deleteOnMaxViews) {
+      await prisma.file.delete({
+        where: {
+          id: file.id,
+        },
+      });
+
+      logger.info(`${file.name} deleted due to reaching max views`, {
+        id: file.id,
+        views: file.views,
+      });
+    }
+
+    return app.render404(req, res, parsedUrl);
+  }
 
   if (file.User?.view.enabled) return res.redirect(`/view/${file.name}`);
 

diff --git a/src/server/routes/urls.ts b/src/server/routes/urls.ts
@@ -2,6 +2,10 @@ import { NextServer } from 'next/dist/server/next';
 import express, { Request, Response } from 'express';
 import { parse } from 'url';
 import { prisma } from '@/lib/db';
+import { config } from '@/lib/config';
+import { log } from '@/lib/logger';
+
+const logger = log('server').c('urls');
 
 export async function urlsRoute(
   this: ReturnType<typeof express>,
@@ -20,7 +24,23 @@ export async function urlsRoute(
   });
   if (!url) return app.render404(req, res, parsedUrl);
 
-  if (url.maxViews && url.views >= url.maxViews) return app.render404(req, res, parsedUrl);
+  if (url.maxViews && url.views >= url.maxViews) {
+    if (config.features.deleteOnMaxViews) {
+      await prisma.url.delete({
+        where: {
+          id: url.id,
+        },
+      });
+
+      logger.info(`${url.code} deleted due to reaching max views`, {
+        id: url.id,
+        views: url.views,
+        vanity: url.vanity ?? 'none',
+      });
+    }
+
+    return app.render404(req, res, parsedUrl);
+  }
 
   await prisma.url.update({
     where: {