const express = require("express");
const Anthropic = require("@anthropic-ai/sdk");

const app = express();
const client = new Anthropic();

app.get("/test", async (req, res) => {
  const persona = req.query.persona; // $ Source
  const query = req.query.query;

  // === messages.create: system as string ===

  // SHOULD ALERT
  const m1 = await client.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    system: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
    messages: [{ role: "user", content: query }],
  });

  // === messages.create: system as TextBlockParam array ===

  // SHOULD ALERT
  const m2 = await client.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    system: [
      {
        type: "text",
        text: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
      },
    ],
    messages: [{ role: "user", content: query }],
  });

  // === messages.create: assistant role content ===

  // SHOULD ALERT
  const m3 = await client.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    messages: [
      {
        role: "assistant",
        content: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
      },
      { role: "user", content: query },
    ],
  });

  // === messages.create: user role content ===

  // SHOULD NOT ALERT
  const m4 = await client.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    messages: [
      {
        role: "user",
        content: query, // OK - user role
      },
    ],
  });

  // === beta.messages.create: system as string ===

  // SHOULD ALERT
  const bm1 = await client.beta.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    system: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
    messages: [{ role: "user", content: query }],
  });

  // === beta.messages.create: system as TextBlockParam array ===

  // SHOULD ALERT
  const bm2 = await client.beta.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    system: [
      {
        type: "text",
        text: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
      },
    ],
    messages: [{ role: "user", content: query }],
  });

  // === beta.messages.create: assistant role content ===

  // SHOULD ALERT
  const bm3 = await client.beta.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    messages: [
      {
        role: "assistant",
        content: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
      },
      { role: "user", content: query },
    ],
  });

  // === beta.agents.create: system ===

  // SHOULD ALERT
  const ba1 = await client.beta.agents.create({
    model: "claude-sonnet-4-20250514",
    system: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
  });

  // === beta.agents.update: system ===

  // SHOULD ALERT
  await client.beta.agents.update("agent_123", {
    system: "Talk like a " + persona, // $ Alert[js/system-prompt-injection]
  });

  // === Barrier: user-role content in shared message array ===

  // SHOULD NOT ALERT — user input placed in { role: "user" } should not
  // taint system messages extracted from the same array.
  const messages = [
    { role: "system", content: "You are a helpful assistant" },
    { role: "user", content: query }, // OK - user role barrier
  ];
  const systemMsg = messages.find((m) => m.role === "system");
  const m6 = await client.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    system: systemMsg.content,
    messages: [{ role: "user", content: query }],
  });

  // === Barrier does NOT suppress: tainted value in system role ===

  // SHOULD ALERT — tainted data goes into system role; barrier on user role
  // must not suppress the system-role taint path.
  const messages2 = [
    { role: "system", content: "Talk like a " + persona },
    { role: "user", content: query },
  ];
  const systemMsg2 = messages2.find((m) => m.role === "system");
  const m7 = await client.messages.create({
    model: "claude-sonnet-4-20250514",
    max_tokens: 1024,
    system: systemMsg2.content, // $ Alert[js/system-prompt-injection]
    messages: [{ role: "user", content: query }],
  });

  // === Sanitizer: constant comparison ===

  // SHOULD NOT ALERT
  if (persona === "pirate") {
    const m5 = await client.messages.create({
      model: "claude-sonnet-4-20250514",
      max_tokens: 1024,
      system: "Talk like a " + persona, // OK - sanitized by constant check
      messages: [{ role: "user", content: query }],
    });
  }

  res.send("done");
});