fix(openapi): CreateManyArgs does not take array as input (#1246)

ymc9 · web-flow · commit 813748160e35 · 2024-04-12T15:04:23.000+08:00
diff --git a/packages/plugins/openapi/src/rpc-generator.ts b/packages/plugins/openapi/src/rpc-generator.ts
@@ -176,7 +176,15 @@ export class RPCOpenAPIGenerator extends OpenAPIGeneratorBase {
                         type: 'object',
                         required: ['data'],
                         properties: {
-                            data: this.ref(`${modelName}CreateManyInput`),
+                            data: this.oneOf(
+                                this.ref(`${modelName}CreateManyInput`),
+                                this.array(this.ref(`${modelName}CreateManyInput`))
+                            ),
+                            skipDuplicates: {
+                                type: 'boolean',
+                                description:
+                                    'Do not insert records with unique fields or ID fields that already exist.',
+                            },
                             meta: this.ref('_Meta'),
                         },
                     },
diff --git a/packages/plugins/openapi/tests/baseline/rpc-3.0.0.baseline.yaml b/packages/plugins/openapi/tests/baseline/rpc-3.0.0.baseline.yaml
@@ -3194,7 +3194,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/UserCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/UserCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/UserCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         UserFindUniqueArgs:
@@ -3374,7 +3382,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/ProfileCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/ProfileCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/ProfileCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         ProfileFindUniqueArgs:
@@ -3554,7 +3570,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/Post_ItemCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/Post_ItemCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/Post_ItemCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         Post_ItemFindUniqueArgs:
diff --git a/packages/plugins/openapi/tests/baseline/rpc-3.1.0.baseline.yaml b/packages/plugins/openapi/tests/baseline/rpc-3.1.0.baseline.yaml
@@ -3248,7 +3248,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/UserCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/UserCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/UserCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         UserFindUniqueArgs:
@@ -3428,7 +3436,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/ProfileCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/ProfileCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/ProfileCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         ProfileFindUniqueArgs:
@@ -3608,7 +3624,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/Post_ItemCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/Post_ItemCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/Post_ItemCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         Post_ItemFindUniqueArgs:
diff --git a/packages/plugins/openapi/tests/baseline/rpc-type-coverage-3.0.0.baseline.yaml b/packages/plugins/openapi/tests/baseline/rpc-type-coverage-3.0.0.baseline.yaml
@@ -2017,7 +2017,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/FooCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/FooCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/FooCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         FooFindUniqueArgs:
diff --git a/packages/plugins/openapi/tests/baseline/rpc-type-coverage-3.1.0.baseline.yaml b/packages/plugins/openapi/tests/baseline/rpc-type-coverage-3.1.0.baseline.yaml
@@ -2049,7 +2049,15 @@ components:
                 - data
             properties:
                 data:
-                    $ref: '#/components/schemas/FooCreateManyInput'
+                    oneOf:
+                        - $ref: '#/components/schemas/FooCreateManyInput'
+                        - type: array
+                          items:
+                              $ref: '#/components/schemas/FooCreateManyInput'
+                skipDuplicates:
+                    type: boolean
+                    description: Do not insert records with unique fields or ID fields that already
+                        exist.
                 meta:
                     $ref: '#/components/schemas/_Meta'
         FooFindUniqueArgs:
diff --git a/packages/schema/src/plugins/zod/transformer.ts b/packages/schema/src/plugins/zod/transformer.ts
@@ -485,7 +485,7 @@ export const ${this.name}ObjectSchema: SchemaType = ${schema} as SchemaType;`;
                 imports.push(
                     `import { ${modelName}CreateManyInputObjectSchema } from '../objects/${modelName}CreateManyInput.schema'`
                 );
-                codeBody += `createMany: z.object({ data: z.union([${modelName}CreateManyInputObjectSchema, z.array(${modelName}CreateManyInputObjectSchema)]) }),`;
+                codeBody += `createMany: z.object({ data: z.union([${modelName}CreateManyInputObjectSchema, z.array(${modelName}CreateManyInputObjectSchema)]), skipDuplicates: z.boolean().optional() }),`;
                 operations.push(['createMany', origModelName]);
             }
 

Original file line number	Diff line number	Diff line change
@@ -485,7 +485,7 @@ export const ${this.name}ObjectSchema: SchemaType = ${schema} as SchemaType;`;
`485`	`485`	`imports.push(`
`486`	`486`	`import { ${modelName}CreateManyInputObjectSchema } from '../objects/${modelName}CreateManyInput.schema'`
`487`	`487`	`);`
`488`		- codeBody += `createMany: z.object({ data: z.union([${modelName}CreateManyInputObjectSchema, z.array(${modelName}CreateManyInputObjectSchema)]) }),`;
	`488`	+ codeBody += `createMany: z.object({ data: z.union([${modelName}CreateManyInputObjectSchema, z.array(${modelName}CreateManyInputObjectSchema)]), skipDuplicates: z.boolean().optional() }),`;
`489`	`489`	`operations.push(['createMany', origModelName]);`
`490`	`490`	`}`
`491`	`491`