Refactor empty string handling across schemas

- Replace direct checks for `notAllowEmptyString` configuration with a new method `shouldEmitAsNotAllowEmptyString` in the `Visitor` class.
- The new method `shouldEmitAsNotAllowEmptyString` encapsulates the check for the `notAllowEmptyString` configuration and the type of the variable, making the code cleaner and more readable.
- The changes have been made across `myzod`, `yup`, and `zod` schemas.

Author: Code-Hex

src/myzod/index.ts

@@ -212,9 +212,8 @@ const generateFieldTypeMyZodSchema = (
     }
     const appliedDirectivesGen = applyDirectives(config, field, gen);
     if (isNonNullType(parentType)) {
-      if (config.notAllowEmptyString === true) {
-        const tsType = visitor.getScalarType(type.name.value);
-        if (tsType === 'string') return `${gen}.min(1)`;
+      if (visitor.shouldEmitAsNotAllowEmptyString(type.name.value)) {
+        return `${gen}.min(1)`;
       }
       return appliedDirectivesGen;
     }

src/visitor.ts

@@ -1,15 +1,19 @@
 import { TsVisitor } from '@graphql-codegen/typescript';
-import { GraphQLSchema, NameNode } from 'graphql';
+import { GraphQLSchema, NameNode, specifiedScalarTypes } from 'graphql';
 
 import { ValidationSchemaPluginConfig } from './config';
 
 export class Visitor extends TsVisitor {
   constructor(
     private scalarDirection: 'input' | 'output' | 'both',
     private schema: GraphQLSchema,
-    config: ValidationSchemaPluginConfig
+    private pluginConfig: ValidationSchemaPluginConfig
   ) {
-    super(schema, config);
+    super(schema, pluginConfig);
+  }
+
+  private isSpecifiedScalarName(scalarName: string) {
+    return specifiedScalarTypes.some(({ name }) => name === scalarName);
   }
 
   public getType(name: string) {
@@ -34,4 +38,16 @@ export class Visitor extends TsVisitor {
     }
     return this.scalars[scalarName][this.scalarDirection];
   }
+
+  public shouldEmitAsNotAllowEmptyString(name: string): boolean {
+    if (this.pluginConfig.notAllowEmptyString !== true) {
+      return false;
+    }
+    const typ = this.getType(name);
+    if (typ?.astNode?.kind !== 'ScalarTypeDefinition' && !this.isSpecifiedScalarName(name)) {
+      return false;
+    }
+    const tsType = this.getScalarType(name);
+    return tsType === 'string';
+  }
 }

src/yup/index.ts

@@ -262,9 +262,8 @@ const generateFieldTypeYupSchema = (
   if (isNamedType(type)) {
     const gen = generateNameNodeYupSchema(config, visitor, type.name);
     if (isNonNullType(parentType)) {
-      if (config.notAllowEmptyString === true) {
-        const tsType = visitor.getScalarType(type.name.value);
-        if (tsType === 'string') return `${gen}.required()`;
+      if (visitor.shouldEmitAsNotAllowEmptyString(type.name.value)) {
+        return `${gen}.required()`;
       }
       return `${gen}.nonNullable()`;
     }

src/zod/index.ts

@@ -222,9 +222,8 @@ const generateFieldTypeZodSchema = (
     }
     const appliedDirectivesGen = applyDirectives(config, field, gen);
     if (isNonNullType(parentType)) {
-      if (config.notAllowEmptyString === true) {
-        const tsType = visitor.getScalarType(type.name.value);
-        if (tsType === 'string') return `${appliedDirectivesGen}.min(1)`;
+      if (visitor.shouldEmitAsNotAllowEmptyString(type.name.value)) {
+        return `${appliedDirectivesGen}.min(1)`;
       }
       return appliedDirectivesGen;
     }