agency-lang 0.0.21 → 0.0.22

package/dist/lib/parsers/function.test.js

@@ -1,5 +1,5 @@
 import { describe, it, expect } from "vitest";
-import { functionParser, docStringParser, graphNodeParser } from "./function.js";
+import { functionParser, docStringParser, graphNodeParser, timeBlockParser, } from "./function.js";
 describe("docStringParser", () => {
     const testCases = [
         // Happy path - basic docstrings
@@ -175,6 +175,9 @@ describe("functionParser", () => {
                             variableName: "x",
                             value: { type: "number", value: "5" },
                         },
+                        {
+                            type: "newLine",
+                        },
                         {
                             type: "assignment",
                             variableName: "y",
@@ -254,11 +257,17 @@ describe("functionParser", () => {
                             variableName: "x",
                             value: { type: "number", value: "5" },
                         },
+                        {
+                            type: "newLine",
+                        },
                         {
                             type: "assignment",
                             variableName: "y",
                             value: { type: "number", value: "10" },
                         },
+                        {
+                            type: "newLine",
+                        },
                     ],
                 },
             },
@@ -333,6 +342,9 @@ describe("functionParser", () => {
                                 segments: [{ type: "text", value: "say hello" }],
                             },
                         },
+                        {
+                            type: "newLine",
+                        },
                     ],
                 },
             },
@@ -356,6 +368,9 @@ describe("functionParser", () => {
                             variableName: "x",
                             value: { type: "number", value: "5" },
                         },
+                        {
+                            type: "newLine",
+                        },
                         {
                             type: "assignment",
                             variableName: "y",
@@ -401,6 +416,9 @@ describe("functionParser", () => {
                             variableName: "x",
                             value: { type: "number", value: "5" },
                         },
+                        {
+                            type: "newLine",
+                        },
                     ],
                 },
             },
@@ -494,6 +512,9 @@ describe("functionParser", () => {
                             variableName: "result",
                             value: { type: "number", value: "42" },
                         },
+                        {
+                            type: "newLine",
+                        },
                         {
                             type: "variableName",
                             value: "result",
@@ -572,43 +593,6 @@ describe("functionParser", () => {
                 },
             },
         },
-        {
-            input: 'def greet(name, greeting) {\n  """Greets someone with a custom greeting"""\n  result = `${greeting} ${name}`\n  result\n}',
-            expected: {
-                success: true,
-                result: {
-                    type: "function",
-                    functionName: "greet",
-                    parameters: [
-                        { type: "functionParameter", name: "name" },
-                        { type: "functionParameter", name: "greeting" },
-                    ],
-                    returnType: null,
-                    docString: {
-                        type: "docString",
-                        value: "Greets someone with a custom greeting",
-                    },
-                    body: [
-                        {
-                            type: "assignment",
-                            variableName: "result",
-                            value: {
-                                type: "prompt",
-                                segments: [
-                                    { type: "interpolation", variableName: "greeting" },
-                                    { type: "text", value: " " },
-                                    { type: "interpolation", variableName: "name" },
-                                ],
-                            },
-                        },
-                        {
-                            type: "variableName",
-                            value: "result",
-                        },
-                    ],
-                },
-            },
-        },
         // Functions with single parameter with type hint
         {
             input: "def add(x: number) { x }",
@@ -1112,7 +1096,7 @@ describe("functionParser", () => {
                         type: "docString",
                         value: "Adds two numbers",
                     },
-                    body: [{ type: "variableName", value: "x" }],
+                    body: [{ type: "variableName", value: "x" }, { type: "newLine" }],
                 },
             },
         },
@@ -1135,7 +1119,7 @@ describe("functionParser", () => {
                         type: "docString",
                         value: "Greets a person by name",
                     },
-                    body: [{ type: "variableName", value: "name" }],
+                    body: [{ type: "variableName", value: "name" }, { type: "newLine" }],
                 },
             },
         },
@@ -1159,7 +1143,7 @@ describe("functionParser", () => {
                         type: "docString",
                         value: "Mix of typed and untyped params",
                     },
-                    body: [{ type: "variableName", value: "typed" }],
+                    body: [{ type: "variableName", value: "typed" }, { type: "newLine" }],
                 },
             },
         },
@@ -1502,7 +1486,7 @@ describe("functionParser", () => {
                         type: "docString",
                         value: "Adds two numbers",
                     },
-                    body: [{ type: "variableName", value: "x" }],
+                    body: [{ type: "variableName", value: "x" }, { type: "newLine" }],
                 },
             },
         },
@@ -1525,7 +1509,7 @@ describe("functionParser", () => {
                         type: "docString",
                         value: "Greets a person",
                     },
-                    body: [{ type: "variableName", value: "name" }],
+                    body: [{ type: "variableName", value: "name" }, { type: "newLine" }],
                 },
             },
         },
@@ -1668,6 +1652,248 @@ describe("functionParser", () => {
         }
     });
 });
+describe("timeBlockParser", () => {
+    const testCases = [
+        // Happy path - basic time blocks
+        {
+            input: "time {\n  x = 1\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "x",
+                            value: { type: "number", value: "1" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        {
+            input: "time {\n  foo = `hello`\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "foo",
+                            value: {
+                                type: "prompt",
+                                segments: [{ type: "text", value: "hello" }],
+                            },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        // Multiple statements in body
+        {
+            input: "time {\n  a = 1\n  b = 2\n  c = 3\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "a",
+                            value: { type: "number", value: "1" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                        {
+                            type: "assignment",
+                            variableName: "b",
+                            value: { type: "number", value: "2" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                        {
+                            type: "assignment",
+                            variableName: "c",
+                            value: { type: "number", value: "3" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        // Empty body
+        {
+            input: "time {\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [],
+                },
+            },
+        },
+        // Whitespace variations
+        {
+            input: "time{\n  x = 1\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "x",
+                            value: { type: "number", value: "1" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        {
+            input: "time  {  \n  x = 1\n  }",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "x",
+                            value: { type: "number", value: "1" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        // Function calls in body
+        {
+            input: "time {\n  result = doSomething()\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "result",
+                            value: {
+                                type: "functionCall",
+                                functionName: "doSomething",
+                                arguments: [],
+                            },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        // Complex body with different statement types
+        {
+            input: "time {\n  x = 1\n  y = `prompt`\n  z = foo()\n}",
+            expected: {
+                success: true,
+                result: {
+                    type: "timeBlock",
+                    body: [
+                        {
+                            type: "assignment",
+                            variableName: "x",
+                            value: { type: "number", value: "1" },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                        {
+                            type: "assignment",
+                            variableName: "y",
+                            value: {
+                                type: "prompt",
+                                segments: [{ type: "text", value: "prompt" }],
+                            },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                        {
+                            type: "assignment",
+                            variableName: "z",
+                            value: {
+                                type: "functionCall",
+                                functionName: "foo",
+                                arguments: [],
+                            },
+                        },
+                        {
+                            type: "newLine",
+                        },
+                    ],
+                },
+            },
+        },
+        // Failure cases - missing braces
+        {
+            input: "time \n  x = 1\n",
+            expected: { success: false },
+        },
+        {
+            input: "time {\n  x = 1",
+            expected: { success: false },
+        },
+        {
+            input: "time \n  x = 1\n}",
+            expected: { success: false },
+        },
+        // Failure cases - empty or malformed
+        {
+            input: "",
+            expected: { success: false },
+        },
+        {
+            input: "time",
+            expected: { success: false },
+        },
+        {
+            input: "time {",
+            expected: { success: false },
+        },
+    ];
+    testCases.forEach(({ input, expected }) => {
+        if (expected.success) {
+            it(`should parse "${input.replace(/\n/g, "\\n")}" successfully`, () => {
+                const result = timeBlockParser(input);
+                expect(result.success).toBe(true);
+                if (result.success) {
+                    expect(result.result).toEqual(expected.result);
+                }
+            });
+        }
+        else {
+            it(`should fail to parse "${input.replace(/\n/g, "\\n")}"`, () => {
+                const result = timeBlockParser(input);
+                expect(result.success).toBe(false);
+            });
+        }
+    });
+});
 describe("graphNodeParser", () => {
     const testCases = [
         // Basic graph nodes without return types
@@ -1733,7 +1959,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "process",
-                    parameters: ["input"],
+                    parameters: [{ type: "functionParameter", name: "input" }],
                     returnType: null,
                     body: [
                         {
@@ -1752,7 +1978,10 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "calculate",
-                    parameters: ["a", "b"],
+                    parameters: [
+                        { type: "functionParameter", name: "a" },
+                        { type: "functionParameter", name: "b" },
+                    ],
                     returnType: null,
                     body: [
                         {
@@ -1812,7 +2041,10 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "add",
-                    parameters: ["x", "y"],
+                    parameters: [
+                        { type: "functionParameter", name: "x" },
+                        { type: "functionParameter", name: "y" },
+                    ],
                     returnType: { type: "primitiveType", value: "number" },
                     body: [
                         {
@@ -1831,7 +2063,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "greet",
-                    parameters: ["name"],
+                    parameters: [{ type: "functionParameter", name: "name" }],
                     returnType: { type: "primitiveType", value: "string" },
                     body: [
                         {
@@ -1853,7 +2085,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "process",
-                    parameters: ["input"],
+                    parameters: [{ type: "functionParameter", name: "input" }],
                     returnType: { type: "primitiveType", value: "boolean" },
                     body: [{ type: "variableName", value: "result" }],
                 },
@@ -1899,7 +2131,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "processArray",
-                    parameters: ["items"],
+                    parameters: [{ type: "functionParameter", name: "items" }],
                     returnType: {
                         type: "arrayType",
                         elementType: { type: "primitiveType", value: "number" },
@@ -1935,7 +2167,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "flexible",
-                    parameters: ["x"],
+                    parameters: [{ type: "functionParameter", name: "x" }],
                     returnType: {
                         type: "unionType",
                         types: [
@@ -1982,7 +2214,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "baz",
-                    parameters: ["x"],
+                    parameters: [{ type: "functionParameter", name: "x" }],
                     returnType: { type: "primitiveType", value: "number" },
                     body: [{ type: "variableName", value: "x" }],
                 },
@@ -2009,7 +2241,7 @@ describe("graphNodeParser", () => {
                 result: {
                     type: "graphNode",
                     nodeName: "getData",
-                    parameters: ["id"],
+                    parameters: [{ type: "functionParameter", name: "id" }],
                     returnType: { type: "typeAliasVariable", aliasName: "UserData" },
                     body: [{ type: "variableName", value: "data" }],
                 },

package/dist/lib/parsers/functionCall.test.js

@@ -152,8 +152,8 @@ describe("functionCallParser", () => {
                         {
                             type: "agencyArray",
                             items: [
-                                { type: "string", value: "hello" },
-                                { type: "string", value: "world" },
+                                { type: "string", segments: [{ type: "text", value: "hello" }] },
+                                { type: "string", segments: [{ type: "text", value: "world" }] },
                             ],
                         },
                     ],
@@ -221,7 +221,7 @@ describe("functionCallParser", () => {
                             entries: [
                                 {
                                     key: "key",
-                                    value: { type: "string", value: "value" },
+                                    value: { type: "string", segments: [{ type: "text", value: "value" }] },
                                 },
                             ],
                         },
@@ -242,7 +242,7 @@ describe("functionCallParser", () => {
                             entries: [
                                 {
                                     key: "name",
-                                    value: { type: "string", value: "Alice" },
+                                    value: { type: "string", segments: [{ type: "text", value: "Alice" }] },
                                 },
                                 {
                                     key: "age",
@@ -349,7 +349,7 @@ describe("functionCallParser", () => {
                             entries: [
                                 {
                                     key: "key",
-                                    value: { type: "string", value: "value" },
+                                    value: { type: "string", segments: [{ type: "text", value: "value" }] },
                                 },
                             ],
                         },
@@ -365,7 +365,7 @@ describe("functionCallParser", () => {
                     type: "functionCall",
                     functionName: "complexCall",
                     arguments: [
-                        { type: "string", value: "test" },
+                        { type: "string", segments: [{ type: "text", value: "test" }] },
                         {
                             type: "agencyArray",
                             items: [],

package/dist/lib/parsers/ifElse.test.d.ts

@@ -0,0 +1 @@
+export {};