refactor: add content type filter function (orval-labs#2907)

Author: froggy1014

packages/core/src/getters/body.ts

@@ -7,7 +7,7 @@ import type {
   OpenApiRequestBodyObject,
   OverrideOutputContentType,
 } from '../types';
-import { camel, isReference, sanitize } from '../utils';
+import { camel, filterByContentType, isReference, sanitize } from '../utils';
 import { getResReqTypes } from './res-req-types';
 
 interface GetBodyOptions {
@@ -29,22 +29,7 @@ export function getBody({
     context,
   );
 
-  const filteredBodyTypes = contentType
-    ? allBodyTypes.filter((type) => {
-        let include = true;
-        let exclude = false;
-
-        if (contentType.include) {
-          include = contentType.include.includes(type.contentType);
-        }
-
-        if (contentType.exclude) {
-          exclude = contentType.exclude.includes(type.contentType);
-        }
-
-        return include && !exclude;
-      })
-    : allBodyTypes;
+  const filteredBodyTypes = filterByContentType(allBodyTypes, contentType);
 
   const imports = filteredBodyTypes.flatMap(({ imports }) => imports);
   const schemas = filteredBodyTypes.flatMap(({ schemas }) => schemas);

packages/core/src/getters/response.ts

@@ -5,7 +5,7 @@ import type {
   OverrideOutputContentType,
   ResReqTypesValue,
 } from '../types';
-import { dedupeUnionType } from '../utils';
+import { dedupeUnionType, filterByContentType } from '../utils';
 import { getResReqTypes } from './res-req-types';
 
 interface GetResponseOptions {
@@ -43,22 +43,7 @@ export function getResponse({
     (type) => `${type.key}-${type.value}`,
   );
 
-  const filteredTypes = contentType
-    ? types.filter((type) => {
-        let include = true;
-        let exclude = false;
-
-        if (contentType.include) {
-          include = contentType.include.includes(type.contentType);
-        }
-
-        if (contentType.exclude) {
-          exclude = contentType.exclude.includes(type.contentType);
-        }
-
-        return include && !exclude;
-      })
-    : types;
+  const filteredTypes = filterByContentType(types, contentType);
 
   const imports = filteredTypes.flatMap(({ imports }) => imports);
   const schemas = filteredTypes.flatMap(({ schemas }) => schemas);

packages/core/src/utils/content-type.ts

@@ -78,3 +78,50 @@ export function getFormDataFieldFileType(
 
   return undefined;
 }
+
+/**
+ * Filter configuration for content types
+ */
+export interface ContentTypeFilter {
+  include?: string[];
+  exclude?: string[];
+}
+
+/**
+ * Filters items by content type based on include/exclude rules
+ *
+ * @param items - Array of items with contentType property
+ * @param filter - Optional filter configuration
+ * @returns Filtered array
+ *
+ * @example
+ * ```ts
+ * const types = [
+ *   { contentType: 'application/json', value: '...' },
+ *   { contentType: 'text/xml', value: '...' }
+ * ];
+ *
+ * // Include only JSON
+ * filterByContentType(types, { include: ['application/json'] });
+ *
+ * // Exclude XML
+ * filterByContentType(types, { exclude: ['text/xml'] });
+ * ```
+ */
+export function filterByContentType<T extends { contentType: string }>(
+  items: T[],
+  filter?: ContentTypeFilter,
+): T[] {
+  if (!filter) {
+    return items;
+  }
+
+  return items.filter((item) => {
+    const shouldInclude =
+      !filter.include || filter.include.includes(item.contentType);
+
+    const shouldExclude = filter.exclude?.includes(item.contentType) ?? false;
+
+    return shouldInclude && !shouldExclude;
+  });
+}

packages/core/src/utils/merge-deep.ts

@@ -1,4 +1,4 @@
-const isObject = (obj: unknown) => obj && typeof obj === 'object';
+import { isObject } from './assertion';
 
 export function mergeDeep<
   T extends Record<string, any>,