fix: new view counting method

Author: diced

src/server/routes/files.dy.ts

@@ -1,10 +1,6 @@
-import { parseRange } from '@/lib/api/range';
-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 { FastifyReply, FastifyRequest } from 'fastify';
+import { rawFileHandler } from './raw/[id]';
 
 type Params = {
   id: string;
@@ -15,14 +11,11 @@ type Query = {
   download?: string;
 };
 
-const logger = log('routes').c('files');
-
 export async function filesRoute(
   req: FastifyRequest<{ Params: Params; Querystring: Query }>,
   res: FastifyReply,
 ) {
   const { id } = req.params;
-  const { pw, download } = req.query;
   const file = await prisma.file.findFirst({
     where: {
       name: decodeURIComponent(id),
@@ -33,120 +26,8 @@ export async function filesRoute(
   });
   if (!file) return res.callNotFound();
 
-  if (file.deletesAt && file.deletesAt <= new Date()) {
-    try {
-      await datasource.delete(file.name);
-      await prisma.file.delete({
-        where: {
-          id: file.id,
-        },
-      });
-    } catch (e) {
-      logger
-        .error('failed to delete file on expiration', {
-          id: file.id,
-        })
-        .error(e as Error);
-    }
-    return res.callNotFound();
-  }
-  if (file.maxViews && file.views >= file.maxViews) {
-    if (!config.features.deleteOnMaxViews) return res.callNotFound();
-    try {
-      await datasource.delete(file.name);
-      await prisma.file.delete({
-        where: {
-          id: file.id,
-        },
-      });
-    } catch (e) {
-      logger
-        .error('failed to delete file on max views', {
-          id: file.id,
-        })
-        .error(e as Error);
-    }
-    return res.callNotFound();
-  }
   if (file.User?.view.enabled) return res.redirect(`/view/${encodeURIComponent(file.name)}`);
   if (file.type.startsWith('text/')) return res.redirect(`/view/${encodeURIComponent(file.name)}`);
-  const stream = await datasource.get(file.name);
-  if (!stream) return res.callNotFound();
-  if (file.password) {
-    if (!pw) return res.redirect(`/view/${encodeURIComponent(file.name)}`);
-    const verified = await verifyPassword(pw as string, file.password!);
-    if (!verified) {
-      logger.warn('password protected file accessed with an incorrect password', { id: file.id, ip: req.ip });
-      return res.callNotFound();
-    }
-  }
-  if (!req.headers.range) {
-    await prisma.file.update({
-      where: {
-        id: file.id,
-      },
-      data: {
-        views: {
-          increment: 1,
-        },
-      },
-    });
-  }
-  const size = file?.size || (await datasource.size(file?.name ?? id));
-  if (req.headers.range) {
-    const [start, end] = parseRange(req.headers.range, size);
-    if (start >= size || end >= size) {
-      const buf = await datasource.get(file?.name ?? id);
-      if (!buf) return res.callNotFound();
-      return res
-        .type(file?.type || 'application/octet-stream')
-        .headers({
-          'Content-Length': size,
-          ...(file?.originalName
-            ? {
-                'Content-Disposition': `${download ? 'attachment; ' : ''}filename="${encodeURIComponent(file.originalName)}"`,
-              }
-            : download && {
-                'Content-Disposition': 'attachment;',
-              }),
-        })
-        .status(416)
-        .send(buf);
-    }
-    const buf = await datasource.range(file?.name ?? id, start || 0, end);
-    if (!buf) return res.callNotFound();
-    return res
-      .type(file?.type || 'application/octet-stream')
-      .headers({
-        'Content-Range': `bytes ${start}-${end}/${size}`,
-        'Accept-Ranges': 'bytes',
-        'Content-Length': end - start + 1,
-        ...(file?.originalName
-          ? {
-              'Content-Disposition': `${download ? 'attachment; ' : ''}filename="${encodeURIComponent(file.originalName)}"`,
-            }
-          : download && {
-              'Content-Disposition': 'attachment;',
-            }),
-      })
-      .status(206)
-      .send(buf);
-  }
-  const buf = await datasource.get(file?.name ?? id);
-  if (!buf) return res.callNotFound();
-  return res
-    .type(file?.type || 'application/octet-stream')
-    .headers({
-      'Content-Length': size,
-      'Accept-Ranges': 'bytes',
-      ...(file?.originalName
-        ? {
-            'Content-Disposition': `${download ? 'attachment; ' : ''}filename="${encodeURIComponent(file.originalName)}"`,
-          }
-        : download && {
-            'Content-Disposition': 'attachment;',
-          }),
-    })
-    .status(200)
-    .send(buf);
+
+  return rawFileHandler(req, res);
 }