From 4c95f48c0418b6315b4701ca84cece361d161767 Mon Sep 17 00:00:00 2001
From: Luc Gagan <lucgagan@gmail.com>
Date: Sat, 11 Nov 2023 22:12:55 -0600
Subject: [PATCH] feat: wrap function response types

---
 src/completeTask.ts | 35 ++++++++++++++++++++++++-----------
 tests/auto.spec.ts  |  9 ++++-----
 2 files changed, 28 insertions(+), 16 deletions(-)

diff --git a/src/completeTask.ts b/src/completeTask.ts
index 3661ebf..53cbd01 100644
--- a/src/completeTask.ts
+++ b/src/completeTask.ts
@@ -65,8 +65,15 @@ export const completeTask = async (
           },
         },
         {
-          function: (args: { attributeName: string; elementId: string }) => {
-            return getLocator(args.elementId).getAttribute(args.attributeName);
+          function: async (args: {
+            attributeName: string;
+            elementId: string;
+          }) => {
+            return {
+              attributeValue: await getLocator(args.elementId).getAttribute(
+                args.attributeName
+              ),
+            };
           },
           name: "locator_getAttribute",
           description: "Returns the matching element's attribute value.",
@@ -91,8 +98,8 @@ export const completeTask = async (
           },
         },
         {
-          function: (args: { elementId: string }) => {
-            return getLocator(args.elementId).innerHTML();
+          function: async (args: { elementId: string }) => {
+            return { innerHTML: await getLocator(args.elementId).innerHTML() };
           },
           name: "locator_innerHTML",
           description: "Returns the element.innerHTML.",
@@ -113,8 +120,8 @@ export const completeTask = async (
           },
         },
         {
-          function: (args: { elementId: string }) => {
-            return getLocator(args.elementId).innerText();
+          function: async (args: { elementId: string }) => {
+            return { innerText: await getLocator(args.elementId).innerText() };
           },
           name: "locator_innerText",
           description: "Returns the element.innerText.",
@@ -159,8 +166,10 @@ export const completeTask = async (
           },
         },
         {
-          function: (args: { elementId: string }) => {
-            return getLocator(args.elementId).inputValue();
+          function: async (args: { elementId: string }) => {
+            return {
+              inputValue: await getLocator(args.elementId).inputValue(),
+            };
           },
           name: "locator_inputValue",
           description:
@@ -183,7 +192,9 @@ export const completeTask = async (
         },
         {
           function: async (args: { elementId: string }) => {
-            return await getLocator(args.elementId).blur();
+            await getLocator(args.elementId).blur();
+
+            return { success: true };
           },
           name: "locator_blur",
           description: "Removes keyboard focus from the current element.",
@@ -467,8 +478,10 @@ export const completeTask = async (
           },
         },
         {
-          function: (args: { url: string }) => {
-            return page.goto(args.url);
+          function: async (args: { url: string }) => {
+            return {
+              url: await page.goto(args.url),
+            };
           },
           name: "page_goto",
           description: "Set a value to the input field.",
diff --git a/tests/auto.spec.ts b/tests/auto.spec.ts
index 5f8e5e7..e507fe9 100644
--- a/tests/auto.spec.ts
+++ b/tests/auto.spec.ts
@@ -22,11 +22,10 @@ test("executes action", async ({ page }) => {
 test("executes click", async ({ page }) => {
   await page.goto("/");
 
-  await auto(
-    "Click the button until the counter value is equal to 2",
-    { page, test },
-    { debug: true }
-  );
+  await auto("Click the button until the counter value is equal to 2", {
+    page,
+    test,
+  });
 
   await expect(page.getByTestId("current-count")).toHaveText("2");
 });