feat: added admin check on prompt pages

webdevcody · Oct 19, 2024 · 8545ffd · 8545ffd
1 parent 20a53d8
commit 8545ffd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/app/prompts/[promptId]/page.tsx b/app/prompts/[promptId]/page.tsx
@@ -18,6 +18,7 @@ export default function editPromtPage({
   });
   console.log(prompt);
   const updatePrompt = useMutation(api.prompts.updatePrompt);
+  const isAdmin = useQuery(api.users.isAdmin);
 
   const [promptName, setPromptName] = useState("");
   const [promptText, setPromptText] = useState("");
@@ -48,6 +49,10 @@ export default function editPromtPage({
     router.push("/prompts");
   };
 
+  if (!isAdmin) {
+    redirect("/");
+  }
+
   return (
     <div className="container mx-auto min-h-screen pb-24 py-12 gap-8">
       <form className="py-6 px-6 h-full" onSubmit={handleSubmit}>

diff --git a/app/prompts/page.tsx b/app/prompts/page.tsx
@@ -12,14 +12,16 @@ import {
 import { api } from "@/convex/_generated/api";
 import { useMutation, useQuery } from "convex/react";
 import Link from "next/link";
-import { useRouter,redirect } from "next/navigation";
+import { useRouter, redirect } from "next/navigation";
 
 const Page = () => {
   const prompts = useQuery(api.prompts.getAllPrompts);
   const enablePrompt = useMutation(api.prompts.enablePrompt);
   const deletePrompt = useMutation(api.prompts.deletePrompt);
   const createPrompt = useMutation(api.prompts.createPrompt);
 
+  const isAdmin = useQuery(api.users.isAdmin);
+
   const router = useRouter();
 
   const handleCreatePrompt = async () => {
@@ -30,6 +32,10 @@ const Page = () => {
     router.push(`/prompts/${newPromptId}`);
   };
 
+  if (!isAdmin) {
+    redirect("/");
+  }
+
   return (
     <div className="container mx-auto min-h-screen py-12 pb-24 gap-8">
       <div className="flex justify-between mb-4">