@argentic/chest-sdk 0.2.0 → 0.4.0

package/client/src/testing.ts

@@ -1,22 +1,30 @@
-import { createHash, createHmac, randomBytes } from "node:crypto";
+import { createHash, randomBytes } from "node:crypto";
 import { createServer, type IncomingMessage, type ServerResponse } from "node:http";
 import type { AddressInfo } from "node:net";
+import type { Alias, Provider } from "./ai.js";
+import type { AiUnavailableReason } from "./errors.js";
 import type { ChestEvent } from "./events.js";
 import { groupIdPattern, memberIdPattern, type Member } from "./member.js";
 import { forget } from "./members.js";
+import { fakeOrigins } from "./api.js";
+import { eventChannel, scheduleChannel, sign, signClaims, type Channel } from "./signed.js";
 
 // For a tool's own tests, never imported by its production code: a member's
 // assertion signed as the Chest signs it, and a Chest's API in the test's
-// process that answers members, groups, files (stat, move, links; an upload
-// it authorises but does not receive), badges, notifications and the
-// acknowledgment of an erasure with the Chest's bounds, quotas and errors;
-// and that delivers an event to the tool, signed as the Chest signs it.
+// process that answers members, groups, files (stat, move, links and
+// uploads, which it serves and takes itself as the team host would, on its
+// own origin), badges, notifications, AI (chat,
+// streamed or not, embeddings, models, usage: deterministic answers, no
+// provider) and the acknowledgment of an erasure with the Chest's bounds,
+// quotas and errors; and that delivers an event or a run of a schedule to
+// the tool, signed as the Chest signs them.
 //
 //   import { fakeChest, withMember } from "@argentic/chest-sdk/testing";
 //   const chest = await fakeChest({ members: [camille], capabilities: ["members", "files", "notifications"] });
 //   const response = await app(withMember(new Request("http://tool/chest"), camille));
 //   assert.equal(chest.notifications[0]?.member, camille.id);
 //   assert.equal(await chest.emit({ type: "access.revoked", data: { id: camille.id } }, request => app(request)), 204);
+//   assert.equal(await chest.run("morning", request => app(request)), 204);
 //   await chest.close();
 
 // A group as a fake Chest keeps it: its identifier, its name, and the
@@ -24,24 +32,51 @@ import { forget } from "./members.js";
 export type FakeGroup = { id: string; name: string; members: string[] };
 // A file as a fake Chest keeps it.
 export type FakeFile = { data: Uint8Array; type: string; updated: string };
+// Someone the tool had who no longer has it: a member of the Chest without
+// access to it, a member who left ("former", by default), or one whose data
+// was erased (their name gone).
+export type FakeFormer = { id: string; name?: string; status?: "no_access" | "former" | "erased" };
 // A notification as a fake Chest keeps it: the member it went to, its text
 // cleaned as the Chest cleans it, its path (/chest when not said) and its
 // key.
 export type FakeNotification = { member: string; title: string; body?: string; path: string; key?: string };
+// An alias a fake Chest maps: the model behind it, its provider (openrouter
+// by default) and its prices in US dollars per million tokens (1 and 2 by
+// default).
+export type FakeAiModel = { alias: Alias; model: string; provider?: Provider; input?: number; output?: number };
+// What a fake model answers: text, or text and tool calls (an id of call_…
+// by default; arguments are JSON text).
+export type FakeAiReply = string | { text?: string; toolCalls?: { name: string; arguments: string; id?: string }[] };
+// The AI of a fake Chest: the aliases the tool declared, each mapped (all
+// four by default: fake-default, fake-fast, fake-smart, fake-embedding); what
+// its models answer to a chat, given the request as the tool sent it (the
+// last user message's text, echoed, by default); the tool's monthly cap in
+// euros (5 by default; a call is refused cap_reached once the spending
+// reaches it, 0 refuses at once); and a reason that makes chat and
+// embeddings unavailable.
+export type FakeAi = { models?: FakeAiModel[]; reply?: (request: Record<string, unknown>) => FakeAiReply; cap?: number; unavailable?: AiUnavailableReason };
+// A call of the tool to the AI of a fake Chest: its path and its body as
+// sent (null for a GET).
+export type FakeAiCall = { path: string; body: unknown };
 
-// What a fake Chest is given: the members who have the tool, those who left
-// it (erased: their data was erased, the name gone), its groups, the
+// What a fake Chest is given: the members who have the tool, those it had
+// who no longer have it (FakeFormer), its groups, the
 // capabilities its version holds (a capability left out answers 403;
-// members, files and notifications by default, members.email to read the
+// members, files, notifications and ai by default, members.email to read the
 // addresses), the events it receives (["member.*"] by default, [] to answer
-// an acknowledgment 403) and the files it keeps.
+// an acknowledgment 403), the files it keeps, its AI, and what the Chest is
+// (the chest module: "Test organization", UTC, English and euros by default;
+// the tool at https://<tool>-chest.chest.test, its public part at
+// https://<tool>.chest.test).
 export type FakeChestOptions = {
   members?: Member[];
-  former?: { id: string; name?: string; erased?: boolean }[];
+  former?: FakeFormer[];
   groups?: FakeGroup[];
   capabilities?: string[];
   receives?: string[];
   files?: Record<string, { data: Uint8Array | string; type?: string }>;
+  ai?: FakeAi;
+  chest?: { organization?: string; timeZone?: string; language?: string; currency?: string; teamUrl?: string; publicUrl?: string | null };
 };
 
 // An event for emit: its type and data; its id (a new evt_… by default) and
@@ -49,13 +84,20 @@ export type FakeChestOptions = {
 // twice.
 export type FakeEvent = { [K in ChestEvent["type"]]: { type: K; data: Extract<ChestEvent, { type: K }>["data"]; id?: string; occurredAt?: string } }[ChestEvent["type"]];
 
-// A fake Chest in the test's process: its address, the token and the tool it
-// set in the environment, what it keeps (members, groups and files a test
+// A run for run: its id (a new run_… by default: name it to deliver the same
+// run twice), the time it stands for (now by default) and its attempt (1 by
+// default).
+export type FakeRun = { id?: string; scheduledAt?: string; attempt?: number };
+
+// A fake Chest in the test's process: its address — its API, and the origin
+// where it serves the links and takes the uploads it signs —, the token and
+// the tool it set in the environment, what it keeps (members, groups and files a test
 // changes or reads; the notifications the tool sent, in the order sent, a
 // replaced one last; each member's badge; the erasures the tool
-// acknowledged), emit, which delivers an event to the tool — POST
+// acknowledged; its calls to AI, in order), emit, which delivers an event to the tool — POST
 // /chest-events of its address, or a handler of Web Requests — and says the
-// status it answered, and close, which stops it and restores the
+// status it answered, run, which delivers a run of a schedule the same way
+// on /chest-schedules, and close, which stops it and restores the
 // environment. Its members are those who have the tool: the others are
 // skipped.
 export type FakeChest = {
@@ -68,45 +110,43 @@ export type FakeChest = {
   notifications: FakeNotification[];
   badges: Map<string, number>;
   acknowledged: string[];
+  ai: FakeAiCall[];
   emit(event: FakeEvent, to: string | ((request: Request) => Response | Promise<Response>)): Promise<number>;
+  run(name: string, to: string | ((request: Request) => Response | Promise<Response>), run?: FakeRun): Promise<number>;
   close(): Promise<void>;
 };
 
-const encode = (value: unknown): string => Buffer.from(JSON.stringify(value)).toString("base64url");
+// What an assertion is signed with and says besides the member: the token
+// (CHEST_TOKEN by default) and the tool (CHEST_TOOL by default) it is for,
+// and when it is issued (now by default). The member is signed as given: a
+// language or a zone the Chest would never send makes member() refuse the
+// assertion, as it refuses the Chest's.
+export type AssertionOptions = { token?: string; tool?: string; now?: Date };
 
 // signAssertion is the Chest-Member value the Chest's front would send for
-// that member: HS256 under the key of the token (CHEST_TOKEN by default), for
-// the tool (CHEST_TOOL by default), valid 60 seconds from now.
-export function signAssertion(member: Member, options: { token?: string; tool?: string; now?: Date } = {}): string {
+// that member: HS256 under the key of the token, for the tool, valid 60
+// seconds from when it is issued.
+export function signAssertion(member: Member, options: AssertionOptions = {}): string {
   const token = options.token ?? process.env["CHEST_TOKEN"];
   const tool = options.tool ?? process.env["CHEST_TOOL"];
   if (!token || !tool) throw new Error("signAssertion needs a token and a tool: start a fakeChest, or name them");
   if (!memberIdPattern.test(member.id) || !member.groups.every(g => groupIdPattern.test(g))) throw new Error("signAssertion needs identifiers of the Chest's shape (mbr_…, grp_…)");
   const iat = Math.floor((options.now ?? new Date()).getTime() / 1000);
-  const body = encode({ alg: "HS256", typ: "JWT" }) + "." + encode({
+  // Under the label of the assertion's shape, as the Chest signs it and
+  // member() reads it.
+  return signClaims("Chest-Member v2", {
     iss: `https://${tool}-chest.chest.test`, aud: tool, iat, exp: iat + 60, sub: member.id,
     given_name: member.firstName, family_name: member.lastName, name: member.name, picture: member.photo ?? "", role: member.role ?? "",
-    admin: member.isAdmin, builder: member.isBuilder, groups: member.groups, ...(member.email === undefined ? {} : { email: member.email }),
-  });
-  // The key as the Chest derives it, and member() reads it: HMAC-SHA256 of
-  // the label of the assertion's shape under the text of the token.
-  const key = createHmac("sha256", Buffer.from(token, "utf8")).update("Chest-Member v2").digest();
-  return body + "." + createHmac("sha256", key).update(body).digest("base64url");
+    admin: member.isAdmin, builder: member.isBuilder, groups: member.groups, time_zone: member.timeZone, ...(member.email === undefined ? {} : { email: member.email }),
+    language: member.language,
+  }, token);
 }
 
-// signEvent is the Chest-Event value the Chest would send with that body:
-// HS256 under the key the token derives for events, for the tool, naming the
-// event and the digest of the body, valid 60 seconds.
-function signEvent(id: string, body: string, options: { token: string; tool: string }): string {
-  const iat = Math.floor(Date.now() / 1000);
-  const signed = encode({ alg: "HS256", typ: "JWT" }) + "." + encode({ aud: options.tool, iat, exp: iat + 60, jti: id, digest: createHash("sha256").update(body).digest("base64url") });
-  const key = createHmac("sha256", Buffer.from(options.token, "utf8")).update("Chest-Event v1").digest();
-  return signed + "." + createHmac("sha256", key).update(signed).digest("base64url");
-}
 
-// withMember is the request carrying that member's assertion: a new Web
-// Request, or the same Node request with its header set.
-export function withMember<R extends Request | IncomingMessage>(request: R, member: Member, options: { token?: string; tool?: string; now?: Date } = {}): R {
+// withMember is the request carrying that member's assertion, signed with
+// the options of signAssertion: a new Web Request, or the same Node request
+// with its header set.
+export function withMember<R extends Request | IncomingMessage>(request: R, member: Member, options: AssertionOptions = {}): R {
   const assertion = signAssertion(member, options);
   if (request instanceof Request) {
     const headers = new Headers(request.headers);
@@ -121,6 +161,19 @@ export function withMember<R extends Request | IncomingMessage>(request: R, memb
 const maxLimit = 500, defaultLimit = 100, maxLookup = 200, callsPerMinute = 600;
 const maxObject = 32 << 20, maxObjects = 10000, maxTotal = 1 << 30;
 const namePattern = /^[A-Za-z0-9][A-Za-z0-9._-]{0,99}(\/[A-Za-z0-9][A-Za-z0-9._-]{0,99}){0,7}$/u;
+// Its links and uploads: their paths on the team host, their lives in
+// seconds, the images it makes thumbnails of, the first bytes an upload of
+// a type must start with, the endings of the names it chooses (chest/toolfiles).
+const linkPath = "/_chest/files/", uploadPath = "/_chest/files/upload/", linkLife = 900, uploadLife = 900;
+const mediaPattern = /^[a-z0-9][a-z0-9!#$&^_.+-]{0,62}\/(\*|[a-z0-9][a-z0-9!#$&^_.+-]{0,62})$/u;
+const thumbnailed = ["image/jpeg", "image/png", "image/gif", "image/webp"];
+const starts = (...heads: string[]) => (data: Buffer): boolean => heads.some(head => data.subarray(0, head.length).equals(Buffer.from(head, "latin1")));
+const signatures: Record<string, (data: Buffer) => boolean> = {
+  "image/jpeg": starts("\xff\xd8\xff"), "image/png": starts("\x89PNG\r\n\x1a\n"), "image/gif": starts("GIF87a", "GIF89a"),
+  "image/webp": data => data.subarray(0, 4).toString("latin1") === "RIFF" && data.subarray(8, 12).toString("latin1") === "WEBP", "application/pdf": starts("%PDF-"),
+};
+const extensions: Record<string, string> = { "image/jpeg": ".jpg", "image/png": ".png", "image/gif": ".gif", "image/webp": ".webp", "image/avif": ".avif", "image/heic": ".heic", "application/pdf": ".pdf", "text/plain": ".txt", "text/csv": ".csv", "application/json": ".json", "application/zip": ".zip", "video/mp4": ".mp4", "audio/mpeg": ".mp3" };
+const newToken = (): string => randomBytes(18).toString("base64url") + "." + randomBytes(12).toString("base64url");
 // Its notifications: recipients and badges a call, text, quotas.
 const maxRecipients = 500, maxTitle = 80, maxText = 280, maxPath = 512, maxCount = 9999;
 const recipientsPerHour = 1000, itemsPerDay = 100, badgesPerMinute = 600;
@@ -130,6 +183,22 @@ const cleanTitle = (s: string): string => s.replace(/[\t\r\n]/gu, " ").replace(/
 const cleanText = (s: string): string => s.replace(/\r\n?/gu, "\n").replace(/\t/gu, " ").replace(/[^\P{Cc}\n]/gu, "").replace(reordering, "").trim();
 const isPath = (p: unknown): boolean => typeof p === "string" && p.length <= maxPath && /^\/chest([/?#][\x21-\x5b\x5d-\x7e]*)?$/u.test(p) && !p.includes("//") && !p.split(/[?#]/u)[0]!.split("/").some(x => /^(\.|%2e){1,2}$/iu.test(x));
 
+// Its AI: aliases in order, bounds of a request, requests a minute.
+const aliasOrder: readonly Alias[] = ["default", "fast", "smart", "embedding"];
+const chatKeys = ["model", "messages", "max_tokens", "stream", "temperature", "top_p", "stop", "tools", "tool_choice", "response_format", "parallel_tool_calls", "seed", "reasoning_effort", "member"];
+const maxAiBody = 10 << 20, maxAiOutput = 128000, maxInputs = 256, aiPerMinute = 60, fakeDimensions = 8;
+// A fake count of tokens: one per 4 characters.
+const tokensOf = (text: string): number => Math.ceil(text.length / 4);
+// A vector of a text, the same for the same text: unit length.
+function vectorOf(text: string, dimensions: number): number[] {
+  const values: number[] = [];
+  for (let block = 0; values.length < dimensions; block++) {
+    for (const b of createHash("sha256").update(block + ":" + text).digest()) values.push(b / 127.5 - 1);
+  }
+  const kept = values.slice(0, dimensions), norm = Math.hypot(...kept) || 1;
+  return kept.map(x => x / norm);
+}
+
 const fold = (s: string): string => s.normalize("NFD").replace(/\p{Mn}/gu, "").toLowerCase();
 
 function send(response: ServerResponse, status: number, value?: unknown, headers: Record<string, string> = {}): void {
@@ -150,10 +219,11 @@ async function body(request: IncomingMessage, limit: number): Promise<Buffer | n
 
 // fakeChest starts a Chest's API on 127.0.0.1 and points the environment at
 // it: CHEST_API, CHEST_TOKEN (a new one) and CHEST_TOOL ("tool" unless the
-// environment names one). What member() and the modules of the SDK read is
-// then this Chest's.
+// environment names one), and what the Chest is (CHEST_ORGANIZATION,
+// CHEST_TIME_ZONE, CHEST_LANGUAGE, as options.chest says). What member(),
+// chest and the modules of the SDK read is then this Chest's.
 export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChest> {
-  const capabilities = new Set(options.capabilities ?? ["members", "files", "notifications"]);
+  const capabilities = new Set(options.capabilities ?? ["members", "files", "notifications", "ai"]);
   const email = capabilities.has("members.email");
   const tool = process.env["CHEST_TOOL"] || "tool";
   const token = randomBytes(32).toString("base64url");
@@ -164,12 +234,12 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
   const receives = options.receives ?? ["member.*"];
   // The erasures the tool was told of, by emit: those it may acknowledge.
   const erasures = new Set<string>();
-  const chest: FakeChest = { api: "", token, tool, members: [...(options.members ?? [])], groups: [...(options.groups ?? [])], files, notifications: [], badges: new Map(), acknowledged: [], emit: async () => 0, close: async () => {} };
+  const chest: FakeChest = { api: "", token, tool, members: [...(options.members ?? [])], groups: [...(options.groups ?? [])], files, notifications: [], badges: new Map(), acknowledged: [], ai: [], emit: async () => 0, run: async () => 0, close: async () => {} };
   const former = [...(options.former ?? [])];
   let window = 0, calls = 0;
-  const shown = (m: Member) => ({ id: m.id, first_name: m.firstName, last_name: m.lastName, name: m.name, photo: m.photo, role: m.role, admin: m.isAdmin, builder: m.isBuilder, groups: m.groups, ...(email && m.email !== undefined ? { email: m.email } : {}) });
+  const shown = (m: Member) => ({ id: m.id, first_name: m.firstName, last_name: m.lastName, name: m.name, photo: m.photo, role: m.role, admin: m.isAdmin, builder: m.isBuilder, groups: m.groups, language: m.language, time_zone: m.timeZone, ...(email && m.email !== undefined ? { email: m.email } : {}) });
   const key = (m: Member) => fold(m.name) + "\u0000" + m.id;
-  const described = (name: string, f: FakeFile) => ({ name, type: f.type, size: f.data.byteLength, updated: f.updated });
+  const described = (name: string, f: FakeFile) => ({ name, type: f.type, size: f.data.byteLength, sha256: createHash("sha256").update(f.data).digest("hex"), updated: f.updated });
 
   async function members(request: IncomingMessage, response: ServerResponse, url: URL): Promise<void> {
     if (!capabilities.has("members")) return send(response, 403, { error: "capability_not_granted" });
@@ -200,7 +270,8 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
       for (const id of new Set(ids as string[])) {
         const m = chest.members.find(x => x.id === id), f = former.find(x => x.id === id);
         if (m) answer.members.push(shown(m));
-        else if (f?.erased) answer.former.push({ id, status: "erased" });
+        else if (f?.status === "erased") answer.former.push({ id, status: "erased" });
+        else if (f?.status === "no_access") answer.former.push({ id, name: f.name ?? "Test member", status: "no_access" });
         else if (f) answer.former.push({ id, ...(f.name ? { name: f.name } : {}), status: "former" });
         else answer.unknown.push(id);
       }
@@ -223,23 +294,30 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
       const names = [...files.keys()].filter(n => n.startsWith(prefix) && n > after).sort();
       return send(response, 200, { files: names.slice(0, 1000).map(n => described(n, files.get(n)!)), next: names.length > 1000 ? names[999] : null });
     }
-    if (request.method === "POST" && url.pathname === "/files/url") {
-      const raw = await body(request, 4096);
-      let name: unknown;
-      try { name = (JSON.parse(raw?.toString() ?? "") as { name?: unknown }).name; } catch { name = undefined; }
-      if (typeof name !== "string") return send(response, 400, { error: "invalid_body" });
-      if (!files.has(name)) return send(response, 404, { error: "not_found" });
-      return send(response, 200, { url: `https://${tool}-chest.chest.test/_chest/files/${Buffer.from(name).toString("base64url")}.fake`, expires_in: 900 });
-    }
-    if (request.method === "POST" && (url.pathname === "/files/move" || url.pathname === "/files/upload-url")) {
+    if (request.method === "POST" && (url.pathname === "/files/url" || url.pathname === "/files/move" || url.pathname === "/files/upload-url")) {
       const raw = await body(request, 4096);
       let command: Record<string, unknown> = {};
       try { command = JSON.parse(raw?.toString() ?? "") as Record<string, unknown>; } catch { command = {}; }
+      if (url.pathname === "/files/url") {
+        const { name, thumbnail, download } = command;
+        if (typeof name !== "string" || !(thumbnail === undefined || thumbnail === 256 || thumbnail === 1024) || !(download === undefined || typeof download === "boolean")) return send(response, 400, { error: "invalid_body" });
+        const object = files.get(name);
+        if (!object) return send(response, 404, { error: "not_found" });
+        if (thumbnail !== undefined && !thumbnailed.includes(object.type)) return send(response, 400, { error: "no_thumbnail" });
+        const token = newToken();
+        links.set(token, { name, object, until: Date.now() + linkLife * 1000, download: download === true });
+        return send(response, 200, { url: `${chest.api}${linkPath}${token}`, expires_in: linkLife });
+      }
       if (url.pathname === "/files/upload-url") {
-        // The upload itself goes from a member's browser to the team host,
-        // which a fake Chest does not play: it only authorises it.
-        if (typeof command["name"] !== "string") return send(response, 400, { error: "invalid_body" });
-        return send(response, 200, { url: `https://${tool}-chest.chest.test/_chest/files/upload/${Buffer.from(command["name"]).toString("base64url")}.fake`, method: "PUT", expires_in: typeof command["expires_in"] === "number" ? command["expires_in"] : 900 });
+        const { name, max_size: maxSize, types, expires_in: expiresIn } = command;
+        const folder = typeof name === "string" && name.endsWith("/");
+        if (typeof name !== "string" || !namePattern.test(folder ? name.slice(0, -1) : name)) return send(response, 400, { error: "invalid_name" });
+        if (!(types === undefined || (Array.isArray(types) && types.length <= 8 && types.every(t => typeof t === "string" && mediaPattern.test(t))))) return send(response, 400, { error: "invalid_type" });
+        if (!(maxSize === undefined || (typeof maxSize === "number" && Number.isSafeInteger(maxSize) && maxSize > 0)) || !(expiresIn === undefined || (typeof expiresIn === "number" && Number.isInteger(expiresIn) && expiresIn > 0 && expiresIn <= uploadLife))) return send(response, 400, { error: "invalid_body" });
+        if ((maxSize ?? 0) > maxObject) return send(response, 413, { error: "too_large" });
+        const token = newToken(), life = (expiresIn as number | undefined) ?? uploadLife;
+        uploads.set(token, { name, maxSize: (maxSize as number | undefined) ?? maxObject, types: (types as string[] | undefined) ?? [], until: Date.now() + life * 1000 });
+        return send(response, 200, { url: `${chest.api}${uploadPath}${token}`, method: "PUT", expires_in: life });
       }
       const from = command["from"], to = command["to"];
       if (typeof from !== "string" || typeof to !== "string" || !namePattern.test(from) || !namePattern.test(to)) return send(response, 400, { error: "invalid_name" });
@@ -271,6 +349,39 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
     send(response, 404, { error: "not_found" });
   }
 
+  // The team host's part of the files, on the fake's own origin: a link it
+  // signed opens the content it was signed for, for its life, as long as
+  // the file is that content — a thumbnail is the image itself, a fake does
+  // not reduce it —; an upload token takes one file, once, within its life,
+  // of the types it names and up to its size, whose first bytes are those of
+  // its type, named by the Chest in a folder. Nobody's session is checked:
+  // the Chest's own front does that.
+  const links = new Map<string, { name: string; object: FakeFile; until: number; download: boolean }>();
+  const uploads = new Map<string, { name: string; maxSize: number; types: string[]; until: number }>();
+  async function front(request: IncomingMessage, response: ServerResponse, url: URL): Promise<void> {
+    if (request.method === "GET" && url.pathname.startsWith(linkPath) && !url.pathname.startsWith(uploadPath)) {
+      const link = links.get(url.pathname.slice(linkPath.length));
+      if (!link || link.until < Date.now() || files.get(link.name) !== link.object) return send(response, 404, { error: "not_found" });
+      const headers: Record<string, string> = { "Content-Type": link.object.type, "Content-Length": String(link.object.data.byteLength), "X-Content-Type-Options": "nosniff" };
+      if (link.download) headers["Content-Disposition"] = `attachment; filename="${link.name.split("/").pop()}"`;
+      return void response.writeHead(200, headers).end(link.object.data);
+    }
+    if (request.method !== "PUT" || !url.pathname.startsWith(uploadPath)) return send(response, 404, { error: "not_found" });
+    const token = url.pathname.slice(uploadPath.length), grant = uploads.get(token);
+    uploads.delete(token);
+    if (!grant || grant.until < Date.now()) return send(response, 403, { error: "invalid_token" });
+    const type = (request.headers["content-type"] ?? "").split(";")[0]!.trim().toLowerCase();
+    if (!mediaPattern.test(type) || type.endsWith("/*") || (grant.types.length > 0 && !grant.types.some(t => t === type || (t.endsWith("/*") && type.startsWith(t.slice(0, -1)))))) return send(response, 415, { error: "type_refused" });
+    const data = await body(request, grant.maxSize);
+    if (data === null) return send(response, 413, { error: "too_large" });
+    if (signatures[type] && !signatures[type]!(data)) return send(response, 400, { error: "type_mismatch" });
+    const name = grant.name.endsWith("/") ? grant.name + randomBytes(10).toString("hex") + (extensions[type] ?? "") : grant.name;
+    const total = [...files.entries()].reduce((sum, [n, f]) => n === name ? sum : sum + f.data.byteLength, 0);
+    if (total + data.length > maxTotal || (!files.has(name) && files.size >= maxObjects)) return send(response, 429, { error: "quota_exceeded" });
+    files.set(name, { data: new Uint8Array(data), type, updated: new Date().toISOString() });
+    send(response, 201, { name, type, size: data.length });
+  }
+
   // The windows of the notification quotas, each from the first call it
   // counts: the tool's recipients this hour, each member's items this day,
   // the tool's badge writes this minute.
@@ -369,6 +480,100 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
     send(response, 200, { delivered: kept, skipped: ids.filter(id => !access(id)) });
   }
 
+  // Its AI: the declared aliases, the spending this month, the requests this
+  // minute.
+  const mapped = (options.ai?.models ?? aliasOrder.map((alias): FakeAiModel => ({ alias, model: "fake-" + alias }))).map(m => ({ alias: m.alias, model: m.model, provider: m.provider ?? "openrouter", input: m.input ?? 1, output: m.output ?? 2 }));
+  const cap = options.ai?.cap ?? 5;
+  const aiMinute: Window = { start: 0, count: 0 };
+  let spent = 0;
+  const month = () => new Date().toISOString().slice(0, 7);
+  const resets = () => {
+    const now = new Date();
+    return new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth() + 1, 1)).toISOString().replace(".000Z", "Z");
+  };
+  const spend = (m: { input: number; output: number }, input: number, output: number): number => {
+    const cost = Math.round((input * m.input + output * m.output) / 1e6 * 1e6) / 1e6;
+    spent = Math.round((spent + cost) * 1e6) / 1e6;
+    return cost;
+  };
+  // echo is the text of the last user message: its text, or its text parts.
+  const echo = (request: Record<string, unknown>): string => {
+    const last = [...request["messages"] as { role?: unknown; content?: unknown }[]].reverse().find(m => m?.role === "user")?.content;
+    return typeof last === "string" ? last : Array.isArray(last) ? last.map(p => (p as { text?: unknown })?.text).filter(t => typeof t === "string").join("") : "";
+  };
+
+  async function aiRoute(request: IncomingMessage, response: ServerResponse, url: URL): Promise<void> {
+    const raw = request.method === "POST" ? await body(request, maxAiBody) : null;
+    let command: Record<string, unknown> | null = null;
+    try {
+      const value = JSON.parse(raw?.toString() ?? "") as unknown;
+      if (value !== null && typeof value === "object" && !Array.isArray(value)) command = value as Record<string, unknown>;
+    } catch {
+      command = null;
+    }
+    chest.ai.push({ path: url.pathname, body: command });
+    if (!capabilities.has("ai")) return send(response, 403, { error: "capability_not_granted" });
+    if (request.method === "GET" && url.pathname === "/ai/models") return send(response, 200, { models: [...mapped].sort((a, b) => aliasOrder.indexOf(a.alias) - aliasOrder.indexOf(b.alias)) });
+    if (request.method === "GET" && url.pathname === "/ai/usage") return send(response, 200, { month: month(), spent, cap, resets: resets() });
+    if (request.method !== "POST" || (url.pathname !== "/ai/chat" && url.pathname !== "/ai/embeddings")) return send(response, 404, { error: "not_found" });
+    if (raw === null) return send(response, 413, { error: "too_large" });
+    const chat = url.pathname === "/ai/chat";
+    if (!command || !Object.keys(command).every(k => chat ? chatKeys.includes(k) : ["model", "input", "dimensions", "member"].includes(k))) return send(response, 400, { error: "invalid_body" });
+    const model = mapped.find(m => m.alias === command!["model"]);
+    if (!model) return send(response, 403, { error: "model_not_allowed" });
+    const member = command["member"];
+    if (member !== undefined && (typeof member !== "string" || !memberIdPattern.test(member))) return send(response, 400, { error: "invalid_body" });
+    const now = Date.now();
+    // refused says the refusal of a valid request, if any: the rate, the
+    // provider, the cap.
+    const refused = (): boolean => {
+      const busy = live(aiMinute, 60_000, now) && aiMinute.count >= aiPerMinute;
+      if (busy) send(response, 429, { error: "rate_limited" }, wait(aiMinute, 60_000, now));
+      else {
+        count(aiMinute, 60_000, now, 1);
+        if (options.ai?.unavailable) send(response, options.ai.unavailable === "provider_key_invalid" ? 502 : 503, { error: options.ai.unavailable });
+        else if (spent >= cap) send(response, 402, { error: "cap_reached", scope: "tool", resets: resets() });
+      }
+      return response.headersSent;
+    };
+    if (!chat) {
+      const input = command["input"], texts = typeof input === "string" ? [input] : input, dimensions = command["dimensions"] ?? fakeDimensions;
+      if (!Array.isArray(texts) || texts.length < 1 || texts.length > maxInputs || !texts.every(t => typeof t === "string") || typeof dimensions !== "number" || !Number.isInteger(dimensions) || dimensions < 1 || dimensions > 4096) return send(response, 400, { error: "invalid_body" });
+      if (refused()) return;
+      const used = texts.reduce((sum: number, t: string) => sum + tokensOf(t), 0);
+      const cost = spend(model, used, 0);
+      return send(response, 200, { object: "list", data: texts.map((t: string, index) => ({ object: "embedding", index, embedding: vectorOf(t, dimensions) })), model: model.model, usage: { prompt_tokens: used, total_tokens: used, cost } });
+    }
+    const messages = command["messages"], limit = command["max_tokens"] ?? 4096;
+    if (!Array.isArray(messages) || messages.length < 1 || !messages.every(m => m !== null && typeof m === "object" && typeof (m as { role?: unknown }).role === "string") || typeof limit !== "number" || !Number.isInteger(limit) || limit < 1 || limit > maxAiOutput || !(command["stream"] === undefined || typeof command["stream"] === "boolean")) return send(response, 400, { error: "invalid_body" });
+    if (refused()) return;
+    const given = options.ai?.reply ? options.ai.reply(command) : echo(command);
+    const text = typeof given === "string" ? given : given.text ?? "";
+    const calls = (typeof given === "string" ? [] : given.toolCalls ?? []).map((c, i) => ({ id: c.id ?? `call_${i + 1}`, type: "function" as const, function: { name: c.name, arguments: c.arguments } }));
+    const input = tokensOf(JSON.stringify(messages)), output = tokensOf(text + calls.map(c => c.function.name + c.function.arguments).join(""));
+    const usage = { prompt_tokens: input, completion_tokens: output, total_tokens: input + output, prompt_tokens_details: { cached_tokens: 0 }, cost: spend(model, input, output) };
+    const finish = calls.length ? "tool_calls" : "stop";
+    const head = { id: "chatcmpl-fake", created: Math.floor(now / 1000), model: model.model };
+    if (command["stream"] !== true) {
+      return send(response, 200, { ...head, object: "chat.completion", choices: [{ index: 0, message: { role: "assistant", content: calls.length && !text ? null : text, ...(calls.length ? { tool_calls: calls } : {}) }, finish_reason: finish }], usage });
+    }
+    // Streamed: the role, the text word by word, each call's name then its
+    // arguments in two halves, the finish reason, the usage, [DONE].
+    const chunk = (choices: unknown[], extra: Record<string, unknown> = {}) => `data: ${JSON.stringify({ ...head, object: "chat.completion.chunk", choices, ...extra })}\n\n`;
+    const delta = (d: Record<string, unknown>, finishReason: string | null = null) => chunk([{ index: 0, delta: d, finish_reason: finishReason }]);
+    const parts = [delta({ role: "assistant", content: "" })];
+    for (const word of text.match(/\S+\s*|\s+/gu) ?? []) parts.push(delta({ content: word }));
+    calls.forEach((c, index) => {
+      const half = Math.ceil(c.function.arguments.length / 2);
+      parts.push(delta({ tool_calls: [{ index, id: c.id, type: "function", function: { name: c.function.name, arguments: "" } }] }));
+      for (const piece of [c.function.arguments.slice(0, half), c.function.arguments.slice(half)]) if (piece) parts.push(delta({ tool_calls: [{ index, function: { arguments: piece } }] }));
+    });
+    parts.push(delta({}, finish), chunk([], { usage }), "data: [DONE]\n\n");
+    response.writeHead(200, { "Content-Type": "text/event-stream", "Cache-Control": "no-cache" });
+    for (const part of parts) response.write(part);
+    response.end();
+  }
+
   // The acknowledgment of an erasure the tool was told of (emit).
   async function erasuresRoute(request: IncomingMessage, response: ServerResponse, url: URL): Promise<void> {
     if (!receives.includes("member.*")) return send(response, 403, { error: "capability_not_granted" });
@@ -383,7 +588,9 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
 
   const server = createServer((request, response) => {
     const url = new URL(request.url ?? "/", "http://127.0.0.1");
-    const route = url.pathname.startsWith("/erasures/") ? erasuresRoute
+    const route = url.pathname.startsWith("/_chest/files/") ? front
+      : url.pathname.startsWith("/erasures/") ? erasuresRoute
+      : url.pathname.startsWith("/ai/") ? aiRoute
       : url.pathname === "/members" || url.pathname.startsWith("/members/") || url.pathname === "/groups" ? members
       : url.pathname === "/files" || url.pathname.startsWith("/files/") ? filesRoute
       : url.pathname === "/badges" || url.pathname.startsWith("/badges/") || url.pathname.startsWith("/notifications") ? notificationsRoute : null;
@@ -391,20 +598,38 @@ export async function fakeChest(options: FakeChestOptions = {}): Promise<FakeChe
     route(request, response, url).catch(() => { if (!response.headersSent) send(response, 503, { error: "unavailable" }); else response.destroy(); });
   });
   await new Promise<void>(resolve => server.listen(0, "127.0.0.1", resolve));
-  const saved = Object.fromEntries(["CHEST_API", "CHEST_TOKEN", "CHEST_TOOL"].map(name => [name, process.env[name]]));
+  const saved = Object.fromEntries(["CHEST_API", "CHEST_TOKEN", "CHEST_TOOL", "CHEST_ORGANIZATION", "CHEST_TIME_ZONE", "CHEST_LANGUAGE", "CHEST_CURRENCY", "CHEST_TEAM_URL", "CHEST_PUBLIC_URL"].map(name => [name, process.env[name]]));
   chest.api = "http://127.0.0.1:" + (server.address() as AddressInfo).port;
-  Object.assign(process.env, { CHEST_API: chest.api, CHEST_TOKEN: token, CHEST_TOOL: tool });
+  // Its links are taken by the files module of this process, for as long as
+  // it runs (api.ts, fakeOrigins).
+  fakeOrigins.add(chest.api);
+  const publicUrl = options.chest?.publicUrl === undefined ? `https://${tool}.chest.test` : options.chest.publicUrl;
+  Object.assign(process.env, { CHEST_API: chest.api, CHEST_TOKEN: token, CHEST_TOOL: tool, CHEST_ORGANIZATION: options.chest?.organization ?? "Test organization", CHEST_TIME_ZONE: options.chest?.timeZone ?? "UTC", CHEST_LANGUAGE: options.chest?.language ?? "en", CHEST_CURRENCY: options.chest?.currency ?? "EUR", CHEST_TEAM_URL: options.chest?.teamUrl ?? `https://${tool}-chest.chest.test` });
+  if (publicUrl === null) delete process.env["CHEST_PUBLIC_URL"];
+  else process.env["CHEST_PUBLIC_URL"] = publicUrl;
   forget();
-  chest.emit = async (event, to) => {
-    const id = event.id ?? "evt_" + Array.from(randomBytes(26), b => "abcdefghijklmnopqrstuvwxyz234567"[b & 31]).join("");
-    const body = JSON.stringify({ id, type: event.type, occurredAt: event.occurredAt ?? new Date().toISOString(), data: event.data });
-    if (event.type === "member.erased") erasures.add(event.data.erasure);
-    const request = new Request(typeof to === "string" ? to.replace(/\/$/u, "") + "/chest-events" : "http://tool.test/chest-events", { method: "POST", headers: { "Content-Type": "application/json", "Chest-Event": signEvent(id, body, { token, tool }) }, body });
+  // post delivers what the Chest posts to the tool on a channel — at its
+  // address, or to a handler — and says the status it answered.
+  const post = async (path: string, channel: Channel, id: string, body: string, to: string | ((request: Request) => Response | Promise<Response>)): Promise<number> => {
+    const request = new Request((typeof to === "string" ? to.replace(/\/$/u, "") : "http://tool.test") + path, { method: "POST", headers: { "Content-Type": "application/json", [channel.header]: sign(channel, id, body, { token, tool }) }, body });
     const answer = typeof to === "string" ? await fetch(request, { redirect: "manual" }) : await to(request);
     await answer.body?.cancel();
     return answer.status;
   };
+  const newId = (prefix: string): string => prefix + Array.from(randomBytes(26), b => "abcdefghijklmnopqrstuvwxyz234567"[b & 31]).join("");
+  chest.emit = async (event, to) => {
+    const id = event.id ?? newId("evt_");
+    const body = JSON.stringify({ id, type: event.type, occurredAt: event.occurredAt ?? new Date().toISOString(), data: event.data });
+    if (event.type === "member.erased") erasures.add(event.data.erasure);
+    return post("/chest-events", eventChannel, id, body, to);
+  };
+  chest.run = async (name, to, run = {}) => {
+    const id = run.id ?? newId("run_");
+    const body = JSON.stringify({ id, name, scheduledAt: run.scheduledAt ?? new Date().toISOString(), attempt: run.attempt ?? 1 });
+    return post("/chest-schedules", scheduleChannel, id, body, to);
+  };
   chest.close = async () => {
+    fakeOrigins.delete(chest.api);
     server.closeAllConnections();
     await new Promise<void>(resolve => server.close(() => resolve()));
     for (const [name, value] of Object.entries(saved)) {

package/dist/index.d.ts

@@ -1,8 +1,11 @@
 export * from "./src/errors.js";
 export * from "./src/member.js";
+export * from "./src/chest.js";
 export * from "./src/database.js";
 export * as files from "./src/files.js";
 export * as members from "./src/members.js";
 export * as notifications from "./src/notifications.js";
 export * as events from "./src/events.js";
+export * as schedules from "./src/schedules.js";
+export * as ai from "./src/ai.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../client/index.ts"],"names":[],"mappings":"AAOA,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,OAAO,KAAK,KAAK,MAAM,gBAAgB,CAAC;AACxC,OAAO,KAAK,OAAO,MAAM,kBAAkB,CAAC;AAC5C,OAAO,KAAK,aAAa,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../client/index.ts"],"names":[],"mappings":"AAOA,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,OAAO,KAAK,KAAK,MAAM,gBAAgB,CAAC;AACxC,OAAO,KAAK,OAAO,MAAM,kBAAkB,CAAC;AAC5C,OAAO,KAAK,aAAa,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,MAAM,iBAAiB,CAAC;AAC1C,OAAO,KAAK,SAAS,MAAM,oBAAoB,CAAC;AAChD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC"}

package/dist/index.js

@@ -1,15 +1,18 @@
 // The package root: every published module a tool's code uses (tool contract
-// v2). Each one is also its own subpath (@argentic/chest-sdk/member,
-// /database, /files, /members, /notifications, /events, /errors), which
-// pulls in nothing else. The files, members, notifications and events APIs
-// are namespaces here, as their names (get, list, stat, move, notify,
-// verify…) are too plain to stand alone. @argentic/chest-sdk/testing is for a tool's tests only, and
+// 0.4). Each one is also its own subpath (@argentic/chest-sdk/member, /chest,
+// /database, /files, /members, /notifications, /events, /schedules, /ai, /errors), which
+// pulls in nothing else. The files, members, notifications, events, schedules and ai
+// APIs are namespaces here, as their names (get, list, stat, move, notify,
+// verify, chat…) are too plain to stand alone. @argentic/chest-sdk/testing is for a tool's tests only, and
 // is not here.
 export * from "./src/errors.js";
 export * from "./src/member.js";
+export * from "./src/chest.js";
 export * from "./src/database.js";
 export * as files from "./src/files.js";
 export * as members from "./src/members.js";
 export * as notifications from "./src/notifications.js";
 export * as events from "./src/events.js";
+export * as schedules from "./src/schedules.js";
+export * as ai from "./src/ai.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../client/index.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAC7E,qEAAqE;AACrE,wEAAwE;AACxE,2EAA2E;AAC3E,sEAAsE;AACtE,qGAAqG;AACrG,eAAe;AACf,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,OAAO,KAAK,KAAK,MAAM,gBAAgB,CAAC;AACxC,OAAO,KAAK,OAAO,MAAM,kBAAkB,CAAC;AAC5C,OAAO,KAAK,aAAa,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../client/index.ts"],"names":[],"mappings":"AAAA,6EAA6E;AAC7E,8EAA8E;AAC9E,yFAAyF;AACzF,qFAAqF;AACrF,2EAA2E;AAC3E,2GAA2G;AAC3G,eAAe;AACf,cAAc,iBAAiB,CAAC;AAChC,cAAc,iBAAiB,CAAC;AAChC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,OAAO,KAAK,KAAK,MAAM,gBAAgB,CAAC;AACxC,OAAO,KAAK,OAAO,MAAM,kBAAkB,CAAC;AAC5C,OAAO,KAAK,aAAa,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,MAAM,iBAAiB,CAAC;AAC1C,OAAO,KAAK,SAAS,MAAM,oBAAoB,CAAC;AAChD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC"}