pass unit test

Author: danadajian

src/config/add-dependent-types-to-only-types.ts

@@ -47,7 +47,7 @@ function getDependentTypeNames(
 ): string[] {
   const namedTypes = getDependentFieldTypeNames(node, config)
     .concat(getDependentUnionNames(node))
-    .concat(getDependentInterfaceNames(node));
+    .concat(getDependentInterfaceNames(node, config));
   const recursivelyFoundTypes = namedTypes
     .map((typeName) => schema.getType(typeName)?.astNode)
     .filter(Boolean)

…config/should-exclude-type-definition.ts …config/should-include-type-definition.tssrc/config/should-exclude-type-definition.ts renamed to src/config/should-include-type-definition.ts src/config/should-exclude-type-definition.ts renamed to src/config/should-include-type-definition.ts

@@ -11,12 +11,11 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import { TypeDefinitionNode } from "graphql";
 import { CodegenConfigWithDefaults } from "./build-config-with-defaults";
 
-export function shouldExcludeTypeDefinition(
-  node: TypeDefinitionNode,
+export function shouldIncludeTypeDefinition(
+  typeName: string,
   config: CodegenConfigWithDefaults,
 ) {
-  return config.onlyTypes && !config.onlyTypes.includes(node.name.value);
+  return !config.onlyTypes || config.onlyTypes.includes(typeName);
 }

src/definitions/enum.ts

@@ -14,7 +14,7 @@ limitations under the License.
 import { EnumTypeDefinitionNode, EnumValueDefinitionNode } from "graphql";
 import { indentMultiline } from "@graphql-codegen/visitor-plugin-common";
 import { buildAnnotations } from "../annotations/build-annotations";
-import { shouldExcludeTypeDefinition } from "../config/should-exclude-type-definition";
+import { shouldIncludeTypeDefinition } from "../config/should-include-type-definition";
 import { CodegenConfigWithDefaults } from "../config/build-config-with-defaults";
 import { sanitizeName } from "../utils/sanitize-name";
 import { GraphQLSchema } from "graphql";
@@ -24,7 +24,7 @@ export function buildEnumTypeDefinition(
   schema: GraphQLSchema,
   config: CodegenConfigWithDefaults,
 ) {
-  if (shouldExcludeTypeDefinition(node, config)) {
+  if (!shouldIncludeTypeDefinition(node.name.value, config)) {
     return "";
   }
 

src/definitions/input.ts

@@ -12,7 +12,7 @@ limitations under the License.
 */
 
 import { GraphQLSchema, InputObjectTypeDefinitionNode } from "graphql";
-import { shouldExcludeTypeDefinition } from "../config/should-exclude-type-definition";
+import { shouldIncludeTypeDefinition } from "../config/should-include-type-definition";
 import { buildTypeMetadata } from "../utils/build-type-metadata";
 import { buildAnnotations } from "../annotations/build-annotations";
 import { indent } from "@graphql-codegen/visitor-plugin-common";
@@ -25,7 +25,7 @@ export function buildInputObjectDefinition(
   schema: GraphQLSchema,
   config: CodegenConfigWithDefaults,
 ) {
-  if (shouldExcludeTypeDefinition(node, config)) {
+  if (!shouldIncludeTypeDefinition(node.name.value, config)) {
     return "";
   }
 

src/definitions/interface.ts

@@ -13,7 +13,7 @@ limitations under the License.
 
 import { GraphQLSchema, InterfaceTypeDefinitionNode } from "graphql";
 import { buildAnnotations } from "../annotations/build-annotations";
-import { shouldExcludeTypeDefinition } from "../config/should-exclude-type-definition";
+import { shouldIncludeTypeDefinition } from "../config/should-include-type-definition";
 import { buildInterfaceFieldDefinition } from "./field";
 import { CodegenConfigWithDefaults } from "../config/build-config-with-defaults";
 import { getDependentInterfaceNames } from "../utils/dependent-type-utils";
@@ -24,7 +24,7 @@ export function buildInterfaceDefinition(
   schema: GraphQLSchema,
   config: CodegenConfigWithDefaults,
 ) {
-  if (shouldExcludeTypeDefinition(node, config)) {
+  if (!shouldIncludeTypeDefinition(node.name.value, config)) {
     return "";
   }
 
@@ -45,7 +45,7 @@ export function buildInterfaceDefinition(
     definitionNode: node,
   });
 
-  const interfacesToInherit = getDependentInterfaceNames(node);
+  const interfacesToInherit = getDependentInterfaceNames(node, config);
   const interfaceInheritance = `${interfacesToInherit.length ? ` : ${interfacesToInherit.join(", ")}` : ""}`;
 
   return `${annotations}interface ${sanitizeName(node.name.value)}${interfaceInheritance} {

src/definitions/object.ts

@@ -18,7 +18,7 @@ import {
   ObjectTypeDefinitionNode,
 } from "graphql";
 import { buildAnnotations } from "../annotations/build-annotations";
-import { shouldExcludeTypeDefinition } from "../config/should-exclude-type-definition";
+import { shouldIncludeTypeDefinition } from "../config/should-include-type-definition";
 import {
   getDependentInterfaceNames,
   getDependentUnionsForType,
@@ -37,7 +37,7 @@ export function buildObjectTypeDefinition(
   schema: GraphQLSchema,
   config: CodegenConfigWithDefaults,
 ) {
-  if (shouldExcludeTypeDefinition(node, config)) {
+  if (!shouldIncludeTypeDefinition(node.name.value, config)) {
     return "";
   }
 
@@ -46,8 +46,8 @@ export function buildObjectTypeDefinition(
     config,
     definitionNode: node,
   });
-  const dependentInterfaces = getDependentInterfaceNames(node);
-  const dependentUnions = getDependentUnionsForType(schema, node);
+  const dependentInterfaces = getDependentInterfaceNames(node, config);
+  const dependentUnions = getDependentUnionsForType(schema, node, config);
   const interfacesToInherit =
     config.unionGeneration === "MARKER_INTERFACE"
       ? dependentInterfaces.concat(dependentUnions)

src/definitions/union.ts

@@ -12,7 +12,7 @@ limitations under the License.
 */
 
 import { GraphQLSchema, UnionTypeDefinitionNode } from "graphql";
-import { shouldExcludeTypeDefinition } from "../config/should-exclude-type-definition";
+import { shouldIncludeTypeDefinition } from "../config/should-include-type-definition";
 import { CodegenConfigWithDefaults } from "../config/build-config-with-defaults";
 import {
   buildAnnotations,
@@ -25,7 +25,7 @@ export function buildUnionTypeDefinition(
   schema: GraphQLSchema,
   config: CodegenConfigWithDefaults,
 ) {
-  if (shouldExcludeTypeDefinition(node, config)) {
+  if (!shouldIncludeTypeDefinition(node.name.value, config)) {
     return "";
   }
   const annotations = buildAnnotations({

src/utils/dependent-type-utils.ts

@@ -21,6 +21,7 @@ import {
 } from "graphql";
 import { CodegenConfigWithDefaults } from "../config/build-config-with-defaults";
 import { getBaseTypeNode } from "@graphql-codegen/visitor-plugin-common";
+import { shouldIncludeTypeDefinition } from "../config/should-include-type-definition";
 
 export function getDependentFieldTypeNames(
   node: TypeDefinitionNode,
@@ -41,9 +42,16 @@ function getFieldTypeName(fieldType: TypeNode) {
   return getBaseTypeNode(fieldType).name.value;
 }
 
-export function getDependentInterfaceNames(node: TypeDefinitionNode) {
+export function getDependentInterfaceNames(
+  node: TypeDefinitionNode,
+  config: CodegenConfigWithDefaults,
+) {
   return "interfaces" in node
-    ? (node.interfaces?.map((interfaceNode) => interfaceNode.name.value) ?? [])
+    ? (node.interfaces
+        ?.map((interfaceNode) => interfaceNode.name.value)
+        .filter((interfaceName) =>
+          shouldIncludeTypeDefinition(interfaceName, config),
+        ) ?? [])
     : [];
 }
 
@@ -56,6 +64,7 @@ export function getDependentUnionNames(node: TypeDefinitionNode) {
 export function getDependentUnionsForType(
   schema: GraphQLSchema,
   node: TypeDefinitionNode,
+  config: CodegenConfigWithDefaults,
 ) {
   const typeMap = schema.getTypeMap();
   const unions = Object.values(typeMap).filter((type) =>
@@ -65,5 +74,6 @@ export function getDependentUnionsForType(
     .filter((union) =>
       union.getTypes().some((type) => type.name === node.name.value),
     )
-    .map((union) => union.name);
+    .map((union) => union.name)
+    .filter((unionName) => shouldIncludeTypeDefinition(unionName, config));
 }