@coursebuilder/adapter-drizzle/src/lib/mysql/schemas/commerce/subscription.ts

Drizzle adapter for Course Builder.

import { relations, sql } from 'drizzle-orm'
import {
	index,
	json,
	MySqlTableFn,
	primaryKey,
	timestamp,
	varchar,
} from 'drizzle-orm/mysql-core'

import { getOrganizationsSchema } from '../org/organizations.js'
import { getMerchantSubscriptionSchema } from './merchant-subscription.js'
import { getProductSchema } from './product.js'

// TODO: do we add a renewal date here? probably just add that stuff to fields
// 	even status feels out of place at the top level maybe
export function getSubscriptionSchema(mysqlTable: MySqlTableFn) {
	return mysqlTable(
		'Subscription',
		{
			id: varchar('id', { length: 191 }).notNull(),
			organizationId: varchar('organizationId', { length: 191 }),
			productId: varchar('productId', { length: 191 }).notNull(),
			createdAt: timestamp('createdAt', { mode: 'date', fsp: 3 })
				.default(sql`CURRENT_TIMESTAMP(3)`)
				.notNull(),
			merchantSubscriptionId: varchar('merchantSubscriptionId', {
				length: 191,
			}).notNull(),
			status: varchar('status', { length: 191 }).default('active').notNull(),
			fields: json('fields').$type<Record<string, any>>().default({}),
		},
		(table) => {
			return {
				subscriptionId: primaryKey({
					columns: [table.id],
					name: 'Subscription_id',
				}),
				organizationIdIdx: index('organizationId_idx').on(table.organizationId),
			}
		},
	)
}

export function getSubscriptionRelationsSchema(mysqlTable: MySqlTableFn) {
	const subscriptions = getSubscriptionSchema(mysqlTable)
	const products = getProductSchema(mysqlTable)

	const organizations = getOrganizationsSchema(mysqlTable)
	const merchantSubscriptions = getMerchantSubscriptionSchema(mysqlTable)

	return relations(subscriptions, ({ many, one }) => ({
		organization: one(organizations, {
			fields: [subscriptions.organizationId],
			references: [organizations.id],
			relationName: 'organization',
		}),
		product: one(products, {
			fields: [subscriptions.productId],
			references: [products.id],
			relationName: 'product',
		}),
		merchantSubscription: one(merchantSubscriptions, {
			fields: [subscriptions.merchantSubscriptionId],
			references: [merchantSubscriptions.id],
			relationName: 'merchantSubscription',
		}),
	}))
}