Fix property content type wrongfully appearing in response schema (#137)

src/openapi.ts

@@ -44,8 +44,8 @@ export function OpenAPIRouter<
   RequestType = IRequest,
   Args extends any[] = any[],
   RouteType = Equal<RequestType, IRequest> extends true
-  ? Route
-  : UniversalRoute<RequestType, Args>
+    ? Route
+    : UniversalRoute<RequestType, Args>
 >(options?: RouterOptions): OpenAPIRouterType<RouteType, Args> {
   const registry: OpenAPIRegistryMerger = new OpenAPIRegistryMerger()
 

src/route.ts

@@ -108,6 +108,9 @@ export class OpenAPIRoute<I = IRequest, A extends any[] = any[]> {
         }
 
         delete value.schema
+        if (value.contentType) {
+          delete value.contentType
+        }
       }
 
       if (value.headers && !isAnyZodType(value.headers)) {

tests/integration/openapi-schema.test.ts

@@ -0,0 +1,22 @@
+import 'isomorphic-fetch'
+import { buildRequest, findError } from '../utils'
+import { ToDoList, todoRouter } from '../router'
+
+describe('openapi schema', () => {
+  test('custom content type', async () => {
+    const request = await todoRouter.handle(
+      buildRequest({ method: 'GET', path: '/openapi.json' })
+    )
+    const resp = await request.json()
+    const respSchema = resp.paths['/contenttype'].get.responses[200]
+
+    expect(respSchema.contentType).toBeUndefined()
+    expect(respSchema.content).toEqual({
+      'text/csv': {
+        schema: {
+          type: 'string',
+        },
+      },
+    })
+  })
+})

tests/router.ts

@@ -108,6 +108,27 @@ export class ToDoGet extends OpenAPIRoute {
   }
 }
 
+export class ContentTypeGet extends OpenAPIRoute {
+  static schema = {
+    responses: {
+      '200': {
+        description: 'Successful Response',
+        contentType: 'text/csv',
+        schema: new Str(),
+      },
+    },
+  }
+
+  async handle(request: Request, env: any, context: any, data: any) {
+    return {
+      todo: {
+        lorem: 'lorem',
+        ipsum: 'ipsum',
+      },
+    }
+  }
+}
+
 export class ToDoCreate extends OpenAPIRoute {
   static schema: OpenAPIRouteSchema = {
     tags: ['ToDo'],
@@ -147,6 +168,7 @@ export const todoRouter = OpenAPIRouter({ openapiVersion: '3' })
 todoRouter.get('/todos', ToDoList)
 todoRouter.get('/todos/:id', ToDoGet)
 todoRouter.post('/todos', ToDoCreate)
+todoRouter.get('/contenttype', ContentTypeGet)
 
 // 404 for everything else
 todoRouter.all('*', () => new Response('Not Found.', { status: 404 }))