import {
  UnifiApiClient,
  UnifiCameraHandler,
  UnifiImageHandler,
  ConsoleLogger,
} from "@komponent/unifi-protect-lib";

let retries = 0;

const start = async () => {
  try {
    const apiClient = new UnifiApiClient(
      ConsoleLogger, // You can use your own logger if you want, just implement the Logger interface
      process.env.CLOUD_KEY_HOST,
      process.env.CLOUD_KEY_USER,
      process.env.CLOUD_KEY_PASSWORD
    );

    const onDoorbellEvent = async (camera: ProtectCameraConfig) => {
      const snapshot = await imageHandler.getSnapshot({
        camera: camera,
      });

      if (snapshot) {
        // TODO: Do your own thing
      }
    };

    const onMotionEvent = async (camera: ProtectCameraConfig) => {
      const snapshot = await imageHandler.getSnapshot({
        camera: camera,
      });

      if (snapshot) {
        // TODO: Do your own thing
      }
    };

    const imageHandler = new UnifiImageHandler(ConsoleLogger, apiClient);
    const cameraHandler = new UnifiCameraHandler(ConsoleLogger, apiClient, {
      refreshInterval: 5,
      onDoorbellEvent: onDoorbellEvent,
      onMotionEvent: onMotionEvent,
    });

    await apiClient.init();

    // Print current status
    apiClient.status();

    const eventListener = await apiClient.listen();

    // Run on each camera websocket update
    eventListener?.on("message", (data: Buffer) =>
      cameraHandler.handleUpdate(data)
    );

    // Restart on close
    eventListener?.on("close", () => {
      if (retries < 5) {
        start();
        retries = 0;
      } else {
        console.error("Too many retries. Closing Websocket for good");
      }
    });
  } catch (error) {
    console.error(error);
  }
};

(async () => {
  await start();
})();