118 lines
3.5 KiB
TypeScript
118 lines
3.5 KiB
TypeScript
import { relations } from "drizzle-orm";
|
|
import {
|
|
pgTable,
|
|
text,
|
|
timestamp,
|
|
boolean,
|
|
integer,
|
|
index,
|
|
} from "drizzle-orm/pg-core";
|
|
|
|
export const user = pgTable("user", {
|
|
id: text("id").primaryKey(),
|
|
name: text("name").notNull(),
|
|
email: text("email").notNull().unique(),
|
|
emailVerified: boolean("email_verified").default(false).notNull(),
|
|
image: text("image"),
|
|
createdAt: timestamp("created_at").notNull(),
|
|
updatedAt: timestamp("updated_at")
|
|
.$onUpdate(() => new Date())
|
|
.notNull(),
|
|
username: text("username").unique(),
|
|
displayUsername: text("display_username"),
|
|
role: text("role"),
|
|
banned: boolean("banned").default(false),
|
|
banReason: text("ban_reason"),
|
|
banExpires: timestamp("ban_expires"),
|
|
});
|
|
|
|
export const account = pgTable(
|
|
"account",
|
|
{
|
|
id: text("id").primaryKey(),
|
|
accountId: text("account_id").notNull(),
|
|
providerId: text("provider_id").notNull(),
|
|
userId: text("user_id")
|
|
.notNull()
|
|
.references(() => user.id, { onDelete: "cascade" }),
|
|
accessToken: text("access_token"),
|
|
refreshToken: text("refresh_token"),
|
|
idToken: text("id_token"),
|
|
accessTokenExpiresAt: timestamp("access_token_expires_at"),
|
|
refreshTokenExpiresAt: timestamp("refresh_token_expires_at"),
|
|
scope: text("scope"),
|
|
password: text("password"),
|
|
createdAt: timestamp("created_at").notNull(),
|
|
updatedAt: timestamp("updated_at")
|
|
.$onUpdate(() => new Date())
|
|
.notNull(),
|
|
},
|
|
(table) => [index("account_userId_idx").on(table.userId)],
|
|
);
|
|
|
|
export const verification = pgTable(
|
|
"verification",
|
|
{
|
|
id: text("id").primaryKey(),
|
|
identifier: text("identifier").notNull(),
|
|
value: text("value").notNull(),
|
|
expiresAt: timestamp("expires_at").notNull(),
|
|
createdAt: timestamp("created_at").notNull(),
|
|
updatedAt: timestamp("updated_at")
|
|
.$onUpdate(() => new Date())
|
|
.notNull(),
|
|
},
|
|
(table) => [index("verification_identifier_idx").on(table.identifier)],
|
|
);
|
|
|
|
export const apikey = pgTable(
|
|
"apikey",
|
|
{
|
|
id: text("id").primaryKey(),
|
|
name: text("name"),
|
|
start: text("start"),
|
|
prefix: text("prefix"),
|
|
key: text("key").notNull(),
|
|
userId: text("user_id")
|
|
.notNull()
|
|
.references(() => user.id, { onDelete: "cascade" }),
|
|
refillInterval: integer("refill_interval"),
|
|
refillAmount: integer("refill_amount"),
|
|
lastRefillAt: timestamp("last_refill_at"),
|
|
enabled: boolean("enabled").default(true),
|
|
rateLimitEnabled: boolean("rate_limit_enabled").default(true),
|
|
rateLimitTimeWindow: integer("rate_limit_time_window").default(86400000),
|
|
rateLimitMax: integer("rate_limit_max").default(10),
|
|
requestCount: integer("request_count").default(0),
|
|
remaining: integer("remaining"),
|
|
lastRequest: timestamp("last_request"),
|
|
expiresAt: timestamp("expires_at"),
|
|
createdAt: timestamp("created_at").notNull(),
|
|
updatedAt: timestamp("updated_at").notNull(),
|
|
permissions: text("permissions"),
|
|
metadata: text("metadata"),
|
|
},
|
|
(table) => [
|
|
index("apikey_key_idx").on(table.key),
|
|
index("apikey_userId_idx").on(table.userId),
|
|
],
|
|
);
|
|
|
|
export const userRelations = relations(user, ({ many }) => ({
|
|
accounts: many(account),
|
|
apikeys: many(apikey),
|
|
}));
|
|
|
|
export const accountRelations = relations(account, ({ one }) => ({
|
|
user: one(user, {
|
|
fields: [account.userId],
|
|
references: [user.id],
|
|
}),
|
|
}));
|
|
|
|
export const apikeyRelations = relations(apikey, ({ one }) => ({
|
|
user: one(user, {
|
|
fields: [apikey.userId],
|
|
references: [user.id],
|
|
}),
|
|
}));
|