@apifuse/provider-sdk 2.1.0-beta.9 → 2.2.0-beta.10

package/src/server/types.ts

@@ -1,13 +1,8 @@
 import { z } from "zod";
 
-import { HttpRetryPreset } from "../types";
+import { HttpRetryPreset } from "../types.js";
 
-export const ConnectionModeSchema = z.enum([
-	"oauth2",
-	"credentials",
-	"platform-managed",
-	"none",
-]);
+export const ConnectionModeSchema = z.enum(["oauth2", "credentials", "platform-managed", "none"]);
 
 export const OperationConnectionSchema = z.object({
 	id: z.string(),
@@ -21,6 +16,7 @@ export const OperationConnectionSchema = z.object({
 export const OperationRequestSchema = z.object({
 	requestId: z.string(),
 	input: z.record(z.string(), z.unknown()),
+	connectionId: z.string().optional(),
 	connection: OperationConnectionSchema.optional(),
 	headers: z.record(z.string(), z.string()).optional(),
 	trace: z.record(z.string(), z.string()).optional(),
@@ -98,18 +94,10 @@ export const AuthFlowErrorResponseSchema = OperationErrorResponseSchema;
 export type ConnectionMode = z.infer<typeof ConnectionModeSchema>;
 export type OperationConnection = z.infer<typeof OperationConnectionSchema>;
 export type OperationRequest = z.infer<typeof OperationRequestSchema>;
-export type OperationSuccessResponse = z.infer<
-	typeof OperationSuccessResponseSchema
->;
-export type OperationErrorResponse = z.infer<
-	typeof OperationErrorResponseSchema
->;
-export type OperationResponse =
-	| OperationSuccessResponse
-	| OperationErrorResponse;
+export type OperationSuccessResponse = z.infer<typeof OperationSuccessResponseSchema>;
+export type OperationErrorResponse = z.infer<typeof OperationErrorResponseSchema>;
+export type OperationResponse = OperationSuccessResponse | OperationErrorResponse;
 export type AuthFlowRequest = z.infer<typeof AuthFlowRequestSchema>;
-export type AuthFlowSuccessResponse = z.infer<
-	typeof AuthFlowSuccessResponseSchema
->;
+export type AuthFlowSuccessResponse = z.infer<typeof AuthFlowSuccessResponseSchema>;
 export type AuthFlowErrorResponse = z.infer<typeof AuthFlowErrorResponseSchema>;
 export type AuthFlowResponse = AuthFlowSuccessResponse | AuthFlowErrorResponse;

package/src/stealth/profiles.ts

@@ -1,5 +1,5 @@
-import { SDKError } from "../errors";
-import type { StealthPlatform, StealthProfile } from "../types";
+import { SDKError } from "../errors.js";
+import type { StealthPlatform, StealthProfile } from "../types.js";
 
 type StealthProfileDefinition = Omit<StealthProfile, "name" | "platform"> & {
 	platform: StealthPlatform;
@@ -81,10 +81,7 @@ const FIREFOX_JA3 =
 const SAFARI_JA3 =
 	"771,4865-4866-4867-49196-49195-52393-49200-49199-49188-49192-159-158-107-103-57-51-157-156-61-60-53-47-255,0-23-65281-10-11-16-5-13-18-51-45-43-27,29-23-24-25,0";
 
-function createProfile(
-	name: string,
-	definition: StealthProfileDefinition,
-): StealthProfile {
+function createProfile(name: string, definition: StealthProfileDefinition): StealthProfile {
 	return {
 		name,
 		platform: definition.platform,
@@ -93,12 +90,8 @@ function createProfile(
 		tlsClientIdentifier: definition.tlsClientIdentifier,
 		ja3: definition.ja3,
 		ja4: definition.ja4,
-		h2Settings: definition.h2Settings
-			? { ...definition.h2Settings }
-			: undefined,
-		headerOrder: definition.headerOrder
-			? [...definition.headerOrder]
-			: undefined,
+		h2Settings: definition.h2Settings ? { ...definition.h2Settings } : undefined,
+		headerOrder: definition.headerOrder ? [...definition.headerOrder] : undefined,
 	};
 }
 
@@ -108,8 +101,7 @@ function extractBrowserMajorVersion(profile: StealthProfile): string {
 		return chromeVersion;
 	}
 
-	const identifierVersion =
-		profile.tlsClientIdentifier?.match(/(\d+)(?!.*\d)/)?.[1];
+	const identifierVersion = profile.tlsClientIdentifier?.match(/(\d+)(?!.*\d)/)?.[1];
 	if (identifierVersion) {
 		return identifierVersion;
 	}
@@ -132,9 +124,7 @@ function toPlatformHeaderValue(platform: StealthPlatform): string {
 	}
 }
 
-export function generateLayer2Headers(
-	profile: StealthProfile,
-): Record<string, string> {
+export function generateLayer2Headers(profile: StealthProfile): Record<string, string> {
 	const headers: Record<string, string> = {
 		"Accept-Language": "ko-KR,ko;q=0.9,en-US;q=0.8",
 	};
@@ -143,16 +133,13 @@ export function generateLayer2Headers(
 	const majorVersion = extractBrowserMajorVersion(profile);
 
 	if (identifier.startsWith("chrome_") || identifier.startsWith("edge_")) {
-		const isEdge =
-			identifier.startsWith("edge_") || /\bEdg\//.test(profile.userAgent);
+		const isEdge = identifier.startsWith("edge_") || /\bEdg\//.test(profile.userAgent);
 		headers["Sec-Ch-Ua"] = isEdge
 			? `"Chromium";v="${majorVersion}", "Microsoft Edge";v="${majorVersion}", "Not)A;Brand";v="99"`
 			: `"Chromium";v="${majorVersion}", "Google Chrome";v="${majorVersion}", "Not)A;Brand";v="99"`;
 		headers["Sec-Ch-Ua-Platform"] = toPlatformHeaderValue(profile.platform);
 		headers["Sec-Ch-Ua-Mobile"] =
-			profile.platform === "android" || profile.platform === "ios"
-				? "?1"
-				: "?0";
+			profile.platform === "android" || profile.platform === "ios" ? "?1" : "?0";
 	}
 
 	return headers;
@@ -301,8 +288,5 @@ export function getStealthProfile(name: string): StealthProfile {
 }
 
 export function listStealthProfiles(): string[] {
-	return [
-		...Object.keys(STEALTH_PROFILES),
-		...Object.keys(STEALTH_PROFILE_ALIASES),
-	];
+	return [...Object.keys(STEALTH_PROFILES), ...Object.keys(STEALTH_PROFILE_ALIASES)];
 }

package/src/stream.ts

@@ -1,4 +1,4 @@
-import type { SseMessage } from "./types";
+import type { SseMessage } from "./types.js";
 
 export interface SseEvent<TData = unknown> {
 	event: string;
@@ -61,8 +61,7 @@ export function done<TData>(
 
 export function encodeSseEvent(input: SseEvent): string {
 	const lines: string[] = [];
-	if (input.id !== undefined)
-		lines.push(`id: ${sseFieldValue(input.id, "id")}`);
+	if (input.id !== undefined) lines.push(`id: ${sseFieldValue(input.id, "id")}`);
 	if (input.event) lines.push(`event: ${sseFieldValue(input.event, "event")}`);
 	if (input.retry !== undefined) {
 		if (!Number.isInteger(input.retry) || input.retry < 0) {
@@ -70,8 +69,7 @@ export function encodeSseEvent(input: SseEvent): string {
 		}
 		lines.push(`retry: ${input.retry}`);
 	}
-	const data =
-		typeof input.data === "string" ? input.data : JSON.stringify(input.data);
+	const data = typeof input.data === "string" ? input.data : JSON.stringify(input.data);
 	for (const line of data.split(/\r?\n/)) {
 		lines.push(`data: ${line}`);
 	}
@@ -85,9 +83,7 @@ function sseFieldValue(value: string, field: "event" | "id"): string {
 	return value;
 }
 
-export async function* readableBytes(
-	body: ReadableStream<Uint8Array>,
-): AsyncIterable<Uint8Array> {
+export async function* readableBytes(body: ReadableStream<Uint8Array>): AsyncIterable<Uint8Array> {
 	const reader = body.getReader();
 	try {
 		for (;;) {
@@ -100,9 +96,7 @@ export async function* readableBytes(
 	}
 }
 
-export async function* readableTextChunks(
-	body: ReadableStream<Uint8Array>,
-): AsyncIterable<string> {
+export async function* readableTextChunks(body: ReadableStream<Uint8Array>): AsyncIterable<string> {
 	const decoder = new TextDecoder();
 	for await (const chunk of readableBytes(body)) {
 		yield decoder.decode(chunk, { stream: true });
@@ -111,9 +105,7 @@ export async function* readableTextChunks(
 	if (tail) yield tail;
 }
 
-export async function* readableLines(
-	body: ReadableStream<Uint8Array>,
-): AsyncIterable<string> {
+export async function* readableLines(body: ReadableStream<Uint8Array>): AsyncIterable<string> {
 	let buffer = "";
 	for await (const chunk of readableTextChunks(body)) {
 		buffer += chunk;
@@ -121,8 +113,7 @@ export async function* readableLines(
 			const index = buffer.search(/\r?\n/);
 			if (index < 0) break;
 			const line = buffer.slice(0, index);
-			const newlineLength =
-				buffer[index] === "\r" && buffer[index + 1] === "\n" ? 2 : 1;
+			const newlineLength = buffer[index] === "\r" && buffer[index + 1] === "\n" ? 2 : 1;
 			buffer = buffer.slice(index + newlineLength);
 			yield line;
 		}
@@ -150,9 +141,7 @@ function parseJson<T>(data: string): T {
 	return JSON.parse(data);
 }
 
-export async function* parseSseStream(
-	body: ReadableStream<Uint8Array>,
-): AsyncIterable<SseMessage> {
+export async function* parseSseStream(body: ReadableStream<Uint8Array>): AsyncIterable<SseMessage> {
 	let eventName = "message";
 	let dataLines: string[] = [];
 	let id: string | undefined;

package/src/testing/index.ts

@@ -1,2 +1,2 @@
-export { describeTransform, snapshotTransform, toMatchShape } from "./helpers";
-export { runStandardTests } from "./run";
+export { describeTransform, snapshotTransform, toMatchShape } from "./helpers.js";
+export { runStandardTests } from "./run.js";

package/src/testing/run.ts

@@ -1,28 +1,23 @@
 import { describe, expect, it } from "bun:test";
 
-import { createProviderCache } from "../runtime/cache";
-import { createTestProviderChoiceContext } from "../runtime/choice";
-import { createMemoryProviderRuntimeState } from "../runtime/state";
-import { createUnsupportedSttClient } from "../runtime/stt";
-import { safeParseSchemaSync } from "../schema";
+import { createProviderCache } from "../runtime/cache.js";
+import { createTestProviderChoiceContext } from "../runtime/choice.js";
+import { createMemoryProviderRuntimeState } from "../runtime/state.js";
+import { createUnsupportedSttClient } from "../runtime/stt.js";
+import { safeParseSchemaSync } from "../schema.js";
 import type {
 	AuthMode,
 	CredentialContext,
 	HttpResponse,
 	ProviderContext,
 	ProviderDefinition,
-} from "../types";
+} from "../types.js";
 
 // Mirrors CONNECTOR_ID_REGEX in ../define.ts, which defineProvider() enforces.
 // A single lowercase segment (no hyphen) is a valid id, so the trailing group
 // is optional (`*`), matching providers like `kakaomap`, `kstartup`, `triple`.
 const CONNECTOR_ID_REGEX = /^[a-z][a-z0-9]*(-[a-z][a-z0-9]*)*$/;
-const VALID_AUTH_MODES = [
-	"none",
-	"platform-managed",
-	"credentials",
-	"oauth2",
-] as const;
+const VALID_AUTH_MODES = ["none", "platform-managed", "credentials", "oauth2"] as const;
 const UPDATE_SNAPSHOT_ARGS = new Set(["-u", "--update-snapshots"]);
 
 export interface StandardTestsManifest {
@@ -118,9 +113,7 @@ function inferFixtureDir(providerId: string): string {
 		.find((file) => file !== undefined);
 
 	if (testFile) {
-		const pathname = testFile.startsWith("file://")
-			? new URL(testFile).pathname
-			: testFile;
+		const pathname = testFile.startsWith("file://") ? new URL(testFile).pathname : testFile;
 		return `${pathname.replace(/\/[^/]+$/, "")}/../__fixtures__`;
 	}
 
@@ -128,13 +121,18 @@ function inferFixtureDir(providerId: string): string {
 }
 
 function jsonResponse(data: unknown): HttpResponse {
+	const body = JSON.stringify(data);
+	const bodyBytes = new TextEncoder().encode(body);
 	return {
 		status: 200,
 		ok: true,
 		headers: {},
 		data,
 		json: async <_T = unknown>() => JSON.parse(JSON.stringify(data)),
-		text: async () => JSON.stringify(data),
+		text: async () => body,
+		arrayBuffer: async () =>
+			bodyBytes.buffer.slice(bodyBytes.byteOffset, bodyBytes.byteOffset + bodyBytes.byteLength),
+		bytes: async () => bodyBytes.slice(0),
 	};
 }
 
@@ -176,16 +174,14 @@ function createSnapshotContext(rawFixture: unknown): ProviderContext {
 			engine: "playwright-stealth",
 			newPage: async () => unsupported("ctx.browser.newPage"),
 			rawPage: async () => unsupported("ctx.browser.rawPage"),
-			withIsolatedContext: async () =>
-				unsupported("ctx.browser.withIsolatedContext"),
+			withIsolatedContext: async () => unsupported("ctx.browser.withIsolatedContext"),
 			solveChallenge: async () => unsupported("ctx.browser.solveChallenge"),
 		},
 		trace: {
 			span: async (_name, fn) => fn(),
 		},
 		auth: {
-			requestField: async (name) =>
-				unsupported(`ctx.auth.requestField(${name})`),
+			requestField: async (name) => unsupported(`ctx.auth.requestField(${name})`),
 		},
 		stt: createUnsupportedSttClient(
 			"Standard test snapshot context does not support ctx.stt.transcribe",
@@ -284,11 +280,7 @@ function parseSchemaFixture(
 		operationName,
 		fieldName,
 		fixture,
-		safeParseSchemaSync(
-			schema,
-			fixture,
-			`operations.${operationName}.fixtures.${fieldName}`,
-		),
+		safeParseSchemaSync(schema, fixture, `operations.${operationName}.fixtures.${fieldName}`),
 	);
 }
 
@@ -314,40 +306,20 @@ export function runStandardTests(
 
 		for (const [operationName, op] of operations) {
 			if (op.fixtures?.request !== undefined) {
-				parseSchemaFixture(
-					operationName,
-					"request",
-					op.input,
-					op.fixtures.request,
-				);
+				parseSchemaFixture(operationName, "request", op.input, op.fixtures.request);
 			}
 
 			if (op.fixtures?.response !== undefined) {
-				parseSchemaFixture(
-					operationName,
-					"response",
-					op.output,
-					op.fixtures.response,
-				);
+				parseSchemaFixture(operationName, "response", op.output, op.fixtures.response);
 			}
 
 			if (isFixtureEnvelope(rawFixture)) {
 				if (rawFixture.request !== undefined) {
-					parseSchemaFixture(
-						operationName,
-						"request",
-						op.input,
-						rawFixture.request,
-					);
+					parseSchemaFixture(operationName, "request", op.input, rawFixture.request);
 				}
 
 				if (rawFixture.response !== undefined) {
-					parseSchemaFixture(
-						operationName,
-						"response",
-						op.output,
-						rawFixture.response,
-					);
+					parseSchemaFixture(operationName, "response", op.output, rawFixture.response);
 				}
 			}
 		}
@@ -403,21 +375,11 @@ export function runStandardTests(
 		it("operation schemas can parse fixture data", () => {
 			for (const [operationName, op] of operations) {
 				if (op.fixtures?.request !== undefined && op.input) {
-					parseSchemaFixture(
-						operationName,
-						"request",
-						op.input,
-						op.fixtures.request,
-					);
+					parseSchemaFixture(operationName, "request", op.input, op.fixtures.request);
 				}
 
 				if (op.fixtures?.response !== undefined && op.output) {
-					parseSchemaFixture(
-						operationName,
-						"response",
-						op.output,
-						op.fixtures.response,
-					);
+					parseSchemaFixture(operationName, "response", op.output, op.fixtures.response);
 				}
 
 				expect(operationName).toBeTruthy();
@@ -463,9 +425,7 @@ export function runStandardTests(
 					await Bun.write(snapshotPath, serialized);
 				}
 
-				const expected: unknown = JSON.parse(
-					await Bun.file(snapshotPath).text(),
-				);
+				const expected: unknown = JSON.parse(await Bun.file(snapshotPath).text());
 				expect(actual).toEqual(expected);
 			});
 		}