From b29bfeb8b1c52c53ff48f03e4eca05b2bd3bcf07 Mon Sep 17 00:00:00 2001
From: diced <pranaco2@gmail.com>
Date: Fri, 3 Sep 2021 17:20:43 -0700
Subject: [PATCH] fix(api): new way to handle non-embedded images to overwrite
 nextjs & supports discord now

---
 src/pages/[...id].tsx | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/pages/[...id].tsx b/src/pages/[...id].tsx
index 7b62cd0..21fb070 100644
--- a/src/pages/[...id].tsx
+++ b/src/pages/[...id].tsx
@@ -4,6 +4,7 @@ import { GetServerSideProps } from 'next';
 import { Box } from '@material-ui/core';
 import config from 'lib/config';
 import prisma from 'lib/prisma';
+import getFile from '../../server/static';
 
 export default function EmbeddedImage({ image, title, username, color, normal, embed }) {
   const dataURL = (route: string) => `${route}/${image.file}`;
@@ -80,10 +81,13 @@ export const getServerSideProps: GetServerSideProps = async (context) => {
 
   if (!image) return { notFound: true };
 
-  if (!image.embed) return { redirect: {
-    permanent: true,
-    destination: '/r/' + image.file,
-  } };
+  if (!image.embed) {
+    const data = await getFile(config.uploader.directory, id);
+    if (!data) return { notFound: true };
+
+    context.res.end(data);
+    return { props: {} };
+  };
 
   const user = await prisma.user.findFirst({
     select: {
@@ -96,11 +100,12 @@ export const getServerSideProps: GetServerSideProps = async (context) => {
     }
   });
   
-  if (!image.mimetype.startsWith('image')) return {
-    redirect: {
-      permanent: true,
-      destination: '/r/' + image.file,
-    }
+  if (!image.mimetype.startsWith('image')) {
+    const data = await getFile(config.uploader.directory, id);
+    if (!data) return { notFound: true };
+
+    context.res.end(data);
+    return { props: {} };
   };
 
   return {