PaperClipAI/server/src/__tests__/fixtures/plugin-worker-invocation-scope.cjs

const readline = require("node:readline");

let nextRequestId = 1;
const pendingNested = new Map();

function send(message) {
  process.stdout.write(`${JSON.stringify(message)}\n`);
}

function sendNestedHostRequest(originalRequest, invocationId) {
  const nestedId = `nested-${nextRequestId++}`;
  const params = originalRequest.params?.params ?? {};
  const mode = params.mode;
  const requestedCompanyId = params.requestedCompanyId;
  const hostMethod = params.hostMethod || "companies.get";
  const nestedParams = hostMethod === "secrets.resolve"
    ? {
        companyId: requestedCompanyId,
        secretRef: {
          type: "secret_ref",
          secretId: params.secretId || "11111111-1111-4111-8111-111111111111",
        },
        configPath: params.configPath || "apiKeyRef",
      }
    : hostMethod === "state.get"
    ? {
        // Company-scoped state key — the shape a proactive gateway loop uses
        // (ctx.state.get with scopeKind "company"). The host derives the
        // requested company from scopeId, not companyId.
        scopeKind: "company",
        scopeId: requestedCompanyId,
        namespace: params.namespace || "ns",
        stateKey: params.stateKey || "key",
      }
    : hostMethod === "events.subscribe"
    ? {
        // The subscribe shape the SDK issues from setup() via
        // ctx.events.on(name, { companyId }, fn): the requested company lives in
        // filter.companyId, NOT a top-level companyId. The host resolver must
        // mirror the SDK gate and read it from there (LOOA-695).
        eventPattern: params.eventPattern || "issue.updated",
        filter: { companyId: requestedCompanyId },
      }
    : {
        companyId: requestedCompanyId,
      };
  const nestedRequest = {
    jsonrpc: "2.0",
    id: nestedId,
    method: hostMethod,
    params: nestedParams,
  };

  if (mode === "echo") {
    nestedRequest.paperclipInvocationId = invocationId;
  } else if (mode === "unknown") {
    nestedRequest.paperclipInvocationId = "unknown-invocation";
  }

  pendingNested.set(nestedId, originalRequest.id);
  send(nestedRequest);
}

const rl = readline.createInterface({
  input: process.stdin,
  crlfDelay: Infinity,
});

rl.on("line", (line) => {
  if (!line.trim()) return;
  const message = JSON.parse(line);

  if (message.id && pendingNested.has(message.id)) {
    const originalId = pendingNested.get(message.id);
    pendingNested.delete(message.id);
    if (message.error) {
      send({
        jsonrpc: "2.0",
        id: originalId,
        error: message.error,
      });
      return;
    }

    send({
      jsonrpc: "2.0",
      id: originalId,
      result: message.result,
    });
    return;
  }

  const method = message && typeof message.method === "string" ? message.method : null;

  if (method === "initialize") {
    send({
      jsonrpc: "2.0",
      id: message.id,
      result: {
        ok: true,
        supportedMethods: ["getData", "performAction", "resourceLifecycle", "workspaceOperation"],
      },
    });
    return;
  }

  if (["getData", "performAction", "resourceLifecycle", "workspaceOperation"].includes(method)) {
    sendNestedHostRequest(message, message.paperclipInvocation?.id);
    return;
  }

  if (method === "shutdown") {
    send({
      jsonrpc: "2.0",
      id: message.id,
      result: {},
    });
    setImmediate(() => process.exit(0));
    return;
  }

  send({
    jsonrpc: "2.0",
    id: message.id,
    error: {
      code: -32601,
      message: `Unhandled method: ${method}`,
    },
  });
});