@coursebuilder/adapter-drizzle 0.0.2 → 0.0.4

package/src/lib/pg.ts

@@ -1,270 +1,297 @@
-import type { AdapterAccount, AdapterSession, AdapterUser } from '@auth/core/adapters'
+import type {
+	AdapterAccount,
+	AdapterSession,
+	AdapterUser,
+} from '@auth/core/adapters'
 import { and, eq } from 'drizzle-orm'
 import {
-  pgTable as defaultPgTableFn,
-  integer,
-  json,
-  PgDatabase,
-  pgEnum,
-  PgTableFn,
-  primaryKey,
-  text,
-  timestamp,
-  varchar,
+	pgTable as defaultPgTableFn,
+	integer,
+	json,
+	PgDatabase,
+	pgEnum,
+	PgTableFn,
+	primaryKey,
+	text,
+	timestamp,
+	varchar,
 } from 'drizzle-orm/pg-core'
 
 import { type CourseBuilderAdapter } from '@coursebuilder/core/adapters'
 
 export function createTables(pgTable: PgTableFn) {
-  const users = pgTable('user', {
-    id: text('id').notNull().primaryKey(),
-    name: text('name'),
-    email: text('email').notNull(),
-    emailVerified: timestamp('emailVerified', { mode: 'date' }),
-    role: pgEnum('role', ['user', 'admin'])('role').default('user'),
-    image: text('image'),
-  })
+	const users = pgTable('user', {
+		id: text('id').notNull().primaryKey(),
+		name: text('name'),
+		email: text('email').notNull(),
+		emailVerified: timestamp('emailVerified', { mode: 'date' }),
+		role: pgEnum('role', ['user', 'admin'])('role').default('user'),
+		image: text('image'),
+	})
 
-  const accounts = pgTable(
-    'account',
-    {
-      userId: text('userId')
-        .notNull()
-        .references(() => users.id, { onDelete: 'cascade' }),
-      type: text('type').$type<AdapterAccount['type']>().notNull(),
-      provider: text('provider').notNull(),
-      providerAccountId: text('providerAccountId').notNull(),
-      refresh_token: text('refresh_token'),
-      access_token: text('access_token'),
-      expires_at: integer('expires_at'),
-      token_type: text('token_type'),
-      scope: text('scope'),
-      id_token: text('id_token'),
-      session_state: text('session_state'),
-    },
-    (account) => ({
-      pk: primaryKey({ columns: [account.provider, account.providerAccountId] }),
-    }),
-  )
+	const accounts = pgTable(
+		'account',
+		{
+			userId: text('userId')
+				.notNull()
+				.references(() => users.id, { onDelete: 'cascade' }),
+			type: text('type').$type<AdapterAccount['type']>().notNull(),
+			provider: text('provider').notNull(),
+			providerAccountId: text('providerAccountId').notNull(),
+			refresh_token: text('refresh_token'),
+			access_token: text('access_token'),
+			expires_at: integer('expires_at'),
+			token_type: text('token_type'),
+			scope: text('scope'),
+			id_token: text('id_token'),
+			session_state: text('session_state'),
+		},
+		(account) => ({
+			pk: primaryKey({
+				columns: [account.provider, account.providerAccountId],
+			}),
+		}),
+	)
 
-  const sessions = pgTable('session', {
-    sessionToken: text('sessionToken').notNull().primaryKey(),
-    userId: text('userId')
-      .notNull()
-      .references(() => users.id, { onDelete: 'cascade' }),
-    expires: timestamp('expires', { mode: 'date' }).notNull(),
-  })
+	const sessions = pgTable('session', {
+		sessionToken: text('sessionToken').notNull().primaryKey(),
+		userId: text('userId')
+			.notNull()
+			.references(() => users.id, { onDelete: 'cascade' }),
+		expires: timestamp('expires', { mode: 'date' }).notNull(),
+	})
 
-  const verificationTokens = pgTable(
-    'verificationToken',
-    {
-      identifier: text('identifier').notNull(),
-      token: text('token').notNull(),
-      expires: timestamp('expires', { mode: 'date' }).notNull(),
-    },
-    (vt) => ({
-      pk: primaryKey({ columns: [vt.identifier, vt.token] }),
-    }),
-  )
+	const verificationTokens = pgTable(
+		'verificationToken',
+		{
+			identifier: text('identifier').notNull(),
+			token: text('token').notNull(),
+			expires: timestamp('expires', { mode: 'date' }).notNull(),
+		},
+		(vt) => ({
+			pk: primaryKey({ columns: [vt.identifier, vt.token] }),
+		}),
+	)
 
-  const contentResource = pgTable('contentResource', {
-    id: varchar('id', { length: 255 }).notNull().primaryKey(),
-    type: varchar('type', { length: 255 }).notNull(),
-    createdById: varchar('createdById', { length: 255 }).notNull(),
-    fields: json('fields').$type<Record<string, any>>().default({}),
-    createdAt: timestamp('createdAt', {
-      mode: 'date',
-      precision: 6,
-      withTimezone: true,
-    }).defaultNow(),
-    updatedAt: timestamp('updatedAt', {
-      mode: 'date',
-      precision: 6,
-      withTimezone: true,
-    }).defaultNow(),
-    deletedAt: timestamp('deletedAt', {
-      mode: 'date',
-      precision: 6,
-      withTimezone: true,
-    }),
-  })
+	const contentResource = pgTable('contentResource', {
+		id: varchar('id', { length: 255 }).notNull().primaryKey(),
+		type: varchar('type', { length: 255 }).notNull(),
+		createdById: varchar('createdById', { length: 255 }).notNull(),
+		fields: json('fields').$type<Record<string, any>>().default({}),
+		createdAt: timestamp('createdAt', {
+			mode: 'date',
+			precision: 6,
+			withTimezone: true,
+		}).defaultNow(),
+		updatedAt: timestamp('updatedAt', {
+			mode: 'date',
+			precision: 6,
+			withTimezone: true,
+		}).defaultNow(),
+		deletedAt: timestamp('deletedAt', {
+			mode: 'date',
+			precision: 6,
+			withTimezone: true,
+		}),
+	})
 
-  const contentResourceResource = pgTable(
-    'contentResourceResource',
-    {
-      resourceOfId: varchar('resourceOfId', { length: 255 }).notNull(),
-      resourceId: varchar('resourceId', { length: 255 }).notNull(),
-      position: integer('position').notNull().default(0),
-      metadata: json('fields').$type<Record<string, any>>().default({}),
-      createdAt: timestamp('createdAt', {
-        mode: 'date',
-        precision: 6,
-        withTimezone: true,
-      }).defaultNow(),
-      updatedAt: timestamp('updatedAt', {
-        mode: 'date',
-        precision: 6,
-        withTimezone: true,
-      }).defaultNow(),
-      deletedAt: timestamp('deletedAt', {
-        mode: 'date',
-        precision: 6,
-        withTimezone: true,
-      }),
-    },
-    (crr) => ({
-      pk: primaryKey({ columns: [crr.resourceOfId, crr.resourceId] }),
-    }),
-  )
+	const contentResourceResource = pgTable(
+		'contentResourceResource',
+		{
+			resourceOfId: varchar('resourceOfId', { length: 255 }).notNull(),
+			resourceId: varchar('resourceId', { length: 255 }).notNull(),
+			position: integer('position').notNull().default(0),
+			metadata: json('fields').$type<Record<string, any>>().default({}),
+			createdAt: timestamp('createdAt', {
+				mode: 'date',
+				precision: 6,
+				withTimezone: true,
+			}).defaultNow(),
+			updatedAt: timestamp('updatedAt', {
+				mode: 'date',
+				precision: 6,
+				withTimezone: true,
+			}).defaultNow(),
+			deletedAt: timestamp('deletedAt', {
+				mode: 'date',
+				precision: 6,
+				withTimezone: true,
+			}),
+		},
+		(crr) => ({
+			pk: primaryKey({ columns: [crr.resourceOfId, crr.resourceId] }),
+		}),
+	)
 
-  return { users, accounts, sessions, verificationTokens, contentResource }
+	return { users, accounts, sessions, verificationTokens, contentResource }
 }
 
 export type DefaultSchema = ReturnType<typeof createTables>
 
 export function pgDrizzleAdapter(
-  client: InstanceType<typeof PgDatabase>,
-  tableFn = defaultPgTableFn,
+	client: InstanceType<typeof PgDatabase>,
+	tableFn = defaultPgTableFn,
 ): CourseBuilderAdapter {
-  const { users, accounts, sessions, verificationTokens, contentResource } = createTables(tableFn)
+	const { users, accounts, sessions, verificationTokens, contentResource } =
+		createTables(tableFn)
 
-  return {
-    async createContentResource(resource) {
-      return client
-        .insert(contentResource)
-        .values({ ...resource, id: crypto.randomUUID() })
-        .returning()
-        .then((res) => res[0] ?? null)
-    },
-    async getContentResource(data) {
-      const result = await client
-        .select()
-        .from(contentResource)
-        .where(eq(contentResource.id, data))
-        .then((res) => res[0] ?? null)
-      return result ?? null
-    },
-    async createUser(data) {
-      return await client
-        .insert(users)
-        .values({ ...data, id: crypto.randomUUID() })
-        .returning()
-        .then((res) => (res[0] as AdapterUser) ?? null)
-    },
-    async getUser(data) {
-      return await client
-        .select()
-        .from(users)
-        .where(eq(users.id, data))
-        .then((res) => res[0] ?? null)
-    },
-    async getUserByEmail(data) {
-      return await client
-        .select()
-        .from(users)
-        .where(eq(users.email, data))
-        .then((res) => res[0] ?? null)
-    },
-    async createSession(data) {
-      return await client
-        .insert(sessions)
-        .values(data)
-        .returning()
-        .then((res) => res[0] as AdapterSession)
-    },
-    async getSessionAndUser(data) {
-      return await client
-        .select({
-          session: sessions,
-          user: users,
-        })
-        .from(sessions)
-        .where(eq(sessions.sessionToken, data))
-        .innerJoin(users, eq(users.id, sessions.userId))
-        .then((res) => res[0] ?? null)
-    },
-    async updateUser(data) {
-      if (!data.id) {
-        throw new Error('No user id.')
-      }
+	return {
+		async updateContentResourceFields(options) {
+			return null
+		},
+		async getVideoResource(id) {
+			//  TODO Implement
+			return null
+		},
+		async createContentResource(resource) {
+			return client
+				.insert(contentResource)
+				.values({ ...resource, id: crypto.randomUUID() })
+				.returning()
+				.then((res) => res[0] ?? null)
+		},
+		async getContentResource(data) {
+			const result = await client
+				.select()
+				.from(contentResource)
+				.where(eq(contentResource.id, data))
+				.then((res) => res[0] ?? null)
+			return result ?? null
+		},
+		async createUser(data) {
+			return await client
+				.insert(users)
+				.values({ ...data, id: crypto.randomUUID() })
+				.returning()
+				.then((res) => (res[0] as AdapterUser) ?? null)
+		},
+		async getUser(data) {
+			return await client
+				.select()
+				.from(users)
+				.where(eq(users.id, data))
+				.then((res) => res[0] ?? null)
+		},
+		async getUserByEmail(data) {
+			return await client
+				.select()
+				.from(users)
+				.where(eq(users.email, data))
+				.then((res) => res[0] ?? null)
+		},
+		async createSession(data) {
+			return await client
+				.insert(sessions)
+				.values(data)
+				.returning()
+				.then((res) => res[0] as AdapterSession)
+		},
+		async getSessionAndUser(data) {
+			return await client
+				.select({
+					session: sessions,
+					user: users,
+				})
+				.from(sessions)
+				.where(eq(sessions.sessionToken, data))
+				.innerJoin(users, eq(users.id, sessions.userId))
+				.then((res) => res[0] ?? null)
+		},
+		async updateUser(data) {
+			if (!data.id) {
+				throw new Error('No user id.')
+			}
 
-      return await client
-        .update(users)
-        .set(data)
-        .where(eq(users.id, data.id))
-        .returning()
-        .then((res) => res[0] as AdapterUser)
-    },
-    async updateSession(data) {
-      return await client
-        .update(sessions)
-        .set(data)
-        .where(eq(sessions.sessionToken, data.sessionToken))
-        .returning()
-        .then((res) => res[0])
-    },
-    async linkAccount(rawAccount) {
-      await client
-        .insert(accounts)
-        .values(rawAccount)
-        .returning()
-        .then((res) => res[0])
-    },
-    async getUserByAccount(account) {
-      const dbAccount =
-        (await client
-          .select()
-          .from(accounts)
-          .where(
-            and(eq(accounts.providerAccountId, account.providerAccountId), eq(accounts.provider, account.provider)),
-          )
-          .leftJoin(users, eq(accounts.userId, users.id))
-          .then((res) => res[0])) ?? null
+			return await client
+				.update(users)
+				.set(data)
+				.where(eq(users.id, data.id))
+				.returning()
+				.then((res) => res[0] as AdapterUser)
+		},
+		async updateSession(data) {
+			return await client
+				.update(sessions)
+				.set(data)
+				.where(eq(sessions.sessionToken, data.sessionToken))
+				.returning()
+				.then((res) => res[0])
+		},
+		async linkAccount(rawAccount) {
+			await client
+				.insert(accounts)
+				.values(rawAccount)
+				.returning()
+				.then((res) => res[0])
+		},
+		async getUserByAccount(account) {
+			const dbAccount =
+				(await client
+					.select()
+					.from(accounts)
+					.where(
+						and(
+							eq(accounts.providerAccountId, account.providerAccountId),
+							eq(accounts.provider, account.provider),
+						),
+					)
+					.leftJoin(users, eq(accounts.userId, users.id))
+					.then((res) => res[0])) ?? null
 
-      return dbAccount?.user ?? null
-    },
-    async deleteSession(sessionToken) {
-      const session = await client
-        .delete(sessions)
-        .where(eq(sessions.sessionToken, sessionToken))
-        .returning()
-        .then((res) => res[0] ?? null)
+			return dbAccount?.user ?? null
+		},
+		async deleteSession(sessionToken) {
+			const session = await client
+				.delete(sessions)
+				.where(eq(sessions.sessionToken, sessionToken))
+				.returning()
+				.then((res) => res[0] ?? null)
 
-      return session
-    },
-    async createVerificationToken(token) {
-      return await client
-        .insert(verificationTokens)
-        .values(token)
-        .returning()
-        .then((res) => res[0])
-    },
-    async useVerificationToken(token) {
-      try {
-        return await client
-          .delete(verificationTokens)
-          .where(and(eq(verificationTokens.identifier, token.identifier), eq(verificationTokens.token, token.token)))
-          .returning()
-          .then((res) => res[0] ?? null)
-      } catch (err) {
-        throw new Error('No verification token found.')
-      }
-    },
-    async deleteUser(id) {
-      await client
-        .delete(users)
-        .where(eq(users.id, id))
-        .returning()
-        .then((res) => res[0] ?? null)
-    },
-    async unlinkAccount(account) {
-      const { type, provider, providerAccountId, userId } = await client
-        .delete(accounts)
-        .where(and(eq(accounts.providerAccountId, account.providerAccountId), eq(accounts.provider, account.provider)))
-        .returning()
-        .then((res) => (res[0] as AdapterAccount) ?? null)
+			return session
+		},
+		async createVerificationToken(token) {
+			return await client
+				.insert(verificationTokens)
+				.values(token)
+				.returning()
+				.then((res) => res[0])
+		},
+		async useVerificationToken(token) {
+			try {
+				return await client
+					.delete(verificationTokens)
+					.where(
+						and(
+							eq(verificationTokens.identifier, token.identifier),
+							eq(verificationTokens.token, token.token),
+						),
+					)
+					.returning()
+					.then((res) => res[0] ?? null)
+			} catch (err) {
+				throw new Error('No verification token found.')
+			}
+		},
+		async deleteUser(id) {
+			await client
+				.delete(users)
+				.where(eq(users.id, id))
+				.returning()
+				.then((res) => res[0] ?? null)
+		},
+		async unlinkAccount(account) {
+			const { type, provider, providerAccountId, userId } = await client
+				.delete(accounts)
+				.where(
+					and(
+						eq(accounts.providerAccountId, account.providerAccountId),
+						eq(accounts.provider, account.provider),
+					),
+				)
+				.returning()
+				.then((res) => (res[0] as AdapterAccount) ?? null)
 
-      return { provider, type, providerAccountId, userId }
-    },
-  }
+			return { provider, type, providerAccountId, userId }
+		},
+	}
 }