adding test + refacto

scripts/test_online.ts

@@ -1,60 +1,88 @@
+import "dotenv/config";
 import axios from "axios";
+import prompts from "prompts";
 
-const agentid = process.env.AGENT_ID;
+(async () => {
+  const agentid =
+    process.env.AGENT_ID ??
+    (
+      await prompts({
+        type: "text",
+        name: "agentId",
+        message: "enter your agent id",
+      })
+    ).agentId;
 
-axios.defaults.headers.common["X-Phantombuster-Key"] = process.env
-  .API_KEY as string;
-axios.defaults.headers.common["Content-Type"] = "application/json";
+  const requester = axios.create({
+    headers: {
+      ["X-Phantombuster-Key"]:
+        process.env.API_KEY ??
+        (
+          await prompts({
+            type: "text",
+            name: "API_KEY",
+            message: "enter your API key",
+          })
+        ).API_KEY,
+      ["Content-Type"]: "application/json",
+    },
+  });
 
-const MAX_TIME_TO_RUN = 10000;
+  const MAX_TIME_TO_RUN = 10000;
 
-type FetchOutputResult = {
-  containerId: string;
-  status: string;
-  output: string;
-  outputPos: number;
-  mostRecentEndedAt: number;
-  isAgentRunning: boolean;
-  canSoftAbort: boolean;
-};
+  type FetchOutputResult = {
+    containerId: string;
+    status: string;
+    output: string;
+    outputPos: number;
+    mostRecentEndedAt: number;
+    isAgentRunning: boolean;
+    canSoftAbort: boolean;
+  };
 
-async function getUntilRunStop(
-  url: string,
-  startTime: number
-): Promise<FetchOutputResult> {
-  const timeRunning = Date.now() - startTime;
+  async function getUntilRunStop(
+    url: string,
+    startTime: number
+  ): Promise<FetchOutputResult> {
+    const timeRunning = Date.now() - startTime;
 
-  if (timeRunning > MAX_TIME_TO_RUN) {
-    throw new Error("TIMEOUT: phantom run for too long");
+    if (timeRunning > MAX_TIME_TO_RUN) {
+      throw new Error("TIMEOUT: phantom run for too long");
+    }
+
+    return await requester
+      .get<FetchOutputResult>(url)
+      .then((res) =>
+        res.data.status !== "finished"
+          ? getUntilRunStop(url, startTime)
+          : res.data
+      );
   }
 
-  return await axios
-    .get<FetchOutputResult>(url)
-    .then((res) =>
-      res.data.status !== "finished"
-        ? getUntilRunStop(url, startTime)
-        : res.data
-    );
-}
-
-axios
-  .post<{ containerId: string }>(
-    "https://api.phantombuster.com/api/v2/agents/launch",
-    { id: agentid }
-  )
-  .then((res) => res.data.containerId)
-  .then(() =>
-    getUntilRunStop(
-      `https://api.phantombuster.com/api/v2/agents/fetch-output?id=${agentid}`,
-      Date.now()
+  await requester
+    .post<{ containerId: string }>(
+      "https://api.phantombuster.com/api/v2/agents/launch",
+      { id: agentid }
     )
-      .then(console.log)
-      .catch(async (err) => {
+    .then((res) => res.data.containerId)
+    .then(async () => {
+      try {
+        await getUntilRunStop(
+          `https://api.phantombuster.com/api/v2/agents/fetch-output?id=${agentid}`,
+          Date.now()
+        )
+          .then((res) => res.output)
+          .then(console.log);
+      } catch (err) {
         console.error(err);
         console.log("stopping agent");
-        await axios.post("https://api.phantombuster.com/api/v2/agents/stop", {
-          id: agentid,
-        });
-        process.exit();
-      })
-  );
+        await requester.post(
+          "https://api.phantombuster.com/api/v2/agents/stop",
+          {
+            id: agentid,
+          }
+        );
+      }
+    })
+    .catch((err) => console.error(err.data, agentid));
+})();