atridad/pollo
1
0
Fork 0
Code Activity Actions

Merge pull request #30 from atridadl/dev

Browse source 
Role changes
This commit is contained in:
Atridad Lahiji 2023-07-29 21:46:42 -06:00 • committed by GitHub
commit 8fb8f5140f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 182 additions and 127 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
prisma/schema.prisma
View file

@ -1,8 +1,3 @@
enum RoleValue {
USER
ADMIN
}
generator client {
provider = "prisma-client-js"
}
@ -46,7 +41,6 @@ model Session {
model User {
id String @id @default(cuid())
role RoleValue @default(USER)
createdAt DateTime @default(now())
name String?
email String? @unique
@ -57,6 +51,9 @@ model User {
rooms Room[]
votes Vote[]
logs Log[]
isAdmin Boolean @default(false)
isVIP Boolean @default(false)
}
model VerificationToken {

2
src/components/Navbar.tsx
View file

@ -85,7 +85,7 @@ const Navbar: React.FC<NavbarProps> = ({ title }) => {
Profile
</Link>
</li>
{sessionData.user.role === "ADMIN" && (
{sessionData.user.isAdmin && (
<li>
<Link
about="Admin Page"

46
src/pages/admin/index.tsx
View file

@ -5,8 +5,8 @@ import { AiOutlineClear } from "react-icons/ai";
import { FaShieldAlt } from "react-icons/fa";
import { IoTrashBinOutline } from "react-icons/io5";
import { SiGithub, SiGoogle } from "react-icons/si";
import { GiStarFormation } from "react-icons/gi";
import { api } from "~/utils/api";
import type { Role } from "~/utils/types";
import { getServerAuthSession } from "../../server/auth";
export const getServerSideProps: GetServerSideProps = async (ctx) => {
@ -22,7 +22,7 @@ export const getServerSideProps: GetServerSideProps = async (ctx) => {
};
}
if (session.user.role !== "ADMIN") {
if (!session.user.isAdmin) {
ctx.res.statusCode = 403;
return {
redirect: {
@ -90,7 +90,8 @@ const AdminBody: React.FC = () => {
id: string;
}[];
id: string;
role: Role;
isAdmin: boolean;
isVIP: boolean;
name: string | null;
email: string | null;
}) => {
@ -119,7 +120,13 @@ const AdminBody: React.FC = () => {
},
});
const setRoleMutation = api.user.setRole.useMutation({
const setAdminMutation = api.user.setAdmin.useMutation({
onSuccess: async () => {
await refetchData();
},
});
const setVIPMutation = api.user.setVIP.useMutation({
onSuccess: async () => {
await refetchData();
},
@ -137,8 +144,12 @@ const AdminBody: React.FC = () => {
await clearSessionsMutation.mutateAsync();
};
const setUserRoleHandler = async (userId: string, role: Role) => {
await setRoleMutation.mutateAsync({ userId, role });
const setAdmin = async (userId: string, value: boolean) => {
await setAdminMutation.mutateAsync({ userId, value });
};
const setVIP = async (userId: string, value: boolean) => {
await setVIPMutation.mutateAsync({ userId, value });
};
const refetchData = async () => {
@ -263,19 +274,28 @@ const AdminBody: React.FC = () => {
</td>
<td>
<button className="m-2">
{ user.role === "ADMIN" ? (
{user.isAdmin ? (
<FaShieldAlt
className="text-xl inline-block text-primary"
onClick={ () =>
void setUserRoleHandler(user.id, "USER")
}
onClick={() => void setAdmin(user.id, false)}
/>
) : (
<FaShieldAlt
className="text-xl inline-block"
onClick={ () =>
void setUserRoleHandler(user.id, "ADMIN")
}
onClick={() => void setAdmin(user.id, true)}
/>
)}
</button>
<button className="m-2">
{user.isVIP ? (
<GiStarFormation
className="text-xl inline-block text-secondary"
onClick={() => void setVIP(user.id, false)}
/>
) : (
<GiStarFormation
className="text-xl inline-block"
onClick={() => void setVIP(user.id, true)}
/>
)}
</button>

6
src/pages/dashboard/index.tsx
View file

@ -8,6 +8,7 @@ import Link from "next/link";
import { useEffect, useState } from "react";
import { FaShieldAlt } from "react-icons/fa";
import { getServerAuthSession } from "~/server/auth";
import { GiStarFormation } from "react-icons/gi";
export const getServerSideProps: GetServerSideProps = async (ctx) => {
const session = await getServerAuthSession(ctx);
@ -58,9 +59,12 @@ const HomePageBody: React.FC = () => {
<>
<h1 className="flex flex-row flex-wrap text-center justify-center items-center gap-1 text-4xl font-bold mx-auto">
Hi, {sessionData?.user.name}!{" "}
{ sessionData?.user.role === "ADMIN" && (
{sessionData?.user.isAdmin && (
<FaShieldAlt className="inline-block text-primary" />
)}
{sessionData?.user.isVIP && (
<GiStarFormation className="inline-block text-secondary" />
)}
</h1>
<div className="tabs tabs-boxed border-2 border-cyan-500 mb-4">
<a

22
src/pages/profile/index.tsx
View file

@ -9,6 +9,7 @@ import { FaShieldAlt } from "react-icons/fa";
import { SiGithub, SiGoogle } from "react-icons/si";
import { api } from "~/utils/api";
import { getServerAuthSession } from "../../server/auth";
import { GiStarFormation } from "react-icons/gi";
export const getServerSideProps: GetServerSideProps = async (ctx) => {
const session = await getServerAuthSession(ctx);
@ -115,15 +116,6 @@ const ProfileBody: React.FC = () => {
<div className="card-body">
<h2 className="card-title">Profile:</h2>
{sessionData.user.image && (
<div className="indicator mx-auto m-4">
{ sessionData.user.role === "ADMIN" && (
<span className="indicator-item indicator-bottom badge badge-primary">
<div className="tooltip tooltip-primary" data-tip="Admin">
<FaShieldAlt className="text-xl" />
</div>
</span>
) }
<Image
className="mx-auto"
src={sessionData.user.image}
@ -132,8 +124,20 @@ const ProfileBody: React.FC = () => {
width={100}
priority
/>
)}
<div className="flex flex-row flex-wrap items-center text-center justify-center gap-4">
{sessionData.user.isAdmin && (
<div className="tooltip tooltip-primary" data-tip="Admin">
<FaShieldAlt className="text-xl text-primary" />
</div>
)}
{sessionData.user.isVIP && (
<div className="tooltip tooltip-secondary" data-tip="VIP">
<GiStarFormation className="inline-block text-xl text-secondary" />
</div>
)}
</div>
{providersLoading ? (
<div className="mx-auto">

14
src/pages/room/[id].tsx
View file

@ -15,6 +15,7 @@ import {
IoReloadOutline,
IoSaveOutline,
} from "react-icons/io5";
import { GiStarFormation } from "react-icons/gi";
import { z } from "zod";
import { api } from "~/utils/api";
import { getServerAuthSession } from "../../server/auth";
@ -109,7 +110,8 @@ const RoomBody: React.FC = ({}) => {
name: sessionData?.user.name || "",
image: sessionData?.user.image || "",
client_id: sessionData?.user.id || "",
role: sessionData?.user.role || "USER",
isAdmin: sessionData?.user.isAdmin || false,
isVIP: sessionData?.user.isVIP || false,
}
);
@ -302,7 +304,7 @@ const RoomBody: React.FC = ({}) => {
<p className="flex flex-row flex-wrap text-center justify-center items-center gap-1 text-md mx-auto">
{presenceItem.data.name}{" "}
{presenceItem.data.role === "ADMIN" && (
{presenceItem.data.isAdmin && (
<div
className="tooltip tooltip-primary"
data-tip="Admin"
@ -310,6 +312,14 @@ const RoomBody: React.FC = ({}) => {
<FaShieldAlt className="inline-block text-primary" />
</div>
)}{" "}
{presenceItem.data.isVIP && (
<div
className="tooltip tooltip-secondary"
data-tip="VIP"
>
<GiStarFormation className="inline-block text-secondary" />
</div>
)}{" "}
{presenceItem.clientId === roomFromDb.userId && (
<div
className="tooltip tooltip-warning"

37
src/server/api/routers/user.ts
View file

@ -4,7 +4,6 @@ import { z } from "zod";
import { Goodbye } from "~/components/templates/Goodbye";
import { env } from "~/env.mjs";
import { createTRPCRouter, protectedProcedure } from "~/server/api/trpc";
import type { Role } from "~/utils/types";
import { fetchCache, invalidateCache, setCache } from "~/server/redis";
@ -54,7 +53,8 @@ export const userRouter = createTRPCRouter({
}[];
id: string;
createdAt: Date;
role: Role;
isAdmin: boolean;
isVIP: boolean;
name: string | null;
email: string | null;
}[]
@ -72,7 +72,8 @@ export const userRouter = createTRPCRouter({
select: {
id: true,
name: true,
role: true,
isAdmin: true,
isVIP: true,
createdAt: true,
email: true,
sessions: {
@ -103,7 +104,7 @@ export const userRouter = createTRPCRouter({
)
.mutation(async ({ ctx, input }) => {
let user: User;
if (input?.userId && ctx.session.user.role === "ADMIN") {
if (input?.userId && ctx.session.user.isAdmin) {
user = await ctx.prisma.user.delete({
where: {
id: input.userId,
@ -149,11 +150,11 @@ export const userRouter = createTRPCRouter({
return !!user;
}),
setRole: protectedProcedure
setAdmin: protectedProcedure
.input(
z.object({
userId: z.string(),
role: z.union([z.literal("ADMIN"), z.literal("USER")]),
value: z.boolean(),
})
)
.mutation(async ({ ctx, input }) => {
@ -162,7 +163,29 @@ export const userRouter = createTRPCRouter({
id: input.userId,
},
data: {
role: input.role,
isAdmin: input.value,
},
});
await invalidateCache(`kv_userlist_admin`);
return !!user;
}),
setVIP: protectedProcedure
.input(
z.object({
userId: z.string(),
value: z.boolean(),
})
)
.mutation(async ({ ctx, input }) => {
const user = await ctx.prisma.user.update({
where: {
id: input.userId,
},
data: {
isVIP: input.value,
},
});

10
src/server/auth.ts
View file

@ -10,7 +10,6 @@ import GoogleProvider from "next-auth/providers/google";
import { Resend } from "resend";
import { env } from "~/env.mjs";
import { prisma } from "~/server/db";
import type { Role } from "~/utils/types";
import { Welcome } from "../components/templates/Welcome";
import { invalidateCache } from "./redis";
@ -26,12 +25,14 @@ declare module "next-auth" {
interface Session extends DefaultSession {
user: {
id: string;
role: Role;
isAdmin: boolean;
isVIP: boolean;
} & DefaultSession["user"];
}
interface User {
role: Role;
isAdmin: boolean;
isVIP: boolean;
}
}
@ -45,7 +46,8 @@ export const authOptions: NextAuthOptions = {
session({ session, user }) {
if (session.user) {
session.user.id = user.id;
session.user.role = user.role;
session.user.isAdmin = user.isAdmin;
session.user.isVIP = user.isVIP;
}
return session;
},

9
src/utils/types.ts
View file

@ -7,15 +7,10 @@ const EventTypes = {
} as const;
export type EventType = BetterEnum<typeof EventTypes>;
const RoleValues = {
ADMIN: "ADMIN",
USER: "USER",
} as const;
export type Role = BetterEnum<typeof RoleValues>;
export interface PresenceItem {
name: string;
image: string;
client_id: string;
role: Role;
isAdmin: boolean;
isVIP: boolean;
}