@simple-auth-kit/cli 1.4.3 → 1.5.1

package/registry/combos/nestjs-drizzle/variants/base/src/{rbac.repository.ts → repositories/rbac.repository.ts}

@@ -1,11 +1,37 @@
-import { ConflictException, Inject, Injectable, NotFoundException } from "@nestjs/common";
-import { and, asc, count, desc, eq, ilike, inArray, type SQL } from "drizzle-orm";
+import {
+  ConflictException,
+  Inject,
+  Injectable,
+  NotFoundException,
+} from "@nestjs/common";
+import {
+  and,
+  asc,
+  count,
+  desc,
+  eq,
+  ilike,
+  inArray,
+  type SQL,
+} from "drizzle-orm";
 import { resolvePermissions } from "@/lib/auth/core/rbac.js";
-import { DRIZZLE_DB, type Database } from "./db.js";
-import { buildPageMeta, normalizeLimit, normalizePage, type Paginated } from "./pagination.js";
-import { PermissionCache } from "./permission-cache.js";
-import { permissionRole, permissionUser, permissions, roleUser, roles, users } from "./schema.js";
-import { toId, toIdOrNull } from "./id.helper.js";
+import { DRIZZLE_DB, type Database } from "../db.js";
+import {
+  buildPageMeta,
+  normalizeLimit,
+  normalizePage,
+  type Paginated,
+} from "../pagination.js";
+import { PermissionCache } from "../permission-cache.js";
+import {
+  permissionRole,
+  permissionUser,
+  permissions,
+  roleUser,
+  roles,
+  users,
+} from "../schema.js";
+import { toId, toIdOrNull } from "../id.helper.js";
 
 export interface UserSummary {
   id: string;
@@ -148,9 +174,15 @@ interface RoleRow {
   isActive: boolean;
 }
 
-const asPermissionSummary = (row: PermissionRow): PermissionSummary => ({ ...row, id: row.id.toString() });
+const asPermissionSummary = (row: PermissionRow): PermissionSummary => ({
+  ...row,
+  id: row.id.toString(),
+});
 
-const asRoleSummary = (row: RoleRow): RoleSummary => ({ ...row, id: row.id.toString() });
+const asRoleSummary = (row: RoleRow): RoleSummary => ({
+  ...row,
+  id: row.id.toString(),
+});
 
 @Injectable()
 export class RbacRepository {
@@ -177,7 +209,9 @@ export class RbacRepository {
    * makes deactivating either one enforcement-changing without unpicking any grant; `isDeleted`
    * on the user themselves takes the whole answer to empty.
    */
-  async resolveAuthzContext(userId: string): Promise<{ roles: string[]; permissions: string[] }> {
+  async resolveAuthzContext(
+    userId: string,
+  ): Promise<{ roles: string[]; permissions: string[] }> {
     const userIdBig = toId(userId);
     const [user] = await this.db
       .select({ id: users.id })
@@ -192,21 +226,44 @@ export class RbacRepository {
       this.db
         .select({ roleSlug: roles.slug, permissionSlug: permissions.slug })
         .from(roleUser)
-        .innerJoin(roles, and(eq(roles.id, roleUser.roleId), eq(roles.isActive, true), eq(roles.isDeleted, false)))
+        .innerJoin(
+          roles,
+          and(
+            eq(roles.id, roleUser.roleId),
+            eq(roles.isActive, true),
+            eq(roles.isDeleted, false),
+          ),
+        )
         .leftJoin(permissionRole, eq(permissionRole.roleId, roles.id))
-        .leftJoin(permissions, and(eq(permissions.id, permissionRole.permissionId), eq(permissions.isActive, true), eq(permissions.isDeleted, false)))
+        .leftJoin(
+          permissions,
+          and(
+            eq(permissions.id, permissionRole.permissionId),
+            eq(permissions.isActive, true),
+            eq(permissions.isDeleted, false),
+          ),
+        )
         .where(eq(roleUser.userId, userIdBig)),
       this.db
         .select({ slug: permissions.slug })
         .from(permissionUser)
-        .innerJoin(permissions, and(eq(permissions.id, permissionUser.permissionId), eq(permissions.isActive, true), eq(permissions.isDeleted, false)))
+        .innerJoin(
+          permissions,
+          and(
+            eq(permissions.id, permissionUser.permissionId),
+            eq(permissions.isActive, true),
+            eq(permissions.isDeleted, false),
+          ),
+        )
         .where(eq(permissionUser.userId, userIdBig)),
     ]);
 
     return {
       roles: [...new Set(roleRows.map((row) => row.roleSlug))].sort(),
       permissions: resolvePermissions(
-        roleRows.map((row) => row.permissionSlug).filter((slug): slug is string => slug !== null),
+        roleRows
+          .map((row) => row.permissionSlug)
+          .filter((slug): slug is string => slug !== null),
         directRows.map((row) => row.slug),
       ),
     };
@@ -222,7 +279,8 @@ export class RbacRepository {
     const limit = normalizeLimit(filter.limit);
 
     const conditions: SQL[] = [eq(users.isDeleted, false)];
-    if (filter.search) conditions.push(ilike(users.email, `%${filter.search}%`));
+    if (filter.search)
+      conditions.push(ilike(users.email, `%${filter.search}%`));
     const where = and(...conditions)!;
 
     const rows = await this.db
@@ -232,7 +290,10 @@ export class RbacRepository {
       .orderBy(desc(users.createdAt), desc(users.id))
       .limit(limit)
       .offset((page - 1) * limit);
-    const [{ value: total }] = await this.db.select({ value: count() }).from(users).where(where);
+    const [{ value: total }] = await this.db
+      .select({ value: count() })
+      .from(users)
+      .where(where);
 
     // One read for the page's role assignments rather than one per user.
     const assignments = rows.length
@@ -254,7 +315,10 @@ export class RbacRepository {
     }
 
     return {
-      items: rows.map((row) => ({ ...row, roles: (rolesByUser.get(row.id.toString()) ?? []).sort() })),
+      items: rows.map((row) => ({
+        ...row,
+        roles: (rolesByUser.get(row.id.toString()) ?? []).sort(),
+      })),
       meta: buildPageMeta(page, limit, total),
     };
   }
@@ -269,9 +333,16 @@ export class RbacRepository {
       .limit(1);
     if (!row) throw new NotFoundException(`user "${userId}" not found`);
 
-    const assignments = await this.db.select({ slug: roles.slug }).from(roleUser).innerJoin(roles, eq(roles.id, roleUser.roleId)).where(eq(roleUser.userId, userIdBig));
+    const assignments = await this.db
+      .select({ slug: roles.slug })
+      .from(roleUser)
+      .innerJoin(roles, eq(roles.id, roleUser.roleId))
+      .where(eq(roleUser.userId, userIdBig));
 
-    return toUserSummary({ ...row, roles: assignments.map((a) => a.slug).sort() });
+    return toUserSummary({
+      ...row,
+      roles: assignments.map((a) => a.slug).sort(),
+    });
   }
 
   /**
@@ -295,7 +366,11 @@ export class RbacRepository {
     },
     actorUserId: string | null,
   ): Promise<UserSummary> {
-    const [existing] = await this.db.select({ id: users.id }).from(users).where(eq(users.email, input.email)).limit(1);
+    const [existing] = await this.db
+      .select({ id: users.id })
+      .from(users)
+      .where(eq(users.email, input.email))
+      .limit(1);
     if (existing) throw new ConflictException("email already registered");
 
     const [user] = await this.db
@@ -320,11 +395,19 @@ export class RbacRepository {
       const grantedRoles = await this.db
         .select({ id: roles.id })
         .from(roles)
-        .where(and(inArray(roles.slug, input.roles), eq(roles.isActive, true), eq(roles.isDeleted, false)));
+        .where(
+          and(
+            inArray(roles.slug, input.roles),
+            eq(roles.isActive, true),
+            eq(roles.isDeleted, false),
+          ),
+        );
       if (grantedRoles.length > 0) {
         await this.db
           .insert(roleUser)
-          .values(grantedRoles.map((role) => ({ userId: user.id, roleId: role.id })))
+          .values(
+            grantedRoles.map((role) => ({ userId: user.id, roleId: role.id })),
+          )
           .onConflictDoNothing({ target: [roleUser.userId, roleUser.roleId] });
         await this.cache.invalidateSubject(user.id.toString());
       }
@@ -363,7 +446,9 @@ export class RbacRepository {
     if (!existing) throw new NotFoundException(`user "${userId}" not found`);
 
     // Only the fields the caller sent are written; an absent one leaves its column alone.
-    const changed = Object.fromEntries(Object.entries(input).filter(([, value]) => value !== undefined));
+    const changed = Object.fromEntries(
+      Object.entries(input).filter(([, value]) => value !== undefined),
+    );
     await this.db
       .update(users)
       .set({ ...changed, updatedBy: toIdOrNull(actorUserId) })
@@ -376,7 +461,11 @@ export class RbacRepository {
    * pattern — the row survives for audit purposes, it just stops appearing in listings and can no
    * longer authenticate (mirrors how `block` already disables login without removing the row).
    */
-  async deleteUser(userId: string, actorUserId: string | null, reason?: string): Promise<void> {
+  async deleteUser(
+    userId: string,
+    actorUserId: string | null,
+    reason?: string,
+  ): Promise<void> {
     const userIdBig = toId(userId);
     const [existing] = await this.db
       .select({ id: users.id })
@@ -387,7 +476,12 @@ export class RbacRepository {
 
     await this.db
       .update(users)
-      .set({ isDeleted: true, deletedAt: new Date(), deletedBy: toIdOrNull(actorUserId), deletedReason: reason ?? null })
+      .set({
+        isDeleted: true,
+        deletedAt: new Date(),
+        deletedBy: toIdOrNull(actorUserId),
+        deletedReason: reason ?? null,
+      })
       .where(eq(users.id, userIdBig));
     await this.cache.invalidateSubject(userId);
   }
@@ -400,7 +494,12 @@ export class RbacRepository {
       .select(PERMISSION_COLUMNS)
       .from(permissions)
       .where(eq(permissions.isDeleted, false))
-      .orderBy(asc(permissions.groupOrder), asc(permissions.group), asc(permissions.order), asc(permissions.slug));
+      .orderBy(
+        asc(permissions.groupOrder),
+        asc(permissions.group),
+        asc(permissions.order),
+        asc(permissions.slug),
+      );
     return rows.map(asPermissionSummary);
   }
 
@@ -413,7 +512,10 @@ export class RbacRepository {
    * Upserted on `slug`, which stays the stable identifier grants and revocations use, so renaming
    * `name`/`displayName` never breaks a grant.
    */
-  async upsertPermission(input: PermissionInput, actorUserId: string | null): Promise<PermissionSummary> {
+  async upsertPermission(
+    input: PermissionInput,
+    actorUserId: string | null,
+  ): Promise<PermissionSummary> {
     const actorId = toIdOrNull(actorUserId);
     // Only the fields the caller sent are written; an absent one leaves its column alone, so a
     // partial edit cannot blank out metadata it did not mention.
@@ -444,7 +546,9 @@ export class RbacRepository {
       // returns none — so the no-op case rewrites the slug with itself.
       .onConflictDoUpdate({
         target: permissions.slug,
-        set: Object.keys(changed).length ? { ...changed, updatedBy: actorId } : { slug: input.slug, updatedBy: actorId },
+        set: Object.keys(changed).length
+          ? { ...changed, updatedBy: actorId }
+          : { slug: input.slug, updatedBy: actorId },
       })
       .returning(PERMISSION_COLUMNS);
 
@@ -454,12 +558,23 @@ export class RbacRepository {
   }
 
   async listRoles(): Promise<RoleSummary[]> {
-    const rows = await this.db.select(ROLE_COLUMNS).from(roles).where(eq(roles.isDeleted, false)).orderBy(asc(roles.order), asc(roles.slug));
+    const rows = await this.db
+      .select(ROLE_COLUMNS)
+      .from(roles)
+      .where(eq(roles.isDeleted, false))
+      .orderBy(asc(roles.order), asc(roles.slug));
     return rows.map(asRoleSummary);
   }
 
   async createRole(
-    input: { slug: string; name?: string; displayName?: string; description?: string | null; isDefault?: boolean; isActive?: boolean },
+    input: {
+      slug: string;
+      name?: string;
+      displayName?: string;
+      description?: string | null;
+      isDefault?: boolean;
+      isActive?: boolean;
+    },
     actorUserId: string | null,
   ): Promise<RoleSummary> {
     const actorId = toIdOrNull(actorUserId);
@@ -481,7 +596,13 @@ export class RbacRepository {
 
   async updateRole(
     roleId: string,
-    input: { name?: string; displayName?: string; description?: string | null; isDefault?: boolean; isActive?: boolean },
+    input: {
+      name?: string;
+      displayName?: string;
+      description?: string | null;
+      isDefault?: boolean;
+      isActive?: boolean;
+    },
     actorUserId: string | null,
   ): Promise<RoleSummary> {
     const roleIdBig = toId(roleId);
@@ -492,7 +613,9 @@ export class RbacRepository {
       .limit(1);
     if (!existing) throw new NotFoundException("role not found");
 
-    const changed = Object.fromEntries(Object.entries(input).filter(([, value]) => value !== undefined));
+    const changed = Object.fromEntries(
+      Object.entries(input).filter(([, value]) => value !== undefined),
+    );
     const [role] = await this.db
       .update(roles)
       .set({ ...changed, updatedBy: toIdOrNull(actorUserId) })
@@ -508,7 +631,11 @@ export class RbacRepository {
    * deleted role simply stops being returned by `listRoles`/`resolveAuthzContext`, rather than
    * cascading a delete that would silently strip a user's other roles' foreign keys.
    */
-  async deleteRole(roleId: string, actorUserId: string | null, reason?: string): Promise<void> {
+  async deleteRole(
+    roleId: string,
+    actorUserId: string | null,
+    reason?: string,
+  ): Promise<void> {
     const roleIdBig = toId(roleId);
     const [existing] = await this.db
       .select({ id: roles.id })
@@ -519,7 +646,12 @@ export class RbacRepository {
 
     await this.db
       .update(roles)
-      .set({ isDeleted: true, deletedAt: new Date(), deletedBy: toIdOrNull(actorUserId), deletedReason: reason ?? null })
+      .set({
+        isDeleted: true,
+        deletedAt: new Date(),
+        deletedBy: toIdOrNull(actorUserId),
+        deletedReason: reason ?? null,
+      })
       .where(eq(roles.id, roleIdBig));
     await this.cache.invalidatePolicy();
   }
@@ -530,7 +662,11 @@ export class RbacRepository {
    * closed-catalog property the routes always had — `@CheckAbility` only accepts slugs from
    * `rbac.defaults.ts`, so an invented slug cannot gate anything shipped by this library.
    */
-  async attachPermissionToRole(roleId: string, permissionSlug: string, actorUserId: string | null): Promise<void> {
+  async attachPermissionToRole(
+    roleId: string,
+    permissionSlug: string,
+    actorUserId: string | null,
+  ): Promise<void> {
     const roleIdBig = toId(roleId);
     const [role] = await this.db
       .select({ id: roles.id })
@@ -543,27 +679,46 @@ export class RbacRepository {
     await this.db
       .insert(permissionRole)
       .values({ permissionId: permission.id, roleId: roleIdBig })
-      .onConflictDoNothing({ target: [permissionRole.permissionId, permissionRole.roleId] });
+      .onConflictDoNothing({
+        target: [permissionRole.permissionId, permissionRole.roleId],
+      });
     await this.cache.invalidatePolicy(); // everyone holding this role is affected
   }
 
   async assignRoleToUser(userId: string, roleSlug: string): Promise<void> {
     const userIdBig = toId(userId);
-    const [role] = await this.db.select({ id: roles.id }).from(roles).where(eq(roles.slug, roleSlug)).limit(1);
+    const [role] = await this.db
+      .select({ id: roles.id })
+      .from(roles)
+      .where(eq(roles.slug, roleSlug))
+      .limit(1);
     if (!role) throw new NotFoundException(`role "${roleSlug}" not found`);
 
-    const [user] = await this.db.select({ id: users.id }).from(users).where(eq(users.id, userIdBig)).limit(1);
+    const [user] = await this.db
+      .select({ id: users.id })
+      .from(users)
+      .where(eq(users.id, userIdBig))
+      .limit(1);
     if (!user) throw new NotFoundException(`user "${userId}" not found`);
 
-    await this.db.insert(roleUser).values({ userId: userIdBig, roleId: role.id }).onConflictDoNothing({ target: [roleUser.userId, roleUser.roleId] });
+    await this.db
+      .insert(roleUser)
+      .values({ userId: userIdBig, roleId: role.id })
+      .onConflictDoNothing({ target: [roleUser.userId, roleUser.roleId] });
     await this.cache.invalidateSubject(userId);
   }
 
   async revokeRoleFromUser(userId: string, roleSlug: string): Promise<void> {
     const userIdBig = toId(userId);
-    const [role] = await this.db.select({ id: roles.id }).from(roles).where(eq(roles.slug, roleSlug)).limit(1);
+    const [role] = await this.db
+      .select({ id: roles.id })
+      .from(roles)
+      .where(eq(roles.slug, roleSlug))
+      .limit(1);
     if (!role) return;
-    await this.db.delete(roleUser).where(and(eq(roleUser.userId, userIdBig), eq(roleUser.roleId, role.id)));
+    await this.db
+      .delete(roleUser)
+      .where(and(eq(roleUser.userId, userIdBig), eq(roleUser.roleId, role.id)));
     await this.cache.invalidateSubject(userId);
   }
 
@@ -579,7 +734,13 @@ export class RbacRepository {
     const defaults = await this.db
       .select({ id: roles.id })
       .from(roles)
-      .where(and(eq(roles.isDefault, true), eq(roles.isActive, true), eq(roles.isDeleted, false)));
+      .where(
+        and(
+          eq(roles.isDefault, true),
+          eq(roles.isActive, true),
+          eq(roles.isDeleted, false),
+        ),
+      );
     if (!defaults.length) return;
     await this.db
       .insert(roleUser)
@@ -589,25 +750,48 @@ export class RbacRepository {
   }
 
   async hasAnyRole(userId: bigint): Promise<boolean> {
-    const [row] = await this.db.select({ value: count() }).from(roleUser).where(eq(roleUser.userId, userId));
+    const [row] = await this.db
+      .select({ value: count() })
+      .from(roleUser)
+      .where(eq(roleUser.userId, userId));
     return (row?.value ?? 0) > 0;
   }
 
-  async grantPermissionToUser(userId: string, permissionSlug: string, actorUserId: string | null): Promise<void> {
+  async grantPermissionToUser(
+    userId: string,
+    permissionSlug: string,
+    actorUserId: string | null,
+  ): Promise<void> {
     const userIdBig = toId(userId);
     const permission = await this.ensurePermission(permissionSlug, actorUserId);
     await this.db
       .insert(permissionUser)
       .values({ userId: userIdBig, permissionId: permission.id })
-      .onConflictDoNothing({ target: [permissionUser.userId, permissionUser.permissionId] });
+      .onConflictDoNothing({
+        target: [permissionUser.userId, permissionUser.permissionId],
+      });
     await this.cache.invalidateSubject(userId);
   }
 
-  async revokePermissionFromUser(userId: string, permissionSlug: string): Promise<void> {
+  async revokePermissionFromUser(
+    userId: string,
+    permissionSlug: string,
+  ): Promise<void> {
     const userIdBig = toId(userId);
-    const [permission] = await this.db.select({ id: permissions.id }).from(permissions).where(eq(permissions.slug, permissionSlug)).limit(1);
+    const [permission] = await this.db
+      .select({ id: permissions.id })
+      .from(permissions)
+      .where(eq(permissions.slug, permissionSlug))
+      .limit(1);
     if (!permission) return;
-    await this.db.delete(permissionUser).where(and(eq(permissionUser.userId, userIdBig), eq(permissionUser.permissionId, permission.id)));
+    await this.db
+      .delete(permissionUser)
+      .where(
+        and(
+          eq(permissionUser.userId, userIdBig),
+          eq(permissionUser.permissionId, permission.id),
+        ),
+      );
     await this.cache.invalidateSubject(userId);
   }
 
@@ -620,13 +804,27 @@ export class RbacRepository {
    * Creates the row for a slug granted before anyone defined it; leaves an existing definition
    * alone. Returns bigint directly: both callers feed the id straight into another database call.
    */
-  private async ensurePermission(slug: string, actorUserId: string | null): Promise<{ id: bigint }> {
-    const [existing] = await this.db.select({ id: permissions.id }).from(permissions).where(eq(permissions.slug, slug)).limit(1);
+  private async ensurePermission(
+    slug: string,
+    actorUserId: string | null,
+  ): Promise<{ id: bigint }> {
+    const [existing] = await this.db
+      .select({ id: permissions.id })
+      .from(permissions)
+      .where(eq(permissions.slug, slug))
+      .limit(1);
     if (existing) return existing;
     const actorId = toIdOrNull(actorUserId);
     const [created] = await this.db
       .insert(permissions)
-      .values({ slug, name: slug, displayName: slug, group: "Custom", createdBy: actorId, updatedBy: actorId })
+      .values({
+        slug,
+        name: slug,
+        displayName: slug,
+        group: "Custom",
+        createdBy: actorId,
+        updatedBy: actorId,
+      })
       .returning({ id: permissions.id });
     return created;
   }