feat(api): ✨ Add Roles API

2026-03-13 13:59:16 +01:00 · 2024-06-07 17:31:17 -10:00 · 2024-06-07 17:31:17 -10:00 · 19823d8eca
commit 19823d8eca
parent 46f41199ac
36 changed files with 7851 additions and 1006 deletions
--- a/packages/database-interface/role.ts
+++ b/packages/database-interface/role.ts
@ -0,0 +1,127 @@
+import {
+    type InferInsertModel,
+    type InferSelectModel,
+    type SQL,
+    and,
+    desc,
+    eq,
+    inArray,
+} from "drizzle-orm";
+import { db } from "~/drizzle/db";
+import { RoleToUsers, Roles } from "~/drizzle/schema";
+
+export class Role {
+    private constructor(private role: InferSelectModel<typeof Roles>) {}
+
+    public static async fromId(id: string | null): Promise<Role | null> {
+        if (!id) return null;
+
+        return await Role.fromSql(eq(Roles.id, id));
+    }
+
+    public static async fromIds(ids: string[]): Promise<Role[]> {
+        return await Role.manyFromSql(inArray(Roles.id, ids));
+    }
+
+    public static async fromSql(
+        sql: SQL<unknown> | undefined,
+        orderBy: SQL<unknown> | undefined = desc(Roles.id),
+    ) {
+        const found = await db.query.Roles.findFirst({
+            where: sql,
+            orderBy,
+        });
+
+        if (!found) return null;
+        return new Role(found);
+    }
+
+    public static async getUserRoles(userId: string) {
+        return (
+            await db.query.RoleToUsers.findMany({
+                where: (role, { eq }) => eq(role.userId, userId),
+                with: {
+                    role: true,
+                },
+            })
+        ).map((r) => new Role(r.role));
+    }
+
+    public static async manyFromSql(
+        sql: SQL<unknown> | undefined,
+        orderBy: SQL<unknown> | undefined = desc(Roles.id),
+        limit?: number,
+        offset?: number,
+        extra?: Parameters<typeof db.query.Users.findMany>[0],
+    ) {
+        const found = await db.query.Roles.findMany({
+            where: sql,
+            orderBy,
+            limit,
+            offset,
+            with: extra?.with,
+        });
+
+        return found.map((s) => new Role(s));
+    }
+
+    public async save(
+        role: Partial<InferSelectModel<typeof Roles>> = this.role,
+    ) {
+        return new Role(
+            (
+                await db
+                    .update(Roles)
+                    .set(role)
+                    .where(eq(Roles.id, this.id))
+                    .returning()
+            )[0],
+        );
+    }
+
+    public async delete() {
+        await db.delete(Roles).where(eq(Roles.id, this.id));
+    }
+
+    public static async new(role: InferInsertModel<typeof Roles>) {
+        return new Role((await db.insert(Roles).values(role).returning())[0]);
+    }
+
+    public async linkUser(userId: string) {
+        await db.insert(RoleToUsers).values({
+            userId,
+            roleId: this.id,
+        });
+    }
+
+    public async unlinkUser(userId: string) {
+        await db
+            .delete(RoleToUsers)
+            .where(
+                and(
+                    eq(RoleToUsers.roleId, this.id),
+                    eq(RoleToUsers.userId, userId),
+                ),
+            );
+    }
+
+    get id() {
+        return this.role.id;
+    }
+
+    public getRole() {
+        return this.role;
+    }
+
+    public toAPI() {
+        return {
+            id: this.id,
+            name: this.role.name,
+            permissions: this.role.permissions,
+            priority: this.role.priority,
+            description: this.role.description,
+            visible: this.role.visible,
+            icon: this.role.icon,
+        };
+    }
+}