@mikeatlast/ponder 0.16.1-fork.1

package/src/drizzle/hex.ts

@@ -0,0 +1,58 @@
+import {
+  type ColumnBaseConfig,
+  type ColumnBuilderBaseConfig,
+  type ColumnBuilderRuntimeConfig,
+  type MakeColumnConfig,
+  entityKind,
+} from "drizzle-orm";
+import {
+  type AnyPgTable,
+  PgColumn,
+  PgColumnBuilder,
+} from "drizzle-orm/pg-core";
+
+export type PgHexBuilderInitial<TName extends string> = PgHexBuilder<{
+  name: TName;
+  dataType: "string";
+  columnType: "PgHex";
+  data: `0x${string}`;
+  driverParam: string;
+  enumValues: undefined;
+  generated: undefined;
+}>;
+
+export class PgHexBuilder<
+  T extends ColumnBuilderBaseConfig<"string", "PgHex">,
+> extends PgColumnBuilder<T> {
+  static readonly [entityKind]: string = "PgHexBuilder";
+
+  constructor(name: T["name"]) {
+    super(name, "string", "PgHex");
+  }
+
+  /** @internal */
+  // @ts-ignore
+  override build<TTableName extends string>(
+    table: AnyPgTable<{ name: TTableName }>,
+  ): PgHex<MakeColumnConfig<T, TTableName>> {
+    return new PgHex<MakeColumnConfig<T, TTableName>>(
+      table,
+      this.config as ColumnBuilderRuntimeConfig<any, any>,
+    );
+  }
+}
+
+export class PgHex<
+  T extends ColumnBaseConfig<"string", "PgHex">,
+> extends PgColumn<T> {
+  static readonly [entityKind]: string = "PgHex";
+
+  getSQLType(): string {
+    return "text";
+  }
+
+  override mapToDriverValue(value: `0x${string}`) {
+    if (value.length % 2 === 0) return value.toLowerCase() as `0x${string}`;
+    return `0x0${value.slice(2)}`.toLowerCase() as `0x${string}`;
+  }
+}

package/src/drizzle/index.ts

@@ -0,0 +1,40 @@
+import { getTableColumns } from "drizzle-orm";
+import {
+  type PgColumn,
+  type PgTable,
+  getTableConfig,
+} from "drizzle-orm/pg-core";
+import { getColumnCasing } from "./kit/index.js";
+
+export const getPrimaryKeyColumns = (
+  table: PgTable,
+): { sql: string; js: string }[] => {
+  const primaryKeys = getTableConfig(table).primaryKeys;
+
+  const findJsName = (column: PgColumn): string => {
+    const name = column.name;
+    for (const [js, column] of Object.entries(getTableColumns(table))) {
+      if (column.name === name) return js;
+    }
+
+    throw "unreachable";
+  };
+
+  if (primaryKeys.length > 0) {
+    return primaryKeys[0]!.columns.map((column) => ({
+      sql: getColumnCasing(column, "snake_case"),
+      js: findJsName(column),
+    }));
+  }
+
+  const pkColumn = Object.values(getTableColumns(table)).find(
+    (c) => c.primary,
+  )!;
+
+  return [
+    {
+      sql: getColumnCasing(pkColumn, "snake_case"),
+      js: findJsName(pkColumn),
+    },
+  ];
+};

package/src/drizzle/json.ts

@@ -0,0 +1,159 @@
+import { type BaseError, BigIntSerializationError } from "@/internal/errors.js";
+import { type ColumnBaseConfig, entityKind } from "drizzle-orm";
+import type {
+  ColumnBuilderBaseConfig,
+  ColumnBuilderRuntimeConfig,
+  MakeColumnConfig,
+} from "drizzle-orm/column-builder";
+import {
+  type AnyPgTable,
+  PgColumn,
+  PgColumnBuilder,
+} from "drizzle-orm/pg-core";
+
+export type PgJsonBuilderInitial<TName extends string> = PgJsonBuilder<{
+  name: TName;
+  dataType: "json";
+  columnType: "PgJson";
+  data: unknown;
+  driverParam: string;
+  enumValues: undefined;
+}>;
+
+export class PgJsonBuilder<
+  T extends ColumnBuilderBaseConfig<"json", "PgJson">,
+> extends PgColumnBuilder<T> {
+  static readonly [entityKind]: string = "PgJsonBuilder";
+
+  constructor(name: T["name"]) {
+    super(name, "json", "PgJson");
+  }
+
+  /** @internal */
+  // @ts-ignore
+  override build<TTableName extends string>(
+    table: AnyPgTable<{ name: TTableName }>,
+  ): PgJson<MakeColumnConfig<T, TTableName>> {
+    return new PgJson<MakeColumnConfig<T, TTableName>>(
+      table,
+      this.config as ColumnBuilderRuntimeConfig<any, any>,
+    );
+  }
+}
+
+export class PgJson<
+  T extends ColumnBaseConfig<"json", "PgJson">,
+> extends PgColumn<T> {
+  static readonly [entityKind]: string = "PgJson";
+
+  getSQLType(): string {
+    return "json";
+  }
+
+  override mapToDriverValue(value: T["data"]): string {
+    try {
+      return JSON.stringify(value);
+    } catch (_error) {
+      let error = _error as Error;
+      if (
+        // node error message
+        error?.message?.includes("Do not know how to serialize a BigInt") ||
+        // bun error message
+        error?.message?.includes("cannot serialize BigInt")
+      ) {
+        error = new BigIntSerializationError(error.message);
+        (error as BaseError).meta.push(
+          "Hint:\n  The JSON column type does not support BigInt values. Use the replaceBigInts() helper function before inserting into the database. Docs: https://ponder.sh/docs/api-reference/ponder-utils#replacebigints",
+        );
+      }
+
+      throw error;
+    }
+  }
+
+  override mapFromDriverValue(value: T["data"] | string): T["data"] {
+    if (typeof value === "string") {
+      try {
+        return JSON.parse(value);
+      } catch {
+        return value as T["data"];
+      }
+    }
+    return value;
+  }
+}
+
+export type PgJsonbBuilderInitial<TName extends string> = PgJsonbBuilder<{
+  name: TName;
+  dataType: "json";
+  columnType: "PgJsonb";
+  data: unknown;
+  driverParam: unknown;
+  enumValues: undefined;
+}>;
+
+export class PgJsonbBuilder<
+  T extends ColumnBuilderBaseConfig<"json", "PgJsonb">,
+> extends PgColumnBuilder<T> {
+  static override readonly [entityKind]: string = "PgJsonbBuilder";
+
+  constructor(name: T["name"]) {
+    super(name, "json", "PgJsonb");
+  }
+
+  /** @internal */
+  // @ts-ignore
+  override build<TTableName extends string>(
+    table: AnyPgTable<{ name: TTableName }>,
+  ): PgJsonb<MakeColumnConfig<T, TTableName>> {
+    return new PgJsonb<MakeColumnConfig<T, TTableName>>(
+      table,
+      this.config as ColumnBuilderRuntimeConfig<any, any>,
+    );
+  }
+}
+
+export class PgJsonb<
+  T extends ColumnBaseConfig<"json", "PgJsonb">,
+> extends PgColumn<T> {
+  static override readonly [entityKind]: string = "PgJsonb";
+
+  // biome-ignore lint/complexity/noUselessConstructor: <explanation>
+  constructor(
+    table: AnyPgTable<{ name: T["tableName"] }>,
+    config: PgJsonbBuilder<T>["config"],
+  ) {
+    super(table, config);
+  }
+
+  getSQLType(): string {
+    return "jsonb";
+  }
+
+  override mapToDriverValue(value: T["data"]): string {
+    try {
+      return JSON.stringify(value);
+    } catch (_error) {
+      let error = _error as Error;
+      if (error?.message?.includes("Do not know how to serialize a BigInt")) {
+        error = new BigIntSerializationError(error.message);
+        (error as BaseError).meta.push(
+          "Hint:\n  The JSON column type does not support BigInt values. Use the replaceBigInts() helper function before inserting into the database. Docs: https://ponder.sh/docs/api-reference/ponder-utils#replacebigints",
+        );
+      }
+
+      throw error;
+    }
+  }
+
+  override mapFromDriverValue(value: T["data"] | string): T["data"] {
+    if (typeof value === "string") {
+      try {
+        return JSON.parse(value);
+      } catch {
+        return value as T["data"];
+      }
+    }
+    return value;
+  }
+}