openclaw/src/auto-reply/reply/runtime-policy-session-key.ts

import { normalizeChatType } from "../../channels/chat-type.js";
import type { OpenClawConfig } from "../../config/types.openclaw.js";
import {
  buildAgentMainSessionKey,
  buildAgentPeerSessionKey,
  normalizeAgentId,
  normalizeMainKey,
  resolveAgentIdFromSessionKey,
} from "../../routing/session-key.js";
import {
  normalizeLowercaseStringOrEmpty,
  normalizeOptionalString,
} from "../../shared/string-coerce.js";
import type { MsgContext } from "../templating.js";

type RuntimePolicyContext = Pick<
  MsgContext,
  | "AccountId"
  | "ChatType"
  | "CommandTargetSessionKey"
  | "From"
  | "NativeDirectUserId"
  | "OriginatingChannel"
  | "OriginatingTo"
  | "Provider"
  | "RuntimePolicySessionKey"
  | "SenderE164"
  | "SenderId"
  | "SenderUsername"
  | "SessionKey"
  | "Surface"
  | "To"
>;

function resolvePolicyChannel(ctx?: RuntimePolicyContext): string | undefined {
  const raw = normalizeOptionalString(ctx?.OriginatingChannel ?? ctx?.Provider ?? ctx?.Surface);
  if (!raw) {
    return undefined;
  }
  const channel = normalizeLowercaseStringOrEmpty(raw);
  return channel && channel !== "webchat" ? channel : undefined;
}

function resolvePolicyDirectPeerId(ctx?: RuntimePolicyContext): string | undefined {
  return normalizeOptionalString(
    ctx?.NativeDirectUserId ??
      ctx?.SenderId ??
      ctx?.SenderE164 ??
      ctx?.SenderUsername ??
      ctx?.OriginatingTo ??
      ctx?.From ??
      ctx?.To,
  );
}

function isMainSessionAlias(params: {
  cfg?: OpenClawConfig;
  agentId: string;
  sessionKey: string;
}): boolean {
  const raw = normalizeLowercaseStringOrEmpty(params.sessionKey);
  if (!raw) {
    return false;
  }
  const agentId = normalizeAgentId(params.agentId);
  const mainKey = normalizeMainKey(params.cfg?.session?.mainKey);
  const agentMainSessionKey = buildAgentMainSessionKey({
    agentId,
    mainKey,
  });
  const agentMainAliasKey = buildAgentMainSessionKey({
    agentId,
    mainKey: "main",
  });
  return (
    raw === "main" ||
    raw === mainKey ||
    raw === agentMainSessionKey ||
    raw === agentMainAliasKey ||
    raw === buildAgentMainSessionKey({ agentId: "main", mainKey }) ||
    raw === buildAgentMainSessionKey({ agentId: "main", mainKey: "main" }) ||
    (params.cfg?.session?.scope === "global" && raw === "global")
  );
}

export function resolveRuntimePolicySessionKey(params: {
  cfg?: OpenClawConfig;
  ctx?: RuntimePolicyContext;
  sessionKey?: string | null;
}): string | undefined {
  const explicitPolicySessionKey = normalizeOptionalString(params.ctx?.RuntimePolicySessionKey);
  if (explicitPolicySessionKey) {
    return explicitPolicySessionKey;
  }
  const sessionKey = normalizeOptionalString(
    params.sessionKey ?? params.ctx?.CommandTargetSessionKey ?? params.ctx?.SessionKey,
  );
  if (!sessionKey) {
    return undefined;
  }

  const agentId = resolveAgentIdFromSessionKey(sessionKey);
  if (!isMainSessionAlias({ cfg: params.cfg, agentId, sessionKey })) {
    return sessionKey;
  }

  if (normalizeChatType(params.ctx?.ChatType) !== "direct") {
    return sessionKey;
  }
  const channel = resolvePolicyChannel(params.ctx);
  const peerId = resolvePolicyDirectPeerId(params.ctx);
  if (!channel || !peerId) {
    return sessionKey;
  }

  return buildAgentPeerSessionKey({
    agentId,
    channel,
    accountId: params.ctx?.AccountId,
    peerKind: "direct",
    peerId,
    dmScope: "per-account-channel-peer",
    identityLinks: params.cfg?.session?.identityLinks,
  });
}