adding tester
This commit is contained in:
@@ -0,0 +1,60 @@
|
||||
import axios from "axios";
|
||||
|
||||
const agentid = process.env.AGENT_ID;
|
||||
|
||||
axios.defaults.headers.common["X-Phantombuster-Key"] = process.env
|
||||
.API_KEY as string;
|
||||
axios.defaults.headers.common["Content-Type"] = "application/json";
|
||||
|
||||
const MAX_TIME_TO_RUN = 10000;
|
||||
|
||||
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;
|
||||
|
||||
if (timeRunning > MAX_TIME_TO_RUN) {
|
||||
throw new Error("TIMEOUT: phantom run for too long");
|
||||
}
|
||||
|
||||
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()
|
||||
)
|
||||
.then(console.log)
|
||||
.catch(async (err) => {
|
||||
console.error(err);
|
||||
console.log("stopping agent");
|
||||
await axios.post("https://api.phantombuster.com/api/v2/agents/stop", {
|
||||
id: agentid,
|
||||
});
|
||||
process.exit();
|
||||
})
|
||||
);
|
||||
Reference in New Issue
Block a user