Using TypeIDs with Drizzle

schema.ts

export const apiKeys = mysqlTable(
  "api_keys",
  {
    id: typeId("api_keys").primaryKey(),
    organizationId: typeId("organizations", "organization_id").notNull(),
    name: varchar("name", { length: 256 }),
    prefix: varchar("prefix", { length: 256 }),
  }
);

typeid.ts

import { TypeID } from "typeid-js";
import { customType } from "drizzle-orm/mysql-core";

export const typeId = (name: string, column: string = "id") => {
  return customType<{
    data: string;
    notNull: true;
  }>({
    dataType: () => {
      return "varchar(36)";
    },
    fromDriver: (val: unknown) => {
      const parsed = TypeID.fromUUID(name, val as string);
      return parsed.toString();
    },
    toDriver: (val) => {
      const parsed = TypeID.fromString(val).toUUID();
      return parsed;
    },
  })(column).notNull().$defaultFn(() => typeid(name).toString())
};