supported import path

Author: Code-Hex

.gitignore

@@ -1,6 +1,6 @@
 node_modules
 dist
 .npmignore
-gen/schemas.ts
+gen/*.ts
 package-lock.json
 .DS_Store

codegen.yml

@@ -1,8 +1,11 @@
 overwrite: true
 schema: "./test.graphql"
 generates:
-  gen/schemas.ts:
+  gen/types.ts:
     plugins:
       - typescript
+  gen/schemas.ts:
+    plugins:
       - ./dist/index.js:
-          schema: yup
+          schema: yup
+          importFrom: ./types

src/index.ts

@@ -11,14 +11,32 @@ export const plugin: PluginFunction<ValidationSchemaPluginConfig> = async (
 ): Promise<Types.PluginOutput> => {
   const { inputObjects, enums, scalars } = retrieveSchema(schema, config);
 
+  const prepend = [importSchema(config.schema)];
+  if (config.importFrom) {
+    prepend.push(
+      buildImportStatement(
+        [
+          // Should put on Scalar here?
+          ...Object.values(enums).map((enumdef) => enumdef.name.value),
+          ...inputObjects.map((inputObject) => inputObject.name.value),
+        ],
+        config.importFrom
+      )
+    );
+  }
   return {
-    prepend: [importSchema(config.schema)],
-    content: [
-      new YupGenerator({ inputObjects, enums, scalars }).generate(),
-    ].join("\n"),
+    prepend,
+    content:
+      "\n" +
+      [new YupGenerator({ inputObjects, enums, scalars }).generate()].join(
+        "\n"
+      ),
   };
 };
 
+const buildImportStatement = (types: string[], importFrom: string): string =>
+  `import { ${types.join(", ")} } from "${importFrom}";`;
+
 const importSchema = (schema?: ValidationSchema): string => {
   if (schema === "yup") {
     return ImportYup();

src/types.ts

@@ -24,6 +24,26 @@ export interface ValidationSchemaPluginConfig extends TypeScriptPluginConfig {
    * ```
    */
   schema?: ValidationSchema;
+
+  /**
+   * @description import types from generated typescript type path
+   * if not given, omit import statement.
+   *
+   * @exampeMarkdown
+   * ```yml
+   * generates:
+   *   path/to/types.ts:
+   *     plugins:
+   *       - typescript
+   *   path/to/schemas.ts:
+   *     plugins:
+   *       - graphql-codegen-validation-schema
+   *     config:
+   *       schema: yup
+   *       importFrom: ./path/to/types
+   * ```
+   */
+  importFrom?: string;
 }
 
 export interface Nodes {