@an5/orm 1.0.8 → 1.1.0

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.
Files changed (73) hide show
  1. package/README.md +174 -175
  2. package/dist/an5Metadata.js +64 -0
  3. package/dist/an5Orm.d.ts +92 -3
  4. package/dist/an5Orm.js +697 -206
  5. package/dist/config.d.ts +9 -1
  6. package/dist/config.js +13 -18
  7. package/dist/errors.d.ts +1 -1
  8. package/dist/errors.js +1 -1
  9. package/dist/generator/src/api.d.ts +35 -0
  10. package/dist/generator/src/api.js +58 -0
  11. package/dist/generator/src/code-generator.d.ts +14 -13
  12. package/dist/generator/src/code-generator.js +249 -225
  13. package/dist/generator/src/config.d.ts +162 -0
  14. package/dist/generator/src/config.js +354 -0
  15. package/dist/generator/src/dialect.d.ts +88 -0
  16. package/dist/generator/src/dialect.js +318 -0
  17. package/dist/generator/src/dotnet-generator.d.ts +21 -11
  18. package/dist/generator/src/dotnet-generator.js +1053 -546
  19. package/dist/generator/src/field-types.d.ts +96 -0
  20. package/dist/generator/src/field-types.js +433 -0
  21. package/dist/generator/src/golang-generator.d.ts +36 -0
  22. package/dist/generator/src/golang-generator.js +1229 -0
  23. package/dist/generator/src/index.d.ts +1 -1
  24. package/dist/generator/src/index.js +137 -88
  25. package/dist/generator/src/metadata-generator.d.ts +11 -8
  26. package/dist/generator/src/metadata-generator.js +105 -90
  27. package/dist/generator/src/parser.d.ts +37 -11
  28. package/dist/generator/src/parser.js +208 -206
  29. package/dist/generator/src/push-apply.d.ts +37 -0
  30. package/dist/generator/src/push-apply.js +109 -0
  31. package/dist/generator/src/push-schema.d.ts +93 -0
  32. package/dist/generator/src/push-schema.js +189 -0
  33. package/dist/generator/src/python-generator.d.ts +37 -9
  34. package/dist/generator/src/python-generator.js +428 -105
  35. package/dist/generator/src/rust-generator.d.ts +38 -0
  36. package/dist/generator/src/rust-generator.js +971 -0
  37. package/dist/generator/src/suggest.d.ts +17 -0
  38. package/dist/generator/src/suggest.js +45 -0
  39. package/dist/generator/src/type-kinds.d.ts +27 -0
  40. package/dist/generator/src/type-kinds.js +72 -0
  41. package/dist/generator/src/types.d.ts +60 -27
  42. package/dist/generator/src/types.js +41 -2
  43. package/dist/index.d.ts +3 -2
  44. package/dist/index.js +19 -8
  45. package/dist/logger.js +1 -1
  46. package/dist/migration-core.d.ts +102 -0
  47. package/dist/migration-core.js +731 -0
  48. package/dist/orm.d.ts +147 -0
  49. package/dist/orm.js +1654 -0
  50. package/dist/provider-support.d.ts +27 -0
  51. package/dist/provider-support.js +50 -0
  52. package/dist/scripts/release-cli.d.ts +2 -2
  53. package/dist/scripts/release-cli.js +115 -115
  54. package/dist/sql-utils.d.ts +3 -1
  55. package/dist/sql-utils.js +94 -24
  56. package/generator/src/api.ts +45 -0
  57. package/generator/src/code-generator.ts +259 -230
  58. package/generator/src/config.ts +496 -0
  59. package/generator/src/dialect.ts +415 -0
  60. package/generator/src/dotnet-generator.ts +1071 -544
  61. package/generator/src/field-types.ts +473 -0
  62. package/generator/src/golang-generator.ts +1267 -0
  63. package/generator/src/index.ts +146 -116
  64. package/generator/src/metadata-generator.ts +116 -101
  65. package/generator/src/parser.ts +234 -202
  66. package/generator/src/push-apply.ts +148 -0
  67. package/generator/src/push-schema.ts +250 -0
  68. package/generator/src/python-generator.ts +480 -114
  69. package/generator/src/rust-generator.ts +974 -0
  70. package/generator/src/suggest.ts +41 -0
  71. package/generator/src/type-kinds.ts +78 -0
  72. package/generator/src/types.ts +77 -29
  73. package/package.json +32 -6
@@ -1,225 +1,249 @@
1
- "use strict";
2
- var __importDefault = (this && this.__importDefault) || function (mod) {
3
- return (mod && mod.__esModule) ? mod : { "default": mod };
4
- };
5
- Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.CodeGenerator = void 0;
7
- const fs_1 = __importDefault(require("fs"));
8
- const path_1 = __importDefault(require("path"));
9
- class CodeGenerator {
10
- constructor(outputDir) {
11
- this.outputDir = outputDir;
12
- if (!fs_1.default.existsSync(this.outputDir)) {
13
- fs_1.default.mkdirSync(this.outputDir, { recursive: true });
14
- }
15
- }
16
- generate(models) {
17
- this.generateBaseTs();
18
- for (const model of models) {
19
- this.generateModelFile(model);
20
- }
21
- this.generateIndexTs(models);
22
- }
23
- generateBaseTs() {
24
- const baseTs = `
25
- export namespace An5 {
26
- export class An5ClientKnownRequestError extends Error {
27
- code!: string;
28
- meta?: any;
29
- constructor(message: string, { code, clientVersion }: { code: string; clientVersion: string }) {
30
- super(message);
31
- this.code = code;
32
- }
33
- }
34
- export type SortOrder = 'asc' | 'desc';
35
- export type StringFilter = { equals?: string; in?: string[]; notIn?: string[]; lt?: string; lte?: string; gt?: string; gte?: string; contains?: string; startsWith?: string; endsWith?: string; not?: string | StringFilter; };
36
- export type StringNullableFilter = { equals?: string | null; in?: (string | null)[]; notIn?: (string | null)[]; lt?: string; lte?: string; gt?: string; gte?: string; contains?: string; startsWith?: string; endsWith?: string; not?: string | StringNullableFilter | null; };
37
- export type NumberFilter = { equals?: number; in?: number[]; notIn?: number[]; lt?: number; lte?: number; gt?: number; gte?: number; not?: number | NumberFilter; };
38
- export type IntFieldUpdateOperationsInput = { set?: number; increment?: number; decrement?: number; multiply?: number; divide?: number; };
39
- export type FloatFieldUpdateOperationsInput = { set?: number; increment?: number; decrement?: number; multiply?: number; divide?: number; };
40
- export type NumberNullableFilter = { equals?: number | null; in?: (number | null)[]; notIn?: (number | null)[]; lt?: number; lte?: number; gt?: number; gte?: number; not?: number | NumberNullableFilter | null; };
41
- export type BooleanFilter = { equals?: boolean; not?: boolean | BooleanFilter; };
42
- export type BooleanNullableFilter = { equals?: boolean | null; not?: boolean | BooleanNullableFilter | null; };
43
- export type DateTimeFilter = { equals?: Date; in?: Date[]; notIn?: Date[]; lt?: Date; lte?: Date; gt?: Date; gte?: Date; not?: Date | DateTimeFilter; };
44
- export type DateTimeNullableFilter = { equals?: Date | null; in?: (Date | null)[]; notIn?: (Date | null)[]; lt?: Date; lte?: Date; gt?: Date; gte?: Date; not?: Date | DateTimeNullableFilter | null; };
45
- }
46
-
47
- export interface TableClient<T, WhereInput = any, Select = any, Include = any, CreateInput = any, UpdateInput = any, FindManyArgs = any, FindFirstArgs = any, FindUniqueArgs = any, CreateArgs = any, UpdateArgs = any, UpsertArgs = any, DeleteArgs = any> {
48
- findMany(args?: FindManyArgs): Promise<T[]>;
49
- vectorSearch(args: { vector: number[]; take?: number; where?: WhereInput; include?: Include; vectorField?: string; distanceMetric?: 'cosine' | 'euclidean' | 'dot'; }): Promise<(T & { distance: number })[]>;
50
- findFirst(args?: FindFirstArgs): Promise<T | null>;
51
- findUnique(args?: FindUniqueArgs): Promise<T | null>;
52
- count(args?: { where?: WhereInput; }): Promise<number>;
53
- create(args: CreateArgs): Promise<T>;
54
- createMany(args: { data: CreateInput[]; skipDuplicates?: boolean; }): Promise<{ count: number }>;
55
- update(args: UpdateArgs): Promise<T>;
56
- updateMany(args: { where?: WhereInput; data: UpdateInput; }): Promise<{ count: number }>;
57
- delete(args: DeleteArgs): Promise<T>;
58
- deleteMany(args?: { where?: WhereInput; }): Promise<{ count: number }>;
59
- aggregate(args: any): Promise<any>;
60
- groupBy(args: any): Promise<any[]>;
61
- upsert(args: UpsertArgs): Promise<T>;
62
- }
63
- `;
64
- fs_1.default.writeFileSync(path_1.default.join(this.outputDir, 'base.ts'), baseTs);
65
- }
66
- generateModelFile(model) {
67
- let content = `import { An5, TableClient } from './base';\n`;
68
- const relModels = [...new Set(model.relations.map(r => r.type))];
69
- for (const relModel of relModels) {
70
- if (relModel !== model.name) {
71
- content += `import { ${relModel}, ${relModel}WhereInput, ${relModel}FindManyArgs, ${relModel}CreateInput, ${relModel}UpdateInput } from './${relModel}';\n`;
72
- }
73
- }
74
- content += `\nexport interface ${model.name} {\n`;
75
- for (const field of model.fields) {
76
- content += ` ${field.name}: ${this.normalizeType(field.type)}${field.isOptional ? ' | null' : ''};\n`;
77
- }
78
- for (const rel of model.relations) {
79
- content += ` ${rel.name}?: ${rel.type}${rel.isArray ? '[]' : ''}${rel.isOptional ? ' | null' : ''};\n`;
80
- }
81
- if (model.relations.length > 0) {
82
- content += ` _count?: { ${model.relations.map(r => `${r.name}: number`).join('; ')} };\n`;
83
- }
84
- content += `}\n\n`;
85
- content += `export type ${model.name}WhereInput = {\n`;
86
- content += ` AND?: ${model.name}WhereInput | ${model.name}WhereInput[];\n`;
87
- content += ` OR?: ${model.name}WhereInput[];\n`;
88
- content += ` NOT?: ${model.name}WhereInput | ${model.name}WhereInput[];\n`;
89
- for (const field of model.fields) {
90
- const filter = this.getFieldFilterType(field);
91
- content += ` ${field.name}?: ${field.type} | ${filter}${field.isOptional ? ' | null' : ''};\n`;
92
- }
93
- for (const rel of model.relations) {
94
- if (rel.isArray) {
95
- content += ` ${rel.name}?: { some?: ${rel.type}WhereInput; none?: ${rel.type}WhereInput; every?: ${rel.type}WhereInput; };\n`;
96
- }
97
- else {
98
- content += ` ${rel.name}?: ${rel.type}WhereInput${rel.isOptional ? ' | null' : ''};\n`;
99
- }
100
- }
101
- if (model.compoundUniques) {
102
- for (const compound of model.compoundUniques) {
103
- const keyName = compound.join('_');
104
- content += ` ${keyName}?: { ${compound.map(f => `${f}: ${model.fields.find(mf => mf.name === f)?.type || 'any'}`).join('; ')} };\n`;
105
- }
106
- }
107
- content += `};\n\n`;
108
- const selectFields = model.fields.map(f => `${f.name}?: boolean`);
109
- const selectRels = model.relations.map(r => `${r.name}?: boolean | ${r.type}FindManyArgs`);
110
- if (model.relations.length > 0) {
111
- selectRels.push(`_count?: boolean | { select?: { ${model.relations.map(r => `${r.name}?: boolean`).join('; ')} } }`);
112
- }
113
- content += `export type ${model.name}Select = { ${[...selectFields, ...selectRels].join('; ')}${[...selectFields, ...selectRels].length > 0 ? ';' : ''} };\n`;
114
- const includeRels = model.relations.map(r => `${r.name}?: boolean | ${r.type}FindManyArgs`);
115
- if (model.relations.length > 0) {
116
- includeRels.push(`_count?: boolean | { select?: { ${model.relations.map(r => `${r.name}?: boolean`).join('; ')} } }`);
117
- }
118
- content += `export type ${model.name}Include = { ${includeRels.join('; ')}${includeRels.length > 0 ? ';' : ''} };\n`;
119
- content += `export type ${model.name}CreateInput = { ${model.fields.map(f => `${f.name}${f.isOptional || f.hasDefault ? '?' : ''}: ${f.type}${f.isOptional ? ' | null' : ''}`).join('; ')}; ${model.relations.map(r => `${r.name}?: { create?: ${r.type}CreateInput | ${r.type}CreateInput[]; connect?: ${r.type}WhereInput | ${r.type}WhereInput[]; }`).join('; ')} };\n`;
120
- content += `export type ${model.name}UpdateInput = { ${model.fields.filter(f => !f.isId).map(f => `${f.name}?: ${f.type === 'number' ? 'number | An5.IntFieldUpdateOperationsInput' : f.type}${f.isOptional ? ' | null' : ''}`).join('; ')}; ${model.relations.map(r => `${r.name}?: { create?: ${r.type}CreateInput | ${r.type}CreateInput[]; connect?: ${r.type}WhereInput | ${r.type}WhereInput[]; set?: ${r.type}WhereInput | ${r.type}WhereInput[]; disconnect?: ${r.type}WhereInput | ${r.type}WhereInput[]; delete?: ${r.type}WhereInput | ${r.type}WhereInput[]; update?: { where: ${r.type}WhereInput; data: ${r.type}UpdateInput; } | { where: ${r.type}WhereInput; data: ${r.type}UpdateInput; }[]; upsert?: { where: ${r.type}WhereInput; create: ${r.type}CreateInput; update: ${r.type}UpdateInput; } | { where: ${r.type}WhereInput; create: ${r.type}CreateInput; update: ${r.type}UpdateInput; }[]; }`).join('; ')} };\n`;
121
- content += `export type ${model.name}FindManyArgs = { where?: ${model.name}WhereInput; orderBy?: any; take?: number; skip?: number; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
122
- content += `export type ${model.name}FindFirstArgs = { where?: ${model.name}WhereInput; orderBy?: any; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
123
- content += `export type ${model.name}FindUniqueArgs = { where?: ${model.name}WhereInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
124
- content += `export type ${model.name}CreateArgs = { data: ${model.name}CreateInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
125
- content += `export type ${model.name}UpdateArgs = { where: ${model.name}WhereInput; data: ${model.name}UpdateInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
126
- content += `export type ${model.name}UpsertArgs = { where: ${model.name}WhereInput; create: ${model.name}CreateInput; update: ${model.name}UpdateInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
127
- content += `export type ${model.name}DeleteArgs = { where: ${model.name}WhereInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
128
- content += `export type ${model.name}TableClient = TableClient<\n ${model.name},\n ${model.name}WhereInput,\n ${model.name}Select,\n ${model.name}Include,\n ${model.name}CreateInput,\n ${model.name}UpdateInput,\n ${model.name}FindManyArgs,\n ${model.name}FindFirstArgs,\n ${model.name}FindUniqueArgs,\n ${model.name}CreateArgs,\n ${model.name}UpdateArgs,\n ${model.name}UpsertArgs,\n ${model.name}DeleteArgs\n>;\n`;
129
- fs_1.default.writeFileSync(path_1.default.join(this.outputDir, `${model.name}.ts`), content);
130
- }
131
- generateIndexTs(models) {
132
- let content = `export * from './base';\n`;
133
- for (const model of models) {
134
- content += `export * from './${model.name}';\n`;
135
- }
136
- content += `\nimport { An5 as BaseAn5 } from './base';\n`;
137
- for (const model of models) {
138
- content += `import * as ${model.name}Types from './${model.name}';\n`;
139
- }
140
- content += `\nexport namespace An5 {\n`;
141
- content += ` export type SortOrder = BaseAn5.SortOrder;\n`;
142
- content += ` export type StringFilter = BaseAn5.StringFilter;\n`;
143
- content += ` export type StringNullableFilter = BaseAn5.StringNullableFilter;\n`;
144
- content += ` export type NumberFilter = BaseAn5.NumberFilter;\n`;
145
- content += ` export type NumberNullableFilter = BaseAn5.NumberNullableFilter;\n`;
146
- content += ` export type BooleanFilter = BaseAn5.BooleanFilter;\n`;
147
- content += ` export type BooleanNullableFilter = BaseAn5.BooleanNullableFilter;\n`;
148
- content += ` export type DateTimeFilter = BaseAn5.DateTimeFilter;\n`;
149
- content += ` export type DateTimeNullableFilter = BaseAn5.DateTimeNullableFilter;\n`;
150
- content += ` export type IntFieldUpdateOperationsInput = BaseAn5.IntFieldUpdateOperationsInput;\n`;
151
- content += ` export type FloatFieldUpdateOperationsInput = BaseAn5.FloatFieldUpdateOperationsInput;\n`;
152
- content += ` export const An5ClientKnownRequestError = BaseAn5.An5ClientKnownRequestError;\n`;
153
- for (const model of models) {
154
- content += ` export type ${model.name} = ${model.name}Types.${model.name};\n`;
155
- content += ` export type ${model.name}WhereInput = ${model.name}Types.${model.name}WhereInput;\n`;
156
- content += ` export type ${model.name}Select = ${model.name}Types.${model.name}Select;\n`;
157
- content += ` export type ${model.name}Include = ${model.name}Types.${model.name}Include;\n`;
158
- content += ` export type ${model.name}CreateInput = ${model.name}Types.${model.name}CreateInput;\n`;
159
- content += ` export type ${model.name}UpdateInput = ${model.name}Types.${model.name}UpdateInput;\n`;
160
- }
161
- content += `}\n`;
162
- content += `\nexport class An5Client {\n $connect(): Promise<void> { return Promise.resolve(); }\n $disconnect(): Promise<void> { return Promise.resolve(); }\n $transaction<R>(fn: (tx: An5Client) => Promise<R>, options?: { timeout?: number }): Promise<R>;\n $transaction<R>(list: Promise<R>[]): Promise<R[]>;\n $transaction(fn: any, options?: any): Promise<any> { return typeof fn === 'function' ? fn(this) : Promise.all(fn); }\n $queryRaw<T = any>(queryParts: TemplateStringsArray | string, ...values: any[]): Promise<T> { return Promise.resolve([] as any); }\n $queryRawUnsafe<R = any>(query: string, ...values: any[]): Promise<R> { return Promise.resolve([] as any); }\n $executeRaw<T = any>(queryParts: TemplateStringsArray | string, ...values: any[]): Promise<any> { return Promise.resolve(0 as any); }\n $executeRawUnsafe(query: string, ...values: any[]): Promise<number> { return Promise.resolve(0); }\n`;
163
- const addedProps = new Set();
164
- for (const model of models) {
165
- const props = this.getAllPropertyVariations(model.name);
166
- for (const prop of props) {
167
- if (!addedProps.has(prop)) {
168
- content += ` ${prop}!: ${model.name}Types.${model.name}TableClient;\n`;
169
- addedProps.add(prop);
170
- }
171
- }
172
- }
173
- content += `}\n`;
174
- fs_1.default.writeFileSync(path_1.default.join(this.outputDir, 'index.ts'), content);
175
- }
176
- getAllPropertyVariations(modelName) {
177
- const variations = new Set();
178
- variations.add(this.toCamelCase(modelName));
179
- const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
180
- for (const acronym of acronyms) {
181
- if (modelName.startsWith(acronym)) {
182
- variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
183
- variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
184
- }
185
- }
186
- return Array.from(variations);
187
- }
188
- getFieldFilterType(field) {
189
- const normalizedType = this.normalizeType(field.type);
190
- if (normalizedType === 'string')
191
- return field.isOptional ? 'An5.StringNullableFilter' : 'An5.StringFilter';
192
- if (normalizedType === 'number')
193
- return field.isOptional ? 'An5.NumberNullableFilter' : 'An5.NumberFilter';
194
- if (normalizedType === 'boolean')
195
- return field.isOptional ? 'An5.BooleanNullableFilter' : 'An5.BooleanFilter';
196
- if (normalizedType === 'Date')
197
- return field.isOptional ? 'An5.DateTimeNullableFilter' : 'An5.DateTimeFilter';
198
- return 'any';
199
- }
200
- normalizeType(type) {
201
- const lowerType = type.toLowerCase();
202
- if (['string', 'varchar', 'nvarchar', 'char', 'nchar', 'text', 'ntext', 'xml', 'guid', 'uuid', 'uniqueidentifier', 'sysname', 'hierarchyid', 'geography', 'geometry', 'sql_variant', 'sqlvariant', 'rowversion', 'variant'].includes(lowerType)) {
203
- return 'string';
204
- }
205
- if (['int', 'integer', 'smallint', 'tinyint', 'bigint', 'long', 'number', 'decimal', 'numeric', 'money', 'smallmoney', 'float', 'real', 'double'].includes(lowerType)) {
206
- return 'number';
207
- }
208
- if (['bool', 'boolean', 'bit'].includes(lowerType)) {
209
- return 'boolean';
210
- }
211
- if (['datetime', 'datetime2', 'datetimeoffset', 'smalldatetime', 'date', 'time', 'timestamp'].includes(lowerType)) {
212
- return 'Date';
213
- }
214
- if (['bytes', 'binary', 'varbinary', 'image'].includes(lowerType)) {
215
- return 'string';
216
- }
217
- return 'any';
218
- }
219
- toCamelCase(str) {
220
- if (!str)
221
- return '';
222
- return str[0].toLowerCase() + str.slice(1);
223
- }
224
- }
225
- exports.CodeGenerator = CodeGenerator;
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.CodeGenerator = void 0;
7
+ const fs_1 = __importDefault(require("fs"));
8
+ const path_1 = __importDefault(require("path"));
9
+ class CodeGenerator {
10
+ constructor(outputDir) {
11
+ this.outputDir = outputDir;
12
+ if (!fs_1.default.existsSync(this.outputDir)) {
13
+ fs_1.default.mkdirSync(this.outputDir, { recursive: true });
14
+ }
15
+ }
16
+ generate(models) {
17
+ this.generateBaseTs();
18
+ for (const model of models) {
19
+ this.generateModelFile(model);
20
+ }
21
+ this.generateIndexTs(models);
22
+ }
23
+ generateBaseTs() {
24
+ const baseTs = `
25
+ import { An5ClientKnownRequestError as An5KnownError } from "@an5/orm";
26
+
27
+ export namespace An5 {
28
+ export const An5ClientKnownRequestError: typeof An5KnownError = An5KnownError;
29
+ export type An5ClientKnownRequestError = An5KnownError;
30
+ export type SortOrder = 'asc' | 'desc';
31
+ export type StringFilter = { equals?: string; in?: string[]; notIn?: string[]; lt?: string; lte?: string; gt?: string; gte?: string; contains?: string; startsWith?: string; endsWith?: string; not?: string | StringFilter; };
32
+ export type StringNullableFilter = { equals?: string | null; in?: (string | null)[]; notIn?: (string | null)[]; lt?: string; lte?: string; gt?: string; gte?: string; contains?: string; startsWith?: string; endsWith?: string; not?: string | StringNullableFilter | null; };
33
+ export type NumberFilter = { equals?: number; in?: number[]; notIn?: number[]; lt?: number; lte?: number; gt?: number; gte?: number; not?: number | NumberFilter; };
34
+ export type IntFieldUpdateOperationsInput = { set?: number; increment?: number; decrement?: number; multiply?: number; divide?: number; };
35
+ export type FloatFieldUpdateOperationsInput = { set?: number; increment?: number; decrement?: number; multiply?: number; divide?: number; };
36
+ export type NumberNullableFilter = { equals?: number | null; in?: (number | null)[]; notIn?: (number | null)[]; lt?: number; lte?: number; gt?: number; gte?: number; not?: number | NumberNullableFilter | null; };
37
+ export type BooleanFilter = { equals?: boolean; not?: boolean | BooleanFilter; };
38
+ export type BooleanNullableFilter = { equals?: boolean | null; not?: boolean | BooleanNullableFilter | null; };
39
+ export type DateTimeFilter = { equals?: Date | string; in?: (Date | string)[]; notIn?: (Date | string)[]; lt?: Date | string; lte?: Date | string; gt?: Date | string; gte?: Date | string; not?: Date | string | DateTimeFilter; };
40
+ export type DateTimeNullableFilter = { equals?: Date | string | null; in?: ((Date | string) | null)[]; notIn?: ((Date | string) | null)[]; lt?: Date | string; lte?: Date | string; gt?: Date | string; gte?: Date | string; not?: Date | string | DateTimeNullableFilter | null; };
41
+ }
42
+
43
+ export interface TableClient<T, WhereInput = any, _Select = any, Include = any, CreateInput = any, UpdateInput = any, FindManyArgs = any, FindFirstArgs = any, FindUniqueArgs = any, CreateArgs = any, UpdateArgs = any, UpsertArgs = any, DeleteArgs = any, AggregateArgs = any, GroupByArgs = any> {
44
+ findMany(args?: FindManyArgs): Promise<T[]>;
45
+ vectorSearch(args: { vector: number[]; take?: number; where?: WhereInput; include?: Include; vectorField?: string; distanceMetric?: 'cosine' | 'euclidean' | 'dot'; }): Promise<(T & { distance: number })[]>;
46
+ findFirst(args?: FindFirstArgs): Promise<T | null>;
47
+ findUnique(args?: FindUniqueArgs): Promise<T | null>;
48
+ count(args?: { where?: WhereInput; }): Promise<number>;
49
+ create(args: CreateArgs): Promise<T>;
50
+ createMany(args: { data: CreateInput[]; skipDuplicates?: boolean; }): Promise<{ count: number }>;
51
+ update(args: UpdateArgs): Promise<T>;
52
+ updateMany(args: { where?: WhereInput; data: UpdateInput; }): Promise<{ count: number }>;
53
+ delete(args: DeleteArgs): Promise<T>;
54
+ deleteMany(args?: { where?: WhereInput; }): Promise<{ count: number }>;
55
+ aggregate(args: AggregateArgs): Promise<any>;
56
+ groupBy(args: GroupByArgs): Promise<any[]>;
57
+ upsert(args: UpsertArgs): Promise<T>;
58
+ }
59
+ `;
60
+ fs_1.default.writeFileSync(path_1.default.join(this.outputDir, 'base.ts'), baseTs);
61
+ }
62
+ generateModelFile(model) {
63
+ let content = `import { An5, TableClient } from './base';\n`;
64
+ const relModels = [...new Set(model.relations.map(r => r.type))];
65
+ for (const relModel of relModels) {
66
+ if (relModel !== model.name) {
67
+ content += `import { ${relModel}, ${relModel}WhereInput, ${relModel}FindManyArgs, ${relModel}CreateInput, ${relModel}UpdateInput } from './${relModel}';\n`;
68
+ }
69
+ }
70
+ content += `\nexport interface ${model.name} {\n`;
71
+ for (const field of model.fields) {
72
+ content += ` ${field.name}: ${this.normalizeType(field.type)}${field.isOptional ? ' | null' : ''};\n`;
73
+ }
74
+ for (const rel of model.relations) {
75
+ content += ` ${rel.name}?: ${rel.type}${rel.isArray ? '[]' : ''}${rel.isOptional ? ' | null' : ''};\n`;
76
+ }
77
+ if (model.relations.length > 0) {
78
+ content += ` _count?: { ${model.relations.map(r => `${r.name}: number`).join('; ')} };\n`;
79
+ }
80
+ content += `}\n\n`;
81
+ content += `export type ${model.name}WhereInput = {\n`;
82
+ content += ` AND?: ${model.name}WhereInput | ${model.name}WhereInput[];\n`;
83
+ content += ` OR?: ${model.name}WhereInput[];\n`;
84
+ content += ` NOT?: ${model.name}WhereInput | ${model.name}WhereInput[];\n`;
85
+ for (const field of model.fields) {
86
+ const filter = this.getFieldFilterType(field);
87
+ content += ` ${field.name}?: ${this.inputType(field.type)} | ${filter}${field.isOptional ? ' | null' : ''};\n`;
88
+ }
89
+ for (const rel of model.relations) {
90
+ if (rel.isArray) {
91
+ content += ` ${rel.name}?: { some?: ${rel.type}WhereInput; none?: ${rel.type}WhereInput; every?: ${rel.type}WhereInput; };\n`;
92
+ }
93
+ else {
94
+ content += ` ${rel.name}?: ${rel.type}WhereInput${rel.isOptional ? ' | null' : ''};\n`;
95
+ }
96
+ }
97
+ if (model.compoundUniques) {
98
+ for (const compound of model.compoundUniques) {
99
+ const keyName = compound.join('_');
100
+ content += ` ${keyName}?: { ${compound.map(f => `${f}: ${model.fields.find(mf => mf.name === f)?.type || 'any'}`).join('; ')} };\n`;
101
+ }
102
+ }
103
+ content += `};\n\n`;
104
+ const selectFields = model.fields.map(f => `${f.name}?: boolean`);
105
+ const selectRels = model.relations.map(r => `${r.name}?: boolean | ${r.type}FindManyArgs`);
106
+ if (model.relations.length > 0) {
107
+ selectRels.push(`_count?: boolean | { select?: { ${model.relations.map(r => `${r.name}?: boolean`).join('; ')} } }`);
108
+ }
109
+ content += `export type ${model.name}Select = { ${[...selectFields, ...selectRels].join('; ')}${[...selectFields, ...selectRels].length > 0 ? ';' : ''} };\n`;
110
+ const includeRels = model.relations.map(r => `${r.name}?: boolean | ${r.type}FindManyArgs`);
111
+ if (model.relations.length > 0) {
112
+ includeRels.push(`_count?: boolean | { select?: { ${model.relations.map(r => `${r.name}?: boolean`).join('; ')} } }`);
113
+ }
114
+ content += `export type ${model.name}Include = { ${includeRels.join('; ')}${includeRels.length > 0 ? ';' : ''} };\n`;
115
+ content += `export type ${model.name}CreateInput = { ${model.fields.map(f => `${f.name}${f.isOptional || f.hasDefault ? '?' : ''}: ${this.inputType(f.type)}${f.isOptional ? ' | null' : ''}`).join('; ')}; ${model.relations.map(r => `${r.name}?: { create?: ${r.type}CreateInput | ${r.type}CreateInput[]; connect?: ${r.type}WhereInput | ${r.type}WhereInput[]; }`).join('; ')} };\n`;
116
+ content += `export type ${model.name}UpdateInput = { ${model.fields.filter(f => !f.isId).map(f => `${f.name}?: ${f.type === 'number' ? 'number | An5.IntFieldUpdateOperationsInput' : this.inputType(f.type)}${f.isOptional ? ' | null' : ''}`).join('; ')}; ${model.relations.map(r => `${r.name}?: { create?: ${r.type}CreateInput | ${r.type}CreateInput[]; connect?: ${r.type}WhereInput | ${r.type}WhereInput[]; set?: ${r.type}WhereInput | ${r.type}WhereInput[]; disconnect?: ${r.type}WhereInput | ${r.type}WhereInput[]; delete?: ${r.type}WhereInput | ${r.type}WhereInput[]; update?: { where: ${r.type}WhereInput; data: ${r.type}UpdateInput; } | { where: ${r.type}WhereInput; data: ${r.type}UpdateInput; }[]; upsert?: { where: ${r.type}WhereInput; create: ${r.type}CreateInput; update: ${r.type}UpdateInput; } | { where: ${r.type}WhereInput; create: ${r.type}CreateInput; update: ${r.type}UpdateInput; }[]; }`).join('; ')} };\n`;
117
+ content += `export type ${model.name}OrderByInput = { ${model.fields.map(f => `${f.name}?: An5.SortOrder`).join('; ')} };\n`;
118
+ content += `export type ${model.name}FindManyArgs = { where?: ${model.name}WhereInput; orderBy?: ${model.name}OrderByInput | ${model.name}OrderByInput[]; take?: number; skip?: number; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
119
+ content += `export type ${model.name}FindFirstArgs = { where?: ${model.name}WhereInput; orderBy?: ${model.name}OrderByInput | ${model.name}OrderByInput[]; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
120
+ content += `export type ${model.name}FindUniqueArgs = { where?: ${model.name}WhereInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
121
+ content += `export type ${model.name}CreateArgs = { data: ${model.name}CreateInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
122
+ content += `export type ${model.name}UpdateArgs = { where: ${model.name}WhereInput; data: ${model.name}UpdateInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
123
+ content += `export type ${model.name}UpsertArgs = { where: ${model.name}WhereInput; create: ${model.name}CreateInput; update: ${model.name}UpdateInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
124
+ content += `export type ${model.name}DeleteArgs = { where: ${model.name}WhereInput; include?: ${model.name}Include; select?: ${model.name}Select; };\n`;
125
+ content += `export type ${model.name}ScalarFieldEnum = ${model.fields.map(f => `'${f.name}'`).join(' | ') || 'never'};\n`;
126
+ content += `export type ${model.name}AggregateArgs = { where?: ${model.name}WhereInput; _count?: true | { _all?: true; ${model.fields.map(f => `${f.name}?: true`).join('; ')} }; _sum?: { ${model.fields.filter(f => this.normalizeType(f.type) === 'number').map(f => `${f.name}?: true`).join('; ')} }; _avg?: { ${model.fields.filter(f => this.normalizeType(f.type) === 'number').map(f => `${f.name}?: true`).join('; ')} }; _min?: { ${model.fields.map(f => `${f.name}?: true`).join('; ')} }; _max?: { ${model.fields.map(f => `${f.name}?: true`).join('; ')} }; };\n`;
127
+ const numericHavingFields = model.fields.filter(f => this.normalizeType(f.type) === 'number').map(f => `${f.name}?: An5.NumberFilter | number`).join('; ');
128
+ const scalarHavingFields = model.fields.map(f => `${f.name}?: any`).join('; ');
129
+ content += `export type ${model.name}AggregateHavingInput = { _count?: { _all?: An5.NumberFilter | number; ${model.fields.map(f => `${f.name}?: An5.NumberFilter | number`).join('; ')} }; _sum?: { ${numericHavingFields} }; _avg?: { ${numericHavingFields} }; _min?: { ${scalarHavingFields} }; _max?: { ${scalarHavingFields} }; };\n`;
130
+ content += `export type ${model.name}GroupByArgs = { by: ${model.name}ScalarFieldEnum | ${model.name}ScalarFieldEnum[]; where?: ${model.name}WhereInput; having?: ${model.name}AggregateHavingInput; orderBy?: ${model.name}OrderByInput | ${model.name}OrderByInput[]; skip?: number; take?: number; _count?: true | { _all?: true; ${model.fields.map(f => `${f.name}?: true`).join('; ')} }; _sum?: { ${model.fields.filter(f => this.normalizeType(f.type) === 'number').map(f => `${f.name}?: true`).join('; ')} }; _avg?: { ${model.fields.filter(f => this.normalizeType(f.type) === 'number').map(f => `${f.name}?: true`).join('; ')} }; _min?: { ${model.fields.map(f => `${f.name}?: true`).join('; ')} }; _max?: { ${model.fields.map(f => `${f.name}?: true`).join('; ')} }; };\n`;
131
+ content += `export type ${model.name}TableClient = TableClient<\n ${model.name},\n ${model.name}WhereInput,\n ${model.name}Select,\n ${model.name}Include,\n ${model.name}CreateInput,\n ${model.name}UpdateInput,\n ${model.name}FindManyArgs,\n ${model.name}FindFirstArgs,\n ${model.name}FindUniqueArgs,\n ${model.name}CreateArgs,\n ${model.name}UpdateArgs,\n ${model.name}UpsertArgs,\n ${model.name}DeleteArgs,\n ${model.name}AggregateArgs,\n ${model.name}GroupByArgs\n>;\n`;
132
+ fs_1.default.writeFileSync(path_1.default.join(this.outputDir, `${model.name}.ts`), content);
133
+ }
134
+ generateIndexTs(models) {
135
+ let content = `export * from './base';\n`;
136
+ for (const model of models) {
137
+ content += `export * from './${model.name}';\n`;
138
+ }
139
+ content += `\nimport { An5 as BaseAn5 } from './base';\n`;
140
+ for (const model of models) {
141
+ content += `import * as ${model.name}Types from './${model.name}';\n`;
142
+ }
143
+ content += `\nexport namespace An5 {\n`;
144
+ content += ` export type SortOrder = BaseAn5.SortOrder;\n`;
145
+ content += ` export type StringFilter = BaseAn5.StringFilter;\n`;
146
+ content += ` export type StringNullableFilter = BaseAn5.StringNullableFilter;\n`;
147
+ content += ` export type NumberFilter = BaseAn5.NumberFilter;\n`;
148
+ content += ` export type NumberNullableFilter = BaseAn5.NumberNullableFilter;\n`;
149
+ content += ` export type BooleanFilter = BaseAn5.BooleanFilter;\n`;
150
+ content += ` export type BooleanNullableFilter = BaseAn5.BooleanNullableFilter;\n`;
151
+ content += ` export type DateTimeFilter = BaseAn5.DateTimeFilter;\n`;
152
+ content += ` export type DateTimeNullableFilter = BaseAn5.DateTimeNullableFilter;\n`;
153
+ content += ` export type IntFieldUpdateOperationsInput = BaseAn5.IntFieldUpdateOperationsInput;\n`;
154
+ content += ` export type FloatFieldUpdateOperationsInput = BaseAn5.FloatFieldUpdateOperationsInput;\n`;
155
+ content += ` export const An5ClientKnownRequestError = BaseAn5.An5ClientKnownRequestError;\n`;
156
+ for (const model of models) {
157
+ content += ` export type ${model.name} = ${model.name}Types.${model.name};\n`;
158
+ content += ` export type ${model.name}WhereInput = ${model.name}Types.${model.name}WhereInput;\n`;
159
+ content += ` export type ${model.name}Select = ${model.name}Types.${model.name}Select;\n`;
160
+ content += ` export type ${model.name}Include = ${model.name}Types.${model.name}Include;\n`;
161
+ content += ` export type ${model.name}CreateInput = ${model.name}Types.${model.name}CreateInput;\n`;
162
+ content += ` export type ${model.name}UpdateInput = ${model.name}Types.${model.name}UpdateInput;\n`;
163
+ content += ` export type ${model.name}OrderByInput = ${model.name}Types.${model.name}OrderByInput;\n`;
164
+ content += ` export type ${model.name}FindManyArgs = ${model.name}Types.${model.name}FindManyArgs;\n`;
165
+ content += ` export type ${model.name}FindFirstArgs = ${model.name}Types.${model.name}FindFirstArgs;\n`;
166
+ content += ` export type ${model.name}FindUniqueArgs = ${model.name}Types.${model.name}FindUniqueArgs;\n`;
167
+ content += ` export type ${model.name}CreateArgs = ${model.name}Types.${model.name}CreateArgs;\n`;
168
+ content += ` export type ${model.name}UpdateArgs = ${model.name}Types.${model.name}UpdateArgs;\n`;
169
+ content += ` export type ${model.name}UpsertArgs = ${model.name}Types.${model.name}UpsertArgs;\n`;
170
+ content += ` export type ${model.name}DeleteArgs = ${model.name}Types.${model.name}DeleteArgs;\n`;
171
+ content += ` export type ${model.name}TableClient = ${model.name}Types.${model.name}TableClient;\n`;
172
+ content += ` export type ${model.name}ScalarFieldEnum = ${model.name}Types.${model.name}ScalarFieldEnum;\n`;
173
+ content += ` export type ${model.name}AggregateArgs = ${model.name}Types.${model.name}AggregateArgs;\n`;
174
+ content += ` export type ${model.name}GroupByArgs = ${model.name}Types.${model.name}GroupByArgs;\n`;
175
+ }
176
+ content += `}\n`;
177
+ content += `\nexport class An5Client {\n $connect(): Promise<void> { return Promise.resolve(); }\n $disconnect(): Promise<void> { return Promise.resolve(); }\n $transaction<R>(fn: (tx: An5Client) => Promise<R>, options?: { timeout?: number }): Promise<R>;\n $transaction<R>(list: Promise<R>[]): Promise<R[]>;\n $transaction(fn: any, _options?: any): Promise<any> { return typeof fn === 'function' ? fn(this) : Promise.all(fn); }\n $begin(): Promise<An5Client> { return Promise.resolve(this); }\n $commit(): Promise<void> { return Promise.resolve(); }\n $rollback(): Promise<void> { return Promise.resolve(); }\n $queryRaw<T = any>(_queryParts: TemplateStringsArray | string, ..._values: any[]): Promise<T> { return Promise.resolve([] as any); }\n $queryRawUnsafe<R = any>(_query: string, ..._values: any[]): Promise<R> { return Promise.resolve([] as any); }\n $executeRaw<_T = any>(_queryParts: TemplateStringsArray | string, ..._values: any[]): Promise<any> { return Promise.resolve(0 as any); }\n $executeRawUnsafe(_query: string, ..._values: any[]): Promise<number> { return Promise.resolve(0); }\n`;
178
+ const addedProps = new Set();
179
+ for (const model of models) {
180
+ const props = this.getAllPropertyVariations(model.name);
181
+ for (const prop of props) {
182
+ if (!addedProps.has(prop)) {
183
+ content += ` ${prop}!: ${model.name}Types.${model.name}TableClient;\n`;
184
+ addedProps.add(prop);
185
+ }
186
+ }
187
+ }
188
+ content += `}\n`;
189
+ fs_1.default.writeFileSync(path_1.default.join(this.outputDir, 'index.ts'), content);
190
+ }
191
+ getAllPropertyVariations(modelName) {
192
+ const variations = new Set();
193
+ variations.add(modelName);
194
+ variations.add(this.toCamelCase(modelName));
195
+ variations.add(modelName + 's');
196
+ variations.add(this.toCamelCase(modelName) + 's');
197
+ const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
198
+ for (const acronym of acronyms) {
199
+ if (modelName.startsWith(acronym)) {
200
+ variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
201
+ variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
202
+ variations.add(acronym.toLowerCase() + modelName.slice(acronym.length) + 's');
203
+ }
204
+ }
205
+ return Array.from(variations);
206
+ }
207
+ getFieldFilterType(field) {
208
+ const normalizedType = this.normalizeType(field.type);
209
+ if (normalizedType === 'string')
210
+ return field.isOptional ? 'An5.StringNullableFilter' : 'An5.StringFilter';
211
+ if (normalizedType === 'number')
212
+ return field.isOptional ? 'An5.NumberNullableFilter' : 'An5.NumberFilter';
213
+ if (normalizedType === 'boolean')
214
+ return field.isOptional ? 'An5.BooleanNullableFilter' : 'An5.BooleanFilter';
215
+ if (normalizedType === 'Date')
216
+ return field.isOptional ? 'An5.DateTimeNullableFilter' : 'An5.DateTimeFilter';
217
+ return 'any';
218
+ }
219
+ // Input positions (create/update/where scalars) accept ISO strings as well
220
+ // as Date objects — Prisma parity for DateTime fields.
221
+ inputType(tsType) {
222
+ return tsType === 'Date' ? 'Date | string' : tsType;
223
+ }
224
+ normalizeType(type) {
225
+ const lowerType = type.toLowerCase();
226
+ if (['string', 'varchar', 'nvarchar', 'char', 'nchar', 'text', 'ntext', 'xml', 'guid', 'uuid', 'uniqueidentifier', 'sysname', 'hierarchyid', 'geography', 'geometry', 'sql_variant', 'sqlvariant', 'rowversion', 'variant'].includes(lowerType)) {
227
+ return 'string';
228
+ }
229
+ if (['int', 'integer', 'smallint', 'tinyint', 'bigint', 'long', 'number', 'decimal', 'numeric', 'money', 'smallmoney', 'float', 'real', 'double'].includes(lowerType)) {
230
+ return 'number';
231
+ }
232
+ if (['bool', 'boolean', 'bit'].includes(lowerType)) {
233
+ return 'boolean';
234
+ }
235
+ if (['datetime', 'datetime2', 'datetimeoffset', 'smalldatetime', 'date', 'time', 'timestamp'].includes(lowerType)) {
236
+ return 'Date';
237
+ }
238
+ if (['bytes', 'binary', 'varbinary', 'image'].includes(lowerType)) {
239
+ return 'string';
240
+ }
241
+ return 'any';
242
+ }
243
+ toCamelCase(str) {
244
+ if (!str)
245
+ return '';
246
+ return str[0].toLowerCase() + str.slice(1);
247
+ }
248
+ }
249
+ exports.CodeGenerator = CodeGenerator;