effect-qb 0.16.0 → 0.17.0

package/src/internal/table.ts

@@ -12,9 +12,13 @@ import {
   resolvePrimaryKeyColumns,
   type DdlExpressionLike,
   type IndexKeySpec,
+  type MatchingColumnArityInput,
+  type NonEmptyColumnInput,
   type NormalizeColumns,
   type ReferentialAction,
   type TableOptionSpec,
+  type ValidateForeignKeyOptionColumns,
+  type ValidateIndexOptionColumns,
   type ValidateKnownColumns,
   type ValidatePrimaryKeyColumns,
   validateOptions
@@ -44,18 +48,32 @@ type InlinePrimaryKeyKeys<Fields extends TableFieldMap> = Extract<{
 type TableDialect<Fields extends TableFieldMap> = Fields[keyof Fields][typeof import("./column-state.js").ColumnTypeId]["dbType"]["dialect"]
 type TableKind = "schema" | "alias"
 type DefaultSchemaName = "public"
-type ClassOptionSpec = Exclude<TableOptionSpec, { readonly kind: "primaryKey" }>
+type FieldColumnName<Fields extends TableFieldMap> = Extract<keyof Fields, string>
+type FieldColumnList<Fields extends TableFieldMap> = readonly [FieldColumnName<Fields>, ...FieldColumnName<Fields>[]]
+type FieldIndexKeySpec<Fields extends TableFieldMap> =
+  | (Extract<IndexKeySpec, { readonly kind: "column" }> & { readonly column: FieldColumnName<Fields> })
+  | Extract<IndexKeySpec, { readonly kind: "expression" }>
+type ClassOptionSpec<Fields extends TableFieldMap = TableFieldMap> =
+  | (Omit<Extract<TableOptionSpec, { readonly kind: "index" }>, "columns" | "include" | "keys"> & {
+      readonly columns?: FieldColumnList<Fields>
+      readonly include?: readonly FieldColumnName<Fields>[]
+      readonly keys?: readonly [FieldIndexKeySpec<Fields>, ...FieldIndexKeySpec<Fields>[]]
+    })
+  | (Omit<Extract<TableOptionSpec, { readonly kind: "unique" }>, "columns"> & {
+      readonly columns: FieldColumnList<Fields>
+    })
+  | (Omit<Extract<TableOptionSpec, { readonly kind: "foreignKey" }>, "columns"> & {
+      readonly columns: FieldColumnList<Fields>
+    })
+  | Extract<TableOptionSpec, { readonly kind: "check" }>
 interface TableOptionBuilderLike<
   Spec extends TableOptionSpec = TableOptionSpec
 > {
-  (
-    table: TableDefinition<any, any, any, "schema", any>
-  ): TableDefinition<any, any, any, "schema", any>
   readonly option: Spec
 }
 
-type ClassTableOption = TableOptionBuilderLike<ClassOptionSpec>
-type ClassDeclaredTableOptions = readonly ClassTableOption[]
+type ClassTableOption<Fields extends TableFieldMap> = TableOptionBuilderLike<ClassOptionSpec<Fields>>
+type ClassDeclaredTableOptions<Fields extends TableFieldMap> = readonly ClassTableOption<Fields>[]
 
 type BuildPrimaryKey<
   Table extends TableDefinition<any, any, any, "schema", any>,
@@ -69,9 +87,13 @@ type OptionInputTable<
   Spec extends TableOptionSpec
 > = Spec extends { readonly kind: "primaryKey"; readonly columns: infer Columns extends readonly string[] }
   ? ValidatePrimaryKeyColumns<Table[typeof TypeId]["fields"], Columns> extends never ? never : Table
-  : Spec extends { readonly columns: infer Columns extends readonly string[] }
-    ? ValidateKnownColumns<Table[typeof TypeId]["fields"], Columns> extends never ? never : Table
-    : Table
+  : Spec extends { readonly kind: "index" }
+    ? ValidateIndexOptionColumns<Table[typeof TypeId]["fields"], Spec> extends never ? never : Table
+    : Spec extends { readonly kind: "foreignKey" }
+      ? ValidateForeignKeyOptionColumns<Table[typeof TypeId]["fields"], Spec> extends never ? never : Table
+      : Spec extends { readonly columns: infer Columns extends readonly string[] }
+        ? ValidateKnownColumns<Table[typeof TypeId]["fields"], Columns> extends never ? never : Table
+        : Table
 
 type ApplyOption<
   Table extends TableDefinition<any, any, any, "schema", any>,
@@ -90,6 +112,26 @@ type ApplyOption<
     "schema"
   >
 
+export type ValidateDeclaredOptions<
+  Table extends TableDefinition<any, any, any, "schema", any>,
+  Options extends DeclaredTableOptions
+> = {
+  readonly [K in keyof Options]: Options[K] extends TableOptionBuilderLike<infer Spec>
+    ? OptionInputTable<Table, Spec> extends never ? never : Options[K]
+    : never
+}
+
+export type ApplyDeclaredOptions<
+  Table extends TableDefinition<any, any, any, "schema", any>,
+  Options extends DeclaredTableOptions
+> = Options extends readonly [infer Head, ...infer Tail]
+  ? Head extends TableOptionBuilderLike<infer Spec>
+    ? Tail extends DeclaredTableOptions
+      ? ApplyDeclaredOptions<ApplyOption<Table, Spec>, Tail>
+      : ApplyOption<Table, Spec>
+    : Table
+  : Table
+
 export type MissingSelfGeneric = "Missing `Self` generic - use `class Self extends Table.Class<Self>(...) {}`"
 
 /** Bound columns keyed by field name for a particular table. */
@@ -132,16 +174,17 @@ export interface TableSchemaNamespace<SchemaName extends string> {
   readonly table: <
     Name extends string,
     Fields extends TableFieldMap,
+    const Options extends DeclaredTableOptions,
     PrimaryKeyColumns extends keyof Fields & string = InlinePrimaryKeyKeys<Fields>
   >(
     name: Name,
     fields: Fields,
-    ...options: DeclaredTableOptions
-  ) => TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>
+    ...options: Options & ValidateDeclaredOptions<TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>, Options>
+  ) => ApplyDeclaredOptions<TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>, Options>
 }
 
 export type DeclaredTableOptions = readonly TableOptionBuilderLike[]
-export type { DdlExpressionLike, IndexKeySpec, NormalizeColumns, ReferentialAction } from "./table-options.js"
+export type { DdlExpressionLike, IndexKeySpec, MatchingColumnArityInput, NonEmptyColumnInput, NormalizeColumns, ReferentialAction } from "./table-options.js"
 
 export type TableDefinition<
   Name extends string,
@@ -192,7 +235,7 @@ export type TableClassStatic<
   >
   readonly [OptionsSymbol]: readonly TableOptionSpec[]
   readonly [DeclaredOptionsSymbol]?: readonly TableOptionSpec[]
-  readonly [options]?: ClassDeclaredTableOptions
+  readonly [options]?: ClassDeclaredTableOptions<Fields>
   readonly tableName: Name
 } & BoundColumns<Name, Fields> & Plan.RowSet<
     BoundColumns<Name, Fields>,
@@ -204,6 +247,14 @@ export type TableClassStatic<
 /** Minimal structural table-like contract used across helper APIs. */
 export type AnyTable = TableDefinition<any, any, any, any, any> | TableClassStatic<any, any, any, any>
 
+type FieldsOfAnyTable<Table extends AnyTable> = Table extends TableDefinition<any, infer Fields extends TableFieldMap, any, any, any>
+  ? Fields
+  : Table extends TableClassStatic<any, infer Fields extends TableFieldMap, any, any>
+    ? Fields
+    : never
+
+type ColumnNamesOfAnyTable<Table extends AnyTable> = Extract<keyof FieldsOfAnyTable<Table>, string>
+
 /** Public table-option builder type used by `Table.index`, `Table.primaryKey`, and friends. */
 export type TableOption<
   Spec extends TableOptionSpec = TableOptionSpec
@@ -341,7 +392,10 @@ const applyDeclaredOptions = <
     return table
   }
   return declaredOptions.reduce<TableDefinition<any, any, any, "schema", any>>(
-    (current, option) => option(current),
+    (current, option) =>
+      (option as unknown as (
+        table: TableDefinition<any, any, any, "schema", any>
+      ) => TableDefinition<any, any, any, "schema", any>)(current),
     table
   ) as unknown as Table
 }
@@ -491,17 +545,17 @@ export function make<
  */
 export const schema = <SchemaName extends string>(
   schemaName: SchemaName
-): TableSchemaNamespace<SchemaName> => ({
-  schemaName,
-  table: <
+): TableSchemaNamespace<SchemaName> => {
+  const table = <
     Name extends string,
     Fields extends TableFieldMap,
+    const Options extends DeclaredTableOptions,
     PrimaryKeyColumns extends keyof Fields & string = InlinePrimaryKeyKeys<Fields>
   >(
     name: Name,
     fields: Fields,
-    ...options: DeclaredTableOptions
-  ): TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName> =>
+    ...options: Options & ValidateDeclaredOptions<TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>, Options>
+  ): ApplyDeclaredOptions<TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>, Options> =>
     applyDeclaredOptions(
       makeTable(
         name,
@@ -512,9 +566,13 @@ export const schema = <SchemaName extends string>(
         schemaName,
         "explicit"
       ) as TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>,
-      options
-    )
-})
+      options as unknown as Options
+    ) as ApplyDeclaredOptions<TableDefinition<Name, Fields, PrimaryKeyColumns, "schema", SchemaName>, Options>
+  return {
+    schemaName,
+    table
+  } as unknown as TableSchemaNamespace<SchemaName>
+}
 
 /**
  * Creates an aliased source from an existing table definition.
@@ -664,7 +722,7 @@ export function Class<
 export const primaryKey = <
   const Columns extends string | readonly string[]
 >(
-  columns: Columns
+  columns: Columns & NonEmptyColumnInput<Columns>
 ): TableOption<{
   readonly kind: "primaryKey"
   readonly columns: NormalizeColumns<Columns>
@@ -677,7 +735,7 @@ export const primaryKey = <
 export const unique = <
   const Columns extends string | readonly string[]
 >(
-  columns: Columns
+  columns: Columns & NonEmptyColumnInput<Columns>
 ): TableOption<{
   readonly kind: "unique"
   readonly columns: NormalizeColumns<Columns>
@@ -690,7 +748,7 @@ export const unique = <
 export const index = <
   const Columns extends string | readonly string[]
 >(
-  columns: Columns
+  columns: Columns & NonEmptyColumnInput<Columns>
 ): TableOption<{
   readonly kind: "index"
   readonly columns: NormalizeColumns<Columns>
@@ -705,9 +763,9 @@ export const foreignKey = <
   TargetTable extends AnyTable,
   const TargetColumns extends string | readonly string[]
 >(
-  columns: LocalColumns,
+  columns: LocalColumns & NonEmptyColumnInput<LocalColumns>,
   target: () => TargetTable,
-  referencedColumns: TargetColumns
+  referencedColumns: TargetColumns & NonEmptyColumnInput<TargetColumns> & MatchingColumnArityInput<LocalColumns, TargetColumns>
 ): TableOption<{
   readonly kind: "foreignKey"
   readonly columns: NormalizeColumns<LocalColumns>
@@ -715,7 +773,7 @@ export const foreignKey = <
     readonly tableName: string
     readonly schemaName?: string
     readonly columns: NormalizeColumns<TargetColumns>
-    readonly knownColumns: readonly string[]
+    readonly knownColumns: readonly ColumnNamesOfAnyTable<TargetTable>[]
   }
 }> => makeOption({
   kind: "foreignKey",
@@ -724,7 +782,7 @@ export const foreignKey = <
     tableName: target()[TypeId].baseName,
     schemaName: target()[TypeId].schemaName,
     columns: normalizeColumnList(referencedColumns) as NormalizeColumns<TargetColumns>,
-    knownColumns: Object.keys(target()[TypeId].fields)
+    knownColumns: Object.keys(target()[TypeId].fields) as unknown as readonly ColumnNamesOfAnyTable<TargetTable>[]
   })
 })
 

package/src/mysql/column.ts

@@ -1,7 +1,7 @@
 import * as Schema from "effect/Schema"
 
 import * as BaseColumn from "../internal/column.js"
-import { makeColumnDefinition, type ColumnDefinition } from "../internal/column-state.js"
+import { makeColumnDefinition, type AnyColumnDefinition, type ColumnDefinition } from "../internal/column-state.js"
 import type * as Expression from "../internal/scalar.js"
 import {
   DecimalStringSchema,
@@ -97,7 +97,23 @@ export const json = <SchemaType extends Schema.Schema.Any>(schema: SchemaType) =
 export const nullable = BaseColumn.nullable
 export const brand = BaseColumn.brand
 export const primaryKey = BaseColumn.primaryKey
-export const unique = BaseColumn.unique
+type UniqueColumn<Column extends AnyColumnDefinition> = ReturnType<typeof BaseColumn.unique<Column>>
+
+type MysqlUniqueOptions = {
+  readonly name?: string
+  readonly nullsNotDistinct?: never
+  readonly deferrable?: never
+  readonly initiallyDeferred?: never
+}
+
+type UniqueModifier = {
+  <Column extends AnyColumnDefinition>(column: Column): UniqueColumn<Column>
+  readonly options: <const Options extends MysqlUniqueOptions>(
+    options: Options
+  ) => <Column extends AnyColumnDefinition>(column: Column) => UniqueColumn<Column>
+}
+
+export const unique = BaseColumn.unique as UniqueModifier
 const default_ = BaseColumn.default_
 export const generated = BaseColumn.generated
 export const driverValueMapping = BaseColumn.driverValueMapping

package/src/mysql/datatypes/index.ts

@@ -50,12 +50,33 @@ type MysqlUuidWitness = Expression.DbType.Base<"mysql", "uuid"> & {
   }
 }
 
+type MysqlJsonWitness = Expression.DbType.Base<"mysql", "json"> & {
+  readonly family: "json"
+  readonly runtime: "json"
+  readonly compareGroup: "json"
+  readonly castTargets: readonly ["json", "text"]
+  readonly driverValueMapping: {
+    readonly toDriver: (value: unknown) => unknown
+  }
+}
+
+mysqlDatatypeModule.json = () => ({
+  ...withMetadata("json"),
+  driverValueMapping: {
+    toDriver: (value: unknown) =>
+      value !== null && typeof value === "object"
+        ? JSON.stringify(value)
+        : value
+  }
+}) as MysqlJsonWitness
+
 export const mysqlDatatypes = mysqlDatatypeModule as DatatypeModule<
   "mysql",
   typeof mysqlDatatypeKinds,
   typeof mysqlDatatypeFamilies
 > & {
   readonly uuid: () => MysqlUuidWitness
+  readonly json: () => MysqlJsonWitness
 }
 
 export type MysqlDatatypeModule = typeof mysqlDatatypes

package/src/mysql/errors/catalog.ts

@@ -51636,17 +51636,17 @@ export const mysqlErrorCatalogByNumber = {
   "MY-015153": [mysqlErrorCatalogBySymbol["ER_WARN_AUDIT_LOG_FILTER_RECOVERY_LOGGING_DISABLED_LOG"]!],
 } as const
 
-const mysqlSymbolPattern = /^(?:[A-Z][A-Z0-9_]*|MY-\d+)$/
-const mysqlNumberPattern = /^(?:\d+|MY-\d+)$/
-
 export type MysqlErrorSymbol = keyof typeof mysqlErrorCatalogBySymbol
 export type MysqlErrorDescriptor<Symbol extends MysqlErrorSymbol = MysqlErrorSymbol> = (typeof mysqlErrorCatalogBySymbol)[Symbol]
 export type MysqlErrorNumber = keyof typeof mysqlErrorCatalogByNumber
 export type MysqlErrorTag<Symbol extends MysqlErrorSymbol = MysqlErrorSymbol> = MysqlErrorDescriptor<Symbol>["tag"]
 export type MysqlErrorDescriptorsByNumber<Number extends MysqlErrorNumber = MysqlErrorNumber> = (typeof mysqlErrorCatalogByNumber)[Number]
 
-export const isMysqlErrorSymbol = (value: string): value is MysqlErrorSymbol => mysqlSymbolPattern.test(value)
-export const isMysqlErrorNumber = (value: string): value is MysqlErrorNumber => mysqlNumberPattern.test(value)
+export const isMysqlErrorSymbol = (value: string): value is MysqlErrorSymbol =>
+  value in mysqlErrorCatalogBySymbol
+
+export const isMysqlErrorNumber = (value: string): value is MysqlErrorNumber =>
+  value in mysqlErrorCatalogByNumber
 
 export const getMysqlErrorDescriptor = <Symbol extends MysqlErrorSymbol>(symbol: Symbol): MysqlErrorDescriptor<Symbol> =>
   mysqlErrorCatalogBySymbol[symbol]

package/src/mysql/errors/normalize.ts

@@ -48,8 +48,8 @@ const asNumber = (value: unknown): number | undefined => {
   if (typeof value === "number" && Number.isFinite(value)) {
     return value
   }
-  if (typeof value === "string" && value.trim() !== "") {
-    const parsed = Number(value)
+  if (typeof value === "string" && /^[+-]?\d+$/.test(value.trim())) {
+    const parsed = Number(value.trim())
     return Number.isFinite(parsed) ? parsed : undefined
   }
   return undefined