Spaces:
Runtime error
Runtime error
| import type { Prisma, User } from "@core/database"; | |
| import type { GoogleProfile } from "@coji/remix-auth-google"; | |
| import { prisma } from "~/db.server"; | |
| import { env } from "~/env.server"; | |
| export type { User } from "@core/database"; | |
| type FindOrCreateMagicLink = { | |
| authenticationMethod: "MAGIC_LINK"; | |
| email: string; | |
| }; | |
| type FindOrCreateGoogle = { | |
| authenticationMethod: "GOOGLE"; | |
| email: User["email"]; | |
| authenticationProfile: GoogleProfile; | |
| authenticationExtraParams: Record<string, unknown>; | |
| }; | |
| type FindOrCreateUser = FindOrCreateMagicLink | FindOrCreateGoogle; | |
| type LoggedInUser = { | |
| user: User; | |
| isNewUser: boolean; | |
| }; | |
| export async function findOrCreateUser( | |
| input: FindOrCreateUser, | |
| ): Promise<LoggedInUser> { | |
| switch (input.authenticationMethod) { | |
| case "GOOGLE": { | |
| return findOrCreateGoogleUser(input); | |
| } | |
| case "MAGIC_LINK": { | |
| return findOrCreateMagicLinkUser(input); | |
| } | |
| } | |
| } | |
| export async function findOrCreateMagicLinkUser( | |
| input: FindOrCreateMagicLink, | |
| ): Promise<LoggedInUser> { | |
| if ( | |
| env.WHITELISTED_EMAILS && | |
| !new RegExp(env.WHITELISTED_EMAILS).test(input.email) | |
| ) { | |
| throw new Error("This email is unauthorized"); | |
| } | |
| const existingUser = await prisma.user.findFirst({ | |
| where: { | |
| email: input.email, | |
| }, | |
| }); | |
| const adminEmailRegex = env.ADMIN_EMAILS | |
| ? new RegExp(env.ADMIN_EMAILS) | |
| : undefined; | |
| const makeAdmin = adminEmailRegex ? adminEmailRegex.test(input.email) : false; | |
| const user = await prisma.user.upsert({ | |
| where: { | |
| email: input.email, | |
| }, | |
| update: { | |
| email: input.email, | |
| }, | |
| create: { | |
| email: input.email, | |
| authenticationMethod: "MAGIC_LINK", | |
| admin: makeAdmin, // only on create, to prevent automatically removing existing admins | |
| }, | |
| }); | |
| return { | |
| user, | |
| isNewUser: !existingUser, | |
| }; | |
| } | |
| export async function findOrCreateGoogleUser({ | |
| email, | |
| authenticationProfile, | |
| authenticationExtraParams, | |
| }: FindOrCreateGoogle): Promise<LoggedInUser> { | |
| const name = authenticationProfile._json.name; | |
| let avatarUrl: string | undefined = undefined; | |
| if (authenticationProfile.photos[0]) { | |
| avatarUrl = authenticationProfile.photos[0].value; | |
| } | |
| const displayName = authenticationProfile.displayName; | |
| const authProfile = authenticationProfile | |
| ? (authenticationProfile as unknown as Prisma.JsonObject) | |
| : undefined; | |
| const authExtraParams = authenticationExtraParams | |
| ? (authenticationExtraParams as unknown as Prisma.JsonObject) | |
| : undefined; | |
| const authIdentifier = `github:${authenticationProfile.id}`; | |
| const existingUser = await prisma.user.findUnique({ | |
| where: { | |
| authIdentifier, | |
| }, | |
| }); | |
| const existingEmailUser = await prisma.user.findUnique({ | |
| where: { | |
| email, | |
| }, | |
| }); | |
| if (existingEmailUser && !existingUser) { | |
| const user = await prisma.user.update({ | |
| where: { | |
| email, | |
| }, | |
| data: { | |
| authenticationProfile: authProfile, | |
| authenticationExtraParams: authExtraParams, | |
| avatarUrl, | |
| authIdentifier, | |
| }, | |
| }); | |
| return { | |
| user, | |
| isNewUser: false, | |
| }; | |
| } | |
| if (existingEmailUser && existingUser) { | |
| const user = await prisma.user.update({ | |
| where: { | |
| id: existingUser.id, | |
| }, | |
| data: {}, | |
| }); | |
| return { | |
| user, | |
| isNewUser: false, | |
| }; | |
| } | |
| const user = await prisma.user.upsert({ | |
| where: { | |
| authIdentifier, | |
| }, | |
| update: {}, | |
| create: { | |
| authenticationProfile: authProfile, | |
| authenticationExtraParams: authExtraParams, | |
| name, | |
| avatarUrl, | |
| displayName, | |
| authIdentifier, | |
| email, | |
| authenticationMethod: "GOOGLE", | |
| UserUsage: { | |
| create: { | |
| availableCredits: 200, | |
| }, | |
| }, | |
| }, | |
| }); | |
| return { | |
| user, | |
| isNewUser: !existingUser, | |
| }; | |
| } | |
| export async function getUserById(id: User["id"]) { | |
| const user = await prisma.user.findUnique({ where: { id } }); | |
| if (!user) { | |
| return null; | |
| } | |
| return { | |
| ...user, | |
| }; | |
| } | |
| export async function getUserLeftCredits(id: User["id"]) { | |
| const userUsage = await prisma.userUsage.findFirst({ where: { userId: id } }); | |
| if (!userUsage) { | |
| return null; | |
| } | |
| return { | |
| ...userUsage, | |
| }; | |
| } | |
| export async function getUserByEmail(email: User["email"]) { | |
| return prisma.user.findUnique({ where: { email } }); | |
| } | |
| export function updateUser({ | |
| id, | |
| marketingEmails, | |
| referralSource, | |
| onboardingComplete, | |
| }: Pick<User, "id" | "onboardingComplete"> & { | |
| marketingEmails?: boolean; | |
| referralSource?: string; | |
| }) { | |
| return prisma.user.update({ | |
| where: { id }, | |
| data: { | |
| marketingEmails, | |
| referralSource, | |
| confirmedBasicDetails: true, | |
| onboardingComplete, | |
| }, | |
| }); | |
| } | |
| export async function grantUserCloudAccess({ | |
| id, | |
| inviteCode, | |
| }: { | |
| id: string; | |
| inviteCode: string; | |
| }) { | |
| return prisma.user.update({ | |
| where: { id }, | |
| data: { | |
| InvitationCode: { | |
| connect: { | |
| code: inviteCode, | |
| }, | |
| }, | |
| }, | |
| }); | |
| } | |