import { relations, sql } from "drizzle-orm";
import { integer, sqliteTable, text, unique } from "drizzle-orm/sqlite-core";

import { monitor } from "../monitors";
import { notification } from "../notifications";
import { page } from "../pages";
import { usersToWorkspaces } from "../users";
import { workspacePlans } from "./constants";

export const workspace = sqliteTable(
  "workspace",
  {
    id: integer("id").primaryKey(),
    slug: text("slug").notNull().unique(), // we love random words
    name: text("name"),

    stripeId: text("stripe_id", { length: 256 }).unique(),
    subscriptionId: text("subscription_id"),
    plan: text("plan", { enum: workspacePlans }),
    endsAt: integer("ends_at", { mode: "timestamp" }),
    paidUntil: integer("paid_until", { mode: "timestamp" }),
    limits: text("limits").default("{}").notNull(),
    createdAt: integer("created_at", { mode: "timestamp" }).default(
      sql`(strftime('%s', 'now'))`,
    ),
    updatedAt: integer("updated_at", { mode: "timestamp" }).default(
      sql`(strftime('%s', 'now'))`,
    ),

    dsn: text("dsn"), // should be removed soon
  },
  (t) => ({
    unique: unique().on(t.id, t.dsn),
  }),
);

export const workspaceRelations = relations(workspace, ({ many }) => ({
  usersToWorkspaces: many(usersToWorkspaces),
  pages: many(page),
  monitors: many(monitor),
  notifications: many(notification),
  // TODO: add checks or monitorRuns
}));