@coursebuilder/adapter-drizzle 0.0.7 → 0.1.0

package/src/lib/pg.ts

@@ -3,7 +3,7 @@ import type {
 	AdapterSession,
 	AdapterUser,
 } from '@auth/core/adapters'
-import { and, eq } from 'drizzle-orm'
+import { and, asc, eq } from 'drizzle-orm'
 import {
 	pgTable as defaultPgTableFn,
 	integer,
@@ -18,6 +18,21 @@ import {
 } from 'drizzle-orm/pg-core'
 
 import { type CourseBuilderAdapter } from '@coursebuilder/core/adapters'
+import {
+	Coupon,
+	MerchantCoupon,
+	MerchantCustomer,
+	MerchantProduct,
+	Price,
+	Product,
+	Purchase,
+	PurchaseUserTransfer,
+	PurchaseUserTransferState,
+	ResourceProgress,
+	UpgradableProduct,
+	User,
+} from '@coursebuilder/core/schemas'
+import { ContentResourceSchema } from '@coursebuilder/core/schemas/content-resource-schema'
 
 export function createTables(pgTable: PgTableFn) {
 	const users = pgTable('user', {
@@ -123,7 +138,14 @@ export function createTables(pgTable: PgTableFn) {
 		}),
 	)
 
-	return { users, accounts, sessions, verificationTokens, contentResource }
+	return {
+		users,
+		accounts,
+		sessions,
+		verificationTokens,
+		contentResource,
+		contentResourceResource,
+	}
 }
 
 export type DefaultSchema = ReturnType<typeof createTables>
@@ -132,10 +154,219 @@ export function pgDrizzleAdapter(
 	client: InstanceType<typeof PgDatabase>,
 	tableFn = defaultPgTableFn,
 ): CourseBuilderAdapter {
-	const { users, accounts, sessions, verificationTokens, contentResource } =
-		createTables(tableFn)
+	const {
+		users,
+		accounts,
+		sessions,
+		verificationTokens,
+		contentResource,
+		contentResourceResource,
+	} = createTables(tableFn)
 
 	return {
+		client,
+		async createPurchase(options): Promise<Purchase> {
+			throw new Error('Method not implemented.')
+		},
+		availableUpgradesForProduct(
+			purchases: any,
+			productId: string,
+		): Promise<
+			{
+				upgradableTo: { id: string; name: string }
+				upgradableFrom: { id: string; name: string }
+			}[]
+		> {
+			return Promise.resolve([])
+		},
+		clearLessonProgressForUser(options: {
+			userId: string
+			lessons: { id: string; slug: string }[]
+		}): Promise<void> {
+			return Promise.resolve(undefined)
+		},
+		completeLessonProgressForUser(options: {
+			userId: string
+			lessonId?: string
+		}): Promise<ResourceProgress | null> {
+			return Promise.resolve(null)
+		},
+		couponForIdOrCode(options: {
+			code?: string
+			couponId?: string
+		}): Promise<(Coupon & { merchantCoupon: MerchantCoupon }) | null> {
+			return Promise.resolve(null)
+		},
+		createMerchantChargeAndPurchase(options: {
+			userId: string
+			productId: string
+			stripeChargeId: string
+			stripeCouponId?: string
+			merchantAccountId: string
+			merchantProductId: string
+			merchantCustomerId: string
+			stripeChargeAmount: number
+			quantity?: number
+			bulk?: boolean
+			checkoutSessionId: string
+			appliedPPPStripeCouponId: string | undefined
+			upgradedFromPurchaseId: string | undefined
+			usedCouponId: string | undefined
+			country?: string
+		}): Promise<Purchase> {
+			throw new Error('Method not implemented.')
+		},
+		findOrCreateMerchantCustomer(options: {
+			user: User
+			identifier: string
+			merchantAccountId: string
+		}): Promise<MerchantCustomer | null> {
+			return Promise.resolve(null)
+		},
+		findOrCreateUser(
+			email: string,
+			name?: string | null,
+		): Promise<{
+			user: User
+			isNewUser: boolean
+		}> {
+			throw new Error('Method not implemented.')
+		},
+		getCoupon(couponIdOrCode: string): Promise<Coupon | null> {
+			return Promise.resolve(null)
+		},
+		getCouponWithBulkPurchases(couponId: string): Promise<
+			| (Coupon & {
+					bulkCouponPurchases: { bulkCouponId: string }[]
+			  })
+			| null
+		> {
+			return Promise.resolve(null)
+		},
+		getDefaultCoupon(productIds?: string[]): Promise<{
+			defaultMerchantCoupon: MerchantCoupon
+			defaultCoupon: Coupon
+		} | null> {
+			return Promise.resolve(null)
+		},
+		getLessonProgressCountsByDate(): Promise<
+			{
+				count: number
+				completedAt: string
+			}[]
+		> {
+			return Promise.resolve([])
+		},
+		getLessonProgressForUser(userId: string): Promise<ResourceProgress[]> {
+			return Promise.resolve([])
+		},
+		getLessonProgresses(): Promise<ResourceProgress[]> {
+			return Promise.resolve([])
+		},
+		getMerchantCharge(merchantChargeId: string): Promise<{
+			id: string
+			identifier: string
+			merchantProductId: string
+		} | null> {
+			return Promise.resolve(null)
+		},
+		getMerchantCoupon(
+			merchantCouponId: string,
+		): Promise<MerchantCoupon | null> {
+			return Promise.resolve(null)
+		},
+		getMerchantProduct(
+			stripeProductId: string,
+		): Promise<MerchantProduct | null> {
+			return Promise.resolve(null)
+		},
+		getPrice(productId: string): Promise<Price | null> {
+			return Promise.resolve(null)
+		},
+		getProduct(productId: string): Promise<Product | null> {
+			return Promise.resolve(null)
+		},
+		getPurchase(purchaseId: string): Promise<Purchase | null> {
+			return Promise.resolve(null)
+		},
+		getPurchaseDetails(
+			purchaseId: string,
+			userId: string,
+		): Promise<{
+			purchase?: Purchase
+			existingPurchase?: Purchase
+			availableUpgrades: UpgradableProduct[]
+		}> {
+			return Promise.resolve({ availableUpgrades: [] })
+		},
+		getPurchaseForStripeCharge(
+			stripeChargeId: string,
+		): Promise<Purchase | null> {
+			return Promise.resolve(null)
+		},
+		getPurchaseUserTransferById(options: { id: string }): Promise<
+			| (PurchaseUserTransfer & {
+					sourceUser: User
+					targetUser: User | null
+					purchase: Purchase
+			  })
+			| null
+		> {
+			return Promise.resolve(null)
+		},
+		getPurchaseWithUser(purchaseId: string): Promise<
+			| (Purchase & {
+					user: User
+			  })
+			| null
+		> {
+			return Promise.resolve(null)
+		},
+		getPurchasesForUser(userId?: string): Promise<
+			(Purchase & {
+				bulkCoupon: { id: string; maxUses: number; usedCount: number } | null
+				product: { id: string; name: string }
+			})[]
+		> {
+			return Promise.resolve([])
+		},
+		getUserById(userId: string): Promise<User | null> {
+			return Promise.resolve(null)
+		},
+		pricesOfPurchasesTowardOneBundle(options: {
+			userId: string | undefined
+			bundleId: string
+		}): Promise<Price[]> {
+			return Promise.resolve([])
+		},
+		toggleLessonProgressForUser(options: {
+			userId: string
+			lessonId?: string
+			lessonSlug?: string
+		}): Promise<ResourceProgress | null> {
+			return Promise.resolve(null)
+		},
+		transferPurchasesToNewUser(options: {
+			merchantCustomerId: string
+			userId: string
+		}): Promise<unknown> {
+			return Promise.resolve(undefined)
+		},
+		updatePurchaseStatusForCharge(
+			chargeId: string,
+			status: 'Valid' | 'Refunded' | 'Disputed' | 'Banned' | 'Restricted',
+		): Promise<Purchase | undefined> {
+			return Promise.resolve(undefined)
+		},
+		updatePurchaseUserTransferTransferState(options: {
+			id: string
+			transferState: PurchaseUserTransferState
+		}): Promise<PurchaseUserTransfer | null> {
+			return Promise.resolve(null)
+		},
+		async addResourceToResource(options) {
+			return null
+		},
 		async updateContentResourceFields(options) {
 			return null
 		},
@@ -143,20 +374,71 @@ export function pgDrizzleAdapter(
 			//  TODO Implement
 			return null
 		},
-		async createContentResource(resource) {
-			return client
-				.insert(contentResource)
-				.values({ ...resource, id: crypto.randomUUID() })
-				.returning()
-				.then((res) => res[0] ?? null)
+		async createContentResource(data) {
+			const id = data.id || crypto.randomUUID()
+
+			await client.insert(contentResource).values({ ...data, id })
+
+			const resource = await client.query.contentResource.findFirst({
+				where: eq(contentResource.id, id),
+				with: {
+					resources: {
+						with: {
+							resource: {
+								with: {
+									resources: {
+										with: {
+											resource: true,
+										},
+										orderBy: asc(contentResourceResource.position),
+									},
+								},
+							},
+						},
+						orderBy: asc(contentResourceResource.position),
+					},
+				},
+			})
+
+			const parsedResource = ContentResourceSchema.safeParse(resource)
+
+			if (!parsedResource.success) {
+				console.error('Error parsing resource', resource)
+				throw new Error('Error parsing resource')
+			}
+
+			return parsedResource.data
 		},
 		async getContentResource(data) {
-			const result = await client
-				.select()
-				.from(contentResource)
-				.where(eq(contentResource.id, data))
-				.then((res) => res[0] ?? null)
-			return result ?? null
+			const resource = await client.query.contentResource.findFirst({
+				where: eq(contentResource.id, data),
+				with: {
+					resources: {
+						with: {
+							resource: {
+								with: {
+									resources: {
+										with: {
+											resource: true,
+										},
+										orderBy: asc(contentResourceResource.position),
+									},
+								},
+							},
+						},
+						orderBy: asc(contentResourceResource.position),
+					},
+				},
+			})
+
+			const parsedResource = ContentResourceSchema.safeParse(resource)
+
+			if (!parsedResource.success) {
+				console.error('Error parsing resource', resource)
+				return null
+			}
+
+			return parsedResource.data
 		},
 		async createUser(data) {
 			return await client

package/src/lib/sqlite.ts

@@ -1,5 +1,5 @@
 import type { AdapterAccount } from '@auth/core/adapters'
-import { and, eq, sql } from 'drizzle-orm'
+import { and, asc, eq, sql } from 'drizzle-orm'
 import {
 	BaseSQLiteDatabase,
 	sqliteTable as defaultSqliteTableFn,
@@ -10,6 +10,21 @@ import {
 } from 'drizzle-orm/sqlite-core'
 
 import { type CourseBuilderAdapter } from '@coursebuilder/core/adapters'
+import {
+	Coupon,
+	MerchantCoupon,
+	MerchantCustomer,
+	MerchantProduct,
+	Price,
+	Product,
+	Purchase,
+	PurchaseUserTransfer,
+	PurchaseUserTransferState,
+	ResourceProgress,
+	UpgradableProduct,
+	User,
+} from '@coursebuilder/core/schemas'
+import { ContentResourceSchema } from '@coursebuilder/core/schemas/content-resource-schema'
 
 import { stripUndefined } from './utils.js'
 
@@ -125,10 +140,219 @@ export function SQLiteDrizzleAdapter(
 	client: InstanceType<typeof BaseSQLiteDatabase>,
 	tableFn = defaultSqliteTableFn,
 ): CourseBuilderAdapter {
-	const { users, accounts, sessions, verificationTokens, contentResource } =
-		createTables(tableFn)
+	const {
+		users,
+		accounts,
+		sessions,
+		verificationTokens,
+		contentResource,
+		contentResourceResource,
+	} = createTables(tableFn)
 
 	return {
+		client,
+		availableUpgradesForProduct(
+			purchases: any,
+			productId: string,
+		): Promise<
+			{
+				upgradableTo: { id: string; name: string }
+				upgradableFrom: { id: string; name: string }
+			}[]
+		> {
+			return Promise.resolve([])
+		},
+		clearLessonProgressForUser(options: {
+			userId: string
+			lessons: { id: string; slug: string }[]
+		}): Promise<void> {
+			return Promise.resolve(undefined)
+		},
+		completeLessonProgressForUser(options: {
+			userId: string
+			lessonId?: string
+		}): Promise<ResourceProgress | null> {
+			return Promise.resolve(null)
+		},
+		couponForIdOrCode(options: {
+			code?: string
+			couponId?: string
+		}): Promise<(Coupon & { merchantCoupon: MerchantCoupon }) | null> {
+			return Promise.resolve(null)
+		},
+		createMerchantChargeAndPurchase(options: {
+			userId: string
+			productId: string
+			stripeChargeId: string
+			stripeCouponId?: string
+			merchantAccountId: string
+			merchantProductId: string
+			merchantCustomerId: string
+			stripeChargeAmount: number
+			quantity?: number
+			bulk?: boolean
+			checkoutSessionId: string
+			appliedPPPStripeCouponId: string | undefined
+			upgradedFromPurchaseId: string | undefined
+			usedCouponId: string | undefined
+			country?: string
+		}): Promise<Purchase> {
+			throw new Error('Method not implemented.')
+		},
+		findOrCreateMerchantCustomer(options: {
+			user: User
+			identifier: string
+			merchantAccountId: string
+		}): Promise<MerchantCustomer | null> {
+			return Promise.resolve(null)
+		},
+		findOrCreateUser(
+			email: string,
+			name?: string | null,
+		): Promise<{
+			user: User
+			isNewUser: boolean
+		}> {
+			throw new Error('Method not implemented.')
+		},
+		getCoupon(couponIdOrCode: string): Promise<Coupon | null> {
+			return Promise.resolve(null)
+		},
+		getCouponWithBulkPurchases(couponId: string): Promise<
+			| (Coupon & {
+					bulkCouponPurchases: { bulkCouponId: string }[]
+			  })
+			| null
+		> {
+			return Promise.resolve(null)
+		},
+		getDefaultCoupon(productIds?: string[]): Promise<{
+			defaultMerchantCoupon: MerchantCoupon
+			defaultCoupon: Coupon
+		} | null> {
+			return Promise.resolve(null)
+		},
+		getLessonProgressCountsByDate(): Promise<
+			{
+				count: number
+				completedAt: string
+			}[]
+		> {
+			return Promise.resolve([])
+		},
+		getLessonProgressForUser(userId: string): Promise<ResourceProgress[]> {
+			return Promise.resolve([])
+		},
+		getLessonProgresses(): Promise<ResourceProgress[]> {
+			return Promise.resolve([])
+		},
+		getMerchantCharge(merchantChargeId: string): Promise<{
+			id: string
+			identifier: string
+			merchantProductId: string
+		} | null> {
+			return Promise.resolve(null)
+		},
+		getMerchantCoupon(
+			merchantCouponId: string,
+		): Promise<MerchantCoupon | null> {
+			return Promise.resolve(null)
+		},
+		getMerchantProduct(
+			stripeProductId: string,
+		): Promise<MerchantProduct | null> {
+			return Promise.resolve(null)
+		},
+		getPrice(productId: string): Promise<Price | null> {
+			return Promise.resolve(null)
+		},
+		getProduct(productId: string): Promise<Product | null> {
+			return Promise.resolve(null)
+		},
+		getPurchase(purchaseId: string): Promise<Purchase | null> {
+			return Promise.resolve(null)
+		},
+		getPurchaseDetails(
+			purchaseId: string,
+			userId: string,
+		): Promise<{
+			purchase?: Purchase
+			existingPurchase?: Purchase
+			availableUpgrades: UpgradableProduct[]
+		}> {
+			return Promise.resolve({ availableUpgrades: [] })
+		},
+		getPurchaseForStripeCharge(
+			stripeChargeId: string,
+		): Promise<Purchase | null> {
+			return Promise.resolve(null)
+		},
+		getPurchaseUserTransferById(options: { id: string }): Promise<
+			| (PurchaseUserTransfer & {
+					sourceUser: User
+					targetUser: User | null
+					purchase: Purchase
+			  })
+			| null
+		> {
+			return Promise.resolve(null)
+		},
+		getPurchaseWithUser(purchaseId: string): Promise<
+			| (Purchase & {
+					user: User
+			  })
+			| null
+		> {
+			return Promise.resolve(null)
+		},
+		getPurchasesForUser(userId?: string): Promise<
+			(Purchase & {
+				bulkCoupon: { id: string; maxUses: number; usedCount: number } | null
+				product: { id: string; name: string }
+			})[]
+		> {
+			return Promise.resolve([])
+		},
+		getUserById(userId: string): Promise<User | null> {
+			return Promise.resolve(null)
+		},
+		pricesOfPurchasesTowardOneBundle(options: {
+			userId: string | undefined
+			bundleId: string
+		}): Promise<Price[]> {
+			return Promise.resolve([])
+		},
+		async createPurchase(options): Promise<Purchase> {
+			throw new Error('Method not implemented.')
+		},
+		toggleLessonProgressForUser(options: {
+			userId: string
+			lessonId?: string
+			lessonSlug?: string
+		}): Promise<ResourceProgress | null> {
+			return Promise.resolve(null)
+		},
+		transferPurchasesToNewUser(options: {
+			merchantCustomerId: string
+			userId: string
+		}): Promise<unknown> {
+			return Promise.resolve(undefined)
+		},
+		updatePurchaseStatusForCharge(
+			chargeId: string,
+			status: 'Valid' | 'Refunded' | 'Disputed' | 'Banned' | 'Restricted',
+		): Promise<Purchase | undefined> {
+			return Promise.resolve(undefined)
+		},
+		updatePurchaseUserTransferTransferState(options: {
+			id: string
+			transferState: PurchaseUserTransferState
+		}): Promise<PurchaseUserTransfer | null> {
+			return Promise.resolve(null)
+		},
+		async addResourceToResource(options) {
+			return null
+		},
 		async updateContentResourceFields(options) {
 			return null
 		},
@@ -136,20 +360,71 @@ export function SQLiteDrizzleAdapter(
 			//  TODO Implement
 			return null
 		},
-		async createContentResource(resource) {
-			return client
-				.insert(contentResource)
-				.values({ ...resource, id: crypto.randomUUID() })
-				.returning()
-				.get()
+		async createContentResource(data) {
+			const id = data.id || crypto.randomUUID()
+
+			await client.insert(contentResource).values({ ...data, id })
+
+			const resource = await client.query.contentResource.findFirst({
+				where: eq(contentResource.id, id),
+				with: {
+					resources: {
+						with: {
+							resource: {
+								with: {
+									resources: {
+										with: {
+											resource: true,
+										},
+										orderBy: asc(contentResourceResource.position),
+									},
+								},
+							},
+						},
+						orderBy: asc(contentResourceResource.position),
+					},
+				},
+			})
+
+			const parsedResource = ContentResourceSchema.safeParse(resource)
+
+			if (!parsedResource.success) {
+				console.error('Error parsing resource', resource)
+				throw new Error('Error parsing resource')
+			}
+
+			return parsedResource.data
 		},
 		async getContentResource(data) {
-			const result = await client
-				.select()
-				.from(contentResource)
-				.where(eq(contentResource.id, data))
-				.get()
-			return result ?? null
+			const resource = await client.query.contentResource.findFirst({
+				where: eq(contentResource.id, data),
+				with: {
+					resources: {
+						with: {
+							resource: {
+								with: {
+									resources: {
+										with: {
+											resource: true,
+										},
+										orderBy: asc(contentResourceResource.position),
+									},
+								},
+							},
+						},
+						orderBy: asc(contentResourceResource.position),
+					},
+				},
+			})
+
+			const parsedResource = ContentResourceSchema.safeParse(resource)
+
+			if (!parsedResource.success) {
+				console.error('Error parsing resource', resource)
+				return null
+			}
+
+			return parsedResource.data
 		},
 		async createUser(data) {
 			return client