Refactor query execution logic to return script results and handle errors properly

src/query/executer/query.executer.service.ts

@@ -22,16 +22,9 @@ export class QueryExecuterService {
     }
 
     const vm = this.createVm(query);
+    const result = await vm.runScript(query.source);
 
-    vm.setFunction("result", (result: string) => {
+    return { message: "Query executed", result, query: queryData };
-      console.log("Query Result:", result);
-    });
-
-    await vm.runScript(query.source);
-
-    // Here you would add the logic to actually execute the query
-    // against your database or data source. This is a placeholder.
-    return { message: "Query executed", query: queryData };
   }
 
   private createVm(query: Query) {

src/vm/vm.class.ts

@@ -29,7 +29,41 @@ export class Vm {
   }
 
   async runScript(script: string) {
-    const compiledScript = await this.context.isolate.compileScript(script);
+    let resolvePromise: (value: any) => void;
-    return compiledScript.run(this.context);
+    let rejectPromise: (reason?: any) => void;
+
+    const resultPromise = new Promise<any>((resolve, reject) => {
+      resolvePromise = resolve;
+      rejectPromise = reject;
+    });
+
+    this.setFunction("result", (...args) => {
+      console.log("Script result:", args);
+      resolvePromise(args);
+    });
+
+    this.setFunction("error", (error: any) => {
+      console.error("Script error:", error);
+      rejectPromise(error);
+    });
+
+    const scriptWithResult = `
+        (async () => {
+            ${script}
+            try {
+                const result = await main()
+                result(result)
+            } catch (e) {
+                error(e)
+            }
+        })()
+    `;
+
+    const compiledScript = await this.context.isolate.compileScript(
+      scriptWithResult
+    );
+    compiledScript.run(this.context);
+
+    return await resultPromise;
   }
 }