Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153import { bigint, boolean, index, int, json, mysqlEnum, mysqlTable, primaryKey, uniqueIndex, varchar,} from "drizzle-orm/mysql-core"import { timestamps, ulid, utc, workspaceColumns } from "../drizzle/types"import { workspaceIndexes } from "./workspace.sql"
export const BlackPlans = ["20", "100", "200"] as constexport const BillingTable = mysqlTable( "billing", { ...workspaceColumns, ...timestamps, customerID: varchar("customer_id", { length: 255 }), paymentMethodID: varchar("payment_method_id", { length: 255 }), paymentMethodType: varchar("payment_method_type", { length: 32 }), paymentMethodLast4: varchar("payment_method_last4", { length: 4 }), balance: bigint("balance", { mode: "number" }).notNull(), monthlyLimit: int("monthly_limit"), monthlyUsage: bigint("monthly_usage", { mode: "number" }), timeMonthlyUsageUpdated: utc("time_monthly_usage_updated"), reload: boolean("reload"), reloadTrigger: int("reload_trigger"), reloadAmount: int("reload_amount"), reloadError: varchar("reload_error", { length: 255 }), timeReloadError: utc("time_reload_error"), timeReloadLockedTill: utc("time_reload_locked_till"), subscription: json("subscription").$type<{ status: "subscribed" seats: number plan: (typeof BlackPlans)[number] useBalance?: boolean coupon?: string }>(), subscriptionID: varchar("subscription_id", { length: 28 }), subscriptionPlan: mysqlEnum("subscription_plan", BlackPlans), timeSubscriptionBooked: utc("time_subscription_booked"), timeSubscriptionSelected: utc("time_subscription_selected"), liteSubscriptionID: varchar("lite_subscription_id", { length: 28 }), lite: json("lite").$type<{ useBalance?: boolean }>(), }, (table) => [ ...workspaceIndexes(table), uniqueIndex("global_customer_id").on(table.customerID), uniqueIndex("global_subscription_id").on(table.subscriptionID), ],)
export const SubscriptionTable = mysqlTable( "subscription", { ...workspaceColumns, ...timestamps, userID: ulid("user_id").notNull(), rollingUsage: bigint("rolling_usage", { mode: "number" }), fixedUsage: bigint("fixed_usage", { mode: "number" }), timeRollingUpdated: utc("time_rolling_updated"), timeFixedUpdated: utc("time_fixed_updated"), }, (table) => [...workspaceIndexes(table), uniqueIndex("workspace_user_id").on(table.workspaceID, table.userID)],)
export const LiteTable = mysqlTable( "lite", { ...workspaceColumns, ...timestamps, userID: ulid("user_id").notNull(), rollingUsage: bigint("rolling_usage", { mode: "number" }), weeklyUsage: bigint("weekly_usage", { mode: "number" }), monthlyUsage: bigint("monthly_usage", { mode: "number" }), timeRollingUpdated: utc("time_rolling_updated"), timeWeeklyUpdated: utc("time_weekly_updated"), timeMonthlyUpdated: utc("time_monthly_updated"), }, (table) => [...workspaceIndexes(table), uniqueIndex("workspace_user_id").on(table.workspaceID, table.userID)],)
export const PaymentTable = mysqlTable( "payment", { ...workspaceColumns, ...timestamps, customerID: varchar("customer_id", { length: 255 }), invoiceID: varchar("invoice_id", { length: 255 }), paymentID: varchar("payment_id", { length: 255 }), amount: bigint("amount", { mode: "number" }).notNull(), timeRefunded: utc("time_refunded"), enrichment: json("enrichment").$type< | { type: "subscription" | "lite" currency?: "inr" couponID?: string } | { type: "credit" } >(), }, (table) => [...workspaceIndexes(table)],)
export const UsageTable = mysqlTable( "usage", { ...workspaceColumns, ...timestamps, model: varchar("model", { length: 255 }).notNull(), provider: varchar("provider", { length: 255 }).notNull(), inputTokens: int("input_tokens").notNull(), outputTokens: int("output_tokens").notNull(), reasoningTokens: int("reasoning_tokens"), cacheReadTokens: int("cache_read_tokens"), cacheWrite5mTokens: int("cache_write_5m_tokens"), cacheWrite1hTokens: int("cache_write_1h_tokens"), cost: bigint("cost", { mode: "number" }).notNull(), keyID: ulid("key_id"), sessionID: varchar("session_id", { length: 30 }), enrichment: json("enrichment").$type<{ plan: "sub" | "byok" | "lite" }>(), }, (table) => [...workspaceIndexes(table), index("usage_time_created").on(table.workspaceID, table.timeCreated)],)
export const CouponType = [ "BUILDATHON", "GO1MONTH50", "GOFREEMONTH", "GO3MONTHS100", "GO6MONTHS100", "GO12MONTHS100",] as constexport const CouponTable = mysqlTable( "coupon", { email: varchar("email", { length: 255 }), type: mysqlEnum("type", CouponType).notNull(), timeRedeemed: utc("time_redeemed"), }, (table) => [primaryKey({ columns: [table.email, table.type] })],)