forked from syntaxbullet/aurorabot
81 lines
2.8 KiB
TypeScript
81 lines
2.8 KiB
TypeScript
import {
|
|
pgTable,
|
|
bigint,
|
|
varchar,
|
|
boolean,
|
|
jsonb,
|
|
timestamp,
|
|
integer,
|
|
primaryKey,
|
|
index,
|
|
} from 'drizzle-orm/pg-core';
|
|
import { relations, type InferSelectModel } from 'drizzle-orm';
|
|
|
|
// --- TYPES ---
|
|
export type Class = InferSelectModel<typeof classes>;
|
|
export type User = InferSelectModel<typeof users>;
|
|
export type UserTimer = InferSelectModel<typeof userTimers>;
|
|
|
|
// --- TABLES ---
|
|
export const classes = pgTable('classes', {
|
|
id: bigint('id', { mode: 'bigint' }).primaryKey(),
|
|
name: varchar('name', { length: 255 }).unique().notNull(),
|
|
balance: bigint('balance', { mode: 'bigint' }).default(0n),
|
|
roleId: varchar('role_id', { length: 255 }),
|
|
});
|
|
|
|
export const users = pgTable('users', {
|
|
id: bigint('id', { mode: 'bigint' }).primaryKey(),
|
|
classId: bigint('class_id', { mode: 'bigint' }).references(() => classes.id),
|
|
username: varchar('username', { length: 255 }).unique().notNull(),
|
|
isActive: boolean('is_active').default(true),
|
|
|
|
// Economy
|
|
balance: bigint('balance', { mode: 'bigint' }).default(0n),
|
|
xp: bigint('xp', { mode: 'bigint' }).default(0n),
|
|
level: integer('level').default(1),
|
|
dailyStreak: integer('daily_streak').default(0),
|
|
|
|
// Metadata
|
|
settings: jsonb('settings').default({}),
|
|
createdAt: timestamp('created_at', { withTimezone: true }).defaultNow(),
|
|
updatedAt: timestamp('updated_at', { withTimezone: true }).defaultNow(),
|
|
}, (table) => [
|
|
index('users_username_idx').on(table.username),
|
|
index('users_balance_idx').on(table.balance),
|
|
index('users_level_xp_idx').on(table.level, table.xp),
|
|
]);
|
|
|
|
export const userTimers = pgTable('user_timers', {
|
|
userId: bigint('user_id', { mode: 'bigint' })
|
|
.references(() => users.id, { onDelete: 'cascade' }).notNull(),
|
|
type: varchar('type', { length: 50 }).notNull(), // 'COOLDOWN', 'EFFECT', 'ACCESS'
|
|
key: varchar('key', { length: 100 }).notNull(), // TimerKey, 'chn_12345', 'xp_boost'
|
|
expiresAt: timestamp('expires_at', { withTimezone: true }).notNull(),
|
|
metadata: jsonb('metadata').default({}),
|
|
}, (table) => [
|
|
primaryKey({ columns: [table.userId, table.type, table.key] }),
|
|
index('user_timers_expires_at_idx').on(table.expiresAt),
|
|
index('user_timers_lookup_idx').on(table.userId, table.type, table.key),
|
|
]);
|
|
|
|
// --- RELATIONS ---
|
|
export const classesRelations = relations(classes, ({ many }) => ({
|
|
users: many(users),
|
|
}));
|
|
|
|
export const usersRelations = relations(users, ({ one, many }) => ({
|
|
class: one(classes, {
|
|
fields: [users.classId],
|
|
references: [classes.id],
|
|
}),
|
|
timers: many(userTimers),
|
|
}));
|
|
|
|
export const userTimersRelations = relations(userTimers, ({ one }) => ({
|
|
user: one(users, {
|
|
fields: [userTimers.userId],
|
|
references: [users.id],
|
|
}),
|
|
}));
|