47 lines
1.3 KiB
TypeScript
47 lines
1.3 KiB
TypeScript
import { logger } from "@zerobyte/core/node";
|
|
import { createControllerSession, type ControllerSession } from "./controller-session";
|
|
|
|
const controllerUrl = process.env.ZEROBYTE_CONTROLLER_URL;
|
|
const agentToken = process.env.ZEROBYTE_AGENT_TOKEN;
|
|
|
|
class Agent {
|
|
private ws: WebSocket | null = null;
|
|
private controllerSession: ControllerSession | null = null;
|
|
|
|
connect() {
|
|
if (!controllerUrl) {
|
|
throw new Error("Env variable ZEROBYTE_CONTROLLER_URL is not set");
|
|
}
|
|
|
|
if (!agentToken) {
|
|
throw new Error("Env variable ZEROBYTE_AGENT_TOKEN is not set");
|
|
}
|
|
|
|
const url = new URL(controllerUrl);
|
|
url.searchParams.set("token", agentToken);
|
|
|
|
this.ws = new WebSocket(url.toString());
|
|
this.controllerSession = createControllerSession(this.ws);
|
|
|
|
this.ws.onopen = () => {
|
|
logger.info("Agent connected to controller");
|
|
this.controllerSession?.onOpen();
|
|
};
|
|
|
|
this.ws.onmessage = (event) => {
|
|
this.controllerSession?.onMessage(event.data);
|
|
};
|
|
this.ws.onclose = () => {
|
|
this.controllerSession?.close();
|
|
this.controllerSession = null;
|
|
this.ws = null;
|
|
logger.info("Agent disconnected from controller");
|
|
};
|
|
this.ws.onerror = (error) => {
|
|
logger.error("Agent encountered an error:", error);
|
|
};
|
|
}
|
|
}
|
|
|
|
const agent = new Agent();
|
|
agent.connect();
|