Dirigent/plugin/rules.ts

export type WhisperGateConfig = {
  enabled?: boolean;
  discordOnly?: boolean;
  listMode?: "human-list" | "agent-list";
  humanList?: string[];
  agentList?: string[];
  channelPoliciesFile?: string;
  // backward compatibility
  bypassUserIds?: string[];
  endSymbols?: string[];
  noReplyProvider: string;
  noReplyModel: string;
};

export type ChannelPolicy = {
  listMode?: "human-list" | "agent-list";
  humanList?: string[];
  agentList?: string[];
  endSymbols?: string[];
};

export type Decision = {
  shouldUseNoReply: boolean;
  shouldInjectEndMarkerPrompt: boolean;
  reason: string;
};

function getLastChar(input: string): string {
  const t = input.trim();
  return t.length ? t[t.length - 1] : "";
}

function resolvePolicy(config: WhisperGateConfig, channelId?: string, channelPolicies?: Record<string, ChannelPolicy>) {
  const globalMode = config.listMode || "human-list";
  const globalHuman = config.humanList || config.bypassUserIds || [];
  const globalAgent = config.agentList || [];
  const globalEnd = config.endSymbols || ["🔚"];

  if (!channelId) {
    return { listMode: globalMode, humanList: globalHuman, agentList: globalAgent, endSymbols: globalEnd };
  }

  const cp = channelPolicies || {};
  const scoped = cp[channelId];
  if (!scoped) {
    return { listMode: globalMode, humanList: globalHuman, agentList: globalAgent, endSymbols: globalEnd };
  }

  return {
    listMode: scoped.listMode || globalMode,
    humanList: scoped.humanList || globalHuman,
    agentList: scoped.agentList || globalAgent,
    endSymbols: scoped.endSymbols || globalEnd,
  };
}

export function evaluateDecision(params: {
  config: WhisperGateConfig;
  channel?: string;
  channelId?: string;
  channelPolicies?: Record<string, ChannelPolicy>;
  senderId?: string;
  content?: string;
}): Decision {
  const { config } = params;

  if (config.enabled === false) {
    return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: false, reason: "disabled" };
  }

  const channel = (params.channel || "").toLowerCase();
  if (config.discordOnly !== false && channel !== "discord") {
    return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: false, reason: "non_discord" };
  }

  // DM bypass: if no conversation info was found in the prompt (no senderId AND no channelId),
  // this is a DM session where untrusted metadata is not injected. Always allow through.
  if (!params.senderId && !params.channelId) {
    return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: true, reason: "dm_no_metadata_bypass" };
  }

  const policy = resolvePolicy(config, params.channelId, params.channelPolicies);

  const mode = policy.listMode;
  const humanList = policy.humanList;
  const agentList = policy.agentList;

  const senderId = params.senderId || "";
  const inHumanList = !!senderId && humanList.includes(senderId);
  const inAgentList = !!senderId && agentList.includes(senderId);

  const lastChar = getLastChar(params.content || "");
  const hasEnd = !!lastChar && policy.endSymbols.includes(lastChar);

  if (mode === "human-list") {
    if (inHumanList) {
      return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: true, reason: "human_list_sender" };
    }
    if (hasEnd) {
      return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: true, reason: `end_symbol:${lastChar}` };
    }
    return { shouldUseNoReply: true, shouldInjectEndMarkerPrompt: false, reason: "rule_match_no_end_symbol" };
  }

  // agent-list mode: listed senders require end symbol; others bypass requirement.
  if (!inAgentList) {
    return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: true, reason: "non_agent_list_sender" };
  }
  if (hasEnd) {
    return { shouldUseNoReply: false, shouldInjectEndMarkerPrompt: true, reason: `end_symbol:${lastChar}` };
  }
  return { shouldUseNoReply: true, shouldInjectEndMarkerPrompt: false, reason: "agent_list_missing_end_symbol" };
}