@m5kdev/backend 0.1.4 → 0.2.0

package/src/modules/auth/auth.repository.ts

@@ -1,672 +0,0 @@
-import { and, count, desc, eq, gte, ne } from "drizzle-orm";
-import type { LibSQLDatabase } from "drizzle-orm/libsql";
-import { ok } from "neverthrow";
-import { v4 as uuidv4 } from "uuid";
-import * as auth from "#modules/auth/auth.db";
-import type {
-  AccountClaim,
-  AccountClaimMagicLink,
-  AccountClaimMagicLinkOutput,
-  AccountClaimOutput,
-  Waitlist,
-  WaitlistOutput,
-} from "#modules/auth/auth.dto";
-import type { ServerResultAsync } from "#modules/base/base.dto";
-import { BaseRepository } from "#modules/base/base.repository";
-
-const schema = { ...auth };
-type Schema = typeof schema;
-type Orm = LibSQLDatabase<Schema>;
-type UserRow = typeof auth.users.$inferSelect;
-
-export class AuthRepository extends BaseRepository<Orm, Schema, Record<string, never>> {
-  async getUserWaitlistCount(userId: string, tx?: Orm): ServerResultAsync<number> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db
-        .select({ count: count() })
-        .from(this.schema.waitlist)
-        .where(eq(this.schema.waitlist.userId, userId));
-
-      return ok(waitlist.count ?? 0);
-    });
-  }
-
-  async getOnboarding(userId: string, tx?: Orm): ServerResultAsync<number> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [user] = await db
-        .select({ onboarding: this.schema.users.onboarding })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.id, userId))
-        .limit(1);
-      if (!user) return this.error("FORBIDDEN");
-
-      return ok(user.onboarding ?? 0);
-    });
-  }
-
-  async setOnboarding(userId: string, onboarding: number, tx?: Orm): ServerResultAsync<number> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      await db
-        .update(this.schema.users)
-        .set({ onboarding })
-        .where(eq(this.schema.users.id, userId));
-      return ok(onboarding);
-    });
-  }
-
-  async getPreferences(userId: string, tx?: Orm): ServerResultAsync<Record<string, unknown>> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [user] = await db
-        .select({ preferences: this.schema.users.preferences })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.id, userId))
-        .limit(1);
-      if (!user) return this.error("FORBIDDEN");
-      const json = user.preferences
-        ? (JSON.parse(user.preferences) as Record<string, unknown>)
-        : {};
-      return ok(json);
-    });
-  }
-
-  async setPreferences(
-    userId: string,
-    preferences: Record<string, unknown>,
-    tx?: Orm
-  ): ServerResultAsync<Record<string, unknown>> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      await db
-        .update(this.schema.users)
-        .set({ preferences: JSON.stringify(preferences) })
-        .where(eq(this.schema.users.id, userId));
-      return ok(preferences);
-    });
-  }
-
-  async getMetadata(userId: string, tx?: Orm): ServerResultAsync<Record<string, unknown>> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [user] = await db
-        .select({ metadata: this.schema.users.metadata })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.id, userId))
-        .limit(1);
-      if (!user) return this.error("FORBIDDEN");
-
-      return ok(user.metadata);
-    });
-  }
-
-  async setMetadata(
-    userId: string,
-    metadata: Record<string, unknown>,
-    tx?: Orm
-  ): ServerResultAsync<Record<string, unknown>> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [user] = await db
-        .select({ metadata: this.schema.users.metadata })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.id, userId))
-        .limit(1);
-      if (!user) return this.error("FORBIDDEN");
-      await db
-        .update(this.schema.users)
-        .set({
-          metadata: {
-            ...user.metadata,
-            ...metadata,
-          },
-        })
-        .where(eq(this.schema.users.id, userId));
-      return ok(metadata);
-    });
-  }
-
-  async getFlags(userId: string, tx?: Orm): ServerResultAsync<string[]> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [user] = await db
-        .select({ flags: this.schema.users.flags })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.id, userId))
-        .limit(1);
-      if (!user) return this.error("FORBIDDEN");
-      const json = user.flags ? (JSON.parse(user.flags) as string[]) : [];
-
-      return ok(json);
-    });
-  }
-
-  async setFlags(userId: string, flags: string[], tx?: Orm): ServerResultAsync<string[]> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      await db
-        .update(this.schema.users)
-        .set({ flags: JSON.stringify(flags) })
-        .where(eq(this.schema.users.id, userId));
-      return ok(flags);
-    });
-  }
-
-  async listAdminWaitlist(tx?: Orm): ServerResultAsync<WaitlistOutput[]> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const waitlist = await db
-        .select({
-          id: this.schema.waitlist.id,
-          name: this.schema.waitlist.name,
-          email: this.schema.waitlist.email,
-          createdAt: this.schema.waitlist.createdAt,
-          updatedAt: this.schema.waitlist.updatedAt,
-          status: this.schema.waitlist.status,
-        })
-        .from(this.schema.waitlist)
-        .where(eq(this.schema.waitlist.type, "WAITLIST"))
-        .orderBy(desc(this.schema.waitlist.createdAt));
-      return ok(waitlist);
-    });
-  }
-
-  async listWaitlist(userId: string, tx?: Orm): ServerResultAsync<Waitlist[]> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const waitlist = await db
-        .select()
-        .from(this.schema.waitlist)
-        .where(
-          and(eq(this.schema.waitlist.userId, userId), eq(this.schema.waitlist.type, "WAITLIST"))
-        );
-      return ok(waitlist);
-    });
-  }
-
-  async addToWaitlist(email: string, tx?: Orm): ServerResultAsync<WaitlistOutput> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db.insert(this.schema.waitlist).values({ email }).returning();
-      return ok(waitlist);
-    });
-  }
-
-  async inviteFromWaitlist(id: string, tx?: Orm): ServerResultAsync<Waitlist> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db
-        .update(this.schema.waitlist)
-        .set({
-          status: "INVITED",
-          code: uuidv4(),
-          expiresAt: new Date(Date.now() + 1000 * 60 * 60 * 24 * 14),
-        })
-        .where(eq(this.schema.waitlist.id, id))
-        .returning();
-      return ok(waitlist);
-    });
-  }
-
-  async inviteToWaitlist(
-    { email, userId, name }: { email: string; userId: string; name?: string },
-    tx?: Orm
-  ): ServerResultAsync<Waitlist> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db
-        .insert(this.schema.waitlist)
-        .values({
-          email,
-          name,
-          status: "INVITED",
-          code: uuidv4(),
-          expiresAt: new Date(Date.now() + 1000 * 60 * 60 * 24),
-          userId: userId,
-        })
-        .returning();
-      return ok(waitlist);
-    });
-  }
-
-  async createInvitationCode(
-    { userId, name }: { userId: string; name?: string },
-    tx?: Orm
-  ): ServerResultAsync<Waitlist> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db
-        .insert(this.schema.waitlist)
-        .values({
-          name,
-          status: "INVITED",
-          code: uuidv4(),
-          expiresAt: new Date(Date.now() + 1000 * 60 * 60 * 24),
-          userId: userId,
-        })
-        .returning();
-      return ok(waitlist);
-    });
-  }
-
-  async joinWaitlist(email: string, tx?: Orm): ServerResultAsync<WaitlistOutput> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db.insert(this.schema.waitlist).values({ email }).returning();
-      return ok(waitlist);
-    });
-  }
-
-  async removeFromWaitlist(id: string, tx?: Orm): ServerResultAsync<WaitlistOutput> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db
-        .update(this.schema.waitlist)
-        .set({ status: "REMOVED" })
-        .where(eq(this.schema.waitlist.id, id))
-        .returning();
-      return ok(waitlist);
-    });
-  }
-
-  async validateWaitlistCode(code: string, tx?: Orm): ServerResultAsync<{ status: string }> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [waitlist] = await db
-        .select()
-        .from(this.schema.waitlist)
-        .where(and(eq(this.schema.waitlist.code, code), eq(this.schema.waitlist.type, "WAITLIST")))
-        .limit(1);
-      if (!waitlist) return ok({ status: "NOT_FOUND" });
-      if (waitlist.expiresAt && waitlist.expiresAt < new Date()) return ok({ status: "EXPIRED" });
-      if (waitlist.status !== "INVITED") return ok({ status: "INVALID" });
-      return ok({ status: "VALID" });
-    });
-  }
-
-  async createAccountClaimCode(
-    { userId, expiresInHours = 24 * 14 }: { userId: string; expiresInHours?: number },
-    tx?: Orm
-  ): ServerResultAsync<AccountClaim> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [claim] = await db
-        .insert(this.schema.waitlist)
-        .values({
-          type: "ACCOUNT_CLAIM",
-          claimUserId: userId,
-          code: uuidv4(),
-          status: "INVITED",
-          expiresAt: new Date(Date.now() + 1000 * 60 * 60 * expiresInHours),
-        })
-        .returning();
-      return ok(claim);
-    });
-  }
-
-  async createClaimableProvisionedUser({
-    name,
-    email,
-    metadata = {},
-    onboarding = 0,
-    role = "user",
-    expiresInHours = 24 * 14,
-  }: {
-    name: string;
-    email: string;
-    metadata?: Record<string, unknown>;
-    onboarding?: number;
-    role?: "user" | "admin" | "agent";
-    expiresInHours?: number;
-  }): ServerResultAsync<{ user: UserRow; claim: AccountClaim }> {
-    return this.throwableAsync(async () => {
-      const normalizedEmail = email.toLowerCase();
-      const [existingUser] = await this.orm
-        .select({ id: this.schema.users.id })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.email, normalizedEmail))
-        .limit(1);
-      if (existingUser) {
-        return this.error("CONFLICT", "Email already in use");
-      }
-
-      const created = await this.orm.transaction(async (tx) => {
-        const [user] = await tx
-          .insert(this.schema.users)
-          .values({
-            name,
-            email: normalizedEmail,
-            emailVerified: false,
-            role,
-            onboarding,
-            metadata,
-          })
-          .returning();
-        if (!user) throw new Error("Failed to create user");
-
-        const organizationId = uuidv4();
-        const [organization] = await tx
-          .insert(this.schema.organizations)
-          .values({
-            id: organizationId,
-            name: organizationId,
-            slug: organizationId,
-          })
-          .returning();
-        if (!organization) throw new Error("Failed to create organization");
-
-        const [member] = await tx
-          .insert(this.schema.members)
-          .values({
-            userId: user.id,
-            organizationId: organization.id,
-            role: "owner",
-          })
-          .returning();
-        if (!member) throw new Error("Failed to create organization membership");
-
-        const [team] = await tx
-          .insert(this.schema.teams)
-          .values({
-            name: organization.id,
-            organizationId: organization.id,
-          })
-          .returning();
-        if (!team) throw new Error("Failed to create team");
-
-        const [teamMember] = await tx
-          .insert(this.schema.teamMembers)
-          .values({
-            userId: user.id,
-            teamId: team.id,
-            role: "owner",
-          })
-          .returning();
-        if (!teamMember) throw new Error("Failed to create team membership");
-
-        const [claim] = await tx
-          .insert(this.schema.waitlist)
-          .values({
-            type: "ACCOUNT_CLAIM",
-            claimUserId: user.id,
-            code: uuidv4(),
-            status: "INVITED",
-            expiresAt: new Date(Date.now() + 1000 * 60 * 60 * expiresInHours),
-          })
-          .returning();
-        if (!claim) throw new Error("Failed to create account claim");
-
-        return { user, claim };
-      });
-
-      return ok(created);
-    });
-  }
-
-  async listAccountClaims(tx?: Orm): ServerResultAsync<AccountClaimOutput[]> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const claims = await db
-        .select({
-          id: this.schema.waitlist.id,
-          claimUserId: this.schema.waitlist.claimUserId,
-          status: this.schema.waitlist.status,
-          expiresAt: this.schema.waitlist.expiresAt,
-          claimedAt: this.schema.waitlist.claimedAt,
-          claimedEmail: this.schema.waitlist.claimedEmail,
-          createdAt: this.schema.waitlist.createdAt,
-          updatedAt: this.schema.waitlist.updatedAt,
-        })
-        .from(this.schema.waitlist)
-        .where(eq(this.schema.waitlist.type, "ACCOUNT_CLAIM"))
-        .orderBy(desc(this.schema.waitlist.createdAt));
-      return ok(claims);
-    });
-  }
-
-  async validateAccountClaimCode(code: string, tx?: Orm): ServerResultAsync<{ status: string }> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [claim] = await db
-        .select()
-        .from(this.schema.waitlist)
-        .where(
-          and(eq(this.schema.waitlist.code, code), eq(this.schema.waitlist.type, "ACCOUNT_CLAIM"))
-        )
-        .limit(1);
-      if (!claim) return ok({ status: "NOT_FOUND" });
-      if (claim.expiresAt && claim.expiresAt < new Date()) return ok({ status: "EXPIRED" });
-      if (claim.status !== "INVITED") return ok({ status: "INVALID" });
-      return ok({ status: "VALID" });
-    });
-  }
-
-  async findAccountClaimByCode(code: string, tx?: Orm): ServerResultAsync<AccountClaim | null> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [claim] = await db
-        .select()
-        .from(this.schema.waitlist)
-        .where(
-          and(
-            eq(this.schema.waitlist.code, code),
-            eq(this.schema.waitlist.type, "ACCOUNT_CLAIM"),
-            eq(this.schema.waitlist.status, "INVITED"),
-            gte(this.schema.waitlist.expiresAt, new Date())
-          )
-        )
-        .limit(1);
-      return ok(claim ?? null);
-    });
-  }
-
-  async findAccountClaimById(id: string, tx?: Orm): ServerResultAsync<AccountClaim | null> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [claim] = await db
-        .select()
-        .from(this.schema.waitlist)
-        .where(and(eq(this.schema.waitlist.id, id), eq(this.schema.waitlist.type, "ACCOUNT_CLAIM")))
-        .limit(1);
-      return ok(claim ?? null);
-    });
-  }
-
-  async findPendingAccountClaimForUser(
-    userId: string,
-    tx?: Orm
-  ): ServerResultAsync<AccountClaim | null> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [claim] = await db
-        .select()
-        .from(this.schema.waitlist)
-        .where(
-          and(
-            eq(this.schema.waitlist.type, "ACCOUNT_CLAIM"),
-            eq(this.schema.waitlist.claimUserId, userId),
-            eq(this.schema.waitlist.status, "INVITED"),
-            gte(this.schema.waitlist.expiresAt, new Date())
-          )
-        )
-        .orderBy(desc(this.schema.waitlist.createdAt))
-        .limit(1);
-      return ok(claim ?? null);
-    });
-  }
-
-  async setAccountClaimEmail(
-    { userId, email }: { userId: string; email: string },
-    tx?: Orm
-  ): ServerResultAsync<{ status: boolean }> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const normalizedEmail = email.toLowerCase();
-      const [existingUser] = await db
-        .select({ id: this.schema.users.id })
-        .from(this.schema.users)
-        .where(and(eq(this.schema.users.email, normalizedEmail), ne(this.schema.users.id, userId)))
-        .limit(1);
-      if (existingUser) {
-        return this.error("BAD_REQUEST", "Email is already in use");
-      }
-
-      const [claim] = await db
-        .select({ id: this.schema.waitlist.id })
-        .from(this.schema.waitlist)
-        .where(
-          and(
-            eq(this.schema.waitlist.type, "ACCOUNT_CLAIM"),
-            eq(this.schema.waitlist.claimUserId, userId),
-            eq(this.schema.waitlist.status, "INVITED"),
-            gte(this.schema.waitlist.expiresAt, new Date())
-          )
-        )
-        .orderBy(desc(this.schema.waitlist.createdAt))
-        .limit(1);
-      if (!claim) {
-        return this.error("BAD_REQUEST", "No pending claim found");
-      }
-
-      await db
-        .update(this.schema.users)
-        .set({
-          email: normalizedEmail,
-          emailVerified: false,
-          updatedAt: new Date(),
-        })
-        .where(eq(this.schema.users.id, userId));
-
-      await db
-        .update(this.schema.waitlist)
-        .set({
-          claimedEmail: normalizedEmail,
-          updatedAt: new Date(),
-        })
-        .where(eq(this.schema.waitlist.id, claim.id));
-
-      return ok({ status: true });
-    });
-  }
-
-  async acceptAccountClaim(userId: string, tx?: Orm): ServerResultAsync<{ status: boolean }> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [claim] = await db
-        .select({ id: this.schema.waitlist.id })
-        .from(this.schema.waitlist)
-        .where(
-          and(
-            eq(this.schema.waitlist.type, "ACCOUNT_CLAIM"),
-            eq(this.schema.waitlist.claimUserId, userId),
-            eq(this.schema.waitlist.status, "INVITED"),
-            gte(this.schema.waitlist.expiresAt, new Date())
-          )
-        )
-        .orderBy(desc(this.schema.waitlist.createdAt))
-        .limit(1);
-      if (!claim) return ok({ status: true });
-
-      const [user] = await db
-        .select({ email: this.schema.users.email })
-        .from(this.schema.users)
-        .where(eq(this.schema.users.id, userId))
-        .limit(1);
-
-      await db
-        .update(this.schema.waitlist)
-        .set({
-          status: "ACCEPTED",
-          claimedAt: new Date(),
-          claimedEmail: user?.email ?? null,
-          updatedAt: new Date(),
-        })
-        .where(eq(this.schema.waitlist.id, claim.id));
-      return ok({ status: true });
-    });
-  }
-
-  async createAccountClaimMagicLink(
-    {
-      claimId,
-      userId,
-      email,
-      token,
-      url,
-      expiresAt,
-    }: {
-      claimId: string;
-      userId: string;
-      email: string;
-      token: string;
-      url: string;
-      expiresAt?: Date;
-    },
-    tx?: Orm
-  ): ServerResultAsync<AccountClaimMagicLink> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [link] = await db
-        .insert(this.schema.accountClaimMagicLinks)
-        .values({
-          claimId,
-          userId,
-          email,
-          token,
-          url,
-          expiresAt: expiresAt ?? null,
-        })
-        .returning();
-      return ok(link);
-    });
-  }
-
-  async listAccountClaimMagicLinks(
-    claimId: string,
-    tx?: Orm
-  ): ServerResultAsync<AccountClaimMagicLinkOutput[]> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const links = await db
-        .select({
-          id: this.schema.accountClaimMagicLinks.id,
-          claimId: this.schema.accountClaimMagicLinks.claimId,
-          userId: this.schema.accountClaimMagicLinks.userId,
-          email: this.schema.accountClaimMagicLinks.email,
-          url: this.schema.accountClaimMagicLinks.url,
-          expiresAt: this.schema.accountClaimMagicLinks.expiresAt,
-          createdAt: this.schema.accountClaimMagicLinks.createdAt,
-        })
-        .from(this.schema.accountClaimMagicLinks)
-        .where(eq(this.schema.accountClaimMagicLinks.claimId, claimId))
-        .orderBy(desc(this.schema.accountClaimMagicLinks.createdAt));
-      return ok(links);
-    });
-  }
-
-  async latestAccountClaimMagicLink(
-    claimId: string,
-    tx?: Orm
-  ): ServerResultAsync<AccountClaimMagicLinkOutput | null> {
-    return this.throwableAsync(async () => {
-      const db = tx ?? this.orm;
-      const [link] = await db
-        .select({
-          id: this.schema.accountClaimMagicLinks.id,
-          claimId: this.schema.accountClaimMagicLinks.claimId,
-          userId: this.schema.accountClaimMagicLinks.userId,
-          email: this.schema.accountClaimMagicLinks.email,
-          url: this.schema.accountClaimMagicLinks.url,
-          expiresAt: this.schema.accountClaimMagicLinks.expiresAt,
-          createdAt: this.schema.accountClaimMagicLinks.createdAt,
-        })
-        .from(this.schema.accountClaimMagicLinks)
-        .where(eq(this.schema.accountClaimMagicLinks.claimId, claimId))
-        .orderBy(desc(this.schema.accountClaimMagicLinks.createdAt))
-        .limit(1);
-      return ok(link ?? null);
-    });
-  }
-}