qweribot/src/db/schema.ts

import type { AccessToken } from "@twurple/auth";
import type { cheers as cheertypes } from "cheers";
import { relations } from "drizzle-orm";
import {
	boolean,
	integer,
	jsonb,
	pgTable,
	timestamp,
	uuid,
	varchar,
} from "drizzle-orm/pg-core";
import type { inventory, items } from "items";
import type { anivBots } from "lib/handleAnivMessage";

export const auth = pgTable("auth", {
	id: integer().primaryKey(),
	accesstoken: jsonb().$type<AccessToken>().notNull(),
});

export const users = pgTable("users", {
	id: integer().primaryKey().notNull(),
	username: varchar().notNull(),
	balance: integer().default(0).notNull(),
	inventory: jsonb().$type<inventory>().default({}).notNull(),
});

export const usersRelations = relations(users, ({ many }) => ({
	timeouts_target: many(timeouts),
	timeouts_shooter: many(timeouts),
	usedItems: many(usedItems),
	cheerEvents: many(cheerEvents),
	cheers: many(cheers),
	anivTimeouts: many(anivTimeouts),
	getLoots: many(getLoots),
}));

export const timeouts = pgTable("timeouts", {
	id: uuid().defaultRandom().primaryKey(),
	user: integer()
		.notNull()
		.references(() => users.id),
	target: integer()
		.notNull()
		.references(() => users.id),
	item: varchar().$type<items | cheertypes>().notNull(),
	created: timestamp().defaultNow().notNull(),
});

export const timeoutsRelations = relations(timeouts, ({ one }) => ({
	user: one(users, {
		fields: [timeouts.user],
		references: [users.id],
		relationName: "shooter",
	}),
	target: one(users, {
		fields: [timeouts.target],
		references: [users.id],
		relationName: "target",
	}),
}));

export const usedItems = pgTable("usedItems", {
	id: uuid().defaultRandom().primaryKey(),
	user: integer()
		.notNull()
		.references(() => users.id),
	item: varchar().$type<items>().notNull(),
	created: timestamp().defaultNow().notNull(),
});

export const usedItemsRelations = relations(usedItems, ({ one }) => ({
	user: one(users, {
		fields: [usedItems.user],
		references: [users.id],
	}),
}));

export const cheerEvents = pgTable("cheerEvents", {
	id: uuid().defaultRandom().primaryKey(),
	user: integer()
		.notNull()
		.references(() => users.id),
	event: varchar().$type<items | cheertypes>().notNull(),
	created: timestamp().defaultNow().notNull(),
});

export const cheerEventsRelations = relations(cheerEvents, ({ one }) => ({
	user: one(users, {
		fields: [cheerEvents.user],
		references: [users.id],
	}),
}));

export const cheers = pgTable("cheers", {
	id: uuid().defaultRandom().primaryKey(),
	user: integer()
		.notNull()
		.references(() => users.id),
	amount: integer().notNull(),
	created: timestamp().defaultNow().notNull(),
});

export const cheersRelations = relations(cheers, ({ one }) => ({
	user: one(users, {
		fields: [cheers.user],
		references: [users.id],
	}),
}));

export const anivTimeouts = pgTable("anivTimeouts", {
	id: uuid().defaultRandom().primaryKey(),
	user: integer()
		.notNull()
		.references(() => users.id),
	message: varchar().notNull(),
	anivBot: varchar().$type<anivBots>().notNull(),
	duration: integer(),
	created: timestamp().defaultNow().notNull(),
	timeout: boolean().default(true),
});

export const anivTimeoutsRelations = relations(anivTimeouts, ({ one }) => ({
	user: one(users, {
		fields: [anivTimeouts.user],
		references: [users.id],
	}),
}));

export type lootTriggers = "getloot" | "superloot";

export const getLoots = pgTable("getLoots", {
	id: uuid().defaultRandom().primaryKey(),
	user: integer()
		.notNull()
		.references(() => users.id),
	qbucks: integer().notNull(),
	items: jsonb().$type<inventory>().notNull(),
	trigger: varchar().$type<lootTriggers>().notNull(),
	created: timestamp().defaultNow().notNull(),
});

export const getLootsRelations = relations(getLoots, ({ one }) => ({
	user: one(users, {
		fields: [getLoots.user],
		references: [users.id],
	}),
}));