import { logger } from "../../lib/logger";
import { isClass } from "../../utils/is-class";
import { SignalEmitter } from "../emitter";
import { SignalGlobalSettings } from "../signals";
import { SignalGenerator, SignalGeneratorClass } from "./types";

export const registerGenerator = async (
  emitter: SignalEmitter,
  signalGenerators: (SignalGeneratorClass | SignalGenerator)[],
  settings: SignalGlobalSettings,
): Promise<VoidFunction> => {
  const _register = (gen: SignalGeneratorClass | SignalGenerator) => {
    logger.debug("Registering generator:", gen.id || (gen as any).name);
    if (isClass(gen)) {
      // Check if Gen is a function and has a constructor
      return new gen(settings).register(emitter);
    } else {
      return gen.register(emitter);
    }
  };

  const cleanupFns = await Promise.all(signalGenerators.map(_register));

  // Return a cleanup function that calls all the cleanup functions (e.g unsubscribes from event listeners)
  return () => cleanupFns.forEach((fn) => fn());
};