openclaw/extensions/whatsapp/src/connection-controller-runtime-context.ts

// Whatsapp plugin module exposes live connection controllers through the channel runtime.
import type { WASocket } from "baileys";
import { getChannelRuntimeContext } from "openclaw/plugin-sdk/channel-runtime-context";
import type { WhatsAppSelfIdentity } from "./identity.js";
import type { ActiveWebListener } from "./inbound/types.js";
import { getOptionalWhatsAppChannelRuntime } from "./runtime.js";

export const WHATSAPP_CONNECTION_CONTROLLER_CAPABILITY = "connection-controller";
export const WHATSAPP_CONNECTION_OWNER_PENDING_CAPABILITY = "connection-owner-pending";

type WhatsAppConnectionControllerHandle = {
  getActiveListener(): ActiveWebListener | null;
  getCurrentSock(): WASocket | null;
  getSelfIdentity(): WhatsAppSelfIdentity | null;
};

export function getWhatsAppConnectionController(
  accountId: string,
): WhatsAppConnectionControllerHandle | null {
  const context = getChannelRuntimeContext({
    channelRuntime: getOptionalWhatsAppChannelRuntime() ?? undefined,
    channelId: "whatsapp",
    accountId,
    capability: WHATSAPP_CONNECTION_CONTROLLER_CAPABILITY,
  });
  return (context as WhatsAppConnectionControllerHandle | undefined) ?? null;
}

export function hasPendingWhatsAppConnectionOwner(accountId: string): boolean {
  return Boolean(
    getChannelRuntimeContext({
      channelRuntime: getOptionalWhatsAppChannelRuntime() ?? undefined,
      channelId: "whatsapp",
      accountId,
      capability: WHATSAPP_CONNECTION_OWNER_PENDING_CAPABILITY,
    }),
  );
}