feat: update middleware config

theodorusclarence · Feb 11, 2023 · c03f562 · c03f562 · vercel · Feb 11, 2023
1 parent d29e348
commit c03f562
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 16 deletions.
diff --git a/src/pages/_middleware.ts → src/middleware.ts b/src/pages/_middleware.ts → src/middleware.ts
@@ -1,8 +1,20 @@
 import { NextRequest, NextResponse } from 'next/server';
 
-import { getUrlBySlug } from '@/lib/notion';
+import { getAllLinkCategories, getUrlBySlug } from '@/lib/notion';
 
 export default async function middleware(req: NextRequest) {
+  if (req.nextUrl.pathname.startsWith('/c/')) {
+    const path = req.nextUrl.pathname.split('/')[2];
+
+    const categories = await getAllLinkCategories();
+
+    if (categories.includes(path)) {
+      return;
+    } else {
+      return NextResponse.redirect('/404');
+    }
+  }
+
   const path = req.nextUrl.pathname.split('/')[1];
   const whitelist = [
     'favicons',

diff --git a/src/pages/c/_middleware.ts b/src/pages/c/_middleware.ts