53 lines
1.3 KiB
TypeScript
53 lines
1.3 KiB
TypeScript
import { eq, or } from "drizzle-orm";
|
|
import * as schema from "@db/schema";
|
|
import type { InsertUser } from "@db/schema";
|
|
import { getDb } from "./connection";
|
|
import { env } from "../lib/env";
|
|
|
|
export async function findUserByUnionId(unionId: string) {
|
|
const rows = await getDb()
|
|
.select()
|
|
.from(schema.users)
|
|
.where(eq(schema.users.unionId, unionId))
|
|
.limit(1);
|
|
return rows.at(0);
|
|
}
|
|
|
|
export async function findUserByLoginIdentifier(identifier: string) {
|
|
const rows = await getDb()
|
|
.select()
|
|
.from(schema.users)
|
|
.where(or(eq(schema.users.unionId, identifier), eq(schema.users.email, identifier)))
|
|
.limit(1);
|
|
return rows.at(0);
|
|
}
|
|
|
|
export async function touchUserLastSignIn(unionId: string) {
|
|
await getDb()
|
|
.update(schema.users)
|
|
.set({ lastSignInAt: new Date() })
|
|
.where(eq(schema.users.unionId, unionId));
|
|
}
|
|
|
|
export async function upsertUser(data: InsertUser) {
|
|
const values = { ...data };
|
|
const updateSet: Partial<InsertUser> = {
|
|
lastSignInAt: new Date(),
|
|
...data,
|
|
};
|
|
|
|
if (
|
|
values.role === undefined &&
|
|
values.unionId &&
|
|
values.unionId === env.ownerUnionId
|
|
) {
|
|
values.role = "admin";
|
|
updateSet.role = "admin";
|
|
}
|
|
|
|
await getDb()
|
|
.insert(schema.users)
|
|
.values(values)
|
|
.onDuplicateKeyUpdate({ set: updateSet });
|
|
}
|