@auth/drizzle-adapter 1.1.0 → 1.2.0

package/src/lib/sqlite.ts

@@ -13,6 +13,7 @@ import type {
   Adapter,
   AdapterAccount,
   AdapterAccountType,
+  AdapterAuthenticator,
   AdapterSession,
   AdapterUser,
   VerificationToken,
@@ -78,16 +79,44 @@ export function defineTables(
         token: text("token").notNull(),
         expires: integer("expires", { mode: "timestamp_ms" }).notNull(),
       },
-      (vt) => ({
-        compositePk: primaryKey({ columns: [vt.identifier, vt.token] }),
+      (verficationToken) => ({
+        compositePk: primaryKey({
+          columns: [verficationToken.identifier, verficationToken.token],
+        }),
       })
     ) satisfies DefaultSQLiteVerificationTokenTable)
 
+  const authenticatorsTable =
+    schema.authenticatorsTable ??
+    (sqliteTable(
+      "authenticator",
+      {
+        credentialID: text("credentialID").notNull().unique(),
+        userId: text("userId")
+          .notNull()
+          .references(() => usersTable.id, { onDelete: "cascade" }),
+        providerAccountId: text("providerAccountId").notNull(),
+        credentialPublicKey: text("credentialPublicKey").notNull(),
+        counter: integer("counter").notNull(),
+        credentialDeviceType: text("credentialDeviceType").notNull(),
+        credentialBackedUp: integer("credentialBackedUp", {
+          mode: "boolean",
+        }).notNull(),
+        transports: text("transports"),
+      },
+      (authenticator) => ({
+        compositePK: primaryKey({
+          columns: [authenticator.userId, authenticator.credentialID],
+        }),
+      })
+    ) satisfies DefaultSQLiteAuthenticatorTable)
+
   return {
     usersTable,
     accountsTable,
     sessionsTable,
     verificationTokensTable,
+    authenticatorsTable,
   }
 }
 
@@ -95,8 +124,13 @@ export function SQLiteDrizzleAdapter(
   client: BaseSQLiteDatabase<"sync" | "async", any, any>,
   schema?: DefaultSQLiteSchema
 ): Adapter {
-  const { usersTable, accountsTable, sessionsTable, verificationTokensTable } =
-    defineTables(schema)
+  const {
+    usersTable,
+    accountsTable,
+    sessionsTable,
+    verificationTokensTable,
+    authenticatorsTable,
+  } = defineTables(schema)
 
   return {
     async createUser(data: AdapterUser) {
@@ -132,7 +166,7 @@ export function SQLiteDrizzleAdapter(
       userId: string
       expires: Date
     }) {
-      return await client.insert(sessionsTable).values(data).returning().get()
+      return client.insert(sessionsTable).values(data).returning().get()
     },
     async getSessionAndUser(sessionToken: string) {
       const result = await client
@@ -207,7 +241,7 @@ export function SQLiteDrizzleAdapter(
         .run()
     },
     async createVerificationToken(data: VerificationToken) {
-      return await client
+      return client
         .insert(verificationTokensTable)
         .values(data)
         .returning()
@@ -243,21 +277,70 @@ export function SQLiteDrizzleAdapter(
         )
         .run()
     },
+    async getAccount(providerAccountId: string, provider: string) {
+      return client
+        .select()
+        .from(accountsTable)
+        .where(
+          and(
+            eq(accountsTable.provider, provider),
+            eq(accountsTable.providerAccountId, providerAccountId)
+          )
+        )
+        .then((res) => res[0] ?? null) as Promise<AdapterAccount | null>
+    },
+    async createAuthenticator(data: AdapterAuthenticator) {
+      return client
+        .insert(authenticatorsTable)
+        .values(data)
+        .returning()
+        .then((res) => res[0] ?? null)
+    },
+    async getAuthenticator(credentialID: string) {
+      return client
+        .select()
+        .from(authenticatorsTable)
+        .where(eq(authenticatorsTable.credentialID, credentialID))
+        .then((res) => res[0] ?? null)
+    },
+    async listAuthenticatorsByUserId(userId: string) {
+      return client
+        .select()
+        .from(authenticatorsTable)
+        .where(eq(authenticatorsTable.userId, userId))
+        .then((res) => res)
+    },
+    async updateAuthenticatorCounter(credentialID: string, newCounter: number) {
+      const authenticator = await client
+        .update(authenticatorsTable)
+        .set({ counter: newCounter })
+        .where(eq(authenticatorsTable.credentialID, credentialID))
+        .returning()
+        .then((res) => res[0])
+
+      if (!authenticator) throw new Error("Authenticator not found.")
+
+      return authenticator
+    },
   }
 }
 
 type DefaultSQLiteColumn<
   T extends {
-    data: string | number | Date
-    dataType: "string" | "number" | "date"
+    data: string | boolean | number | Date
+    dataType: "string" | "boolean" | "number" | "date"
     notNull: boolean
-    columnType: "SQLiteText" | "SQLiteTimestamp" | "SQLiteInteger"
+    columnType:
+      | "SQLiteText"
+      | "SQLiteBoolean"
+      | "SQLiteTimestamp"
+      | "SQLiteInteger"
   },
 > = SQLiteColumn<{
   name: string
   columnType: T["columnType"]
   data: T["data"]
-  driverParam: string | number
+  driverParam: string | number | boolean
   notNull: T["notNull"]
   hasDefault: boolean
   enumValues: any
@@ -429,9 +512,66 @@ export type DefaultSQLiteVerificationTokenTable = SQLiteTableWithColumns<{
   schema: string | undefined
 }>
 
+export type DefaultSQLiteAuthenticatorTable = SQLiteTableWithColumns<{
+  name: string
+  columns: {
+    credentialID: DefaultSQLiteColumn<{
+      columnType: "SQLiteText"
+      data: string
+      notNull: true
+      dataType: "string"
+    }>
+    userId: DefaultSQLiteColumn<{
+      columnType: "SQLiteText"
+      data: string
+      notNull: true
+      dataType: "string"
+    }>
+    providerAccountId: DefaultSQLiteColumn<{
+      columnType: "SQLiteText"
+      data: string
+      notNull: true
+      dataType: "string"
+    }>
+    credentialPublicKey: DefaultSQLiteColumn<{
+      columnType: "SQLiteText"
+      data: string
+      notNull: true
+      dataType: "string"
+    }>
+    counter: DefaultSQLiteColumn<{
+      columnType: "SQLiteInteger"
+      data: number
+      notNull: true
+      dataType: "number"
+    }>
+    credentialDeviceType: DefaultSQLiteColumn<{
+      columnType: "SQLiteText"
+      data: string
+      notNull: true
+      dataType: "string"
+    }>
+    credentialBackedUp: DefaultSQLiteColumn<{
+      columnType: "SQLiteBoolean"
+      data: boolean
+      notNull: true
+      dataType: "boolean"
+    }>
+    transports: DefaultSQLiteColumn<{
+      columnType: "SQLiteText"
+      data: string
+      notNull: false
+      dataType: "string"
+    }>
+  }
+  dialect: "sqlite"
+  schema: string | undefined
+}>
+
 export type DefaultSQLiteSchema = {
   usersTable: DefaultSQLiteUsersTable
   accountsTable: DefaultSQLiteAccountsTable
   sessionsTable?: DefaultSQLiteSessionsTable
   verificationTokensTable?: DefaultSQLiteVerificationTokenTable
+  authenticatorsTable?: DefaultSQLiteAuthenticatorTable
 }