@auth/drizzle-adapter 0.0.0-manual.ffc00566 → 0.2.0

package/lib/sqlite.js

@@ -1,69 +1,66 @@
-import { integer, sqliteTable, text, primaryKey, } from "drizzle-orm/sqlite-core";
 import { eq, and } from "drizzle-orm";
-/** @internal */
-export const users = sqliteTable("users", {
-    id: text("id").notNull().primaryKey(),
-    name: text("name"),
-    email: text("email").notNull(),
-    emailVerified: integer("emailVerified", { mode: "timestamp_ms" }),
-    image: text("image"),
-});
-/** @internal */
-export const accounts = sqliteTable("accounts", {
-    userId: text("userId")
-        .notNull()
-        .references(() => users.id, { onDelete: "cascade" }),
-    type: text("type").$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) => ({
-    compoundKey: primaryKey(account.provider, account.providerAccountId),
-}));
-/** @internal */
-export const sessions = sqliteTable("sessions", {
-    sessionToken: text("sessionToken").notNull().primaryKey(),
-    userId: text("userId")
-        .notNull()
-        .references(() => users.id, { onDelete: "cascade" }),
-    expires: integer("expires", { mode: "timestamp_ms" }).notNull(),
-});
-/** @internal */
-export const verificationTokens = sqliteTable("verificationToken", {
-    identifier: text("identifier").notNull(),
-    token: text("token").notNull(),
-    expires: integer("expires", { mode: "timestamp_ms" }).notNull(),
-}, (vt) => ({
-    compoundKey: primaryKey(vt.identifier, vt.token),
-}));
-/** @internal */
-export const schema = { users, accounts, sessions, verificationTokens };
-/** @internal */
-export function SQLiteDrizzleAdapter(client) {
+import { integer, sqliteTable as defaultSqliteTableFn, text, primaryKey, } from "drizzle-orm/sqlite-core";
+export function createTables(sqliteTable) {
+    const users = sqliteTable("users", {
+        id: text("id").notNull().primaryKey(),
+        name: text("name"),
+        email: text("email").notNull(),
+        emailVerified: integer("emailVerified", { mode: "timestamp_ms" }),
+        image: text("image"),
+    });
+    const accounts = sqliteTable("accounts", {
+        userId: text("userId")
+            .notNull()
+            .references(() => users.id, { onDelete: "cascade" }),
+        type: text("type").$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) => ({
+        compoundKey: primaryKey(account.provider, account.providerAccountId),
+    }));
+    const sessions = sqliteTable("sessions", {
+        sessionToken: text("sessionToken").notNull().primaryKey(),
+        userId: text("userId")
+            .notNull()
+            .references(() => users.id, { onDelete: "cascade" }),
+        expires: integer("expires", { mode: "timestamp_ms" }).notNull(),
+    });
+    const verificationTokens = sqliteTable("verificationToken", {
+        identifier: text("identifier").notNull(),
+        token: text("token").notNull(),
+        expires: integer("expires", { mode: "timestamp_ms" }).notNull(),
+    }, (vt) => ({
+        compoundKey: primaryKey(vt.identifier, vt.token),
+    }));
+    return { users, accounts, sessions, verificationTokens };
+}
+export function SQLiteDrizzleAdapter(client, tableFn = defaultSqliteTableFn) {
+    const { users, accounts, sessions, verificationTokens } = createTables(tableFn);
     return {
-        createUser: (data) => {
+        createUser(data) {
             return client
                 .insert(users)
                 .values({ ...data, id: crypto.randomUUID() })
                 .returning()
                 .get();
         },
-        getUser: (data) => {
+        getUser(data) {
             return client.select().from(users).where(eq(users.id, data)).get() ?? null;
         },
-        getUserByEmail: (data) => {
+        getUserByEmail(data) {
             return (client.select().from(users).where(eq(users.email, data)).get() ?? null);
         },
-        createSession: (data) => {
+        createSession(data) {
             return client.insert(sessions).values(data).returning().get();
         },
-        getSessionAndUser: (data) => {
+        getSessionAndUser(data) {
             return (client
                 .select({
                 session: sessions,
@@ -74,7 +71,7 @@ export function SQLiteDrizzleAdapter(client) {
                 .innerJoin(users, eq(users.id, sessions.userId))
                 .get() ?? null);
         },
-        updateUser: (data) => {
+        updateUser(data) {
             if (!data.id) {
                 throw new Error("No user id.");
             }
@@ -85,7 +82,7 @@ export function SQLiteDrizzleAdapter(client) {
                 .returning()
                 .get();
         },
-        updateSession: (data) => {
+        updateSession(data) {
             return client
                 .update(sessions)
                 .set(data)
@@ -93,7 +90,7 @@ export function SQLiteDrizzleAdapter(client) {
                 .returning()
                 .get();
         },
-        linkAccount: (rawAccount) => {
+        linkAccount(rawAccount) {
             const updatedAccount = client
                 .insert(accounts)
                 .values(rawAccount)
@@ -112,7 +109,7 @@ export function SQLiteDrizzleAdapter(client) {
             };
             return account;
         },
-        getUserByAccount: (account) => {
+        getUserByAccount(account) {
             const results = client
                 .select()
                 .from(accounts)
@@ -121,17 +118,17 @@ export function SQLiteDrizzleAdapter(client) {
                 .get();
             return results?.users ?? null;
         },
-        deleteSession: (sessionToken) => {
+        deleteSession(sessionToken) {
             return (client
                 .delete(sessions)
                 .where(eq(sessions.sessionToken, sessionToken))
                 .returning()
                 .get() ?? null);
         },
-        createVerificationToken: (token) => {
+        createVerificationToken(token) {
             return client.insert(verificationTokens).values(token).returning().get();
         },
-        useVerificationToken: (token) => {
+        useVerificationToken(token) {
             try {
                 return (client
                     .delete(verificationTokens)
@@ -143,10 +140,10 @@ export function SQLiteDrizzleAdapter(client) {
                 throw new Error("No verification token found.");
             }
         },
-        deleteUser: (id) => {
+        deleteUser(id) {
             return client.delete(users).where(eq(users.id, id)).returning().get();
         },
-        unlinkAccount: (account) => {
+        unlinkAccount(account) {
             client
                 .delete(accounts)
                 .where(and(eq(accounts.providerAccountId, account.providerAccountId), eq(accounts.provider, account.provider)))

package/lib/utils.d.ts

@@ -1,9 +1,12 @@
-import { AnyMySqlTable, MySqlDatabase } from "drizzle-orm/mysql-core";
-import { AnyPgTable, PgDatabase } from "drizzle-orm/pg-core";
-import { AnySQLiteTable, BaseSQLiteDatabase } from "drizzle-orm/sqlite-core";
-import { DefaultSchema as PgSchema } from "./pg";
-import { DefaultSchema as MySqlSchema } from "./mysql";
-import { DefaultSchema as SQLiteSchema } from "./sqlite";
+import { MySqlDatabase } from "drizzle-orm/mysql-core";
+import { PgDatabase } from "drizzle-orm/pg-core";
+import { BaseSQLiteDatabase } from "drizzle-orm/sqlite-core";
+import type { AnyMySqlTable, MySqlTableFn } from "drizzle-orm/mysql-core";
+import type { AnyPgTable, PgTableFn } from "drizzle-orm/pg-core";
+import type { AnySQLiteTable, SQLiteTableFn } from "drizzle-orm/sqlite-core";
+import type { DefaultSchema as PgSchema } from "./pg.js";
+import type { DefaultSchema as MySqlSchema } from "./mysql.js";
+import type { DefaultSchema as SQLiteSchema } from "./sqlite.js";
 export type AnyMySqlDatabase = MySqlDatabase<any, any>;
 export type AnyPgDatabase = PgDatabase<any, any, any>;
 export type AnySQLiteDatabase = BaseSQLiteDatabase<any, any, any, any>;
@@ -14,6 +17,7 @@ export interface MinimumSchema {
 }
 export type SqlFlavorOptions = AnyMySqlDatabase | AnyPgDatabase | AnySQLiteDatabase;
 export type ClientFlavors<Flavor> = Flavor extends AnyMySqlDatabase ? MinimumSchema["mysql"] : Flavor extends AnyPgDatabase ? MinimumSchema["pg"] : Flavor extends AnySQLiteDatabase ? MinimumSchema["sqlite"] : never;
+export type TableFn<Flavor> = Flavor extends AnyMySqlDatabase ? MySqlTableFn : Flavor extends AnyPgDatabase ? PgTableFn : Flavor extends AnySQLiteDatabase ? AnySQLiteTable : SQLiteTableFn;
 export declare function isMySqlDatabase(db: any): db is MySqlDatabase<any, any, any, any>;
 export declare function isPgDatabase(db: any): db is PgDatabase<any, any, any>;
 export declare function isSQLiteDatabase(db: any): db is AnySQLiteDatabase;

package/lib/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/lib/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAA;AACrE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAA;AAC5D,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAC5E,OAAO,EAAE,aAAa,IAAI,QAAQ,EAAE,MAAM,MAAM,CAAA;AAChD,OAAO,EAAE,aAAa,IAAI,WAAW,EAAE,MAAM,SAAS,CAAA;AACtD,OAAO,EAAE,aAAa,IAAI,YAAY,EAAE,MAAM,UAAU,CAAA;AAExD,MAAM,MAAM,gBAAgB,GAAG,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;AACtD,MAAM,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;AACrD,MAAM,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;AAEtE,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;IAClD,EAAE,EAAE,QAAQ,GAAG,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAA;IACzC,MAAM,EAAE,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;CACtD;AAED,MAAM,MAAM,gBAAgB,GACxB,gBAAgB,GAChB,aAAa,GACb,iBAAiB,CAAA;AAErB,MAAM,MAAM,aAAa,CAAC,MAAM,IAAI,MAAM,SAAS,gBAAgB,GAC/D,aAAa,CAAC,OAAO,CAAC,GACtB,MAAM,SAAS,aAAa,GAC5B,aAAa,CAAC,IAAI,CAAC,GACnB,MAAM,SAAS,iBAAiB,GAChC,aAAa,CAAC,QAAQ,CAAC,GACvB,KAAK,CAAA;AAET,wBAAgB,eAAe,CAC7B,EAAE,EAAE,GAAG,GACN,EAAE,IAAI,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAEzC;AAED,wBAAgB,YAAY,CAAC,EAAE,EAAE,GAAG,GAAG,EAAE,IAAI,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAErE;AAED,wBAAgB,gBAAgB,CAAC,EAAE,EAAE,GAAG,GAAG,EAAE,IAAI,iBAAiB,CAEjE"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/lib/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAA;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAA;AAChD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAE5D,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAA;AACzE,OAAO,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAA;AAChE,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAA;AAC5E,OAAO,KAAK,EAAE,aAAa,IAAI,QAAQ,EAAE,MAAM,SAAS,CAAA;AACxD,OAAO,KAAK,EAAE,aAAa,IAAI,WAAW,EAAE,MAAM,YAAY,CAAA;AAC9D,OAAO,KAAK,EAAE,aAAa,IAAI,YAAY,EAAE,MAAM,aAAa,CAAA;AAEhE,MAAM,MAAM,gBAAgB,GAAG,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAAA;AACtD,MAAM,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;AACrD,MAAM,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;AAEtE,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;IAClD,EAAE,EAAE,QAAQ,GAAG,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAA;IACzC,MAAM,EAAE,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;CACtD;AAED,MAAM,MAAM,gBAAgB,GACxB,gBAAgB,GAChB,aAAa,GACb,iBAAiB,CAAA;AAErB,MAAM,MAAM,aAAa,CAAC,MAAM,IAAI,MAAM,SAAS,gBAAgB,GAC/D,aAAa,CAAC,OAAO,CAAC,GACtB,MAAM,SAAS,aAAa,GAC5B,aAAa,CAAC,IAAI,CAAC,GACnB,MAAM,SAAS,iBAAiB,GAChC,aAAa,CAAC,QAAQ,CAAC,GACvB,KAAK,CAAA;AAET,MAAM,MAAM,OAAO,CAAC,MAAM,IAAI,MAAM,SAAS,gBAAgB,GACzD,YAAY,GACZ,MAAM,SAAS,aAAa,GAC5B,SAAS,GACT,MAAM,SAAS,iBAAiB,GAChC,cAAc,GACd,aAAa,CAAA;AAEjB,wBAAgB,eAAe,CAC7B,EAAE,EAAE,GAAG,GACN,EAAE,IAAI,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAEzC;AAED,wBAAgB,YAAY,CAAC,EAAE,EAAE,GAAG,GAAG,EAAE,IAAI,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAErE;AAED,wBAAgB,gBAAgB,CAAC,EAAE,EAAE,GAAG,GAAG,EAAE,IAAI,iBAAiB,CAEjE"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@auth/drizzle-adapter",
-  "version": "0.0.0-manual.ffc00566",
+  "version": "0.2.0",
   "description": "Drizzle adapter for Auth.js.",
   "homepage": "https://authjs.dev",
   "repository": "https://github.com/nextauthjs/next-auth",
@@ -36,7 +36,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@auth/core": "0.10.0"
+    "@auth/core": "0.10.1"
   },
   "devDependencies": {
     "@types/better-sqlite3": "^7.6.4",
@@ -47,8 +47,8 @@
     "jest": "^27.4.3",
     "mysql2": "^3.2.0",
     "postgres": "^3.3.4",
-    "@next-auth/tsconfig": "0.0.0",
-    "@next-auth/adapter-test": "0.0.0"
+    "@next-auth/adapter-test": "0.0.0",
+    "@next-auth/tsconfig": "0.0.0"
   },
   "jest": {
     "preset": "@next-auth/adapter-test/jest"

package/src/index.ts

@@ -16,20 +16,24 @@
  * @module @auth/drizzle-adapter
  */
 
-import { mySqlDrizzleAdapter } from "./lib/mysql"
-import { pgDrizzleAdapter } from "./lib/pg"
-import { SQLiteDrizzleAdapter } from "./lib/sqlite"
+import { MySqlTableFn } from "drizzle-orm/mysql-core/index.js"
+import { PgTableFn } from "drizzle-orm/pg-core/index.js"
+import { SQLiteTableFn } from "drizzle-orm/sqlite-core/index.js"
+import { mySqlDrizzleAdapter } from "./lib/mysql.js"
+import { pgDrizzleAdapter } from "./lib/pg.js"
+import { SQLiteDrizzleAdapter } from "./lib/sqlite.js"
 import {
   isMySqlDatabase,
   isPgDatabase,
   isSQLiteDatabase,
   SqlFlavorOptions,
-} from "./lib/utils"
+  TableFn,
+} from "./lib/utils.js"
 
 import type { Adapter } from "@auth/core/adapters"
 
 /**
- * Add the adapter to your `app/api/[...nextauth]/route.js` next-auth configuration object.
+ * Add the adapter to your `pages/api/[...nextauth].ts` next-auth configuration object.
  *
  * ```ts title="pages/api/auth/[...nextauth].ts"
  * import NextAuth from "next-auth"
@@ -47,6 +51,10 @@ import type { Adapter } from "@auth/core/adapters"
  *   ],
  * })
  * ```
+ * 
+ * :::info
+ * If you're using multi-project schemas, you can pass your table function as a second argument
+ * :::
  *
  * ## Setup
  *
@@ -249,19 +257,20 @@ import type { Adapter } from "@auth/core/adapters"
  *
  **/
 export function DrizzleAdapter<SqlFlavor extends SqlFlavorOptions>(
-  db: SqlFlavor
+  db: SqlFlavor,
+  table?: TableFn<SqlFlavor>
 ): Adapter {
   if (isMySqlDatabase(db)) {
     // We need to cast to unknown since the type overlaps (PScale is MySQL based)
-    return mySqlDrizzleAdapter(db)
+    return mySqlDrizzleAdapter(db, table as MySqlTableFn)
   }
 
   if (isPgDatabase(db)) {
-    return pgDrizzleAdapter(db)
+    return pgDrizzleAdapter(db, table as PgTableFn)
   }
 
   if (isSQLiteDatabase(db)) {
-    return SQLiteDrizzleAdapter(db)
+    return SQLiteDrizzleAdapter(db, table as SQLiteTableFn)
   }
 
   throw new Error("Unsupported database type in Auth.js Drizzle adapter.")

package/src/lib/mysql.ts

@@ -2,82 +2,89 @@ import { and, eq } from "drizzle-orm"
 import {
   int,
   timestamp,
-  mysqlTable,
+  mysqlTable as defaultMySqlTableFn,
   primaryKey,
   varchar,
+  MySqlTableFn,
 } from "drizzle-orm/mysql-core"
+
 import type { Adapter, AdapterAccount } from "@auth/core/adapters"
-import { MySql2Database } from "drizzle-orm/mysql2"
+import type { MySql2Database } from "drizzle-orm/mysql2"
+
+export function createTables(mySqlTable: MySqlTableFn) {
+  const users = mySqlTable("users", {
+    id: varchar("id", { length: 255 }).notNull().primaryKey(),
+    name: varchar("name", { length: 255 }),
+    email: varchar("email", { length: 255 }).notNull(),
+    emailVerified: timestamp("emailVerified", {
+      mode: "date",
+      fsp: 3,
+    }).defaultNow(),
+    image: varchar("image", { length: 255 }),
+  })
 
-/** @internal */
-export const users = mysqlTable("users", {
-  id: varchar("id", { length: 255 }).notNull().primaryKey(),
-  name: varchar("name", { length: 255 }),
-  email: varchar("email", { length: 255 }).notNull(),
-  emailVerified: timestamp("emailVerified", {
-    mode: "date",
-    fsp: 3,
-  }).defaultNow(),
-  image: varchar("image", { length: 255 }),
-})
+  const accounts = mySqlTable(
+    "accounts",
+    {
+      userId: varchar("userId", { length: 255 })
+        .notNull()
+        .references(() => users.id, { onDelete: "cascade" }),
+      type: varchar("type", { length: 255 })
+        .$type<AdapterAccount["type"]>()
+        .notNull(),
+      provider: varchar("provider", { length: 255 }).notNull(),
+      providerAccountId: varchar("providerAccountId", {
+        length: 255,
+      }).notNull(),
+      refresh_token: varchar("refresh_token", { length: 255 }),
+      access_token: varchar("access_token", { length: 255 }),
+      expires_at: int("expires_at"),
+      token_type: varchar("token_type", { length: 255 }),
+      scope: varchar("scope", { length: 255 }),
+      id_token: varchar("id_token", { length: 255 }),
+      session_state: varchar("session_state", { length: 255 }),
+    },
+    (account) => ({
+      compoundKey: primaryKey(account.provider, account.providerAccountId),
+    })
+  )
 
-/** @internal */
-export const accounts = mysqlTable(
-  "accounts",
-  {
+  const sessions = mySqlTable("sessions", {
+    sessionToken: varchar("sessionToken", { length: 255 })
+      .notNull()
+      .primaryKey(),
     userId: varchar("userId", { length: 255 })
       .notNull()
       .references(() => users.id, { onDelete: "cascade" }),
-    type: varchar("type", { length: 255 })
-      .$type<AdapterAccount["type"]>()
-      .notNull(),
-    provider: varchar("provider", { length: 255 }).notNull(),
-    providerAccountId: varchar("providerAccountId", { length: 255 }).notNull(),
-    refresh_token: varchar("refresh_token", { length: 255 }),
-    access_token: varchar("access_token", { length: 255 }),
-    expires_at: int("expires_at"),
-    token_type: varchar("token_type", { length: 255 }),
-    scope: varchar("scope", { length: 255 }),
-    id_token: varchar("id_token", { length: 255 }),
-    session_state: varchar("session_state", { length: 255 }),
-  },
-  (account) => ({
-    compoundKey: primaryKey(account.provider, account.providerAccountId),
+    expires: timestamp("expires", { mode: "date" }).notNull(),
   })
-)
 
-/** @internal */
-export const sessions = mysqlTable("sessions", {
-  sessionToken: varchar("sessionToken", { length: 255 }).notNull().primaryKey(),
-  userId: varchar("userId", { length: 255 })
-    .notNull()
-    .references(() => users.id, { onDelete: "cascade" }),
-  expires: timestamp("expires", { mode: "date" }).notNull(),
-})
+  const verificationTokens = mySqlTable(
+    "verificationToken",
+    {
+      identifier: varchar("identifier", { length: 255 }).notNull(),
+      token: varchar("token", { length: 255 }).notNull(),
+      expires: timestamp("expires", { mode: "date" }).notNull(),
+    },
+    (vt) => ({
+      compoundKey: primaryKey(vt.identifier, vt.token),
+    })
+  )
 
-/** @internal */
-export const verificationTokens = mysqlTable(
-  "verificationToken",
-  {
-    identifier: varchar("identifier", { length: 255 }).notNull(),
-    token: varchar("token", { length: 255 }).notNull(),
-    expires: timestamp("expires", { mode: "date" }).notNull(),
-  },
-  (vt) => ({
-    compoundKey: primaryKey(vt.identifier, vt.token),
-  })
-)
+  return { users, accounts, sessions, verificationTokens }
+}
 
-/** @internal */
-export const schema = { users, accounts, sessions, verificationTokens }
-export type DefaultSchema = typeof schema
+export type DefaultSchema = ReturnType<typeof createTables>
 
-/** @internal */
 export function mySqlDrizzleAdapter(
-  client: MySql2Database<Record<string, never>>
+  client: MySql2Database<Record<string, never>>,
+  tableFn = defaultMySqlTableFn
 ): Adapter {
+  const { users, accounts, sessions, verificationTokens } =
+    createTables(tableFn)
+
   return {
-    createUser: async (data) => {
+    async createUser(data) {
       const id = crypto.randomUUID()
 
       await client.insert(users).values({ ...data, id })
@@ -88,7 +95,7 @@ export function mySqlDrizzleAdapter(
         .where(eq(users.id, id))
         .then((res) => res[0])
     },
-    getUser: async (data) => {
+    async getUser(data) {
       const thing =
         (await client
           .select()
@@ -98,7 +105,7 @@ export function mySqlDrizzleAdapter(
 
       return thing
     },
-    getUserByEmail: async (data) => {
+    async getUserByEmail(data) {
       const user =
         (await client
           .select()
@@ -108,7 +115,7 @@ export function mySqlDrizzleAdapter(
 
       return user
     },
-    createSession: async (data) => {
+    async createSession(data) {
       await client.insert(sessions).values(data)
 
       return await client
@@ -117,7 +124,7 @@ export function mySqlDrizzleAdapter(
         .where(eq(sessions.sessionToken, data.sessionToken))
         .then((res) => res[0])
     },
-    getSessionAndUser: async (data) => {
+    async getSessionAndUser(data) {
       const sessionAndUser =
         (await client
           .select({
@@ -131,7 +138,7 @@ export function mySqlDrizzleAdapter(
 
       return sessionAndUser
     },
-    updateUser: async (data) => {
+    async updateUser(data) {
       if (!data.id) {
         throw new Error("No user id.")
       }
@@ -144,7 +151,7 @@ export function mySqlDrizzleAdapter(
         .where(eq(users.id, data.id))
         .then((res) => res[0])
     },
-    updateSession: async (data) => {
+    async updateSession(data) {
       await client
         .update(sessions)
         .set(data)
@@ -156,13 +163,13 @@ export function mySqlDrizzleAdapter(
         .where(eq(sessions.sessionToken, data.sessionToken))
         .then((res) => res[0])
     },
-    linkAccount: async (rawAccount) => {
+    async linkAccount(rawAccount) {
       await client
         .insert(accounts)
         .values(rawAccount)
         .then((res) => res[0])
     },
-    getUserByAccount: async (account) => {
+    async getUserByAccount(account) {
       const dbAccount =
         (await client
           .select()
@@ -182,7 +189,7 @@ export function mySqlDrizzleAdapter(
 
       return dbAccount.users
     },
-    deleteSession: async (sessionToken) => {
+    async deleteSession(sessionToken) {
       const session =
         (await client
           .select()
@@ -196,7 +203,7 @@ export function mySqlDrizzleAdapter(
 
       return session
     },
-    createVerificationToken: async (token) => {
+    async createVerificationToken(token) {
       await client.insert(verificationTokens).values(token)
 
       return await client
@@ -205,7 +212,7 @@ export function mySqlDrizzleAdapter(
         .where(eq(verificationTokens.identifier, token.identifier))
         .then((res) => res[0])
     },
-    useVerificationToken: async (token) => {
+    async useVerificationToken(token) {
       try {
         const deletedToken =
           (await client
@@ -233,7 +240,7 @@ export function mySqlDrizzleAdapter(
         throw new Error("No verification token found.")
       }
     },
-    deleteUser: async (id) => {
+    async deleteUser(id) {
       const user = await client
         .select()
         .from(users)
@@ -244,7 +251,7 @@ export function mySqlDrizzleAdapter(
 
       return user
     },
-    unlinkAccount: async (account) => {
+    async unlinkAccount(account) {
       await client
         .delete(accounts)
         .where(