import { TelemetryClient } from "@copilotkit/shared";
import { EndpointType, LangGraphPlatformEndpoint } from "./runtime/remote-actions";
import { createHash } from "node:crypto";
import { CopilotRuntime, resolveEndpointType } from "./runtime/copilot-runtime";
import { RuntimeInstanceCreatedInfo } from "@copilotkit/shared/src/telemetry/events";
const packageJson = require("../../package.json");

const telemetryClient = new TelemetryClient({
  packageName: packageJson.name,
  packageVersion: packageJson.version,
});

export function getRuntimeInstanceTelemetryInfo(
  runtime: CopilotRuntime,
): RuntimeInstanceCreatedInfo {
  const endpointsInfo = runtime.remoteEndpointDefinitions.reduce(
    (acc, endpoint) => {
      let info = { ...acc };

      const endpointType = resolveEndpointType(endpoint);
      if (!info.endpointTypes.includes(endpointType)) {
        info = {
          ...info,
          endpointTypes: [...info.endpointTypes, endpointType],
        };
      }

      if (endpointType === EndpointType.LangGraphPlatform) {
        // When type is resolved, recreating a const with casting of type
        const ep = endpoint as LangGraphPlatformEndpoint;
        info = {
          ...info,
          agentsAmount: ep.agents.length,
          hashedKey: ep.langsmithApiKey
            ? createHash("sha256").update(ep.langsmithApiKey).digest("hex")
            : null,
        };
      }

      return info;
    },
    { endpointTypes: [], agentsAmount: null, hashedKey: null },
  );

  return {
    actionsAmount: runtime.actions.length,
    endpointsAmount: runtime.remoteEndpointDefinitions.length,
    endpointTypes: endpointsInfo.endpointTypes,
    agentsAmount: endpointsInfo.agentsAmount,
    hashedLgcKey: endpointsInfo.hashedKey,
  };
}

export default telemetryClient;