fix: create a custom scalars map with some Hasura specific types

Author: ValentinH

src/TreeToTS/templates/customScalarsMap.ts

@@ -0,0 +1,7 @@
+export const customScalarsMap: Record<string, string> = {
+  uuid: 'string',
+  timestamptz: 'string',
+  json: 'string',
+  jsonb: 'string',
+  numeric: 'number',
+};

src/TreeToTS/templates/resolveValueTypes.ts

@@ -1,6 +1,7 @@
 import { Options, ParserField } from '@/Models';
 import { Helpers, TypeDefinition, TypeSystemDefinition } from '@/Models/Spec';
 import { truthyType } from '@/TreeToTS/templates/truthy';
+import { customScalarsMap } from './customScalarsMap';
 
 export const VALUETYPES = 'ValueTypes';
 
@@ -16,6 +17,7 @@ const typeScriptMap: Record<GqlTypes, TSTypes> = {
   ID: 'string',
   String: 'string',
 };
+
 const toTypeScriptPrimitive = (a: GqlTypes): string => typeScriptMap[a] || a;
 
 const plusDescription = (description?: string, prefix = ''): string =>
@@ -72,6 +74,11 @@ const resolveValueTypeFromRoot = (i: ParserField, rootNodes: ParserField[], enum
     return '';
   }
 
+  if (i.data.type === TypeDefinition.ScalarTypeDefinition) {
+    const type = customScalarsMap[i.name] || 'unknown';
+    return `${plusDescription(i.description)}["${i.name}"]:${type}`;
+  }
+
   if (!i.args || !i.args.length) {
     return `${plusDescription(i.description)}["${i.name}"]:unknown`;
   }

src/TreeToTS/templates/returnedModelTypes.ts

@@ -1,6 +1,7 @@
 import { Options, ParserField } from '@/Models';
 import { TypeDefinition, TypeSystemDefinition } from '@/Models/Spec';
 import { TYPES } from './returnedTypes';
+import { customScalarsMap } from './customScalarsMap';
 
 export const MODEL_TYPES = 'ModelTypes';
 
@@ -11,6 +12,7 @@ const typeScriptMap: Record<string, string> = {
   ID: 'string',
   String: 'string',
 };
+
 const toTypeScriptPrimitive = (a: string): string => typeScriptMap[a] || `${MODEL_TYPES}["${a}"]`;
 
 const plusDescription = (description?: string, prefix = ''): string =>
@@ -57,7 +59,8 @@ const resolveTypeFromRoot = (i: ParserField, rootNodes: ParserField[]): string =
     return '';
   }
   if (i.data.type === TypeDefinition.ScalarTypeDefinition) {
-    return `${plusDescription(i.description)}["${i.name}"]:any`;
+    const type = customScalarsMap[i.name] || 'any';
+    return `${plusDescription(i.description)}["${i.name}"]:${type}`;
   }
   if (!i.args || !i.args.length) {
     return ``;

src/TreeToTS/templates/returnedTypes.ts

@@ -1,5 +1,6 @@
 import { Options, ParserField } from '@/Models';
 import { Helpers, TypeDefinition, TypeSystemDefinition } from '@/Models/Spec';
+import { customScalarsMap } from './customScalarsMap';
 
 export const TYPES = 'GraphQLTypes';
 
@@ -10,6 +11,7 @@ const typeScriptMap: Record<string, string> = {
   ID: 'string',
   String: 'string',
 };
+
 const toTypeScriptPrimitive = (a: string): string => typeScriptMap[a] || `${TYPES}["${a}"]`;
 
 const plusDescription = (description?: string, prefix = ''): string =>
@@ -72,6 +74,10 @@ export const resolveTypeFromRoot = (i: ParserField, rootNodes: ParserField[]): s
   if (i.data.type === Helpers.Comment) {
     return `// ${i.description}`;
   }
+  if (i.data.type === TypeDefinition.ScalarTypeDefinition) {
+    const type = customScalarsMap[i.name] || 'any';
+    return `${plusDescription(i.description)}["${i.name}"]:${type}`;
+  }
   if (!i.args || !i.args.length) {
     return `${plusDescription(i.description)}["${i.name}"]:any`;
   }