import * as plugins from './plugins.js';
import { AgentProxy } from './classes.agentproxy.js';
import { DeviceProxy } from './classes.deviceproxy.js';

export type TAutomationEvent = {
  triggerId: string;
  kind: plugins.shxInterfaces.data.TAutomationTriggerKind;
  payload: Record<string, unknown>;
};

export type TAutomationHandler = (
  eventArg: TAutomationEvent,
  contextArg: ShxAutomationContext
) => Promise<void> | void;

export type TToolPlanExecutor = (
  planArg: plugins.shxInterfaces.data.IToolPlan
) => Promise<plugins.shxInterfaces.data.IToolPlanResult>;

export interface IShxAutomationContextOptions {
  callerId: string;
  executePlan?: TToolPlanExecutor;
}

export interface IAutomationRegistration {
  trigger: plugins.shxInterfaces.data.IAutomationTrigger;
  handler: TAutomationHandler;
}

export class ShxAutomationContext {
  public devices = new DeviceProxy(this);
  public agents = new AgentProxy(this);
  private registrations: IAutomationRegistration[] = [];

  constructor(public options: IShxAutomationContextOptions) {}

  public on(
    triggerArg: plugins.shxInterfaces.data.IAutomationTrigger,
    handlerArg: TAutomationHandler
  ) {
    this.registrations.push({
      trigger: triggerArg,
      handler: handlerArg,
    });
    return triggerArg;
  }

  public getRegistrations() {
    return [...this.registrations];
  }

  public async runTrigger(eventArg: TAutomationEvent) {
    const matchingRegistrations = this.registrations.filter((registrationArg) => {
      return registrationArg.trigger.id === eventArg.triggerId || registrationArg.trigger.kind === eventArg.kind;
    });
    for (const registration of matchingRegistrations) {
      await registration.handler(eventArg, this);
    }
  }

  public async executePlan(planArg: plugins.shxInterfaces.data.IToolPlan) {
    if (this.options.executePlan) {
      return this.options.executePlan(planArg);
    }
    return {
      planId: planArg.id,
      results: planArg.calls.map((callArg: plugins.shxInterfaces.data.IToolCall) => ({
        callId: callArg.id,
        status: 'suggested' as const,
        output: {
          message: 'No hub executor attached. Plan was recorded as a suggestion.',
        },
      })),
    };
  }
}