73 lines
1.8 KiB
TypeScript
73 lines
1.8 KiB
TypeScript
import { sql } from "drizzle-orm";
|
|
import {
|
|
boolean,
|
|
foreignKey,
|
|
index,
|
|
jsonb,
|
|
numeric,
|
|
pgTable,
|
|
text,
|
|
unique,
|
|
} from "drizzle-orm/pg-core";
|
|
import { features } from "../../featureModels/featureTable.js";
|
|
import { organizations } from "../../orgModels/orgTable.js";
|
|
import type {
|
|
DbOverageAllowed,
|
|
DbSpendLimit,
|
|
DbUsageAlert,
|
|
} from "../billingControls/customerBillingControls.js";
|
|
import { customers } from "../cusTable.js";
|
|
|
|
export const entities = pgTable(
|
|
"entities",
|
|
{
|
|
id: text(),
|
|
org_id: text("org_id"),
|
|
created_at: numeric({ mode: "number" }).notNull(),
|
|
internal_id: text("internal_id").primaryKey().notNull(),
|
|
internal_customer_id: text("internal_customer_id").notNull(),
|
|
env: text(),
|
|
name: text(),
|
|
deleted: boolean().default(false).notNull(),
|
|
internal_feature_id: text("internal_feature_id"),
|
|
spend_limits: jsonb().$type<DbSpendLimit[]>(),
|
|
usage_alerts: jsonb().$type<DbUsageAlert[]>(),
|
|
overage_allowed: jsonb().$type<DbOverageAllowed[]>(),
|
|
|
|
// Optional...
|
|
feature_id: text("feature_id"),
|
|
},
|
|
(table) => [
|
|
foreignKey({
|
|
columns: [table.internal_customer_id],
|
|
foreignColumns: [customers.internal_id],
|
|
name: "entities_internal_customer_id_fkey",
|
|
}).onDelete("cascade"),
|
|
foreignKey({
|
|
columns: [table.internal_feature_id],
|
|
foreignColumns: [features.internal_id],
|
|
name: "entities_internal_feature_id_fkey",
|
|
}).onDelete("cascade"),
|
|
foreignKey({
|
|
columns: [table.org_id],
|
|
foreignColumns: [organizations.id],
|
|
name: "entities_org_id_fkey",
|
|
}).onDelete("cascade"),
|
|
|
|
unique("entity_id_constraint").on(
|
|
table.org_id,
|
|
table.env,
|
|
table.internal_customer_id,
|
|
table.id,
|
|
),
|
|
index("idx_entities_internal_customer_id").using(
|
|
"hash",
|
|
table.internal_customer_id,
|
|
),
|
|
index("idx_entities_customer_internal_desc").on(
|
|
table.internal_customer_id,
|
|
sql`${table.internal_id} DESC`,
|
|
),
|
|
],
|
|
);
|