fix: handle map in workflows

packages/cli/src/commands/test.ts

@@ -15,6 +15,7 @@ import {
   defaultWorkflowManifest,
   parseManifestFileOption,
   parseLogFileOption,
+  typesHandler,
 } from "../lib";
 import { createLogger } from "./utils/createLogger";
 
@@ -143,7 +144,10 @@ const _run = async (options: WorkflowCommandOptions) => {
         fs.writeFileSync(outputFile, yaml.stringify(printableOutput, null, 2));
         break;
       case "json":
-        fs.writeFileSync(outputFile, JSON.stringify(printableOutput, null, 2));
+        fs.writeFileSync(
+          outputFile,
+          JSON.stringify(printableOutput, typesHandler, 2)
+        );
         break;
       default:
         throw new Error(

packages/cli/src/lib/helpers/workflow-validator.ts

@@ -14,6 +14,20 @@ export function cueExists(logger: Logger): boolean {
   return stdout ? stdout.startsWith("cue version ") : false;
 }
 
+export const typesHandler = (_: unknown, value: unknown): unknown => {
+  if (value instanceof Map) {
+    return Array.from(value).reduce(
+      (obj: Record<string, unknown>, [key, value]) => {
+        obj[key] = value;
+        return obj;
+      },
+      {}
+    );
+  }
+
+  return value;
+};
+
 export function validateOutput(
   output: WorkflowOutput,
   validateScriptPath: string,
@@ -30,20 +44,22 @@ export function validateOutput(
   const stepId = id.substring(index + 1);
 
   const selector = `${jobId}.\\$${stepId}`;
-  const jsonOutput = `${TMPDIR}/${id}.json`;
+  const tempOutputPath = `${TMPDIR}/${id}.json`;
 
-  fs.writeFileSync(
-    jsonOutput,
-    JSON.stringify({ data, error: error?.message }, null, 2)
+  const outputData = JSON.stringify(
+    { data, error: error?.message },
+    typesHandler,
+    2
   );
+  fs.writeFileSync(tempOutputPath, outputData);
 
   const { stderr } = runCommandSync(
-    `cue vet -d ${selector} ${validateScriptPath} ${jsonOutput}`,
+    `cue vet -d ${selector} ${validateScriptPath} ${tempOutputPath}`,
     logger
   );
 
-  if (fs.existsSync(jsonOutput)) {
-    fs.unlinkSync(jsonOutput);
+  if (fs.existsSync(tempOutputPath)) {
+    fs.unlinkSync(tempOutputPath);
   }
 
   if (!stderr) {

packages/cli/src/lib/workflow/util.ts

@@ -4,6 +4,7 @@ import { WorkflowOutput } from "./types";
 import path from "path";
 import fs from "fs";
 import { WorkflowJobs } from "@polywrap/polywrap-manifest-types-js";
+import { typesHandler } from "../helpers";
 
 export const validateJobNames = (
   jobs: WorkflowJobs | undefined,
@@ -44,7 +45,7 @@ export function printJobOutput(output: WorkflowOutput): void {
   console.log(`Job status: ${output.status}`);
 
   if (output.data !== undefined) {
-    console.log(`Data: ${JSON.stringify(output.data, null, 2)}`);
+    console.log(`Data: ${JSON.stringify(output.data, typesHandler, 2)}`);
   }
 
   if (output.error) {

packages/test-cases/cases/cli/test/003-json-output/polywrap.test.yaml

@@ -31,3 +31,12 @@ jobs:
             args:
               x: "$cases.0.data"
               y: "$cases.0.data"
+      case3:
+        steps:
+          - uri: fs/../run-test-wrapper/build
+            method: returnMap
+            args:
+              map:
+                nested_map:
+                  Hello: 1
+                  Heyo: 50

packages/test-cases/cases/cli/test/run-test-wrapper/src/index.ts

@@ -2,6 +2,7 @@ import {
   Args_add,
   Args_addInBox,
   Args_addFromEnv,
+  Args_returnMap,
   Num,
   Env
 } from "./wrap";
@@ -16,4 +17,8 @@ export function addInBox(args: Args_addInBox): Num {
 
 export function addFromEnv(args: Args_addFromEnv, env: Env): i32 {
   return args.x + env.value;
-}
+}
+
+export function returnMap(args: Args_returnMap): Map<string, Map<string, i32>> {
+  return args.map;
+}

packages/test-cases/cases/cli/test/run-test-wrapper/src/schema.graphql

@@ -12,6 +12,10 @@ type Module {
   addFromEnv(
     x: Int32!
   ): Int32! @env(required: true)
+
+  returnMap(
+    map: Map! @annotate(type: "Map<String!, Map<String!, Int!>!>!")
+  ): Map! @annotate(type: "Map<String!, Map<String!, Int!>!>!")
 }
 
 type Num {