chore: import upstream snapshot with attribution

examples/redteam-auth/auth/get-token.js

@@ -0,0 +1,34 @@
+export default async function getAuth(context) {
+  const tokenUrl = 'https://example-app.promptfoo.app/oauth/token';
+  const clientId = process.env.PROMPTFOO_TARGET_CLIENT_ID || context.vars.clientId;
+  const clientSecret = process.env.PROMPTFOO_TARGET_CLIENT_SECRET || context.vars.clientSecret;
+  const scopes = process.env.PROMPTFOO_TARGET_SCOPES || context.vars.scopes || '';
+
+  const body = new URLSearchParams({
+    grant_type: 'client_credentials',
+    client_id: clientId,
+    client_secret: clientSecret,
+  });
+
+  if (scopes) {
+    body.set('scope', scopes);
+  }
+
+  const response = await fetch(tokenUrl, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/x-www-form-urlencoded',
+    },
+    body,
+  });
+
+  if (!response.ok) {
+    throw new Error(`Token request failed with status ${response.status}`);
+  }
+
+  const data = await response.json();
+  return {
+    token: data.access_token,
+    expiration: data.expires_in ? Date.now() + data.expires_in * 1000 : undefined,
+  };
+}

examples/redteam-auth/auth/get-token.py

@@ -0,0 +1,41 @@
+import json
+import os
+import time
+import urllib.parse
+import urllib.request
+
+
+def get_auth(context):
+    token_url = "https://example-app.promptfoo.app/oauth/token"
+    vars = context.get("vars", {})
+    client_id = os.environ.get("PROMPTFOO_TARGET_CLIENT_ID") or vars.get("clientId")
+    client_secret = os.environ.get("PROMPTFOO_TARGET_CLIENT_SECRET") or vars.get(
+        "clientSecret"
+    )
+    scopes = os.environ.get("PROMPTFOO_TARGET_SCOPES") or vars.get("scopes")
+
+    body = {
+        "grant_type": "client_credentials",
+        "client_id": client_id,
+        "client_secret": client_secret,
+    }
+    if scopes:
+        body["scope"] = scopes
+
+    request = urllib.request.Request(
+        token_url,
+        data=urllib.parse.urlencode(body).encode("utf-8"),
+        headers={"Content-Type": "application/x-www-form-urlencoded"},
+        method="POST",
+    )
+    with urllib.request.urlopen(request) as response:
+        data = json.loads(response.read().decode("utf-8"))
+
+    return {
+        "token": data["access_token"],
+        "expiration": (
+            int(data["expires_in"]) * 1000 + int(time.time() * 1000)
+            if data.get("expires_in")
+            else None
+        ),
+    }

examples/redteam-auth/auth/get-token.ts

@@ -0,0 +1,47 @@
+type AuthContext = {
+  vars: Record<string, string>;
+};
+
+type AuthResponse = {
+  access_token: string;
+  expires_in?: number;
+};
+
+export default async function getAuth(context: AuthContext) {
+  const tokenUrl = 'https://example-app.promptfoo.app/oauth/token';
+  const clientId = process.env.PROMPTFOO_TARGET_CLIENT_ID || context.vars.clientId;
+  const clientSecret = process.env.PROMPTFOO_TARGET_CLIENT_SECRET || context.vars.clientSecret;
+  const scopes = process.env.PROMPTFOO_TARGET_SCOPES || context.vars.scopes || '';
+
+  const body = new URLSearchParams({
+    grant_type: 'client_credentials',
+    client_id: clientId,
+    client_secret: clientSecret,
+  });
+
+  if (scopes) {
+    body.set('scope', scopes);
+  }
+
+  const response = await fetch(tokenUrl, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/x-www-form-urlencoded',
+    },
+    body,
+  });
+
+  if (!response.ok) {
+    throw new Error(`Token request failed with status ${response.status}`);
+  }
+
+  const data = (await response.json()) as AuthResponse;
+  return {
+    token: data.access_token,
+    expiration: data.expires_in ? Date.now() + data.expires_in * 1000 : undefined,
+  };
+}
+
+export async function buildAuth(context: AuthContext) {
+  return getAuth(context);
+}