@s2-dev/streamstore 0.15.13 → 0.16.0

package/src/index.ts

@@ -1,11 +1,66 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-export * from "./lib/config.js";
-export * as files from "./lib/files.js";
-export { HTTPClient } from "./lib/http.js";
-export type { Fetcher, HTTPClientOptions } from "./lib/http.js";
-export * from "./sdk/sdk.js";
-
-export * from "./index.extras.js";
+export type {
+	IssueAccessTokenArgs,
+	ListAccessTokensArgs,
+	RevokeAccessTokenArgs,
+} from "./accessTokens";
+export type {
+	CreateBasinArgs,
+	DeleteBasinArgs,
+	GetBasinConfigArgs,
+	ListBasinsArgs,
+	ReconfigureBasinArgs,
+} from "./basins";
+export { S2Error } from "./error";
+export type {
+	AccessTokenInfo,
+	AccessTokenScope,
+	AccountMetricSet,
+	Accumulation,
+	BasinConfig,
+	BasinInfo,
+	BasinMetricSet,
+	BasinReconfiguration,
+	BasinScope,
+	CreateStreamRequest,
+	DeleteOnEmptyConfig,
+	Gauge,
+	IssueAccessTokenResponse,
+	Label,
+	ListAccessTokensResponse,
+	ListBasinsResponse,
+	ListStreamsResponse,
+	Metric,
+	MetricSetResponse,
+	Operation,
+	PermittedOperationGroups,
+	ReadWritePermissions,
+	ResourceSet,
+	RetentionPolicy,
+	StorageClass,
+	StreamConfig,
+	StreamInfo,
+	StreamMetricSet,
+	TimeseriesInterval,
+	TimestampingConfig,
+	TimestampingMode,
+} from "./generated/types.gen";
+export type {
+	AccountMetricsArgs,
+	BasinMetricsArgs,
+	StreamMetricsArgs,
+} from "./metrics";
+export { S2 } from "./s2";
+export type {
+	AppendArgs,
+	ReadArgs,
+	ReadBatch,
+	SequencedRecord,
+} from "./stream";
+export type {
+	CreateStreamArgs,
+	DeleteStreamArgs,
+	GetStreamConfigArgs,
+	ListStreamsArgs,
+	ReconfigureStreamArgs,
+} from "./streams";
+export { AppendRecord } from "./utils";

package/src/lib/event-stream.ts

@@ -0,0 +1,167 @@
+/*
+ * Code from Speakeasy (https://speakeasy.com).
+ */
+
+export type SseMessage<T> = {
+	data?: T | undefined;
+	event?: string | undefined;
+	id?: string | undefined;
+	retry?: number | undefined;
+};
+
+type ParseResult<T> =
+	| {
+			done: true;
+			value?: undefined;
+			batch?: undefined;
+	  }
+	| {
+			done: false;
+			batch?: false;
+			value?: T;
+	  }
+	| {
+			done: false;
+			batch: true;
+			value: T[];
+	  };
+
+export class EventStream<T>
+	extends ReadableStream<T>
+	implements AsyncDisposable
+{
+	constructor(
+		responseBody: ReadableStream<Uint8Array>,
+		parse: (x: SseMessage<string>) => ParseResult<T>,
+	) {
+		const upstream = responseBody.getReader();
+		let buffer: Uint8Array = new Uint8Array();
+		super({
+			async pull(downstream) {
+				try {
+					while (true) {
+						const match = findBoundary(buffer);
+						if (!match) {
+							const chunk = await upstream.read();
+							if (chunk.done) return downstream.close();
+							buffer = concatBuffer(buffer, chunk.value);
+							continue;
+						}
+						const message = buffer.slice(0, match.index);
+						buffer = buffer.slice(match.index + match.length);
+						const item = parseMessage(message, parse);
+						if (item) {
+							if (item.batch) {
+								for (const chunk of item.value) {
+									downstream.enqueue(chunk);
+								}
+							} else if (item.value) {
+								downstream.enqueue(item.value);
+							} else if (item.done) {
+								await upstream.cancel("done");
+								downstream.close();
+							}
+						}
+					}
+				} catch (e) {
+					downstream.error(e);
+					await upstream.cancel(e);
+				}
+			},
+			cancel: (reason) => upstream.cancel(reason),
+		});
+	}
+
+	async [Symbol.asyncDispose]() {
+		await this.cancel("disposed");
+	}
+
+	// Polyfill for older browsers
+	[Symbol.asyncIterator](): AsyncIterableIterator<T> {
+		const fn = (ReadableStream.prototype as any)[Symbol.asyncIterator];
+		if (typeof fn === "function") return fn.call(this);
+		const reader = this.getReader();
+		return {
+			next: async () => {
+				const r = await reader.read();
+				if (r.done) {
+					reader.releaseLock();
+					return { done: true, value: undefined };
+				}
+				return { done: false, value: r.value };
+			},
+			throw: async (e) => {
+				await reader.cancel(e);
+				reader.releaseLock();
+				return { done: true, value: undefined };
+			},
+			return: async () => {
+				await reader.cancel("done");
+				reader.releaseLock();
+				return { done: true, value: undefined };
+			},
+			[Symbol.asyncIterator]() {
+				return this;
+			},
+		};
+	}
+}
+
+function concatBuffer(a: Uint8Array, b: Uint8Array): Uint8Array {
+	const c = new Uint8Array(a.length + b.length);
+	c.set(a, 0);
+	c.set(b, a.length);
+	return c;
+}
+
+/** Finds the first (CR,LF,CR,LF) or (CR,CR) or (LF,LF) */
+function findBoundary(
+	buf: Uint8Array,
+): { index: number; length: number } | null {
+	const len = buf.length;
+	for (let i = 0; i < len; i++) {
+		if (
+			i <= len - 4 &&
+			buf[i] === 13 &&
+			buf[i + 1] === 10 &&
+			buf[i + 2] === 13 &&
+			buf[i + 3] === 10
+		) {
+			return { index: i, length: 4 };
+		}
+		if (i <= len - 2 && buf[i] === 13 && buf[i + 1] === 13) {
+			return { index: i, length: 2 };
+		}
+		if (i <= len - 2 && buf[i] === 10 && buf[i + 1] === 10) {
+			return { index: i, length: 2 };
+		}
+	}
+	return null;
+}
+function parseMessage<T>(
+	chunk: Uint8Array,
+	parse: (x: SseMessage<string>) => ParseResult<T>,
+) {
+	const text = new TextDecoder().decode(chunk);
+	const lines = text.split(/\r\n|\r|\n/);
+	const dataLines: string[] = [];
+	const ret: SseMessage<string> = {};
+	let ignore = true;
+	for (const line of lines) {
+		if (!line || line.startsWith(":")) continue;
+		ignore = false;
+		const i = line.indexOf(":");
+		const field = line.slice(0, i);
+		const value = line[i + 1] === " " ? line.slice(i + 2) : line.slice(i + 1);
+		if (field === "data") dataLines.push(value);
+		else if (field === "event") ret.event = value;
+		else if (field === "id") ret.id = value;
+		else if (field === "retry") {
+			const n = Number(value);
+			if (!isNaN(n)) ret.retry = n;
+		}
+	}
+	if (ignore) return;
+	if (dataLines.length) ret.data = dataLines.join("\n");
+	return parse(ret);
+}

package/src/metrics.ts

@@ -0,0 +1,106 @@
+import type { DataToObject, S2RequestOptions } from "./common";
+import { S2Error } from "./error";
+import {
+	type AccountMetricsData,
+	accountMetrics,
+	type BasinMetricsData,
+	basinMetrics,
+	type StreamMetricsData,
+	streamMetrics,
+} from "./generated";
+import type { Client } from "./generated/client/types.gen";
+
+export interface AccountMetricsArgs extends DataToObject<AccountMetricsData> {}
+export interface BasinMetricsArgs extends DataToObject<BasinMetricsData> {}
+export interface StreamMetricsArgs extends DataToObject<StreamMetricsData> {}
+
+export class S2Metrics {
+	readonly client: Client;
+
+	constructor(client: Client) {
+		this.client = client;
+	}
+
+	/**
+	 * Account-level metrics.
+	 *
+	 * @param args.set Metric set to return
+	 * @param args.start Optional start timestamp (Unix seconds)
+	 * @param args.end Optional end timestamp (Unix seconds)
+	 * @param args.interval Optional aggregation interval for timeseries sets
+	 */
+	public async account(args: AccountMetricsArgs, options?: S2RequestOptions) {
+		const response = await accountMetrics({
+			client: this.client,
+			query: args,
+			...options,
+		});
+
+		if (response.error) {
+			throw new S2Error({
+				message: response.error.message,
+				code: response.error.code ?? undefined,
+				status: response.response.status,
+			});
+		}
+
+		return response.data;
+	}
+
+	/**
+	 * Basin-level metrics.
+	 *
+	 * @param args.basin Basin name
+	 * @param args.set Metric set to return
+	 * @param args.start Optional start timestamp (Unix seconds)
+	 * @param args.end Optional end timestamp (Unix seconds)
+	 * @param args.interval Optional aggregation interval for timeseries sets
+	 */
+	public async basin(args: BasinMetricsArgs, options?: S2RequestOptions) {
+		const response = await basinMetrics({
+			client: this.client,
+			path: args,
+			query: args,
+			...options,
+		});
+
+		if (response.error) {
+			throw new S2Error({
+				message: response.error.message,
+				code: response.error.code ?? undefined,
+				status: response.response.status,
+			});
+		}
+
+		return response.data;
+	}
+
+	/**
+	 * Stream-level metrics.
+	 *
+	 * @param args.basin Basin name
+	 * @param args.stream Stream name
+	 * @param args.set Metric set to return
+	 * @param args.start Optional start timestamp (Unix seconds)
+	 * @param args.end Optional end timestamp (Unix seconds)
+	 * @param args.interval Optional aggregation interval for timeseries sets
+	 */
+	public async stream(args: StreamMetricsArgs, options?: S2RequestOptions) {
+		const response = await streamMetrics({
+			client: this.client,
+			path: args,
+			query: args,
+			...options,
+		});
+
+		if (response.error) {
+			throw new S2Error({
+				message: response.error.message,
+				code: response.error.code ?? undefined,
+				status: response.response.status,
+			});
+		}
+
+		return response.data;
+	}
+}

package/src/s2.ts

@@ -0,0 +1,65 @@
+import { Redacted } from "effect";
+import { S2AccessTokens } from "./accessTokens";
+import { S2Basin } from "./basin";
+import { S2Basins } from "./basins";
+import type { S2ClientOptions } from "./common";
+import { createClient, createConfig } from "./generated/client";
+import type { Client } from "./generated/client/types.gen";
+import { S2Metrics } from "./metrics";
+
+const defaultBaseUrl = "https://aws.s2.dev/v1";
+const defaultMakeBasinBaseUrl = (basin: string) =>
+	`https://${basin}.b.aws.s2.dev/v1`;
+
+/**
+ * Top-level S2 SDK client.
+ *
+ * - Authenticates with an access token and exposes account-scoped helpers for basins, streams, access tokens and metrics.
+ */
+export class S2 {
+	private readonly accessToken: Redacted.Redacted;
+	private readonly client: Client;
+	private readonly makeBasinBaseUrl: (basin: string) => string;
+
+	/**
+	 * Account-scoped basin management operations.
+	 *
+	 * - List, create, delete and reconfigure basins.
+	 */
+	public readonly basins: S2Basins;
+	/** Manage access tokens for the account (list, issue, revoke). */
+	public readonly accessTokens: S2AccessTokens;
+	/** Account, basin and stream level metrics. */
+	public readonly metrics: S2Metrics;
+
+	/**
+	 * Create a new S2 client.
+	 *
+	 * @param options Access token configuration.
+	 */
+	constructor(options: S2ClientOptions) {
+		this.accessToken = Redacted.make(options.accessToken);
+		this.client = createClient(
+			createConfig({
+				baseUrl: options.baseUrl ?? defaultBaseUrl,
+				auth: () => Redacted.value(this.accessToken),
+			}),
+		);
+		this.basins = new S2Basins(this.client);
+		this.accessTokens = new S2AccessTokens(this.client);
+		this.metrics = new S2Metrics(this.client);
+		this.makeBasinBaseUrl = options.makeBasinBaseUrl ?? defaultMakeBasinBaseUrl;
+	}
+
+	/**
+	 * Create a basin-scoped client bound to a specific basin name.
+	 *
+	 * @param name Basin name.
+	 */
+	public basin(name: string) {
+		return new S2Basin(name, {
+			accessToken: this.accessToken,
+			baseUrl: this.makeBasinBaseUrl(name),
+		});
+	}
+}