feat: update getNodeSuggestions function to include dataTypes parameter for enhanced functionality

Author: nicosammito

src/suggestion/getNodeSuggestions.ts

@@ -1,13 +1,11 @@
-import {FunctionDefinition, NodeFunction} from "@code0-tech/sagittarius-graphql-types";
+import {DataType, FunctionDefinition, NodeFunction} from "@code0-tech/sagittarius-graphql-types";
 import {createCompilerHost, getSharedTypeDeclarations} from "../utils";
-import {DATA_TYPES} from "../../test/data";
 
 /**
  * Suggests NodeFunctions based on a given type and a list of available FunctionDefinitions.
  * Returns functions whose return type is compatible with the target type.
  */
-//TODO: add dataTypes
-export function getNodeSuggestions(type: string, functions: FunctionDefinition[]): NodeFunction[] {
+export function getNodeSuggestions(type: string, functions: FunctionDefinition[], dataTypes: DataType[]): NodeFunction[] {
     if (!type || !functions || functions.length === 0) {
         return [];
     }
@@ -18,7 +16,7 @@ export function getNodeSuggestions(type: string, functions: FunctionDefinition[]
         return match[1].split(',').map(s => s.trim()).filter(Boolean).length;
     }
 
-    const sharedTypes = getSharedTypeDeclarations(DATA_TYPES);
+    const sharedTypes = getSharedTypeDeclarations(dataTypes);
     const sourceCode = `
         ${sharedTypes}
         type TargetType = ${type};

test/nodeSuggestion.test.ts

@@ -1,12 +1,12 @@
 import {describe, expect, it} from "vitest";
 import {getNodeSuggestions} from "../src/suggestion/getNodeSuggestions";
-import {FUNCTION_SIGNATURES} from "./data";
+import {DATA_TYPES, FUNCTION_SIGNATURES} from "./data";
 
 describe("getNodeSuggestions", () => {
     it("should suggest functions with compatible return types and prioritize exact matches", () => {
 
         // We are looking for suggestions for a 'number' type
-        const suggestions = getNodeSuggestions("STRING", FUNCTION_SIGNATURES);
+        const suggestions = getNodeSuggestions("STRING", FUNCTION_SIGNATURES, DATA_TYPES);
 
         const map = suggestions.map(s => s.functionDefinition?.id)