feat: add enumsPrefix option (#56)

Similar to `typesPrefix`, the option `enumsPrefix` is used to add a prefix to all enum type usages

Also add union type in unit tests to verify the behaviour

Fixes #51

Author: ardeois

README.md

@@ -105,6 +105,26 @@ Setting the `typesPrefix` to `Api.` will create the following mock data
 export const aUser = (overrides?: Partial<Api.User>): Api.User => {
 ```
 
+### enumsPrefix (`string`, defaultValue: '')
+
+Similar to `typesPrefix`, but for enum types
+
+```
+declare namespace Api {
+ enum Status {
+  ...
+ }
+}
+```
+
+Setting the `enumsPrefix` to `Api.` will create the following mock data
+
+```
+export const aUser = (overrides?: Partial<User>): User => {
+   status: overrides && overrides.hasOwnProperty('status') ? overrides.status! : Api.Status.Online,
+}
+```
+
 ## Example of usage
 
 **codegen.yml**

src/index.ts

@@ -17,6 +17,7 @@ type Options<T = TypeNode> = {
     terminateCircularRelationships: boolean;
     prefix: string | undefined;
     typesPrefix: string;
+    enumsPrefix: string;
     currentType: T;
     customScalars?: ScalarMap;
 };
@@ -108,7 +109,7 @@ const getNamedType = (opts: Options<NamedTypeNode>): string | number | boolean =
                         // It's an enum
                         const typenameConverter = createNameConverter(opts.typenamesConvention);
                         const value = foundType.values ? foundType.values[0] : '';
-                        return `${typenameConverter(foundType.name)}.${updateTextCase(
+                        return `${typenameConverter(foundType.name, opts.enumsPrefix)}.${updateTextCase(
                             value,
                             opts.enumValuesConvention,
                         )}`;
@@ -242,23 +243,34 @@ const getImportTypes = ({
     types,
     typesFile,
     typesPrefix,
+    enumsPrefix,
 }: {
     typenamesConvention: NamingConvention;
     definitions: any;
     types: TypeItem[];
     typesFile: string;
     typesPrefix: string;
+    enumsPrefix: string;
 }) => {
     const typenameConverter = createNameConverter(typenamesConvention);
-    const enumTypes = types.filter(({ type }) => type === 'enum');
-    const typeImports = definitions
-        .filter(({ typeName }: { typeName: string }) => !!typeName)
-        .map(({ typeName }: { typeName: string }) => typenameConverter(typeName, typesPrefix));
+    const typeImports = typesPrefix?.endsWith('.')
+        ? [typesPrefix.slice(0, -1)]
+        : definitions
+              .filter(({ typeName }: { typeName: string }) => !!typeName)
+              .map(({ typeName }: { typeName: string }) => typenameConverter(typeName, typesPrefix));
+    const enumTypes = enumsPrefix?.endsWith('.')
+        ? [enumsPrefix.slice(0, -1)]
+        : types.filter(({ type }) => type === 'enum').map(({ name }) => typenameConverter(name, enumsPrefix));
+
+    typeImports.push(...enumTypes);
+
+    function onlyUnique(value, index, self) {
+        return self.indexOf(value) === index;
+    }
 
-    typeImports.push(...enumTypes.map(({ name }) => typenameConverter(name)));
     return typesFile
         ? `/* eslint-disable @typescript-eslint/no-use-before-define,@typescript-eslint/no-unused-vars,no-prototype-builtins */
-import { ${typeImports.join(', ')} } from '${typesFile}';\n`
+import { ${typeImports.filter(onlyUnique).join(', ')} } from '${typesFile}';\n`
         : '';
 };
 
@@ -281,6 +293,7 @@ export interface TypescriptMocksPluginConfig {
     scalars?: ScalarMap;
     terminateCircularRelationships?: boolean;
     typesPrefix?: string;
+    enumsPrefix?: string;
 }
 
 interface TypeItem {
@@ -339,6 +352,7 @@ export const plugin: PluginFunction<TypescriptMocksPluginConfig> = (schema, docu
                         terminateCircularRelationships: !!config.terminateCircularRelationships,
                         prefix: config.prefix,
                         typesPrefix: config.typesPrefix,
+                        enumsPrefix: config.enumsPrefix,
                         currentType: node.type,
                         customScalars: config.scalars,
                     });
@@ -365,6 +379,7 @@ export const plugin: PluginFunction<TypescriptMocksPluginConfig> = (schema, docu
                                       terminateCircularRelationships: !!config.terminateCircularRelationships,
                                       prefix: config.prefix,
                                       typesPrefix: config.typesPrefix,
+                                      enumsPrefix: config.enumsPrefix,
                                       currentType: field.type,
                                       customScalars: config.scalars,
                                   });
@@ -453,6 +468,7 @@ export const plugin: PluginFunction<TypescriptMocksPluginConfig> = (schema, docu
         types,
         typesFile,
         typesPrefix: config.typesPrefix,
+        enumsPrefix: config.enumsPrefix,
     });
     // List of function that will generate the mock.
     // We generate it after having visited because we need to distinct types from enums