import { Async, Log } from "../deps.ts";
import { getGlobalSession } from "../bot/session.ts";
import { sdGetConfig } from "../sd.ts";

const logger = () => Log.getLogger();

export const runningWorkers = new Set<string>();

/**
 * Periodically ping the workers to see if they are alive.
 */
export async function pingWorkers(): Promise<never> {
  while (true) {
    try {
      const config = await getGlobalSession();
      for (const worker of config.workers) {
        const status = await sdGetConfig(worker.api).catch(() => null);
        const wasRunning = runningWorkers.has(worker.name);
        if (status) {
          runningWorkers.add(worker.name);
          if (!wasRunning) logger().info(`Worker ${worker.name} is online`);
        } else {
          runningWorkers.delete(worker.name);
          if (wasRunning) logger().warning(`Worker ${worker.name} went offline`);
        }
      }
      await Async.delay(60 * 1000);
    } catch (err) {
      logger().warning(`Pinging workers failed: ${err}`);
    }
  }
}