@@ -24,7 +24,7 @@ export function generateProcedure(
2424 ) } .${ prismaMethod } (input as any))) as ProcReturns<
2525 "query",
2626 Proc,
27- (typeof ${ upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
27+ (typeof $Schema. $ {upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
2828 ReturnType<PrismaClient["${ lowerCaseFirst ( modelName ) } "]["${ opType } "]>
2929 >,
3030 ` ) ;
@@ -36,7 +36,7 @@ export function generateProcedure(
3636 ) } .${ prismaMethod } (input as any))) as ProcReturns<
3737 "mutation",
3838 Proc,
39- (typeof ${ upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
39+ (typeof $Schema. $ {upperCaseFirst ( modelName ) } InputSchema)["${ opType . replace ( 'OrThrow' , '' ) } "],
4040 ReturnType<PrismaClient["${ lowerCaseFirst ( modelName ) } "]["${ opType } "]>
4141 >,
4242 ` ) ;
@@ -247,8 +247,8 @@ export function generateRouterTypingImports(sourceFile: SourceFile, model: Model
247247}
248248
249249// 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';` ) ;
252252}
253253
254254export function generateHelperImport ( sourceFile : SourceFile ) {
@@ -257,54 +257,55 @@ export function generateHelperImport(sourceFile: SourceFile) {
257257
258258export const getInputSchemaByOpName = ( opName : string , modelName : string ) => {
259259 let inputType ;
260+ const capModelName = upperCaseFirst ( modelName ) ;
260261 switch ( opName ) {
261262 case 'findUnique' :
262- inputType = `${ modelName } InputSchema.findUnique` ;
263+ inputType = `$Schema. ${ capModelName } InputSchema.findUnique` ;
263264 break ;
264265 case 'findFirst' :
265- inputType = `${ modelName } InputSchema.findFirst` ;
266+ inputType = `$Schema. ${ capModelName } InputSchema.findFirst` ;
266267 break ;
267268 case 'findMany' :
268- inputType = `${ modelName } InputSchema.findMany` ;
269+ inputType = `$Schema. ${ capModelName } InputSchema.findMany` ;
269270 break ;
270271 case 'findRaw' :
271- inputType = `${ modelName } InputSchema.findRawObject` ;
272+ inputType = `$Schema. ${ capModelName } InputSchema.findRawObject` ;
272273 break ;
273274 case 'createOne' :
274- inputType = `${ modelName } InputSchema.create` ;
275+ inputType = `$Schema. ${ capModelName } InputSchema.create` ;
275276 break ;
276277 case 'createMany' :
277- inputType = `${ modelName } InputSchema.createMany` ;
278+ inputType = `$Schema. ${ capModelName } InputSchema.createMany` ;
278279 break ;
279280 case 'deleteOne' :
280- inputType = `${ modelName } InputSchema.delete` ;
281+ inputType = `$Schema. ${ capModelName } InputSchema.delete` ;
281282 break ;
282283 case 'updateOne' :
283- inputType = `${ modelName } InputSchema.update` ;
284+ inputType = `$Schema. ${ capModelName } InputSchema.update` ;
284285 break ;
285286 case 'deleteMany' :
286- inputType = `${ modelName } InputSchema.deleteMany` ;
287+ inputType = `$Schema. ${ capModelName } InputSchema.deleteMany` ;
287288 break ;
288289 case 'updateMany' :
289- inputType = `${ modelName } InputSchema.updateMany` ;
290+ inputType = `$Schema. ${ capModelName } InputSchema.updateMany` ;
290291 break ;
291292 case 'upsertOne' :
292- inputType = `${ modelName } InputSchema.upsert` ;
293+ inputType = `$Schema. ${ capModelName } InputSchema.upsert` ;
293294 break ;
294295 case 'aggregate' :
295- inputType = `${ modelName } InputSchema.aggregate` ;
296+ inputType = `$Schema. ${ capModelName } InputSchema.aggregate` ;
296297 break ;
297298 case 'aggregateRaw' :
298- inputType = `${ modelName } InputSchema.aggregateRawObject` ;
299+ inputType = `$Schema. ${ capModelName } InputSchema.aggregateRawObject` ;
299300 break ;
300301 case 'groupBy' :
301- inputType = `${ modelName } InputSchema.groupBy` ;
302+ inputType = `$Schema. ${ capModelName } InputSchema.groupBy` ;
302303 break ;
303304 case 'count' :
304- inputType = `${ modelName } InputSchema.count` ;
305+ inputType = `$Schema. ${ capModelName } InputSchema.count` ;
305306 break ;
306307 default :
307- console . log ( 'getInputTypeByOpName: ' , { opName , modelName } ) ;
308+ break ;
308309 }
309310 return inputType ;
310311} ;
0 commit comments