@s2-dev/streamstore 0.3.14 → 0.4.2

package/mcp-server/tools/streamRead.d.ts

@@ -0,0 +1,8 @@
+import * as operations from "../../models/operations/index.js";
+import { ToolDefinition } from "../tools.js";
+declare const args: {
+    request: import("zod").ZodType<operations.ReadRequest, import("zod").ZodTypeDef, unknown>;
+};
+export declare const tool$streamRead: ToolDefinition<typeof args>;
+export {};
+//# sourceMappingURL=streamRead.d.ts.map

package/mcp-server/tools/streamRead.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamRead.d.ts","sourceRoot":"","sources":["../../src/mcp-server/tools/streamRead.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,UAAU,MAAM,kCAAkC,CAAC;AAC/D,OAAO,EAAgB,cAAc,EAAE,MAAM,aAAa,CAAC;AAE3D,QAAA,MAAM,IAAI;;CAET,CAAC;AAEF,eAAO,MAAM,eAAe,EAAE,cAAc,CAAC,OAAO,IAAI,CAyBvD,CAAC"}

package/mcp-server/tools/streamRead.js

@@ -0,0 +1,65 @@
+"use strict";
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tool$streamRead = void 0;
+const streamRead_js_1 = require("../../funcs/streamRead.js");
+const operations = __importStar(require("../../models/operations/index.js"));
+const tools_js_1 = require("../tools.js");
+const args = {
+    request: operations.ReadRequest$inboundSchema,
+};
+exports.tool$streamRead = {
+    name: "stream_read",
+    description: `Retrieve records.
+
+Retrieve a batch of records, or set \`Accept: text/event-stream\` to stream using server-sent events.`,
+    scopes: ["read"],
+    args,
+    tool: async (client, args, ctx) => {
+        const [result, apiCall] = await (0, streamRead_js_1.streamRead)(client, args.request, { fetchOptions: { signal: ctx.signal } }).$inspect();
+        if (!result.ok) {
+            return {
+                content: [{ type: "text", text: result.error.message }],
+                isError: true,
+            };
+        }
+        const value = result.value;
+        return (0, tools_js_1.formatResult)(value, apiCall);
+    },
+};
+//# sourceMappingURL=streamRead.js.map

package/mcp-server/tools/streamRead.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamRead.js","sourceRoot":"","sources":["../../src/mcp-server/tools/streamRead.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,6DAAuD;AACvD,6EAA+D;AAC/D,0CAA2D;AAE3D,MAAM,IAAI,GAAG;IACX,OAAO,EAAE,UAAU,CAAC,yBAAyB;CAC9C,CAAC;AAEW,QAAA,eAAe,GAAgC;IAC1D,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE;;sGAEuF;IACpG,MAAM,EAAE,CAAC,MAAM,CAAC;IAChB,IAAI;IACJ,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAChC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,IAAA,0BAAU,EACxC,MAAM,EACN,IAAI,CAAC,OAAO,EACZ,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,CACzC,CAAC,QAAQ,EAAE,CAAC;QAEb,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBACvD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAE3B,OAAO,IAAA,uBAAY,EAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC;CACF,CAAC"}

package/mcp-server/tools.d.ts

@@ -0,0 +1,25 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { objectOutputType, ZodRawShape, ZodTypeAny } from "zod";
+import { S2Core } from "../core.js";
+import { ConsoleLogger } from "./console-logger.js";
+import { MCPScope } from "./scopes.js";
+export type ToolDefinition<Args extends undefined | ZodRawShape = undefined> = Args extends ZodRawShape ? {
+    name: string;
+    description: string;
+    scopes?: MCPScope[];
+    args: Args;
+    tool: (client: S2Core, args: objectOutputType<Args, ZodTypeAny>, extra: RequestHandlerExtra) => CallToolResult | Promise<CallToolResult>;
+} : {
+    name: string;
+    description: string;
+    scopes?: MCPScope[];
+    args?: undefined;
+    tool: (client: S2Core, extra: RequestHandlerExtra) => CallToolResult | Promise<CallToolResult>;
+};
+export declare function formatResult(value: unknown, init: {
+    response?: Response | undefined;
+}): Promise<CallToolResult>;
+export declare function createRegisterTool(logger: ConsoleLogger, server: McpServer, sdk: S2Core, allowedScopes: Set<MCPScope>): <A extends ZodRawShape | undefined>(tool: ToolDefinition<A>) => void;
+//# sourceMappingURL=tools.d.ts.map

package/mcp-server/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/mcp-server/tools.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;AACnF,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,KAAK,CAAC;AAChE,OAAO,EAAE,MAAM,EAAE,MAAM,YAAY,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAGvC,MAAM,MAAM,cAAc,CAAC,IAAI,SAAS,SAAS,GAAG,WAAW,GAAG,SAAS,IACzE,IAAI,SAAS,WAAW,GAAG;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC;IACpB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,CACJ,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,gBAAgB,CAAC,IAAI,EAAE,UAAU,CAAC,EACxC,KAAK,EAAE,mBAAmB,KACvB,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;CAC/C,GACC;IACA,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC;IACpB,IAAI,CAAC,EAAE,SAAS,CAAC;IACjB,IAAI,EAAE,CACJ,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,mBAAmB,KACvB,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;CAC/C,CAAC;AAEN,wBAAsB,YAAY,CAChC,KAAK,EAAE,OAAO,EACd,IAAI,EAAE;IAAE,QAAQ,CAAC,EAAE,QAAQ,GAAG,SAAS,CAAA;CAAE,GACxC,OAAO,CAAC,cAAc,CAAC,CAkCzB;AAkBD,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,aAAa,EACrB,MAAM,EAAE,SAAS,EACjB,GAAG,EAAE,MAAM,EACX,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,GAC3B,CAAC,CAAC,SAAS,WAAW,GAAG,SAAS,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,KAAK,IAAI,CAmBtE"}

package/mcp-server/tools.js

@@ -0,0 +1,74 @@
+"use strict";
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatResult = formatResult;
+exports.createRegisterTool = createRegisterTool;
+const shared_js_1 = require("./shared.js");
+async function formatResult(value, init) {
+    if (typeof value === "undefined") {
+        return { content: [] };
+    }
+    const { response } = init;
+    const contentType = response?.headers.get("content-type") ?? "";
+    let content = [];
+    if (contentType.search(/\bjson\b/g)) {
+        content = [{ type: "text", text: JSON.stringify(value) }];
+    }
+    else if (contentType.startsWith("text/event-stream")
+        && (0, shared_js_1.isAsyncIterable)(value)) {
+        content = await consumeSSE(value);
+    }
+    else if (contentType.startsWith("text/") && typeof value === "string") {
+        content = [{ type: "text", text: value }];
+    }
+    else if ((0, shared_js_1.isBinaryData)(value) && contentType.startsWith("image/")) {
+        const data = await (0, shared_js_1.valueToBase64)(value);
+        content = data == null
+            ? []
+            : [{ type: "image", data, mimeType: contentType }];
+    }
+    else {
+        return {
+            content: [{
+                    type: "text",
+                    text: `Unsupported content type: "${contentType}"`,
+                }],
+            isError: true,
+        };
+    }
+    return { content };
+}
+async function consumeSSE(value) {
+    const content = [];
+    for await (const chunk of value) {
+        if (typeof chunk === "string") {
+            content.push({ type: "text", text: chunk });
+        }
+        else {
+            content.push({ type: "text", text: JSON.stringify(chunk) });
+        }
+    }
+    return content;
+}
+function createRegisterTool(logger, server, sdk, allowedScopes) {
+    return (tool) => {
+        const toolScopes = tool.scopes ?? [];
+        if (!toolScopes.every((s) => allowedScopes.has(s))) {
+            return;
+        }
+        if (tool.args) {
+            server.tool(tool.name, tool.description, tool.args, async (args, ctx) => {
+                return tool.tool(sdk, args, ctx);
+            });
+        }
+        else {
+            server.tool(tool.name, tool.description, async (ctx) => {
+                return tool.tool(sdk, ctx);
+            });
+        }
+        logger.debug("Registered tool", { name: tool.name });
+    };
+}
+//# sourceMappingURL=tools.js.map

package/mcp-server/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/mcp-server/tools.ts"],"names":[],"mappings":";AAAA;;GAEG;;AAkCH,oCAqCC;AAkBD,gDAwBC;AAxGD,2CAA2E;AAyBpE,KAAK,UAAU,YAAY,CAChC,KAAc,EACd,IAAyC;IAEzC,IAAI,OAAO,KAAK,KAAK,WAAW,EAAE,CAAC;QACjC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IACzB,CAAC;IAED,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC1B,MAAM,WAAW,GAAG,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAChE,IAAI,OAAO,GAA8B,EAAE,CAAC;IAE5C,IAAI,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QACpC,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;SAAM,IACL,WAAW,CAAC,UAAU,CAAC,mBAAmB,CAAC;WACxC,IAAA,2BAAe,EAAC,KAAK,CAAC,EACzB,CAAC;QACD,OAAO,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;SAAM,IAAI,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxE,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5C,CAAC;SAAM,IAAI,IAAA,wBAAY,EAAC,KAAK,CAAC,IAAI,WAAW,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,GAAG,MAAM,IAAA,yBAAa,EAAC,KAAK,CAAC,CAAC;QACxC,OAAO,GAAG,IAAI,IAAI,IAAI;YACpB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;IACvD,CAAC;SAAM,CAAC;QACN,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,8BAA8B,WAAW,GAAG;iBACnD,CAAC;YACF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,UAAU,CACvB,KAA6B;IAE7B,MAAM,OAAO,GAA8B,EAAE,CAAC;IAE9C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;QAChC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAgB,kBAAkB,CAChC,MAAqB,EACrB,MAAiB,EACjB,GAAW,EACX,aAA4B;IAE5B,OAAO,CAAoC,IAAuB,EAAQ,EAAE;QAC1E,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBACtE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACnC,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;gBACrD,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC7B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC;AACJ,CAAC"}

package/models/components/endpoints.d.ts

@@ -0,0 +1,67 @@
+type Authority = string;
+type BasinName = string;
+/**
+ * S2 cloud environment to connect with.
+ */
+export declare enum S2Cloud {
+    /** S2 running on AWS */
+    Aws = "aws"
+}
+/**
+ * Endpoint for connecting to an S2 basin.
+ */
+export declare enum BasinEndpointKind {
+    ParentZone = "ParentZone",
+    Direct = "Direct"
+}
+export type BasinEndpoint = {
+    kind: BasinEndpointKind.ParentZone;
+    authority: Authority;
+} | {
+    kind: BasinEndpointKind.Direct;
+    authority: Authority;
+};
+/**
+ * Endpoints for the S2 environment.
+ *
+ * You can find the S2 endpoints in our [documentation](https://s2.dev/docs/interface/endpoints)
+ */
+export declare class S2Endpoints {
+    /** Used by `AccountService` requests */
+    account: Authority;
+    /** Used by `BasinService` and `StreamService` requests */
+    basin: BasinEndpoint;
+    constructor(account: Authority, basin: BasinEndpoint);
+    /**
+     * Get S2 endpoints for the specified cloud.
+     */
+    static forCloud(cloud: S2Cloud): S2Endpoints;
+    /**
+     * Get S2 endpoints for the specified cell.
+     */
+    static forCell(cloud: S2Cloud, cellId: string): S2Endpoints;
+    /**
+     * Get S2 endpoints from environment variables.
+     *
+     * The following environment variables are used:
+     * - `S2_CLOUD`: Valid S2 cloud name. Defaults to AWS.
+     * - `S2_ACCOUNT_ENDPOINT`: Overrides the account endpoint.
+     * - `S2_BASIN_ENDPOINT`: Overrides the basin endpoint. The prefix `"{basin}."` indicates the
+     *   basin endpoint is `ParentZone` else `Direct`.
+     */
+    static fromEnv(): S2Endpoints;
+}
+/**
+ * Client kind for different S2 services
+ */
+export type ClientKind = {
+    kind: 'Account';
+} | {
+    kind: 'Basin';
+    basin: BasinName;
+};
+export declare const ClientKind: {
+    toAuthority(kind: ClientKind, endpoints: S2Endpoints): Authority;
+};
+export {};
+//# sourceMappingURL=endpoints.d.ts.map

package/models/components/endpoints.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"endpoints.d.ts","sourceRoot":"","sources":["../../src/models/components/endpoints.ts"],"names":[],"mappings":"AAAA,KAAK,SAAS,GAAG,MAAM,CAAC;AAGxB,KAAK,SAAS,GAAG,MAAM,CAAC;AAExB;;GAEG;AACH,oBAAY,OAAO;IACf,wBAAwB;IACxB,GAAG,QAAQ;CACd;AAED;;GAEG;AACH,oBAAY,iBAAiB;IACzB,UAAU,eAAe;IACzB,MAAM,WAAW;CACpB;AAED,MAAM,MAAM,aAAa,GAAG;IACxB,IAAI,EAAE,iBAAiB,CAAC,UAAU,CAAC;IACnC,SAAS,EAAE,SAAS,CAAC;CACxB,GAAG;IACA,IAAI,EAAE,iBAAiB,CAAC,MAAM,CAAC;IAC/B,SAAS,EAAE,SAAS,CAAC;CACxB,CAAC;AASF;;;;GAIG;AACH,qBAAa,WAAW;IACpB,wCAAwC;IACjC,OAAO,EAAE,SAAS,CAAC;IAC1B,0DAA0D;IACnD,KAAK,EAAE,aAAa,CAAC;gBAEhB,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa;IAKpD;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,WAAW;IAW5C;;OAEG;IACH,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,WAAW;IAW3D;;;;;;;;OAQG;IACH,MAAM,CAAC,OAAO,IAAI,WAAW;CA8ChC;AAED;;GAEG;AACH,MAAM,MAAM,UAAU,GAChB;IAAE,IAAI,EAAE,SAAS,CAAA;CAAE,GACnB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,SAAS,CAAA;CAAE,CAAC;AAE1C,eAAO,MAAM,UAAU;sBACD,UAAU,aAAa,WAAW,GAAG,SAAS;CAanE,CAAC"}

package/models/components/endpoints.js

@@ -0,0 +1,126 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClientKind = exports.S2Endpoints = exports.BasinEndpointKind = exports.S2Cloud = void 0;
+/**
+ * S2 cloud environment to connect with.
+ */
+var S2Cloud;
+(function (S2Cloud) {
+    /** S2 running on AWS */
+    S2Cloud["Aws"] = "aws";
+})(S2Cloud || (exports.S2Cloud = S2Cloud = {}));
+/**
+ * Endpoint for connecting to an S2 basin.
+ */
+var BasinEndpointKind;
+(function (BasinEndpointKind) {
+    BasinEndpointKind["ParentZone"] = "ParentZone";
+    BasinEndpointKind["Direct"] = "Direct";
+})(BasinEndpointKind || (exports.BasinEndpointKind = BasinEndpointKind = {}));
+class S2EndpointError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'S2EndpointError';
+    }
+}
+/**
+ * Endpoints for the S2 environment.
+ *
+ * You can find the S2 endpoints in our [documentation](https://s2.dev/docs/interface/endpoints)
+ */
+class S2Endpoints {
+    constructor(account, basin) {
+        this.account = account;
+        this.basin = basin;
+    }
+    /**
+     * Get S2 endpoints for the specified cloud.
+     */
+    static forCloud(cloud) {
+        const authority = `${cloud}.s2.dev`;
+        return new S2Endpoints(authority, {
+            kind: BasinEndpointKind.ParentZone,
+            authority: `b.${cloud}.s2.dev`
+        });
+    }
+    /**
+     * Get S2 endpoints for the specified cell.
+     */
+    static forCell(cloud, cellId) {
+        const cellEndpoint = `${cellId}.o.${cloud}.s2.dev`;
+        return new S2Endpoints(cellEndpoint, {
+            kind: BasinEndpointKind.Direct,
+            authority: cellEndpoint
+        });
+    }
+    /**
+     * Get S2 endpoints from environment variables.
+     *
+     * The following environment variables are used:
+     * - `S2_CLOUD`: Valid S2 cloud name. Defaults to AWS.
+     * - `S2_ACCOUNT_ENDPOINT`: Overrides the account endpoint.
+     * - `S2_BASIN_ENDPOINT`: Overrides the basin endpoint. The prefix `"{basin}."` indicates the
+     *   basin endpoint is `ParentZone` else `Direct`.
+     */
+    static fromEnv() {
+        let cloud;
+        try {
+            const cloudEnv = process.env['S2_CLOUD'] || S2Cloud.Aws;
+            if (Object.values(S2Cloud).includes(cloudEnv)) {
+                cloud = cloudEnv;
+            }
+            else {
+                throw new S2EndpointError(`Invalid S2_CLOUD: ${cloudEnv}`);
+            }
+        }
+        catch (err) {
+            const error = err;
+            throw new S2EndpointError(`Invalid S2_CLOUD: ${error.message}`);
+        }
+        const endpoints = S2Endpoints.forCloud(cloud);
+        const accountEndpoint = process.env['S2_ACCOUNT_ENDPOINT'];
+        if (accountEndpoint !== undefined) {
+            if (typeof accountEndpoint !== 'string') {
+                throw new S2EndpointError('Invalid S2_ACCOUNT_ENDPOINT: not string');
+            }
+            endpoints.account = accountEndpoint;
+        }
+        const basinEndpoint = process.env['S2_BASIN_ENDPOINT'];
+        if (basinEndpoint !== undefined) {
+            if (typeof basinEndpoint !== 'string') {
+                throw new S2EndpointError('Invalid S2_BASIN_ENDPOINT: not string');
+            }
+            if (basinEndpoint.startsWith('{basin}.')) {
+                const parentZone = basinEndpoint.substring(7);
+                endpoints.basin = {
+                    kind: BasinEndpointKind.ParentZone,
+                    authority: parentZone
+                };
+            }
+            else {
+                endpoints.basin = {
+                    kind: BasinEndpointKind.Direct,
+                    authority: basinEndpoint
+                };
+            }
+        }
+        return endpoints;
+    }
+}
+exports.S2Endpoints = S2Endpoints;
+exports.ClientKind = {
+    toAuthority(kind, endpoints) {
+        switch (kind.kind) {
+            case 'Account':
+                return endpoints.account;
+            case 'Basin':
+                switch (endpoints.basin.kind) {
+                    case BasinEndpointKind.ParentZone:
+                        return `${kind.basin}.${endpoints.basin.authority}`;
+                    case BasinEndpointKind.Direct:
+                        return endpoints.basin.authority;
+                }
+        }
+    }
+};
+//# sourceMappingURL=endpoints.js.map

package/models/components/endpoints.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"endpoints.js","sourceRoot":"","sources":["../../src/models/components/endpoints.ts"],"names":[],"mappings":";;;AAKA;;GAEG;AACH,IAAY,OAGX;AAHD,WAAY,OAAO;IACf,wBAAwB;IACxB,sBAAW,CAAA;AACf,CAAC,EAHW,OAAO,uBAAP,OAAO,QAGlB;AAED;;GAEG;AACH,IAAY,iBAGX;AAHD,WAAY,iBAAiB;IACzB,8CAAyB,CAAA;IACzB,sCAAiB,CAAA;AACrB,CAAC,EAHW,iBAAiB,iCAAjB,iBAAiB,QAG5B;AAUD,MAAM,eAAgB,SAAQ,KAAK;IAC/B,YAAY,OAAe;QACvB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,iBAAiB,CAAC;IAClC,CAAC;CACJ;AAED;;;;GAIG;AACH,MAAa,WAAW;IAMpB,YAAY,OAAkB,EAAE,KAAoB;QAChD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,KAAc;QAC1B,MAAM,SAAS,GAAG,GAAG,KAAK,SAAS,CAAC;QACpC,OAAO,IAAI,WAAW,CAClB,SAAS,EACT;YACI,IAAI,EAAE,iBAAiB,CAAC,UAAU;YAClC,SAAS,EAAE,KAAK,KAAK,SAAS;SACjC,CACJ,CAAC;IACN,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,OAAO,CAAC,KAAc,EAAE,MAAc;QACzC,MAAM,YAAY,GAAG,GAAG,MAAM,MAAM,KAAK,SAAS,CAAC;QACnD,OAAO,IAAI,WAAW,CAClB,YAAY,EACZ;YACI,IAAI,EAAE,iBAAiB,CAAC,MAAM;YAC9B,SAAS,EAAE,YAAY;SAC1B,CACJ,CAAC;IACN,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,OAAO;QACV,IAAI,KAAc,CAAC;QACnB,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC;YACxD,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,QAAmB,CAAC,EAAE,CAAC;gBACvD,KAAK,GAAG,QAAmB,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACJ,MAAM,IAAI,eAAe,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;YAC/D,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,GAAY,CAAC;YAC3B,MAAM,IAAI,eAAe,CAAC,qBAAqB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAE9C,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAC3D,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;YAChC,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;gBACtC,MAAM,IAAI,eAAe,CAAC,yCAAyC,CAAC,CAAC;YACzE,CAAC;YACD,SAAS,CAAC,OAAO,GAAG,eAAe,CAAC;QACxC,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACvD,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YAC9B,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACpC,MAAM,IAAI,eAAe,CAAC,uCAAuC,CAAC,CAAC;YACvE,CAAC;YAED,IAAI,aAAa,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBACvC,MAAM,UAAU,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC9C,SAAS,CAAC,KAAK,GAAG;oBACd,IAAI,EAAE,iBAAiB,CAAC,UAAU;oBAClC,SAAS,EAAE,UAAU;iBACxB,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,SAAS,CAAC,KAAK,GAAG;oBACd,IAAI,EAAE,iBAAiB,CAAC,MAAM;oBAC9B,SAAS,EAAE,aAAa;iBAC3B,CAAC;YACN,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AA9FD,kCA8FC;AASY,QAAA,UAAU,GAAG;IACtB,WAAW,CAAC,IAAgB,EAAE,SAAsB;QAChD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAChB,KAAK,SAAS;gBACV,OAAO,SAAS,CAAC,OAAO,CAAC;YAC7B,KAAK,OAAO;gBACR,QAAQ,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC3B,KAAK,iBAAiB,CAAC,UAAU;wBAC7B,OAAO,GAAG,IAAI,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;oBACxD,KAAK,iBAAiB,CAAC,MAAM;wBACzB,OAAO,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC;gBACzC,CAAC;QACT,CAAC;IACL,CAAC;CACJ,CAAC"}

package/package.json

@@ -1,7 +1,10 @@
 {
   "name": "@s2-dev/streamstore",
-  "version": "0.3.14",
+  "version": "0.4.2",
   "author": "Speakeasy",
+  "bin": {
+    "mcp": "bin/mcp-server.js"
+  },
   "main": "./index.js",
   "sideEffects": false,
   "repository": {
@@ -10,18 +13,29 @@
   },
   "scripts": {
     "lint": "eslint --cache --max-warnings=0 src",
-    "build": "tsc",
+    "build:mcp": "bun src/mcp-server/build.mts",
+    "build": "npm run build:mcp && tsc",
     "prepublishOnly": "npm run build"
   },
   "peerDependencies": {
+    "@modelcontextprotocol/sdk": "^1.5.0",
     "zod": ">= 3"
   },
+  "peerDependenciesMeta": {
+    "@modelcontextprotocol/sdk": {"optional":true}
+  },
   "devDependencies": {
     "@eslint/js": "^9.19.0",
+    "@modelcontextprotocol/sdk": "^1.5.0",
+    "@stricli/core": "^1.1.1",
+    "@types/express": "^4.17.21",
     "@types/jsonpath": "^0.2.4",
     "@types/node": "^22.13.1",
     "@types/uuid": "^9.0.8",
+    "bun": "^1.2.2",
+    "bun-types": "^1.2.2",
     "eslint": "^9.19.0",
+    "express": "^4.21.2",
     "globals": "^15.14.0",
     "typescript": "^5.4.5",
     "typescript-eslint": "^8.22.0",

package/src/funcs/accountCreateBasin.ts

@@ -22,16 +22,17 @@ import {
 import * as errors from "../models/errors/index.js";
 import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
 import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
 
 /**
  * Create a new basin.
  */
-export async function accountCreateBasin(
+export function accountCreateBasin(
   client: S2Core,
   request: operations.CreateBasinRequest,
   options?: RequestOptions,
-): Promise<
+): APIPromise<
   Result<
     components.BasinInfo,
     | errors.ErrorResponse
@@ -45,6 +46,35 @@ export async function accountCreateBasin(
     | RequestTimeoutError
     | ConnectionError
   >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: S2Core,
+  request: operations.CreateBasinRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.BasinInfo,
+      | errors.ErrorResponse
+      | errors.RetryableError
+      | errors.RetryableError
+      | APIError
+      | SDKValidationError
+      | UnexpectedClientError
+      | InvalidRequestError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | ConnectionError
+    >,
+    APICall,
+  ]
 > {
   const parsed = safeParse(
     request,
@@ -52,7 +82,7 @@ export async function accountCreateBasin(
     "Input validation failed",
   );
   if (!parsed.ok) {
-    return parsed;
+    return [parsed, { status: "invalid" }];
   }
   const payload = parsed.value;
   const body = encodeJSON("body", payload.CreateBasinRequest, {
@@ -83,6 +113,7 @@ export async function accountCreateBasin(
   const requestSecurity = resolveGlobalSecurity(securityInput);
 
   const context = {
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
     operationID: "create_basin",
     oAuth2Scopes: [],
 
@@ -105,7 +136,7 @@ export async function accountCreateBasin(
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
   }, options);
   if (!requestRes.ok) {
-    return requestRes;
+    return [requestRes, { status: "invalid" }];
   }
   const req = requestRes.value;
 
@@ -116,7 +147,7 @@ export async function accountCreateBasin(
     retryCodes: context.retryCodes,
   });
   if (!doResult.ok) {
-    return doResult;
+    return [doResult, { status: "request-error", request: req }];
   }
   const response = doResult.value;
 
@@ -145,8 +176,8 @@ export async function accountCreateBasin(
     M.fail("5XX"),
   )(response, { extraFields: responseFields });
   if (!result.ok) {
-    return result;
+    return [result, { status: "complete", request: req, response }];
   }
 
-  return result;
+  return [result, { status: "complete", request: req, response }];
 }

package/src/funcs/accountDeleteBasin.ts

@@ -22,16 +22,17 @@ import {
 import * as errors from "../models/errors/index.js";
 import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
 import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
 import { Result } from "../types/fp.js";
 
 /**
  * Delete a basin.
  */
-export async function accountDeleteBasin(
+export function accountDeleteBasin(
   client: S2Core,
   request: operations.DeleteBasinRequest,
   options?: RequestOptions,
-): Promise<
+): APIPromise<
   Result<
     void,
     | errors.ErrorResponse
@@ -46,6 +47,36 @@ export async function accountDeleteBasin(
     | RequestTimeoutError
     | ConnectionError
   >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: S2Core,
+  request: operations.DeleteBasinRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      void,
+      | errors.ErrorResponse
+      | errors.NotFoundError
+      | errors.RetryableError
+      | errors.RetryableError
+      | APIError
+      | SDKValidationError
+      | UnexpectedClientError
+      | InvalidRequestError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | ConnectionError
+    >,
+    APICall,
+  ]
 > {
   const parsed = safeParse(
     request,
@@ -53,7 +84,7 @@ export async function accountDeleteBasin(
     "Input validation failed",
   );
   if (!parsed.ok) {
-    return parsed;
+    return [parsed, { status: "invalid" }];
   }
   const payload = parsed.value;
   const body = null;
@@ -76,6 +107,7 @@ export async function accountDeleteBasin(
   const requestSecurity = resolveGlobalSecurity(securityInput);
 
   const context = {
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
     operationID: "delete_basin",
     oAuth2Scopes: [],
 
@@ -98,7 +130,7 @@ export async function accountDeleteBasin(
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
   }, options);
   if (!requestRes.ok) {
-    return requestRes;
+    return [requestRes, { status: "invalid" }];
   }
   const req = requestRes.value;
 
@@ -109,7 +141,7 @@ export async function accountDeleteBasin(
     retryCodes: context.retryCodes,
   });
   if (!doResult.ok) {
-    return doResult;
+    return [doResult, { status: "request-error", request: req }];
   }
   const response = doResult.value;
 
@@ -140,8 +172,8 @@ export async function accountDeleteBasin(
     M.fail("5XX"),
   )(response, { extraFields: responseFields });
   if (!result.ok) {
-    return result;
+    return [result, { status: "complete", request: req, response }];
   }
 
-  return result;
+  return [result, { status: "complete", request: req, response }];
 }