Merge pull request #17 from code0-tech/feat/#16

Value suggestion should also work with dataTypes

Author: Nico Sammito

src/suggestion/getValueSuggestions.ts

@@ -1,14 +1,20 @@
 import ts from "typescript";
-import {LiteralValue} from "@code0-tech/sagittarius-graphql-types";
-import {createCompilerHost} from "../utils";
+import {DataType, LiteralValue} from "@code0-tech/sagittarius-graphql-types";
+import {createCompilerHost, getSharedTypeDeclarations} from "../utils";
 
 /**
  * Extracts possible literal values from a type string to provide suggestions.
  */
-export const getValueSuggestions = (type: string): LiteralValue[] => {
+export const getValueSuggestions = (
+    type?: string,
+    dataTypes?: DataType[]
+): LiteralValue[] => {
     if (!type) return [];
 
-    const sourceCode = `type T = ${type}; const val: T = {} as any;`;
+    const sourceCode = `
+        ${getSharedTypeDeclarations(dataTypes)}
+        type T = ${type}; const val: T = {} as any;
+    `;
     const fileName = "index.ts";
     const host = createCompilerHost(fileName, sourceCode);
     const sourceFile = host.getSourceFile(fileName)!;

src/utils.ts

@@ -75,12 +75,12 @@ export const DEFAULT_COMPILER_OPTIONS: ts.CompilerOptions = {
 /**
  * Extracts and returns common type and generic declarations from DATA_TYPES.
  */
-export function getSharedTypeDeclarations(dataTypes: DataType[]): string {
-    const genericDeclarations = Array.from(new Set(dataTypes.flatMap(dt => dt.genericKeys || [])))
+export function getSharedTypeDeclarations(dataTypes?: DataType[]): string {
+    const genericDeclarations = Array.from(new Set(dataTypes?.flatMap(dt => dt.genericKeys || [])))
         .map(g => `type ${g} = any;`)
         .join("\n");
 
-    const typeAliasDeclarations = dataTypes.map(dt =>
+    const typeAliasDeclarations = dataTypes?.map(dt =>
         `type ${dt.identifier}${dt.genericKeys ? `<${dt.genericKeys.join(",")}>` : ""} = ${dt.type};`
     ).join("\n");