drizzle-arktype 0.1.2 → 0.1.3-08e4e66

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "drizzle-arktype",
3
- "version": "0.1.2",
3
+ "version": "0.1.3-08e4e66",
4
4
  "description": "Generate arktype schemas from Drizzle ORM schemas",
5
5
  "type": "module",
6
6
  "scripts": {
@@ -9,7 +9,8 @@
9
9
  "test:types": "cd tests && tsc",
10
10
  "pack": "(cd dist && npm pack --pack-destination ..) && rm -f package.tgz && mv *.tgz package.tgz",
11
11
  "publish": "npm publish package.tgz",
12
- "test": "vitest run"
12
+ "test": "vitest run",
13
+ "bench:types": "tsx ./benchmarks/types.ts"
13
14
  },
14
15
  "exports": {
15
16
  ".": {
@@ -56,19 +57,21 @@
56
57
  "license": "Apache-2.0",
57
58
  "peerDependencies": {
58
59
  "arktype": ">=2.0.0",
59
- "drizzle-orm": ">=0.36.0"
60
+ "drizzle-orm": ">=1.0.0-beta.1"
60
61
  },
61
62
  "devDependencies": {
63
+ "@ark/attest": "^0.45.8",
62
64
  "@rollup/plugin-typescript": "^11.1.0",
63
65
  "@types/node": "^18.15.10",
64
66
  "arktype": "^2.1.10",
65
67
  "cpy": "^10.1.0",
66
68
  "drizzle-orm": "link:../drizzle-orm/dist",
67
- "json-rules-engine": "7.3.0",
69
+ "json-rules-engine": "7.3.1",
68
70
  "rimraf": "^5.0.0",
69
- "rollup": "^3.20.7",
71
+ "rollup": "^3.29.5",
72
+ "tsx": "^4.19.3",
70
73
  "vite-tsconfig-paths": "^4.3.2",
71
- "vitest": "^1.6.0",
74
+ "vitest": "^3.1.3",
72
75
  "zx": "^7.2.2"
73
76
  }
74
77
  }
@@ -1,19 +1,20 @@
1
1
  import type { Type } from 'arktype';
2
- import type { Table, View } from 'drizzle-orm';
2
+ import type { InferInsertModel, InferSelectModel, Table, View } from 'drizzle-orm';
3
+ import type { CockroachEnum } from 'drizzle-orm/cockroach-core';
3
4
  import type { PgEnum } from 'drizzle-orm/pg-core';
4
5
  import type { BuildRefine, BuildSchema, NoUnknownKeys } from "./schema.types.internal.cjs";
5
6
  export interface CreateSelectSchema {
6
7
  <TTable extends Table>(table: TTable): BuildSchema<'select', TTable['_']['columns'], undefined>;
7
- <TTable extends Table, TRefine extends BuildRefine<TTable['_']['columns']>>(table: TTable, refine?: NoUnknownKeys<TRefine, TTable['$inferSelect']>): BuildSchema<'select', TTable['_']['columns'], TRefine>;
8
+ <TTable extends Table, TRefine extends BuildRefine<TTable['_']['columns']>>(table: TTable, refine?: NoUnknownKeys<TRefine, InferSelectModel<TTable>>): BuildSchema<'select', TTable['_']['columns'], TRefine>;
8
9
  <TView extends View>(view: TView): BuildSchema<'select', TView['_']['selectedFields'], undefined>;
9
10
  <TView extends View, TRefine extends BuildRefine<TView['_']['selectedFields']>>(view: TView, refine: NoUnknownKeys<TRefine, TView['$inferSelect']>): BuildSchema<'select', TView['_']['selectedFields'], TRefine>;
10
- <TEnum extends PgEnum<any>>(enum_: TEnum): Type<TEnum['enumValues'][number], {}>;
11
+ <TEnum extends PgEnum<any> | CockroachEnum<any>>(enum_: TEnum): Type<TEnum['enumValues'][number]>;
11
12
  }
12
13
  export interface CreateInsertSchema {
13
14
  <TTable extends Table>(table: TTable): BuildSchema<'insert', TTable['_']['columns'], undefined>;
14
- <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof TTable['$inferInsert']>>>(table: TTable, refine?: NoUnknownKeys<TRefine, TTable['$inferInsert']>): BuildSchema<'insert', TTable['_']['columns'], TRefine>;
15
+ <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof InferInsertModel<TTable>>>>(table: TTable, refine?: NoUnknownKeys<TRefine, InferInsertModel<TTable>>): BuildSchema<'insert', TTable['_']['columns'], TRefine>;
15
16
  }
16
17
  export interface CreateUpdateSchema {
17
18
  <TTable extends Table>(table: TTable): BuildSchema<'update', TTable['_']['columns'], undefined>;
18
- <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof TTable['$inferInsert']>>>(table: TTable, refine?: TRefine): BuildSchema<'update', TTable['_']['columns'], TRefine>;
19
+ <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof InferInsertModel<TTable>>>>(table: TTable, refine?: TRefine): BuildSchema<'update', TTable['_']['columns'], TRefine>;
19
20
  }
@@ -1,19 +1,20 @@
1
1
  import type { Type } from 'arktype';
2
- import type { Table, View } from 'drizzle-orm';
2
+ import type { InferInsertModel, InferSelectModel, Table, View } from 'drizzle-orm';
3
+ import type { CockroachEnum } from 'drizzle-orm/cockroach-core';
3
4
  import type { PgEnum } from 'drizzle-orm/pg-core';
4
5
  import type { BuildRefine, BuildSchema, NoUnknownKeys } from "./schema.types.internal.mjs";
5
6
  export interface CreateSelectSchema {
6
7
  <TTable extends Table>(table: TTable): BuildSchema<'select', TTable['_']['columns'], undefined>;
7
- <TTable extends Table, TRefine extends BuildRefine<TTable['_']['columns']>>(table: TTable, refine?: NoUnknownKeys<TRefine, TTable['$inferSelect']>): BuildSchema<'select', TTable['_']['columns'], TRefine>;
8
+ <TTable extends Table, TRefine extends BuildRefine<TTable['_']['columns']>>(table: TTable, refine?: NoUnknownKeys<TRefine, InferSelectModel<TTable>>): BuildSchema<'select', TTable['_']['columns'], TRefine>;
8
9
  <TView extends View>(view: TView): BuildSchema<'select', TView['_']['selectedFields'], undefined>;
9
10
  <TView extends View, TRefine extends BuildRefine<TView['_']['selectedFields']>>(view: TView, refine: NoUnknownKeys<TRefine, TView['$inferSelect']>): BuildSchema<'select', TView['_']['selectedFields'], TRefine>;
10
- <TEnum extends PgEnum<any>>(enum_: TEnum): Type<TEnum['enumValues'][number], {}>;
11
+ <TEnum extends PgEnum<any> | CockroachEnum<any>>(enum_: TEnum): Type<TEnum['enumValues'][number]>;
11
12
  }
12
13
  export interface CreateInsertSchema {
13
14
  <TTable extends Table>(table: TTable): BuildSchema<'insert', TTable['_']['columns'], undefined>;
14
- <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof TTable['$inferInsert']>>>(table: TTable, refine?: NoUnknownKeys<TRefine, TTable['$inferInsert']>): BuildSchema<'insert', TTable['_']['columns'], TRefine>;
15
+ <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof InferInsertModel<TTable>>>>(table: TTable, refine?: NoUnknownKeys<TRefine, InferInsertModel<TTable>>): BuildSchema<'insert', TTable['_']['columns'], TRefine>;
15
16
  }
16
17
  export interface CreateUpdateSchema {
17
18
  <TTable extends Table>(table: TTable): BuildSchema<'update', TTable['_']['columns'], undefined>;
18
- <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof TTable['$inferInsert']>>>(table: TTable, refine?: TRefine): BuildSchema<'update', TTable['_']['columns'], TRefine>;
19
+ <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof InferInsertModel<TTable>>>>(table: TTable, refine?: TRefine): BuildSchema<'update', TTable['_']['columns'], TRefine>;
19
20
  }
package/schema.types.d.ts CHANGED
@@ -1,19 +1,20 @@
1
1
  import type { Type } from 'arktype';
2
- import type { Table, View } from 'drizzle-orm';
2
+ import type { InferInsertModel, InferSelectModel, Table, View } from 'drizzle-orm';
3
+ import type { CockroachEnum } from 'drizzle-orm/cockroach-core';
3
4
  import type { PgEnum } from 'drizzle-orm/pg-core';
4
5
  import type { BuildRefine, BuildSchema, NoUnknownKeys } from './schema.types.internal.js';
5
6
  export interface CreateSelectSchema {
6
7
  <TTable extends Table>(table: TTable): BuildSchema<'select', TTable['_']['columns'], undefined>;
7
- <TTable extends Table, TRefine extends BuildRefine<TTable['_']['columns']>>(table: TTable, refine?: NoUnknownKeys<TRefine, TTable['$inferSelect']>): BuildSchema<'select', TTable['_']['columns'], TRefine>;
8
+ <TTable extends Table, TRefine extends BuildRefine<TTable['_']['columns']>>(table: TTable, refine?: NoUnknownKeys<TRefine, InferSelectModel<TTable>>): BuildSchema<'select', TTable['_']['columns'], TRefine>;
8
9
  <TView extends View>(view: TView): BuildSchema<'select', TView['_']['selectedFields'], undefined>;
9
10
  <TView extends View, TRefine extends BuildRefine<TView['_']['selectedFields']>>(view: TView, refine: NoUnknownKeys<TRefine, TView['$inferSelect']>): BuildSchema<'select', TView['_']['selectedFields'], TRefine>;
10
- <TEnum extends PgEnum<any>>(enum_: TEnum): Type<TEnum['enumValues'][number], {}>;
11
+ <TEnum extends PgEnum<any> | CockroachEnum<any>>(enum_: TEnum): Type<TEnum['enumValues'][number]>;
11
12
  }
12
13
  export interface CreateInsertSchema {
13
14
  <TTable extends Table>(table: TTable): BuildSchema<'insert', TTable['_']['columns'], undefined>;
14
- <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof TTable['$inferInsert']>>>(table: TTable, refine?: NoUnknownKeys<TRefine, TTable['$inferInsert']>): BuildSchema<'insert', TTable['_']['columns'], TRefine>;
15
+ <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof InferInsertModel<TTable>>>>(table: TTable, refine?: NoUnknownKeys<TRefine, InferInsertModel<TTable>>): BuildSchema<'insert', TTable['_']['columns'], TRefine>;
15
16
  }
16
17
  export interface CreateUpdateSchema {
17
18
  <TTable extends Table>(table: TTable): BuildSchema<'update', TTable['_']['columns'], undefined>;
18
- <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof TTable['$inferInsert']>>>(table: TTable, refine?: TRefine): BuildSchema<'update', TTable['_']['columns'], TRefine>;
19
+ <TTable extends Table, TRefine extends BuildRefine<Pick<TTable['_']['columns'], keyof InferInsertModel<TTable>>>>(table: TTable, refine?: TRefine): BuildSchema<'update', TTable['_']['columns'], TRefine>;
19
20
  }
@@ -1,24 +1,22 @@
1
1
  import type { Type, type } from 'arktype';
2
- import type { Assume, Column, DrizzleTypeError, SelectedFieldsFlat, Simplify, Table, View } from 'drizzle-orm';
3
- import type { ArktypeNullable, ArktypeOptional, GetArktypeType, GetEnumValuesFromColumn, HandleColumn } from "./column.types.cjs";
4
- import type { GetSelection, RemoveNever } from "./utils.cjs";
2
+ import type { Column, DrizzleTypeError, SelectedFieldsFlat, Simplify, Table, View } from 'drizzle-orm';
3
+ import type { ArktypeNullable, ArktypeOptional, GetArktypeType, HandleColumn } from "./column.types.cjs";
4
+ import type { ColumnIsGeneratedAlwaysAs, GetSelection } from "./utils.cjs";
5
5
  export interface Conditions {
6
6
  never: (column?: Column) => boolean;
7
7
  optional: (column: Column) => boolean;
8
8
  nullable: (column: Column) => boolean;
9
9
  }
10
10
  type GenericSchema = type.cast<unknown> | [type.cast<unknown>, '?'];
11
- export type BuildRefineColumns<TColumns extends Record<string, any>> = Simplify<RemoveNever<{
12
- [K in keyof TColumns]: TColumns[K] extends infer TColumn extends Column ? GetArktypeType<TColumn['_']['data'], TColumn['_']['columnType'], GetEnumValuesFromColumn<TColumn>> extends infer TSchema extends GenericSchema ? TSchema : Type<any, {}> : TColumns[K] extends infer TObject extends SelectedFieldsFlat<Column> | Table | View ? BuildRefineColumns<GetSelection<TObject>> : TColumns[K];
11
+ type BuildRefineField<T> = T extends GenericSchema ? ((schema: T) => GenericSchema) | GenericSchema : never;
12
+ export type BuildRefine<TColumns extends Record<string, any>> = {
13
+ [K in keyof TColumns as TColumns[K] extends Column | SelectedFieldsFlat<Column> | Table | View ? K : never]?: TColumns[K] extends Column ? BuildRefineField<GetArktypeType<TColumns[K]>> : BuildRefine<GetSelection<TColumns[K]>>;
14
+ };
15
+ type HandleRefinement<TType extends 'select' | 'insert' | 'update', TRefinement, TColumn extends Column> = TRefinement extends (schema: any) => GenericSchema ? (TColumn['_']['notNull'] extends true ? ReturnType<TRefinement> : ArktypeNullable<ReturnType<TRefinement>>) extends infer TSchema ? TType extends 'update' ? ArktypeOptional<TSchema> : TSchema : Type<any> : TRefinement;
16
+ type IsRefinementDefined<TRefinements extends Record<string | symbol | number, any> | undefined, TKey extends string | symbol | number> = TRefinements extends object ? TRefinements[TKey] extends GenericSchema | ((schema: any) => any) ? true : false : false;
17
+ export type BuildSchema<TType extends 'select' | 'insert' | 'update', TColumns extends Record<string, any>, TRefinements extends Record<string, any> | undefined> = type.instantiate<Simplify<{
18
+ readonly [K in keyof TColumns as ColumnIsGeneratedAlwaysAs<TColumns[K]> extends true ? TType extends 'select' ? K : never : K]: TColumns[K] extends infer TColumn extends Column ? IsRefinementDefined<TRefinements, K> extends true ? HandleRefinement<TType, TRefinements[K & keyof TRefinements], TColumn> : HandleColumn<TType, TColumn> : TColumns[K] extends infer TNested extends SelectedFieldsFlat<Column> | Table | View ? BuildSchema<TType, GetSelection<TNested>, TRefinements extends object ? TRefinements[K & keyof TRefinements] : undefined> : any;
13
19
  }>>;
14
- export type BuildRefine<TColumns extends Record<string, any>> = BuildRefineColumns<TColumns> extends infer TBuildColumns ? {
15
- [K in keyof TBuildColumns]?: TBuildColumns[K] extends GenericSchema ? ((schema: TBuildColumns[K]) => GenericSchema) | GenericSchema : TBuildColumns[K] extends Record<string, any> ? Simplify<BuildRefine<TBuildColumns[K]>> : never;
16
- } : never;
17
- type HandleRefinement<TType extends 'select' | 'insert' | 'update', TRefinement extends GenericSchema | ((schema: GenericSchema) => GenericSchema), TColumn extends Column> = TRefinement extends (schema: any) => GenericSchema ? (TColumn['_']['notNull'] extends true ? ReturnType<TRefinement> : ArktypeNullable<ReturnType<TRefinement>>) extends infer TSchema ? TType extends 'update' ? ArktypeOptional<TSchema> : TSchema : Type<any, {}> : TRefinement;
18
- type IsRefinementDefined<TRefinements, TKey extends string> = TKey extends keyof TRefinements ? TRefinements[TKey] extends GenericSchema | ((schema: any) => any) ? true : false : false;
19
- export type BuildSchema<TType extends 'select' | 'insert' | 'update', TColumns extends Record<string, any>, TRefinements extends Record<string, any> | undefined> = type.instantiate<Simplify<RemoveNever<{
20
- readonly [K in keyof TColumns]: TColumns[K] extends infer TColumn extends Column ? TRefinements extends object ? IsRefinementDefined<TRefinements, Assume<K, string>> extends true ? HandleRefinement<TType, TRefinements[Assume<K, keyof TRefinements>], TColumn> : HandleColumn<TType, TColumn> : HandleColumn<TType, TColumn> : TColumns[K] extends infer TObject extends SelectedFieldsFlat<Column> | Table | View ? BuildSchema<TType, GetSelection<TObject>, TRefinements extends object ? TRefinements[Assume<K, keyof TRefinements>] extends infer TNestedRefinements extends object ? TNestedRefinements : undefined : undefined> : any;
21
- }>>>;
22
20
  export type NoUnknownKeys<TRefinement extends Record<string, any>, TCompare extends Record<string, any>> = {
23
21
  [K in keyof TRefinement]: K extends keyof TCompare ? TRefinement[K] extends Record<string, GenericSchema> ? NoUnknownKeys<TRefinement[K], TCompare[K]> : TRefinement[K] : DrizzleTypeError<`Found unknown key in refinement: "${K & string}"`>;
24
22
  };
@@ -1,24 +1,22 @@
1
1
  import type { Type, type } from 'arktype';
2
- import type { Assume, Column, DrizzleTypeError, SelectedFieldsFlat, Simplify, Table, View } from 'drizzle-orm';
3
- import type { ArktypeNullable, ArktypeOptional, GetArktypeType, GetEnumValuesFromColumn, HandleColumn } from "./column.types.mjs";
4
- import type { GetSelection, RemoveNever } from "./utils.mjs";
2
+ import type { Column, DrizzleTypeError, SelectedFieldsFlat, Simplify, Table, View } from 'drizzle-orm';
3
+ import type { ArktypeNullable, ArktypeOptional, GetArktypeType, HandleColumn } from "./column.types.mjs";
4
+ import type { ColumnIsGeneratedAlwaysAs, GetSelection } from "./utils.mjs";
5
5
  export interface Conditions {
6
6
  never: (column?: Column) => boolean;
7
7
  optional: (column: Column) => boolean;
8
8
  nullable: (column: Column) => boolean;
9
9
  }
10
10
  type GenericSchema = type.cast<unknown> | [type.cast<unknown>, '?'];
11
- export type BuildRefineColumns<TColumns extends Record<string, any>> = Simplify<RemoveNever<{
12
- [K in keyof TColumns]: TColumns[K] extends infer TColumn extends Column ? GetArktypeType<TColumn['_']['data'], TColumn['_']['columnType'], GetEnumValuesFromColumn<TColumn>> extends infer TSchema extends GenericSchema ? TSchema : Type<any, {}> : TColumns[K] extends infer TObject extends SelectedFieldsFlat<Column> | Table | View ? BuildRefineColumns<GetSelection<TObject>> : TColumns[K];
11
+ type BuildRefineField<T> = T extends GenericSchema ? ((schema: T) => GenericSchema) | GenericSchema : never;
12
+ export type BuildRefine<TColumns extends Record<string, any>> = {
13
+ [K in keyof TColumns as TColumns[K] extends Column | SelectedFieldsFlat<Column> | Table | View ? K : never]?: TColumns[K] extends Column ? BuildRefineField<GetArktypeType<TColumns[K]>> : BuildRefine<GetSelection<TColumns[K]>>;
14
+ };
15
+ type HandleRefinement<TType extends 'select' | 'insert' | 'update', TRefinement, TColumn extends Column> = TRefinement extends (schema: any) => GenericSchema ? (TColumn['_']['notNull'] extends true ? ReturnType<TRefinement> : ArktypeNullable<ReturnType<TRefinement>>) extends infer TSchema ? TType extends 'update' ? ArktypeOptional<TSchema> : TSchema : Type<any> : TRefinement;
16
+ type IsRefinementDefined<TRefinements extends Record<string | symbol | number, any> | undefined, TKey extends string | symbol | number> = TRefinements extends object ? TRefinements[TKey] extends GenericSchema | ((schema: any) => any) ? true : false : false;
17
+ export type BuildSchema<TType extends 'select' | 'insert' | 'update', TColumns extends Record<string, any>, TRefinements extends Record<string, any> | undefined> = type.instantiate<Simplify<{
18
+ readonly [K in keyof TColumns as ColumnIsGeneratedAlwaysAs<TColumns[K]> extends true ? TType extends 'select' ? K : never : K]: TColumns[K] extends infer TColumn extends Column ? IsRefinementDefined<TRefinements, K> extends true ? HandleRefinement<TType, TRefinements[K & keyof TRefinements], TColumn> : HandleColumn<TType, TColumn> : TColumns[K] extends infer TNested extends SelectedFieldsFlat<Column> | Table | View ? BuildSchema<TType, GetSelection<TNested>, TRefinements extends object ? TRefinements[K & keyof TRefinements] : undefined> : any;
13
19
  }>>;
14
- export type BuildRefine<TColumns extends Record<string, any>> = BuildRefineColumns<TColumns> extends infer TBuildColumns ? {
15
- [K in keyof TBuildColumns]?: TBuildColumns[K] extends GenericSchema ? ((schema: TBuildColumns[K]) => GenericSchema) | GenericSchema : TBuildColumns[K] extends Record<string, any> ? Simplify<BuildRefine<TBuildColumns[K]>> : never;
16
- } : never;
17
- type HandleRefinement<TType extends 'select' | 'insert' | 'update', TRefinement extends GenericSchema | ((schema: GenericSchema) => GenericSchema), TColumn extends Column> = TRefinement extends (schema: any) => GenericSchema ? (TColumn['_']['notNull'] extends true ? ReturnType<TRefinement> : ArktypeNullable<ReturnType<TRefinement>>) extends infer TSchema ? TType extends 'update' ? ArktypeOptional<TSchema> : TSchema : Type<any, {}> : TRefinement;
18
- type IsRefinementDefined<TRefinements, TKey extends string> = TKey extends keyof TRefinements ? TRefinements[TKey] extends GenericSchema | ((schema: any) => any) ? true : false : false;
19
- export type BuildSchema<TType extends 'select' | 'insert' | 'update', TColumns extends Record<string, any>, TRefinements extends Record<string, any> | undefined> = type.instantiate<Simplify<RemoveNever<{
20
- readonly [K in keyof TColumns]: TColumns[K] extends infer TColumn extends Column ? TRefinements extends object ? IsRefinementDefined<TRefinements, Assume<K, string>> extends true ? HandleRefinement<TType, TRefinements[Assume<K, keyof TRefinements>], TColumn> : HandleColumn<TType, TColumn> : HandleColumn<TType, TColumn> : TColumns[K] extends infer TObject extends SelectedFieldsFlat<Column> | Table | View ? BuildSchema<TType, GetSelection<TObject>, TRefinements extends object ? TRefinements[Assume<K, keyof TRefinements>] extends infer TNestedRefinements extends object ? TNestedRefinements : undefined : undefined> : any;
21
- }>>>;
22
20
  export type NoUnknownKeys<TRefinement extends Record<string, any>, TCompare extends Record<string, any>> = {
23
21
  [K in keyof TRefinement]: K extends keyof TCompare ? TRefinement[K] extends Record<string, GenericSchema> ? NoUnknownKeys<TRefinement[K], TCompare[K]> : TRefinement[K] : DrizzleTypeError<`Found unknown key in refinement: "${K & string}"`>;
24
22
  };
@@ -1,24 +1,22 @@
1
1
  import type { Type, type } from 'arktype';
2
- import type { Assume, Column, DrizzleTypeError, SelectedFieldsFlat, Simplify, Table, View } from 'drizzle-orm';
3
- import type { ArktypeNullable, ArktypeOptional, GetArktypeType, GetEnumValuesFromColumn, HandleColumn } from './column.types.js';
4
- import type { GetSelection, RemoveNever } from './utils.js';
2
+ import type { Column, DrizzleTypeError, SelectedFieldsFlat, Simplify, Table, View } from 'drizzle-orm';
3
+ import type { ArktypeNullable, ArktypeOptional, GetArktypeType, HandleColumn } from './column.types.js';
4
+ import type { ColumnIsGeneratedAlwaysAs, GetSelection } from './utils.js';
5
5
  export interface Conditions {
6
6
  never: (column?: Column) => boolean;
7
7
  optional: (column: Column) => boolean;
8
8
  nullable: (column: Column) => boolean;
9
9
  }
10
10
  type GenericSchema = type.cast<unknown> | [type.cast<unknown>, '?'];
11
- export type BuildRefineColumns<TColumns extends Record<string, any>> = Simplify<RemoveNever<{
12
- [K in keyof TColumns]: TColumns[K] extends infer TColumn extends Column ? GetArktypeType<TColumn['_']['data'], TColumn['_']['columnType'], GetEnumValuesFromColumn<TColumn>> extends infer TSchema extends GenericSchema ? TSchema : Type<any, {}> : TColumns[K] extends infer TObject extends SelectedFieldsFlat<Column> | Table | View ? BuildRefineColumns<GetSelection<TObject>> : TColumns[K];
11
+ type BuildRefineField<T> = T extends GenericSchema ? ((schema: T) => GenericSchema) | GenericSchema : never;
12
+ export type BuildRefine<TColumns extends Record<string, any>> = {
13
+ [K in keyof TColumns as TColumns[K] extends Column | SelectedFieldsFlat<Column> | Table | View ? K : never]?: TColumns[K] extends Column ? BuildRefineField<GetArktypeType<TColumns[K]>> : BuildRefine<GetSelection<TColumns[K]>>;
14
+ };
15
+ type HandleRefinement<TType extends 'select' | 'insert' | 'update', TRefinement, TColumn extends Column> = TRefinement extends (schema: any) => GenericSchema ? (TColumn['_']['notNull'] extends true ? ReturnType<TRefinement> : ArktypeNullable<ReturnType<TRefinement>>) extends infer TSchema ? TType extends 'update' ? ArktypeOptional<TSchema> : TSchema : Type<any> : TRefinement;
16
+ type IsRefinementDefined<TRefinements extends Record<string | symbol | number, any> | undefined, TKey extends string | symbol | number> = TRefinements extends object ? TRefinements[TKey] extends GenericSchema | ((schema: any) => any) ? true : false : false;
17
+ export type BuildSchema<TType extends 'select' | 'insert' | 'update', TColumns extends Record<string, any>, TRefinements extends Record<string, any> | undefined> = type.instantiate<Simplify<{
18
+ readonly [K in keyof TColumns as ColumnIsGeneratedAlwaysAs<TColumns[K]> extends true ? TType extends 'select' ? K : never : K]: TColumns[K] extends infer TColumn extends Column ? IsRefinementDefined<TRefinements, K> extends true ? HandleRefinement<TType, TRefinements[K & keyof TRefinements], TColumn> : HandleColumn<TType, TColumn> : TColumns[K] extends infer TNested extends SelectedFieldsFlat<Column> | Table | View ? BuildSchema<TType, GetSelection<TNested>, TRefinements extends object ? TRefinements[K & keyof TRefinements] : undefined> : any;
13
19
  }>>;
14
- export type BuildRefine<TColumns extends Record<string, any>> = BuildRefineColumns<TColumns> extends infer TBuildColumns ? {
15
- [K in keyof TBuildColumns]?: TBuildColumns[K] extends GenericSchema ? ((schema: TBuildColumns[K]) => GenericSchema) | GenericSchema : TBuildColumns[K] extends Record<string, any> ? Simplify<BuildRefine<TBuildColumns[K]>> : never;
16
- } : never;
17
- type HandleRefinement<TType extends 'select' | 'insert' | 'update', TRefinement extends GenericSchema | ((schema: GenericSchema) => GenericSchema), TColumn extends Column> = TRefinement extends (schema: any) => GenericSchema ? (TColumn['_']['notNull'] extends true ? ReturnType<TRefinement> : ArktypeNullable<ReturnType<TRefinement>>) extends infer TSchema ? TType extends 'update' ? ArktypeOptional<TSchema> : TSchema : Type<any, {}> : TRefinement;
18
- type IsRefinementDefined<TRefinements, TKey extends string> = TKey extends keyof TRefinements ? TRefinements[TKey] extends GenericSchema | ((schema: any) => any) ? true : false : false;
19
- export type BuildSchema<TType extends 'select' | 'insert' | 'update', TColumns extends Record<string, any>, TRefinements extends Record<string, any> | undefined> = type.instantiate<Simplify<RemoveNever<{
20
- readonly [K in keyof TColumns]: TColumns[K] extends infer TColumn extends Column ? TRefinements extends object ? IsRefinementDefined<TRefinements, Assume<K, string>> extends true ? HandleRefinement<TType, TRefinements[Assume<K, keyof TRefinements>], TColumn> : HandleColumn<TType, TColumn> : HandleColumn<TType, TColumn> : TColumns[K] extends infer TObject extends SelectedFieldsFlat<Column> | Table | View ? BuildSchema<TType, GetSelection<TObject>, TRefinements extends object ? TRefinements[Assume<K, keyof TRefinements>] extends infer TNestedRefinements extends object ? TNestedRefinements : undefined : undefined> : any;
21
- }>>>;
22
20
  export type NoUnknownKeys<TRefinement extends Record<string, any>, TCompare extends Record<string, any>> = {
23
21
  [K in keyof TRefinement]: K extends keyof TCompare ? TRefinement[K] extends Record<string, GenericSchema> ? NoUnknownKeys<TRefinement[K], TCompare[K]> : TRefinement[K] : DrizzleTypeError<`Found unknown key in refinement: "${K & string}"`>;
24
22
  };
package/utils.d.cts CHANGED
@@ -9,14 +9,12 @@ export declare function isWithEnum(column: Column): column is typeof column & {
9
9
  export declare const isPgEnum: (entity: any) => entity is PgEnum<[string, ...string[]]>;
10
10
  type Literal = type.infer<typeof literalSchema>;
11
11
  export type Json = Literal | Record<string, any> | any[];
12
- export type IsEnumDefined<TEnum extends [string, ...string[]] | undefined> = [string, ...string[]] extends TEnum ? false : undefined extends TEnum ? false : true;
13
- export type IsNever<T> = [T] extends [never] ? true : false;
14
- export type ColumnIsGeneratedAlwaysAs<TColumn extends Column> = TColumn['_']['identity'] extends 'always' ? true : TColumn['_']['generated'] extends undefined ? false : TColumn['_']['generated'] extends infer TGenerated extends {
15
- type: string;
16
- } ? TGenerated['type'] extends 'byDefault' ? false : true : true;
17
- export type RemoveNever<T> = {
18
- [K in keyof T as T[K] extends never ? never : K]: T[K];
19
- };
12
+ export type ColumnIsGeneratedAlwaysAs<TColumn> = TColumn extends Column ? TColumn['_']['identity'] extends 'always' ? true : TColumn['_'] extends {
13
+ generated: undefined;
14
+ } ? false : TColumn['_'] extends {
15
+ generated: {
16
+ type: 'byDefault';
17
+ };
18
+ } ? false : true : false;
20
19
  export type GetSelection<T extends SelectedFieldsFlat<Column> | Table | View> = T extends Table ? T['_']['columns'] : T extends View ? T['_']['selectedFields'] : T;
21
- export type IsUnknown<T> = unknown extends T ? [T] extends [null] ? false : true : false;
22
20
  export {};
package/utils.d.mts CHANGED
@@ -9,14 +9,12 @@ export declare function isWithEnum(column: Column): column is typeof column & {
9
9
  export declare const isPgEnum: (entity: any) => entity is PgEnum<[string, ...string[]]>;
10
10
  type Literal = type.infer<typeof literalSchema>;
11
11
  export type Json = Literal | Record<string, any> | any[];
12
- export type IsEnumDefined<TEnum extends [string, ...string[]] | undefined> = [string, ...string[]] extends TEnum ? false : undefined extends TEnum ? false : true;
13
- export type IsNever<T> = [T] extends [never] ? true : false;
14
- export type ColumnIsGeneratedAlwaysAs<TColumn extends Column> = TColumn['_']['identity'] extends 'always' ? true : TColumn['_']['generated'] extends undefined ? false : TColumn['_']['generated'] extends infer TGenerated extends {
15
- type: string;
16
- } ? TGenerated['type'] extends 'byDefault' ? false : true : true;
17
- export type RemoveNever<T> = {
18
- [K in keyof T as T[K] extends never ? never : K]: T[K];
19
- };
12
+ export type ColumnIsGeneratedAlwaysAs<TColumn> = TColumn extends Column ? TColumn['_']['identity'] extends 'always' ? true : TColumn['_'] extends {
13
+ generated: undefined;
14
+ } ? false : TColumn['_'] extends {
15
+ generated: {
16
+ type: 'byDefault';
17
+ };
18
+ } ? false : true : false;
20
19
  export type GetSelection<T extends SelectedFieldsFlat<Column> | Table | View> = T extends Table ? T['_']['columns'] : T extends View ? T['_']['selectedFields'] : T;
21
- export type IsUnknown<T> = unknown extends T ? [T] extends [null] ? false : true : false;
22
20
  export {};
package/utils.d.ts CHANGED
@@ -9,14 +9,12 @@ export declare function isWithEnum(column: Column): column is typeof column & {
9
9
  export declare const isPgEnum: (entity: any) => entity is PgEnum<[string, ...string[]]>;
10
10
  type Literal = type.infer<typeof literalSchema>;
11
11
  export type Json = Literal | Record<string, any> | any[];
12
- export type IsEnumDefined<TEnum extends [string, ...string[]] | undefined> = [string, ...string[]] extends TEnum ? false : undefined extends TEnum ? false : true;
13
- export type IsNever<T> = [T] extends [never] ? true : false;
14
- export type ColumnIsGeneratedAlwaysAs<TColumn extends Column> = TColumn['_']['identity'] extends 'always' ? true : TColumn['_']['generated'] extends undefined ? false : TColumn['_']['generated'] extends infer TGenerated extends {
15
- type: string;
16
- } ? TGenerated['type'] extends 'byDefault' ? false : true : true;
17
- export type RemoveNever<T> = {
18
- [K in keyof T as T[K] extends never ? never : K]: T[K];
19
- };
12
+ export type ColumnIsGeneratedAlwaysAs<TColumn> = TColumn extends Column ? TColumn['_']['identity'] extends 'always' ? true : TColumn['_'] extends {
13
+ generated: undefined;
14
+ } ? false : TColumn['_'] extends {
15
+ generated: {
16
+ type: 'byDefault';
17
+ };
18
+ } ? false : true : false;
20
19
  export type GetSelection<T extends SelectedFieldsFlat<Column> | Table | View> = T extends Table ? T['_']['columns'] : T extends View ? T['_']['selectedFields'] : T;
21
- export type IsUnknown<T> = unknown extends T ? [T] extends [null] ? false : true : false;
22
20
  export {};