@@ -24,7 +24,7 @@ export function generateProcedure(
24
24
) } .${ prismaMethod } (input as any))) as ProcReturns<
25
25
"query",
26
26
Proc,
27
- (typeof ${ upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
27
+ (typeof $Schema. $ {upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
28
28
ReturnType<PrismaClient["${ lowerCaseFirst ( modelName ) } "]["${ opType } "]>
29
29
>,
30
30
` ) ;
@@ -36,7 +36,7 @@ export function generateProcedure(
36
36
) } .${ prismaMethod } (input as any))) as ProcReturns<
37
37
"mutation",
38
38
Proc,
39
- (typeof ${ upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
39
+ (typeof $Schema. $ {upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
40
40
ReturnType<PrismaClient["${ lowerCaseFirst ( modelName ) } "]["${ opType } "]>
41
41
>,
42
42
` ) ;
@@ -247,8 +247,8 @@ export function generateRouterTypingImports(sourceFile: SourceFile, model: Model
247
247
}
248
248
249
249
// eslint-disable-next-line @typescript-eslint/no-unused-vars
250
- export function generateRouterSchemaImports ( sourceFile : SourceFile , name : string , zodSchemasImport : string ) {
251
- sourceFile . addStatements ( `import { ${ name } InputSchema } from '${ zodSchemasImport } /input';` ) ;
250
+ export function generateRouterSchemaImport ( sourceFile : SourceFile , zodSchemasImport : string ) {
251
+ sourceFile . addStatements ( `import $Schema from '${ zodSchemasImport } /input';` ) ;
252
252
}
253
253
254
254
export function generateHelperImport ( sourceFile : SourceFile ) {
@@ -257,54 +257,55 @@ export function generateHelperImport(sourceFile: SourceFile) {
257
257
258
258
export const getInputSchemaByOpName = ( opName : string , modelName : string ) => {
259
259
let inputType ;
260
+ const capModelName = upperCaseFirst ( modelName ) ;
260
261
switch ( opName ) {
261
262
case 'findUnique' :
262
- inputType = `${ modelName } InputSchema.findUnique` ;
263
+ inputType = `$Schema. ${ capModelName } InputSchema.findUnique` ;
263
264
break ;
264
265
case 'findFirst' :
265
- inputType = `${ modelName } InputSchema.findFirst` ;
266
+ inputType = `$Schema. ${ capModelName } InputSchema.findFirst` ;
266
267
break ;
267
268
case 'findMany' :
268
- inputType = `${ modelName } InputSchema.findMany` ;
269
+ inputType = `$Schema. ${ capModelName } InputSchema.findMany` ;
269
270
break ;
270
271
case 'findRaw' :
271
- inputType = `${ modelName } InputSchema.findRawObject` ;
272
+ inputType = `$Schema. ${ capModelName } InputSchema.findRawObject` ;
272
273
break ;
273
274
case 'createOne' :
274
- inputType = `${ modelName } InputSchema.create` ;
275
+ inputType = `$Schema. ${ capModelName } InputSchema.create` ;
275
276
break ;
276
277
case 'createMany' :
277
- inputType = `${ modelName } InputSchema.createMany` ;
278
+ inputType = `$Schema. ${ capModelName } InputSchema.createMany` ;
278
279
break ;
279
280
case 'deleteOne' :
280
- inputType = `${ modelName } InputSchema.delete` ;
281
+ inputType = `$Schema. ${ capModelName } InputSchema.delete` ;
281
282
break ;
282
283
case 'updateOne' :
283
- inputType = `${ modelName } InputSchema.update` ;
284
+ inputType = `$Schema. ${ capModelName } InputSchema.update` ;
284
285
break ;
285
286
case 'deleteMany' :
286
- inputType = `${ modelName } InputSchema.deleteMany` ;
287
+ inputType = `$Schema. ${ capModelName } InputSchema.deleteMany` ;
287
288
break ;
288
289
case 'updateMany' :
289
- inputType = `${ modelName } InputSchema.updateMany` ;
290
+ inputType = `$Schema. ${ capModelName } InputSchema.updateMany` ;
290
291
break ;
291
292
case 'upsertOne' :
292
- inputType = `${ modelName } InputSchema.upsert` ;
293
+ inputType = `$Schema. ${ capModelName } InputSchema.upsert` ;
293
294
break ;
294
295
case 'aggregate' :
295
- inputType = `${ modelName } InputSchema.aggregate` ;
296
+ inputType = `$Schema. ${ capModelName } InputSchema.aggregate` ;
296
297
break ;
297
298
case 'aggregateRaw' :
298
- inputType = `${ modelName } InputSchema.aggregateRawObject` ;
299
+ inputType = `$Schema. ${ capModelName } InputSchema.aggregateRawObject` ;
299
300
break ;
300
301
case 'groupBy' :
301
- inputType = `${ modelName } InputSchema.groupBy` ;
302
+ inputType = `$Schema. ${ capModelName } InputSchema.groupBy` ;
302
303
break ;
303
304
case 'count' :
304
- inputType = `${ modelName } InputSchema.count` ;
305
+ inputType = `$Schema. ${ capModelName } InputSchema.count` ;
305
306
break ;
306
307
default :
307
- console . log ( 'getInputTypeByOpName: ' , { opName , modelName } ) ;
308
+ break ;
308
309
}
309
310
return inputType ;
310
311
} ;
0 commit comments