@keboola/api-client 51.0.0 → 52.0.0

package/dist/queryService/types.d.cts

@@ -1,2 +1,2 @@
-import { $ as GetQueryResutsQuery, J as GetExportResultsQuery, K as QueryJobPollOptions, Q as GetQueryResultsPath, X as GetQueriesPath, Y as GetQueriesHistoryQuery, Z as GetQueryJobPath, at as QueryResult, et as JobStatusResponse, it as QueryJobStatus, nt as PostQueryBody, ot as Statements, q as ExportResultsResponse, rt as PostQueryPath, tt as PostCancelQueryJobPath } from "../index-qJ2IjXIM.cjs";
+import { $ as GetExportResultsQuery, Q as ExportResultsResponse, Z as QueryJobPollOptions, at as JobStatusResponse, ct as PostQueryPath, dt as Statements, et as GetQueriesHistoryQuery, it as GetQueryResutsQuery, lt as QueryJobStatus, nt as GetQueryJobPath, ot as PostCancelQueryJobPath, rt as GetQueryResultsPath, st as PostQueryBody, tt as GetQueriesPath, ut as QueryResult } from "../index-B7el5pJU.cjs";
 export { ExportResultsResponse, GetExportResultsQuery, GetQueriesHistoryQuery, GetQueriesPath, GetQueryJobPath, GetQueryResultsPath, GetQueryResutsQuery, JobStatusResponse, PostCancelQueryJobPath, PostQueryBody, PostQueryPath, type QueryJobPollOptions, QueryJobStatus, QueryResult, Statements };

package/dist/queryService/types.d.ts

@@ -1,2 +1,2 @@
-import { $ as GetQueryResutsQuery, J as GetExportResultsQuery, K as QueryJobPollOptions, Q as GetQueryResultsPath, X as GetQueriesPath, Y as GetQueriesHistoryQuery, Z as GetQueryJobPath, at as QueryResult, et as JobStatusResponse, it as QueryJobStatus, nt as PostQueryBody, ot as Statements, q as ExportResultsResponse, rt as PostQueryPath, tt as PostCancelQueryJobPath } from "../index-BsT_cBJd.js";
+import { $ as GetExportResultsQuery, Q as ExportResultsResponse, Z as QueryJobPollOptions, at as JobStatusResponse, ct as PostQueryPath, dt as Statements, et as GetQueriesHistoryQuery, it as GetQueryResutsQuery, lt as QueryJobStatus, nt as GetQueryJobPath, ot as PostCancelQueryJobPath, rt as GetQueryResultsPath, st as PostQueryBody, tt as GetQueriesPath, ut as QueryResult } from "../index-C7JcNvbd.js";
 export { ExportResultsResponse, GetExportResultsQuery, GetQueriesHistoryQuery, GetQueriesPath, GetQueryJobPath, GetQueryResultsPath, GetQueryResutsQuery, JobStatusResponse, PostCancelQueryJobPath, PostQueryBody, PostQueryPath, type QueryJobPollOptions, QueryJobStatus, QueryResult, Statements };

package/dist/queryService-C3BQ6GKW.cjs

@@ -0,0 +1,214 @@
+const require_createFetchClient = require("./createFetchClient-CDKSkd7i.cjs");
+const require_RowLevelSecurityError = require("./RowLevelSecurityError-RK_NhhAx.cjs");
+const require_ValidationError = require("./ValidationError-BXcFX4nD.cjs");
+const require_createGenericFetchClient = require("./createGenericFetchClient-BuYzzV_D.cjs");
+const require_createOpenapiFetchClient = require("./createOpenapiFetchClient-dkssExkU.cjs");
+const require_transientRetries = require("./transientRetries-CBorl4Ue.cjs");
+const require_auth = require("./auth-AehUDNRK.cjs");
+// src/clients/queryService/auth.ts
+const PRINTABLE_ASCII_WITHOUT_SPACES = /^[\x21-\x7E]+$/;
+const requireToken = (token) => {
+	if (typeof token !== "string" || !PRINTABLE_ASCII_WITHOUT_SPACES.test(token)) throw new require_ValidationError.ValidationError("token must be non-empty printable ASCII without spaces");
+	return token;
+};
+const readToken = async (source) => requireToken(typeof source === "function" ? await source() : source);
+const hasCallback = (credential) => typeof credential.token === "function" || credential.type === "app-user-token" && typeof credential.storageToken === "function";
+/**
+* Fixed headers for a credential with only string tokens.
+*
+* - `app-user-token` tokens are checked here, so an empty user token throws at client creation.
+* - `sapi-token` / `bearer` strings go out unchecked: service clients send an empty `bearer` token
+*   that their own middleware replaces.
+*/
+const staticHeaders = (credential) => {
+	const { token } = credential;
+	if (typeof token !== "string") return {};
+	if (credential.type === "app-user-token") return typeof credential.storageToken === "string" ? {
+		[require_createFetchClient.KeboolaHttpHeader.STORAGE_API_TOKEN]: requireToken(credential.storageToken),
+		[require_createFetchClient.APP_USER_TOKEN_HEADER]: requireToken(token)
+	} : {};
+	return require_auth.createAuthHeaders({
+		...credential,
+		token
+	});
+};
+const resolvedHeaders = async (credential) => {
+	const token = await readToken(credential.token);
+	if (credential.type === "app-user-token") return {
+		[require_createFetchClient.KeboolaHttpHeader.STORAGE_API_TOKEN]: await readToken(credential.storageToken),
+		[require_createFetchClient.APP_USER_TOKEN_HEADER]: token
+	};
+	if (credential.type === "bearer" && (!Number.isInteger(credential.projectId) || credential.projectId <= 0)) throw new require_ValidationError.ValidationError("projectId must be a positive integer");
+	return require_auth.createAuthHeaders({
+		...credential,
+		token
+	});
+};
+const CREDENTIAL_HEADER_NAMES = [
+	require_createFetchClient.KeboolaHttpHeader.STORAGE_API_TOKEN,
+	require_createFetchClient.KeboolaHttpHeader.MANAGEMENT_API_TOKEN,
+	require_createFetchClient.KeboolaHttpHeader.PROJECT_ID,
+	require_createFetchClient.HttpHeader.AUTHORIZATION,
+	require_createFetchClient.APP_USER_TOKEN_HEADER
+];
+const createCredentialMiddleware = (provider) => (next) => async (request) => {
+	const credentialHeaders = await resolvedHeaders(await provider());
+	const { headers } = request.request;
+	for (const name of CREDENTIAL_HEADER_NAMES) headers.delete(name);
+	for (const [name, value] of Object.entries(credentialHeaders)) headers.set(name, value);
+	return next(request);
+};
+/**
+* The Query Service client's credential setup for its `auth`.
+*
+* 1. A credential with only string tokens: default headers, set once (as `createAuthHeaders`).
+* 2. A credential with a token callback, or a provider: no default headers; the innermost
+*    middleware resolves and checks every token on every attempt.
+*/
+const resolveAuth = (auth, middlewares) => {
+	if (typeof auth === "function") return {
+		headers: void 0,
+		middlewares: [...middlewares, createCredentialMiddleware(auth)]
+	};
+	if (hasCallback(auth)) return {
+		headers: void 0,
+		middlewares: [...middlewares, createCredentialMiddleware(() => auth)]
+	};
+	return {
+		headers: staticHeaders(auth),
+		middlewares
+	};
+};
+// src/clients/queryService/poller.ts
+const DEFAULT_POLL_INTERVAL = 2e3;
+const createJobPoller = ({ getQueryJob }) => ({ init, interval, maxAttempts, timeout, signal }) => {
+	const retries = require_transientRetries.transientRetries();
+	return require_transientRetries.poller({
+		pollKey: ({ id }) => ["queryService.job", { id }],
+		init,
+		pollFn: async ({ init: { id }, signal: pollSignal }) => {
+			const job = await getQueryJob(id, pollSignal);
+			retries.reset();
+			return job;
+		},
+		isRetryable: retries.isRetryable,
+		until: ({ data }) => require_transientRetries.jobStatus.query.isTerminal(data.status),
+		failed: (job) => {
+			if (job.status !== "failed") return void 0;
+			const failedStatements = job.statements.filter((statement) => statement.status === "failed").map((statement) => ({
+				id: statement.id,
+				error: statement.error
+			}));
+			return new require_RowLevelSecurityError.QueryJobError(failedStatements[0]?.error ?? "Job failed", job.queryJobId, { failedStatements });
+		},
+		backoff: {
+			type: "constant",
+			interval: interval ?? DEFAULT_POLL_INTERVAL
+		},
+		maxAttempts,
+		timeout,
+		signal
+	});
+};
+// src/clients/queryService/validateStatus.ts
+const isRecord = (value) => typeof value === "object" && value !== null;
+/**
+* Status validation for Query Service, which reads a row-level security refusal as `RowLevelSecurityError`.
+*
+* - The fixed `exception` text is the only signal: `X-KBC-RLS-Mode` is not exposed through CORS,
+*   and a credentials 403 on a governed workspace carries that header too.
+*/
+const validateQueryServiceStatus = (apiResponse) => {
+	const { response, data } = apiResponse;
+	if (response.status === require_createFetchClient.HttpStatus.FORBIDDEN && isRecord(data) && data.exception === "Row-level security does not allow this query") return new require_RowLevelSecurityError.RowLevelSecurityError(apiResponse);
+	return require_createFetchClient.defaultValidateStatus(apiResponse);
+};
+// src/clients/queryService/queryServiceClient.ts
+const createQueryServiceClient = ({ baseUrl, middlewares = [], auth }) => {
+	const client = require_createOpenapiFetchClient.createOpenapiFetchClient({
+		baseUrl,
+		...resolveAuth(auth, middlewares),
+		validateStatus: validateQueryServiceStatus
+	});
+	const genericClient = require_createGenericFetchClient.createGenericFetchClient({
+		baseUrl,
+		middlewares
+	});
+	const createQueryJob = async (branchId, workspaceId, body) => {
+		const { data } = await client.post("/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries", {
+			path: {
+				branchId,
+				workspaceId
+			},
+			body
+		});
+		return data;
+	};
+	const cancelQueryJob = async (queryJobId) => {
+		const { data } = await client.post("/api/v1/queries/{queryJobId}/cancel", {
+			path: { queryJobId },
+			body: { reason: "Cancelled by user" }
+		});
+		return data;
+	};
+	const getQueryJob = async (queryJobId, signal) => {
+		const { data } = await client.get("/api/v1/queries/{queryJobId}", { path: { queryJobId } }, { signal });
+		return data;
+	};
+	const getQueryResults = async (queryJobId, statementId, query) => {
+		const { data } = await client.get("/api/v1/queries/{queryJobId}/{statementId}/results", {
+			path: {
+				queryJobId,
+				statementId
+			},
+			query
+		});
+		return data;
+	};
+	const getQueryHistory = async (branchId, workspaceId, query) => {
+		const { data } = await client.get("/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries", {
+			path: {
+				branchId,
+				workspaceId
+			},
+			query
+		});
+		return data;
+	};
+	const exportResults = async (queryJobId, statementId, query, signal) => {
+		const { data, response } = await client.get("/api/v1/queries/{queryJobId}/{statementId}/export", {
+			path: {
+				queryJobId,
+				statementId
+			},
+			query
+		}, { signal });
+		return {
+			mimeType: response.headers.get(require_createFetchClient.HttpHeader.CONTENT_TYPE) ?? "text/csv",
+			content: data,
+			fileName: require_createFetchClient.parseContentDispositionHeader(response.headers.get(require_createFetchClient.HttpHeader.CONTENT_DISPOSITION))?.filename ?? "export.csv"
+		};
+	};
+	const healthCheck = async (signal) => {
+		const { data } = await genericClient.get("/health-check", {}, { signal });
+		return data;
+	};
+	return {
+		healthCheck,
+		poller: createJobPoller({ getQueryJob }),
+		createQueryJob,
+		cancelQueryJob,
+		getQueryJob,
+		getQueryResults,
+		getQueryHistory,
+		exportResults
+	};
+};
+Object.defineProperty(exports, "createQueryServiceClient", {
+	enumerable: true,
+	get: function() {
+		return createQueryServiceClient;
+	}
+});
+
+//# sourceMappingURL=queryService-C3BQ6GKW.cjs.map

package/dist/queryService-C3BQ6GKW.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"queryService-C3BQ6GKW.cjs","names":["ValidationError","KeboolaHttpHeader","APP_USER_TOKEN_HEADER","createAuthHeaders","HttpHeader","transientRetries","poller","jobStatus","QueryJobError","HttpStatus","RowLevelSecurityError","defaultValidateStatus","createOpenapiFetchClient","createGenericFetchClient","HttpHeader","parseContentDispositionHeader"],"sources":["../src/clients/queryService/auth.ts","../src/clients/queryService/poller.ts","../src/clients/queryService/validateStatus.ts","../src/clients/queryService/queryServiceClient.ts"],"sourcesContent":["import { ValidationError } from '../../errors';\nimport { HttpHeader, type MiddlewareFn } from '../../fetchClient';\nimport { APP_USER_TOKEN_HEADER, KeboolaHttpHeader } from '../../internal';\nimport { createAuthHeaders } from '../auth';\n\n/** A token value, or a callback that returns the current one (read before every request). */\nexport type TokenSource = string | (() => string | Promise<string>);\n\n/** A data app user's token, sent as `X-Kbc-App-User-Token` together with the app's Storage token. */\nexport type AppUserTokenAuth = {\n  type: 'app-user-token';\n  token: TokenSource;\n  storageToken: TokenSource;\n};\n\n/** One credential; a request carries exactly that credential's headers. */\nexport type QueryServiceCredential =\n  | { type: 'sapi-token'; token: TokenSource }\n  | { type: 'bearer'; token: TokenSource; projectId: number }\n  | AppUserTokenAuth;\n\n/** Resolved before every request, retries included. */\nexport type QueryServiceCredentialProvider = () =>\n  | QueryServiceCredential\n  | Promise<QueryServiceCredential>;\n\nconst PRINTABLE_ASCII_WITHOUT_SPACES = /^[\\x21-\\x7E]+$/;\n\nconst requireToken = (token: unknown): string => {\n  if (typeof token !== 'string' || !PRINTABLE_ASCII_WITHOUT_SPACES.test(token)) {\n    throw new ValidationError('token must be non-empty printable ASCII without spaces');\n  }\n  return token;\n};\n\nconst readToken = async (source: TokenSource) =>\n  requireToken(typeof source === 'function' ? await source() : source);\n\nconst hasCallback = (credential: QueryServiceCredential) =>\n  typeof credential.token === 'function' ||\n  (credential.type === 'app-user-token' && typeof credential.storageToken === 'function');\n\n/**\n * Fixed headers for a credential with only string tokens.\n *\n * - `app-user-token` tokens are checked here, so an empty user token throws at client creation.\n * - `sapi-token` / `bearer` strings go out unchecked: service clients send an empty `bearer` token\n *   that their own middleware replaces.\n */\nconst staticHeaders = (credential: QueryServiceCredential): Record<string, string> => {\n  const { token } = credential;\n  if (typeof token !== 'string') return {};\n  if (credential.type === 'app-user-token') {\n    return typeof credential.storageToken === 'string'\n      ? {\n          [KeboolaHttpHeader.STORAGE_API_TOKEN]: requireToken(credential.storageToken),\n          [APP_USER_TOKEN_HEADER]: requireToken(token),\n        }\n      : {};\n  }\n  return createAuthHeaders({ ...credential, token });\n};\n\nconst resolvedHeaders = async (\n  credential: QueryServiceCredential,\n): Promise<Record<string, string>> => {\n  const token = await readToken(credential.token);\n  if (credential.type === 'app-user-token') {\n    return {\n      [KeboolaHttpHeader.STORAGE_API_TOKEN]: await readToken(credential.storageToken),\n      [APP_USER_TOKEN_HEADER]: token,\n    };\n  }\n  if (\n    credential.type === 'bearer' &&\n    (!Number.isInteger(credential.projectId) || credential.projectId <= 0)\n  ) {\n    throw new ValidationError('projectId must be a positive integer');\n  }\n  return createAuthHeaders({ ...credential, token });\n};\n\nconst CREDENTIAL_HEADER_NAMES = [\n  KeboolaHttpHeader.STORAGE_API_TOKEN,\n  KeboolaHttpHeader.MANAGEMENT_API_TOKEN,\n  KeboolaHttpHeader.PROJECT_ID,\n  HttpHeader.AUTHORIZATION,\n  APP_USER_TOKEN_HEADER,\n];\n\nconst createCredentialMiddleware =\n  (provider: QueryServiceCredentialProvider): MiddlewareFn =>\n  (next) =>\n  async (request) => {\n    const credentialHeaders = await resolvedHeaders(await provider());\n    const { headers } = request.request;\n    for (const name of CREDENTIAL_HEADER_NAMES) headers.delete(name);\n    for (const [name, value] of Object.entries(credentialHeaders)) headers.set(name, value);\n    return next(request);\n  };\n\n/**\n * The Query Service client's credential setup for its `auth`.\n *\n * 1. A credential with only string tokens: default headers, set once (as `createAuthHeaders`).\n * 2. A credential with a token callback, or a provider: no default headers; the innermost\n *    middleware resolves and checks every token on every attempt.\n */\nexport const resolveAuth = (\n  auth: QueryServiceCredential | QueryServiceCredentialProvider,\n  middlewares: MiddlewareFn[],\n) => {\n  if (typeof auth === 'function') {\n    return { headers: undefined, middlewares: [...middlewares, createCredentialMiddleware(auth)] };\n  }\n  if (hasCallback(auth)) {\n    return {\n      headers: undefined,\n      middlewares: [...middlewares, createCredentialMiddleware(() => auth)],\n    };\n  }\n  return { headers: staticHeaders(auth), middlewares };\n};\n","import { jobStatus } from '../../domain/jobStatus';\nimport { QueryJobError } from '../../errors';\nimport { poller } from '../../utils/async';\nimport { transientRetries } from '../transientRetries';\n\nimport type { GetQueryJobPath, JobStatusResponse } from './types';\n\nconst DEFAULT_POLL_INTERVAL = 2000;\n\nexport type QueryJobPollOptions = {\n  interval?: number;\n  maxAttempts?: number;\n  timeout?: number;\n  signal?: AbortSignal;\n};\n\ntype GetQueryJob = (\n  queryJobId: GetQueryJobPath['queryJobId'],\n  signal?: AbortSignal,\n) => Promise<JobStatusResponse>;\n\n/**\n * Waits on a query job.\n *\n * - `init` is the job, or the request that yields it, so a caller hands over what `createQueryJob` gave it.\n * - Terminal is the job's own terminal set, not its statements': a `completed` job can still hold a\n *   statement that failed, and what that means is the caller's business.\n */\ntype JobRef = { id: GetQueryJobPath['queryJobId'] };\n\nexport const createJobPoller =\n  ({ getQueryJob }: { getQueryJob: GetQueryJob }) =>\n  ({\n    init,\n    interval,\n    maxAttempts,\n    timeout,\n    signal,\n  }: QueryJobPollOptions & { init: JobRef | ((signal: AbortSignal) => Promise<JobRef>) }) => {\n    const retries = transientRetries();\n\n    return poller({\n      pollKey: ({ id }) => ['queryService.job', { id }] as const,\n      init,\n      pollFn: async ({ init: { id }, signal: pollSignal }) => {\n        const job = await getQueryJob(id, pollSignal);\n        retries.reset();\n        return job;\n      },\n      isRetryable: retries.isRetryable,\n      until: ({ data }) => jobStatus.query.isTerminal(data.status),\n      failed: (job) => {\n        if (job.status !== 'failed') return undefined;\n        const failedStatements = job.statements\n          .filter((statement) => statement.status === 'failed')\n          .map((statement) => ({ id: statement.id, error: statement.error }));\n\n        return new QueryJobError(failedStatements[0]?.error ?? 'Job failed', job.queryJobId, {\n          failedStatements,\n        });\n      },\n      backoff: { type: 'constant', interval: interval ?? DEFAULT_POLL_INTERVAL },\n      maxAttempts,\n      timeout,\n      signal,\n    });\n  };\n","import { RowLevelSecurityError } from '../../errors';\nimport { defaultValidateStatus, type ValidateStatusFn } from '../../fetchClient';\nimport { HttpStatus, ROW_LEVEL_SECURITY_DENIED_MESSAGE } from '../../internal';\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === 'object' && value !== null;\n\n/**\n * Status validation for Query Service, which reads a row-level security refusal as `RowLevelSecurityError`.\n *\n * - The fixed `exception` text is the only signal: `X-KBC-RLS-Mode` is not exposed through CORS,\n *   and a credentials 403 on a governed workspace carries that header too.\n */\nexport const validateQueryServiceStatus: ValidateStatusFn = (apiResponse) => {\n  const { response, data } = apiResponse;\n\n  if (\n    response.status === HttpStatus.FORBIDDEN &&\n    isRecord(data) &&\n    data.exception === ROW_LEVEL_SECURITY_DENIED_MESSAGE\n  ) {\n    return new RowLevelSecurityError(apiResponse);\n  }\n\n  return defaultValidateStatus(apiResponse);\n};\n","import {\n  createGenericFetchClient,\n  createOpenapiFetchClient,\n  HttpHeader,\n  type MiddlewareFn,\n  parseContentDispositionHeader,\n} from '../../fetchClient';\nimport type { ClientInitOptions } from '../types';\n\nimport type { paths } from './__generated__/schema';\nimport {\n  type QueryServiceCredential,\n  type QueryServiceCredentialProvider,\n  resolveAuth,\n} from './auth';\nimport { createJobPoller } from './poller';\nimport type {\n  ExportResultsResponse,\n  GetExportResultsQuery,\n  GetQueriesHistoryQuery,\n  GetQueriesPath,\n  GetQueryJobPath,\n  GetQueryResultsPath,\n  GetQueryResutsQuery,\n  PostCancelQueryJobPath,\n  PostQueryBody,\n  PostQueryPath,\n} from './types';\nimport { validateQueryServiceStatus } from './validateStatus';\n\nexport type QueryServiceClientInitOptions = Omit<ClientInitOptions, 'auth' | 'middlewares'> & {\n  auth: QueryServiceCredential | QueryServiceCredentialProvider;\n  middlewares?: MiddlewareFn[];\n};\n\nexport const createQueryServiceClient = ({\n  baseUrl,\n  middlewares = [],\n  auth,\n}: QueryServiceClientInitOptions) => {\n  const client = createOpenapiFetchClient<paths>({\n    baseUrl,\n    ...resolveAuth(auth, middlewares),\n    validateStatus: validateQueryServiceStatus,\n  });\n  const genericClient = createGenericFetchClient({ baseUrl, middlewares });\n\n  const createQueryJob = async (\n    branchId: PostQueryPath['branchId'],\n    workspaceId: PostQueryPath['workspaceId'],\n    body: PostQueryBody,\n  ) => {\n    const { data } = await client.post(\n      '/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries',\n      { path: { branchId, workspaceId }, body },\n    );\n\n    return data;\n  };\n\n  const cancelQueryJob = async (queryJobId: PostCancelQueryJobPath['queryJobId']) => {\n    const { data } = await client.post('/api/v1/queries/{queryJobId}/cancel', {\n      path: { queryJobId },\n      body: { reason: 'Cancelled by user' },\n    });\n\n    return data;\n  };\n\n  const getQueryJob = async (queryJobId: GetQueryJobPath['queryJobId'], signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/api/v1/queries/{queryJobId}',\n      { path: { queryJobId } },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const getQueryResults = async (\n    queryJobId: GetQueryResultsPath['queryJobId'],\n    statementId: GetQueryResultsPath['statementId'],\n    query?: GetQueryResutsQuery,\n  ) => {\n    const { data } = await client.get('/api/v1/queries/{queryJobId}/{statementId}/results', {\n      path: { queryJobId, statementId },\n      query,\n    });\n\n    return data;\n  };\n\n  const getQueryHistory = async (\n    branchId: GetQueriesPath['branchId'],\n    workspaceId: GetQueriesPath['workspaceId'],\n    query?: GetQueriesHistoryQuery,\n  ) => {\n    const { data } = await client.get(\n      '/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries',\n      {\n        path: { branchId, workspaceId },\n        query,\n      },\n    );\n\n    return data;\n  };\n\n  const exportResults = async (\n    queryJobId: GetQueryResultsPath['queryJobId'],\n    statementId: GetQueryResultsPath['statementId'],\n    query: GetExportResultsQuery,\n    signal?: AbortSignal,\n  ): Promise<ExportResultsResponse> => {\n    const { data, response } = await client.get(\n      '/api/v1/queries/{queryJobId}/{statementId}/export',\n      {\n        path: { queryJobId, statementId },\n        query,\n      },\n      { signal },\n    );\n\n    const mimeType = response.headers.get(HttpHeader.CONTENT_TYPE) ?? 'text/csv';\n    const contentDisposition = parseContentDispositionHeader(\n      response.headers.get(HttpHeader.CONTENT_DISPOSITION),\n    );\n\n    return {\n      mimeType,\n      content: data,\n      fileName: contentDisposition?.filename ?? 'export.csv',\n    };\n  };\n\n  const healthCheck = async (signal?: AbortSignal) => {\n    const { data } = await genericClient.get('/health-check', {}, { signal });\n\n    return data;\n  };\n\n  // Every wait on a query job is the same wait, so it is the client's rather than a caller's.\n  const poller = createJobPoller({ getQueryJob });\n\n  return {\n    healthCheck,\n    poller,\n    createQueryJob,\n    cancelQueryJob,\n    getQueryJob,\n    getQueryResults,\n    getQueryHistory,\n    exportResults,\n  };\n};\n"],"mappings":";;;;;;;;AA0BA,MAAM,iCAAiC;AAEvC,MAAM,gBAAgB,UAA2B;CAC/C,IAAI,OAAO,UAAU,YAAY,CAAC,+BAA+B,KAAK,KAAK,GACzE,MAAM,IAAIA,wBAAAA,gBAAgB,wDAAwD;CAEpF,OAAO;AACT;AAEA,MAAM,YAAY,OAAO,WACvB,aAAa,OAAO,WAAW,aAAa,MAAM,OAAO,IAAI,MAAM;AAErE,MAAM,eAAe,eACnB,OAAO,WAAW,UAAU,cAC3B,WAAW,SAAS,oBAAoB,OAAO,WAAW,iBAAiB;;;;;;;;AAS9E,MAAM,iBAAiB,eAA+D;CACpF,MAAM,EAAE,UAAU;CAClB,IAAI,OAAO,UAAU,UAAU,OAAO,CAAC;CACvC,IAAI,WAAW,SAAS,kBACtB,OAAO,OAAO,WAAW,iBAAiB,WACtC;GACGC,0BAAAA,kBAAkB,oBAAoB,aAAa,WAAW,YAAY;GAC1EC,0BAAAA,wBAAwB,aAAa,KAAK;CAC7C,IACA,CAAC;CAEP,OAAOC,aAAAA,kBAAkB;EAAE,GAAG;EAAY;CAAM,CAAC;AACnD;AAEA,MAAM,kBAAkB,OACtB,eACoC;CACpC,MAAM,QAAQ,MAAM,UAAU,WAAW,KAAK;CAC9C,IAAI,WAAW,SAAS,kBACtB,OAAO;GACJF,0BAAAA,kBAAkB,oBAAoB,MAAM,UAAU,WAAW,YAAY;GAC7EC,0BAAAA,wBAAwB;CAC3B;CAEF,IACE,WAAW,SAAS,aACnB,CAAC,OAAO,UAAU,WAAW,SAAS,KAAK,WAAW,aAAa,IAEpE,MAAM,IAAIF,wBAAAA,gBAAgB,sCAAsC;CAElE,OAAOG,aAAAA,kBAAkB;EAAE,GAAG;EAAY;CAAM,CAAC;AACnD;AAEA,MAAM,0BAA0B;CAC9BF,0BAAAA,kBAAkB;CAClBA,0BAAAA,kBAAkB;CAClBA,0BAAAA,kBAAkB;CAClBG,0BAAAA,WAAW;CACXF,0BAAAA;AACF;AAEA,MAAM,8BACH,cACA,SACD,OAAO,YAAY;CACjB,MAAM,oBAAoB,MAAM,gBAAgB,MAAM,SAAS,CAAC;CAChE,MAAM,EAAE,YAAY,QAAQ;CAC5B,KAAK,MAAM,QAAQ,yBAAyB,QAAQ,OAAO,IAAI;CAC/D,KAAK,MAAM,CAAC,MAAM,UAAU,OAAO,QAAQ,iBAAiB,GAAG,QAAQ,IAAI,MAAM,KAAK;CACtF,OAAO,KAAK,OAAO;AACrB;;;;;;;;AASF,MAAa,eACX,MACA,gBACG;CACH,IAAI,OAAO,SAAS,YAClB,OAAO;EAAE,SAAS,KAAA;EAAW,aAAa,CAAC,GAAG,aAAa,2BAA2B,IAAI,CAAC;CAAE;CAE/F,IAAI,YAAY,IAAI,GAClB,OAAO;EACL,SAAS,KAAA;EACT,aAAa,CAAC,GAAG,aAAa,iCAAiC,IAAI,CAAC;CACtE;CAEF,OAAO;EAAE,SAAS,cAAc,IAAI;EAAG;CAAY;AACrD;;ACnHA,MAAM,wBAAwB;AAuB9B,MAAa,mBACV,EAAE,mBACF,EACC,MACA,UACA,aACA,SACA,aACyF;CACzF,MAAM,UAAUG,yBAAAA,iBAAiB;CAEjC,OAAOC,yBAAAA,OAAO;EACZ,UAAU,EAAE,SAAS,CAAC,oBAAoB,EAAE,GAAG,CAAC;EAChD;EACA,QAAQ,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,iBAAiB;GACtD,MAAM,MAAM,MAAM,YAAY,IAAI,UAAU;GAC5C,QAAQ,MAAM;GACd,OAAO;EACT;EACA,aAAa,QAAQ;EACrB,QAAQ,EAAE,WAAWC,yBAAAA,UAAU,MAAM,WAAW,KAAK,MAAM;EAC3D,SAAS,QAAQ;GACf,IAAI,IAAI,WAAW,UAAU,OAAO,KAAA;GACpC,MAAM,mBAAmB,IAAI,WAC1B,QAAQ,cAAc,UAAU,WAAW,QAAQ,CAAC,CACpD,KAAK,eAAe;IAAE,IAAI,UAAU;IAAI,OAAO,UAAU;GAAM,EAAE;GAEpE,OAAO,IAAIC,8BAAAA,cAAc,iBAAiB,EAAE,EAAE,SAAS,cAAc,IAAI,YAAY,EACnF,iBACF,CAAC;EACH;EACA,SAAS;GAAE,MAAM;GAAY,UAAU,YAAY;EAAsB;EACzE;EACA;EACA;CACF,CAAC;AACH;;AC9DF,MAAM,YAAY,UAChB,OAAO,UAAU,YAAY,UAAU;;;;;;;AAQzC,MAAa,8BAAgD,gBAAgB;CAC3E,MAAM,EAAE,UAAU,SAAS;CAE3B,IACE,SAAS,WAAWC,0BAAAA,WAAW,aAC/B,SAAS,IAAI,KACb,KAAK,cAAA,gDAEL,OAAO,IAAIC,8BAAAA,sBAAsB,WAAW;CAG9C,OAAOC,0BAAAA,sBAAsB,WAAW;AAC1C;;ACUA,MAAa,4BAA4B,EACvC,SACA,cAAc,CAAC,GACf,WACmC;CACnC,MAAM,SAASC,iCAAAA,yBAAgC;EAC7C;EACA,GAAG,YAAY,MAAM,WAAW;EAChC,gBAAgB;CAClB,CAAC;CACD,MAAM,gBAAgBC,iCAAAA,yBAAyB;EAAE;EAAS;CAAY,CAAC;CAEvE,MAAM,iBAAiB,OACrB,UACA,aACA,SACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,KAC5B,gEACA;GAAE,MAAM;IAAE;IAAU;GAAY;GAAG;EAAK,CAC1C;EAEA,OAAO;CACT;CAEA,MAAM,iBAAiB,OAAO,eAAqD;EACjF,MAAM,EAAE,SAAS,MAAM,OAAO,KAAK,uCAAuC;GACxE,MAAM,EAAE,WAAW;GACnB,MAAM,EAAE,QAAQ,oBAAoB;EACtC,CAAC;EAED,OAAO;CACT;CAEA,MAAM,cAAc,OAAO,YAA2C,WAAyB;EAC7F,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,gCACA,EAAE,MAAM,EAAE,WAAW,EAAE,GACvB,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,kBAAkB,OACtB,YACA,aACA,UACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,IAAI,sDAAsD;GACtF,MAAM;IAAE;IAAY;GAAY;GAChC;EACF,CAAC;EAED,OAAO;CACT;CAEA,MAAM,kBAAkB,OACtB,UACA,aACA,UACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,gEACA;GACE,MAAM;IAAE;IAAU;GAAY;GAC9B;EACF,CACF;EAEA,OAAO;CACT;CAEA,MAAM,gBAAgB,OACpB,YACA,aACA,OACA,WACmC;EACnC,MAAM,EAAE,MAAM,aAAa,MAAM,OAAO,IACtC,qDACA;GACE,MAAM;IAAE;IAAY;GAAY;GAChC;EACF,GACA,EAAE,OAAO,CACX;EAOA,OAAO;GACL,UANe,SAAS,QAAQ,IAAIC,0BAAAA,WAAW,YAAY,KAAK;GAOhE,SAAS;GACT,UAPyBC,0BAAAA,8BACzB,SAAS,QAAQ,IAAID,0BAAAA,WAAW,mBAAmB,CAMxB,CAAC,EAAE,YAAY;EAC5C;CACF;CAEA,MAAM,cAAc,OAAO,WAAyB;EAClD,MAAM,EAAE,SAAS,MAAM,cAAc,IAAI,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC;EAExE,OAAO;CACT;CAKA,OAAO;EACL;EACA,QAJa,gBAAgB,EAAE,YAAY,CAItC;EACL;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}

package/dist/queryService-DQHtHy2-.js

@@ -0,0 +1,209 @@
+import { _ as HttpStatus, c as defaultValidateStatus, f as parseContentDispositionHeader, m as APP_USER_TOKEN_HEADER, s as HttpHeader, v as KeboolaHttpHeader } from "./createFetchClient-Dc_ZMuq7.js";
+import { r as QueryJobError, t as RowLevelSecurityError } from "./RowLevelSecurityError-CObFqj3E.js";
+import { t as ValidationError } from "./ValidationError-BsdFDGY2.js";
+import { t as createGenericFetchClient } from "./createGenericFetchClient-Ba4E4T-Y.js";
+import { t as createOpenapiFetchClient } from "./createOpenapiFetchClient-DSk29o_s.js";
+import { a as poller, n as jobStatus, t as transientRetries } from "./transientRetries-BORv47mH.js";
+import { t as createAuthHeaders } from "./auth-ZBQWCJPM.js";
+// src/clients/queryService/auth.ts
+const PRINTABLE_ASCII_WITHOUT_SPACES = /^[\x21-\x7E]+$/;
+const requireToken = (token) => {
+	if (typeof token !== "string" || !PRINTABLE_ASCII_WITHOUT_SPACES.test(token)) throw new ValidationError("token must be non-empty printable ASCII without spaces");
+	return token;
+};
+const readToken = async (source) => requireToken(typeof source === "function" ? await source() : source);
+const hasCallback = (credential) => typeof credential.token === "function" || credential.type === "app-user-token" && typeof credential.storageToken === "function";
+/**
+* Fixed headers for a credential with only string tokens.
+*
+* - `app-user-token` tokens are checked here, so an empty user token throws at client creation.
+* - `sapi-token` / `bearer` strings go out unchecked: service clients send an empty `bearer` token
+*   that their own middleware replaces.
+*/
+const staticHeaders = (credential) => {
+	const { token } = credential;
+	if (typeof token !== "string") return {};
+	if (credential.type === "app-user-token") return typeof credential.storageToken === "string" ? {
+		[KeboolaHttpHeader.STORAGE_API_TOKEN]: requireToken(credential.storageToken),
+		[APP_USER_TOKEN_HEADER]: requireToken(token)
+	} : {};
+	return createAuthHeaders({
+		...credential,
+		token
+	});
+};
+const resolvedHeaders = async (credential) => {
+	const token = await readToken(credential.token);
+	if (credential.type === "app-user-token") return {
+		[KeboolaHttpHeader.STORAGE_API_TOKEN]: await readToken(credential.storageToken),
+		[APP_USER_TOKEN_HEADER]: token
+	};
+	if (credential.type === "bearer" && (!Number.isInteger(credential.projectId) || credential.projectId <= 0)) throw new ValidationError("projectId must be a positive integer");
+	return createAuthHeaders({
+		...credential,
+		token
+	});
+};
+const CREDENTIAL_HEADER_NAMES = [
+	KeboolaHttpHeader.STORAGE_API_TOKEN,
+	KeboolaHttpHeader.MANAGEMENT_API_TOKEN,
+	KeboolaHttpHeader.PROJECT_ID,
+	HttpHeader.AUTHORIZATION,
+	APP_USER_TOKEN_HEADER
+];
+const createCredentialMiddleware = (provider) => (next) => async (request) => {
+	const credentialHeaders = await resolvedHeaders(await provider());
+	const { headers } = request.request;
+	for (const name of CREDENTIAL_HEADER_NAMES) headers.delete(name);
+	for (const [name, value] of Object.entries(credentialHeaders)) headers.set(name, value);
+	return next(request);
+};
+/**
+* The Query Service client's credential setup for its `auth`.
+*
+* 1. A credential with only string tokens: default headers, set once (as `createAuthHeaders`).
+* 2. A credential with a token callback, or a provider: no default headers; the innermost
+*    middleware resolves and checks every token on every attempt.
+*/
+const resolveAuth = (auth, middlewares) => {
+	if (typeof auth === "function") return {
+		headers: void 0,
+		middlewares: [...middlewares, createCredentialMiddleware(auth)]
+	};
+	if (hasCallback(auth)) return {
+		headers: void 0,
+		middlewares: [...middlewares, createCredentialMiddleware(() => auth)]
+	};
+	return {
+		headers: staticHeaders(auth),
+		middlewares
+	};
+};
+// src/clients/queryService/poller.ts
+const DEFAULT_POLL_INTERVAL = 2e3;
+const createJobPoller = ({ getQueryJob }) => ({ init, interval, maxAttempts, timeout, signal }) => {
+	const retries = transientRetries();
+	return poller({
+		pollKey: ({ id }) => ["queryService.job", { id }],
+		init,
+		pollFn: async ({ init: { id }, signal: pollSignal }) => {
+			const job = await getQueryJob(id, pollSignal);
+			retries.reset();
+			return job;
+		},
+		isRetryable: retries.isRetryable,
+		until: ({ data }) => jobStatus.query.isTerminal(data.status),
+		failed: (job) => {
+			if (job.status !== "failed") return void 0;
+			const failedStatements = job.statements.filter((statement) => statement.status === "failed").map((statement) => ({
+				id: statement.id,
+				error: statement.error
+			}));
+			return new QueryJobError(failedStatements[0]?.error ?? "Job failed", job.queryJobId, { failedStatements });
+		},
+		backoff: {
+			type: "constant",
+			interval: interval ?? DEFAULT_POLL_INTERVAL
+		},
+		maxAttempts,
+		timeout,
+		signal
+	});
+};
+// src/clients/queryService/validateStatus.ts
+const isRecord = (value) => typeof value === "object" && value !== null;
+/**
+* Status validation for Query Service, which reads a row-level security refusal as `RowLevelSecurityError`.
+*
+* - The fixed `exception` text is the only signal: `X-KBC-RLS-Mode` is not exposed through CORS,
+*   and a credentials 403 on a governed workspace carries that header too.
+*/
+const validateQueryServiceStatus = (apiResponse) => {
+	const { response, data } = apiResponse;
+	if (response.status === HttpStatus.FORBIDDEN && isRecord(data) && data.exception === "Row-level security does not allow this query") return new RowLevelSecurityError(apiResponse);
+	return defaultValidateStatus(apiResponse);
+};
+// src/clients/queryService/queryServiceClient.ts
+const createQueryServiceClient = ({ baseUrl, middlewares = [], auth }) => {
+	const client = createOpenapiFetchClient({
+		baseUrl,
+		...resolveAuth(auth, middlewares),
+		validateStatus: validateQueryServiceStatus
+	});
+	const genericClient = createGenericFetchClient({
+		baseUrl,
+		middlewares
+	});
+	const createQueryJob = async (branchId, workspaceId, body) => {
+		const { data } = await client.post("/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries", {
+			path: {
+				branchId,
+				workspaceId
+			},
+			body
+		});
+		return data;
+	};
+	const cancelQueryJob = async (queryJobId) => {
+		const { data } = await client.post("/api/v1/queries/{queryJobId}/cancel", {
+			path: { queryJobId },
+			body: { reason: "Cancelled by user" }
+		});
+		return data;
+	};
+	const getQueryJob = async (queryJobId, signal) => {
+		const { data } = await client.get("/api/v1/queries/{queryJobId}", { path: { queryJobId } }, { signal });
+		return data;
+	};
+	const getQueryResults = async (queryJobId, statementId, query) => {
+		const { data } = await client.get("/api/v1/queries/{queryJobId}/{statementId}/results", {
+			path: {
+				queryJobId,
+				statementId
+			},
+			query
+		});
+		return data;
+	};
+	const getQueryHistory = async (branchId, workspaceId, query) => {
+		const { data } = await client.get("/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries", {
+			path: {
+				branchId,
+				workspaceId
+			},
+			query
+		});
+		return data;
+	};
+	const exportResults = async (queryJobId, statementId, query, signal) => {
+		const { data, response } = await client.get("/api/v1/queries/{queryJobId}/{statementId}/export", {
+			path: {
+				queryJobId,
+				statementId
+			},
+			query
+		}, { signal });
+		return {
+			mimeType: response.headers.get(HttpHeader.CONTENT_TYPE) ?? "text/csv",
+			content: data,
+			fileName: parseContentDispositionHeader(response.headers.get(HttpHeader.CONTENT_DISPOSITION))?.filename ?? "export.csv"
+		};
+	};
+	const healthCheck = async (signal) => {
+		const { data } = await genericClient.get("/health-check", {}, { signal });
+		return data;
+	};
+	return {
+		healthCheck,
+		poller: createJobPoller({ getQueryJob }),
+		createQueryJob,
+		cancelQueryJob,
+		getQueryJob,
+		getQueryResults,
+		getQueryHistory,
+		exportResults
+	};
+};
+export { createQueryServiceClient as t };
+
+//# sourceMappingURL=queryService-DQHtHy2-.js.map

package/dist/queryService-DQHtHy2-.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queryService-DQHtHy2-.js","names":[],"sources":["../src/clients/queryService/auth.ts","../src/clients/queryService/poller.ts","../src/clients/queryService/validateStatus.ts","../src/clients/queryService/queryServiceClient.ts"],"sourcesContent":["import { ValidationError } from '../../errors';\nimport { HttpHeader, type MiddlewareFn } from '../../fetchClient';\nimport { APP_USER_TOKEN_HEADER, KeboolaHttpHeader } from '../../internal';\nimport { createAuthHeaders } from '../auth';\n\n/** A token value, or a callback that returns the current one (read before every request). */\nexport type TokenSource = string | (() => string | Promise<string>);\n\n/** A data app user's token, sent as `X-Kbc-App-User-Token` together with the app's Storage token. */\nexport type AppUserTokenAuth = {\n  type: 'app-user-token';\n  token: TokenSource;\n  storageToken: TokenSource;\n};\n\n/** One credential; a request carries exactly that credential's headers. */\nexport type QueryServiceCredential =\n  | { type: 'sapi-token'; token: TokenSource }\n  | { type: 'bearer'; token: TokenSource; projectId: number }\n  | AppUserTokenAuth;\n\n/** Resolved before every request, retries included. */\nexport type QueryServiceCredentialProvider = () =>\n  | QueryServiceCredential\n  | Promise<QueryServiceCredential>;\n\nconst PRINTABLE_ASCII_WITHOUT_SPACES = /^[\\x21-\\x7E]+$/;\n\nconst requireToken = (token: unknown): string => {\n  if (typeof token !== 'string' || !PRINTABLE_ASCII_WITHOUT_SPACES.test(token)) {\n    throw new ValidationError('token must be non-empty printable ASCII without spaces');\n  }\n  return token;\n};\n\nconst readToken = async (source: TokenSource) =>\n  requireToken(typeof source === 'function' ? await source() : source);\n\nconst hasCallback = (credential: QueryServiceCredential) =>\n  typeof credential.token === 'function' ||\n  (credential.type === 'app-user-token' && typeof credential.storageToken === 'function');\n\n/**\n * Fixed headers for a credential with only string tokens.\n *\n * - `app-user-token` tokens are checked here, so an empty user token throws at client creation.\n * - `sapi-token` / `bearer` strings go out unchecked: service clients send an empty `bearer` token\n *   that their own middleware replaces.\n */\nconst staticHeaders = (credential: QueryServiceCredential): Record<string, string> => {\n  const { token } = credential;\n  if (typeof token !== 'string') return {};\n  if (credential.type === 'app-user-token') {\n    return typeof credential.storageToken === 'string'\n      ? {\n          [KeboolaHttpHeader.STORAGE_API_TOKEN]: requireToken(credential.storageToken),\n          [APP_USER_TOKEN_HEADER]: requireToken(token),\n        }\n      : {};\n  }\n  return createAuthHeaders({ ...credential, token });\n};\n\nconst resolvedHeaders = async (\n  credential: QueryServiceCredential,\n): Promise<Record<string, string>> => {\n  const token = await readToken(credential.token);\n  if (credential.type === 'app-user-token') {\n    return {\n      [KeboolaHttpHeader.STORAGE_API_TOKEN]: await readToken(credential.storageToken),\n      [APP_USER_TOKEN_HEADER]: token,\n    };\n  }\n  if (\n    credential.type === 'bearer' &&\n    (!Number.isInteger(credential.projectId) || credential.projectId <= 0)\n  ) {\n    throw new ValidationError('projectId must be a positive integer');\n  }\n  return createAuthHeaders({ ...credential, token });\n};\n\nconst CREDENTIAL_HEADER_NAMES = [\n  KeboolaHttpHeader.STORAGE_API_TOKEN,\n  KeboolaHttpHeader.MANAGEMENT_API_TOKEN,\n  KeboolaHttpHeader.PROJECT_ID,\n  HttpHeader.AUTHORIZATION,\n  APP_USER_TOKEN_HEADER,\n];\n\nconst createCredentialMiddleware =\n  (provider: QueryServiceCredentialProvider): MiddlewareFn =>\n  (next) =>\n  async (request) => {\n    const credentialHeaders = await resolvedHeaders(await provider());\n    const { headers } = request.request;\n    for (const name of CREDENTIAL_HEADER_NAMES) headers.delete(name);\n    for (const [name, value] of Object.entries(credentialHeaders)) headers.set(name, value);\n    return next(request);\n  };\n\n/**\n * The Query Service client's credential setup for its `auth`.\n *\n * 1. A credential with only string tokens: default headers, set once (as `createAuthHeaders`).\n * 2. A credential with a token callback, or a provider: no default headers; the innermost\n *    middleware resolves and checks every token on every attempt.\n */\nexport const resolveAuth = (\n  auth: QueryServiceCredential | QueryServiceCredentialProvider,\n  middlewares: MiddlewareFn[],\n) => {\n  if (typeof auth === 'function') {\n    return { headers: undefined, middlewares: [...middlewares, createCredentialMiddleware(auth)] };\n  }\n  if (hasCallback(auth)) {\n    return {\n      headers: undefined,\n      middlewares: [...middlewares, createCredentialMiddleware(() => auth)],\n    };\n  }\n  return { headers: staticHeaders(auth), middlewares };\n};\n","import { jobStatus } from '../../domain/jobStatus';\nimport { QueryJobError } from '../../errors';\nimport { poller } from '../../utils/async';\nimport { transientRetries } from '../transientRetries';\n\nimport type { GetQueryJobPath, JobStatusResponse } from './types';\n\nconst DEFAULT_POLL_INTERVAL = 2000;\n\nexport type QueryJobPollOptions = {\n  interval?: number;\n  maxAttempts?: number;\n  timeout?: number;\n  signal?: AbortSignal;\n};\n\ntype GetQueryJob = (\n  queryJobId: GetQueryJobPath['queryJobId'],\n  signal?: AbortSignal,\n) => Promise<JobStatusResponse>;\n\n/**\n * Waits on a query job.\n *\n * - `init` is the job, or the request that yields it, so a caller hands over what `createQueryJob` gave it.\n * - Terminal is the job's own terminal set, not its statements': a `completed` job can still hold a\n *   statement that failed, and what that means is the caller's business.\n */\ntype JobRef = { id: GetQueryJobPath['queryJobId'] };\n\nexport const createJobPoller =\n  ({ getQueryJob }: { getQueryJob: GetQueryJob }) =>\n  ({\n    init,\n    interval,\n    maxAttempts,\n    timeout,\n    signal,\n  }: QueryJobPollOptions & { init: JobRef | ((signal: AbortSignal) => Promise<JobRef>) }) => {\n    const retries = transientRetries();\n\n    return poller({\n      pollKey: ({ id }) => ['queryService.job', { id }] as const,\n      init,\n      pollFn: async ({ init: { id }, signal: pollSignal }) => {\n        const job = await getQueryJob(id, pollSignal);\n        retries.reset();\n        return job;\n      },\n      isRetryable: retries.isRetryable,\n      until: ({ data }) => jobStatus.query.isTerminal(data.status),\n      failed: (job) => {\n        if (job.status !== 'failed') return undefined;\n        const failedStatements = job.statements\n          .filter((statement) => statement.status === 'failed')\n          .map((statement) => ({ id: statement.id, error: statement.error }));\n\n        return new QueryJobError(failedStatements[0]?.error ?? 'Job failed', job.queryJobId, {\n          failedStatements,\n        });\n      },\n      backoff: { type: 'constant', interval: interval ?? DEFAULT_POLL_INTERVAL },\n      maxAttempts,\n      timeout,\n      signal,\n    });\n  };\n","import { RowLevelSecurityError } from '../../errors';\nimport { defaultValidateStatus, type ValidateStatusFn } from '../../fetchClient';\nimport { HttpStatus, ROW_LEVEL_SECURITY_DENIED_MESSAGE } from '../../internal';\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === 'object' && value !== null;\n\n/**\n * Status validation for Query Service, which reads a row-level security refusal as `RowLevelSecurityError`.\n *\n * - The fixed `exception` text is the only signal: `X-KBC-RLS-Mode` is not exposed through CORS,\n *   and a credentials 403 on a governed workspace carries that header too.\n */\nexport const validateQueryServiceStatus: ValidateStatusFn = (apiResponse) => {\n  const { response, data } = apiResponse;\n\n  if (\n    response.status === HttpStatus.FORBIDDEN &&\n    isRecord(data) &&\n    data.exception === ROW_LEVEL_SECURITY_DENIED_MESSAGE\n  ) {\n    return new RowLevelSecurityError(apiResponse);\n  }\n\n  return defaultValidateStatus(apiResponse);\n};\n","import {\n  createGenericFetchClient,\n  createOpenapiFetchClient,\n  HttpHeader,\n  type MiddlewareFn,\n  parseContentDispositionHeader,\n} from '../../fetchClient';\nimport type { ClientInitOptions } from '../types';\n\nimport type { paths } from './__generated__/schema';\nimport {\n  type QueryServiceCredential,\n  type QueryServiceCredentialProvider,\n  resolveAuth,\n} from './auth';\nimport { createJobPoller } from './poller';\nimport type {\n  ExportResultsResponse,\n  GetExportResultsQuery,\n  GetQueriesHistoryQuery,\n  GetQueriesPath,\n  GetQueryJobPath,\n  GetQueryResultsPath,\n  GetQueryResutsQuery,\n  PostCancelQueryJobPath,\n  PostQueryBody,\n  PostQueryPath,\n} from './types';\nimport { validateQueryServiceStatus } from './validateStatus';\n\nexport type QueryServiceClientInitOptions = Omit<ClientInitOptions, 'auth' | 'middlewares'> & {\n  auth: QueryServiceCredential | QueryServiceCredentialProvider;\n  middlewares?: MiddlewareFn[];\n};\n\nexport const createQueryServiceClient = ({\n  baseUrl,\n  middlewares = [],\n  auth,\n}: QueryServiceClientInitOptions) => {\n  const client = createOpenapiFetchClient<paths>({\n    baseUrl,\n    ...resolveAuth(auth, middlewares),\n    validateStatus: validateQueryServiceStatus,\n  });\n  const genericClient = createGenericFetchClient({ baseUrl, middlewares });\n\n  const createQueryJob = async (\n    branchId: PostQueryPath['branchId'],\n    workspaceId: PostQueryPath['workspaceId'],\n    body: PostQueryBody,\n  ) => {\n    const { data } = await client.post(\n      '/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries',\n      { path: { branchId, workspaceId }, body },\n    );\n\n    return data;\n  };\n\n  const cancelQueryJob = async (queryJobId: PostCancelQueryJobPath['queryJobId']) => {\n    const { data } = await client.post('/api/v1/queries/{queryJobId}/cancel', {\n      path: { queryJobId },\n      body: { reason: 'Cancelled by user' },\n    });\n\n    return data;\n  };\n\n  const getQueryJob = async (queryJobId: GetQueryJobPath['queryJobId'], signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/api/v1/queries/{queryJobId}',\n      { path: { queryJobId } },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const getQueryResults = async (\n    queryJobId: GetQueryResultsPath['queryJobId'],\n    statementId: GetQueryResultsPath['statementId'],\n    query?: GetQueryResutsQuery,\n  ) => {\n    const { data } = await client.get('/api/v1/queries/{queryJobId}/{statementId}/results', {\n      path: { queryJobId, statementId },\n      query,\n    });\n\n    return data;\n  };\n\n  const getQueryHistory = async (\n    branchId: GetQueriesPath['branchId'],\n    workspaceId: GetQueriesPath['workspaceId'],\n    query?: GetQueriesHistoryQuery,\n  ) => {\n    const { data } = await client.get(\n      '/api/v1/branches/{branchId}/workspaces/{workspaceId}/queries',\n      {\n        path: { branchId, workspaceId },\n        query,\n      },\n    );\n\n    return data;\n  };\n\n  const exportResults = async (\n    queryJobId: GetQueryResultsPath['queryJobId'],\n    statementId: GetQueryResultsPath['statementId'],\n    query: GetExportResultsQuery,\n    signal?: AbortSignal,\n  ): Promise<ExportResultsResponse> => {\n    const { data, response } = await client.get(\n      '/api/v1/queries/{queryJobId}/{statementId}/export',\n      {\n        path: { queryJobId, statementId },\n        query,\n      },\n      { signal },\n    );\n\n    const mimeType = response.headers.get(HttpHeader.CONTENT_TYPE) ?? 'text/csv';\n    const contentDisposition = parseContentDispositionHeader(\n      response.headers.get(HttpHeader.CONTENT_DISPOSITION),\n    );\n\n    return {\n      mimeType,\n      content: data,\n      fileName: contentDisposition?.filename ?? 'export.csv',\n    };\n  };\n\n  const healthCheck = async (signal?: AbortSignal) => {\n    const { data } = await genericClient.get('/health-check', {}, { signal });\n\n    return data;\n  };\n\n  // Every wait on a query job is the same wait, so it is the client's rather than a caller's.\n  const poller = createJobPoller({ getQueryJob });\n\n  return {\n    healthCheck,\n    poller,\n    createQueryJob,\n    cancelQueryJob,\n    getQueryJob,\n    getQueryResults,\n    getQueryHistory,\n    exportResults,\n  };\n};\n"],"mappings":";;;;;;;;AA0BA,MAAM,iCAAiC;AAEvC,MAAM,gBAAgB,UAA2B;CAC/C,IAAI,OAAO,UAAU,YAAY,CAAC,+BAA+B,KAAK,KAAK,GACzE,MAAM,IAAI,gBAAgB,wDAAwD;CAEpF,OAAO;AACT;AAEA,MAAM,YAAY,OAAO,WACvB,aAAa,OAAO,WAAW,aAAa,MAAM,OAAO,IAAI,MAAM;AAErE,MAAM,eAAe,eACnB,OAAO,WAAW,UAAU,cAC3B,WAAW,SAAS,oBAAoB,OAAO,WAAW,iBAAiB;;;;;;;;AAS9E,MAAM,iBAAiB,eAA+D;CACpF,MAAM,EAAE,UAAU;CAClB,IAAI,OAAO,UAAU,UAAU,OAAO,CAAC;CACvC,IAAI,WAAW,SAAS,kBACtB,OAAO,OAAO,WAAW,iBAAiB,WACtC;GACG,kBAAkB,oBAAoB,aAAa,WAAW,YAAY;GAC1E,wBAAwB,aAAa,KAAK;CAC7C,IACA,CAAC;CAEP,OAAO,kBAAkB;EAAE,GAAG;EAAY;CAAM,CAAC;AACnD;AAEA,MAAM,kBAAkB,OACtB,eACoC;CACpC,MAAM,QAAQ,MAAM,UAAU,WAAW,KAAK;CAC9C,IAAI,WAAW,SAAS,kBACtB,OAAO;GACJ,kBAAkB,oBAAoB,MAAM,UAAU,WAAW,YAAY;GAC7E,wBAAwB;CAC3B;CAEF,IACE,WAAW,SAAS,aACnB,CAAC,OAAO,UAAU,WAAW,SAAS,KAAK,WAAW,aAAa,IAEpE,MAAM,IAAI,gBAAgB,sCAAsC;CAElE,OAAO,kBAAkB;EAAE,GAAG;EAAY;CAAM,CAAC;AACnD;AAEA,MAAM,0BAA0B;CAC9B,kBAAkB;CAClB,kBAAkB;CAClB,kBAAkB;CAClB,WAAW;CACX;AACF;AAEA,MAAM,8BACH,cACA,SACD,OAAO,YAAY;CACjB,MAAM,oBAAoB,MAAM,gBAAgB,MAAM,SAAS,CAAC;CAChE,MAAM,EAAE,YAAY,QAAQ;CAC5B,KAAK,MAAM,QAAQ,yBAAyB,QAAQ,OAAO,IAAI;CAC/D,KAAK,MAAM,CAAC,MAAM,UAAU,OAAO,QAAQ,iBAAiB,GAAG,QAAQ,IAAI,MAAM,KAAK;CACtF,OAAO,KAAK,OAAO;AACrB;;;;;;;;AASF,MAAa,eACX,MACA,gBACG;CACH,IAAI,OAAO,SAAS,YAClB,OAAO;EAAE,SAAS,KAAA;EAAW,aAAa,CAAC,GAAG,aAAa,2BAA2B,IAAI,CAAC;CAAE;CAE/F,IAAI,YAAY,IAAI,GAClB,OAAO;EACL,SAAS,KAAA;EACT,aAAa,CAAC,GAAG,aAAa,iCAAiC,IAAI,CAAC;CACtE;CAEF,OAAO;EAAE,SAAS,cAAc,IAAI;EAAG;CAAY;AACrD;;ACnHA,MAAM,wBAAwB;AAuB9B,MAAa,mBACV,EAAE,mBACF,EACC,MACA,UACA,aACA,SACA,aACyF;CACzF,MAAM,UAAU,iBAAiB;CAEjC,OAAO,OAAO;EACZ,UAAU,EAAE,SAAS,CAAC,oBAAoB,EAAE,GAAG,CAAC;EAChD;EACA,QAAQ,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,iBAAiB;GACtD,MAAM,MAAM,MAAM,YAAY,IAAI,UAAU;GAC5C,QAAQ,MAAM;GACd,OAAO;EACT;EACA,aAAa,QAAQ;EACrB,QAAQ,EAAE,WAAW,UAAU,MAAM,WAAW,KAAK,MAAM;EAC3D,SAAS,QAAQ;GACf,IAAI,IAAI,WAAW,UAAU,OAAO,KAAA;GACpC,MAAM,mBAAmB,IAAI,WAC1B,QAAQ,cAAc,UAAU,WAAW,QAAQ,CAAC,CACpD,KAAK,eAAe;IAAE,IAAI,UAAU;IAAI,OAAO,UAAU;GAAM,EAAE;GAEpE,OAAO,IAAI,cAAc,iBAAiB,EAAE,EAAE,SAAS,cAAc,IAAI,YAAY,EACnF,iBACF,CAAC;EACH;EACA,SAAS;GAAE,MAAM;GAAY,UAAU,YAAY;EAAsB;EACzE;EACA;EACA;CACF,CAAC;AACH;;AC9DF,MAAM,YAAY,UAChB,OAAO,UAAU,YAAY,UAAU;;;;;;;AAQzC,MAAa,8BAAgD,gBAAgB;CAC3E,MAAM,EAAE,UAAU,SAAS;CAE3B,IACE,SAAS,WAAW,WAAW,aAC/B,SAAS,IAAI,KACb,KAAK,cAAA,gDAEL,OAAO,IAAI,sBAAsB,WAAW;CAG9C,OAAO,sBAAsB,WAAW;AAC1C;;ACUA,MAAa,4BAA4B,EACvC,SACA,cAAc,CAAC,GACf,WACmC;CACnC,MAAM,SAAS,yBAAgC;EAC7C;EACA,GAAG,YAAY,MAAM,WAAW;EAChC,gBAAgB;CAClB,CAAC;CACD,MAAM,gBAAgB,yBAAyB;EAAE;EAAS;CAAY,CAAC;CAEvE,MAAM,iBAAiB,OACrB,UACA,aACA,SACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,KAC5B,gEACA;GAAE,MAAM;IAAE;IAAU;GAAY;GAAG;EAAK,CAC1C;EAEA,OAAO;CACT;CAEA,MAAM,iBAAiB,OAAO,eAAqD;EACjF,MAAM,EAAE,SAAS,MAAM,OAAO,KAAK,uCAAuC;GACxE,MAAM,EAAE,WAAW;GACnB,MAAM,EAAE,QAAQ,oBAAoB;EACtC,CAAC;EAED,OAAO;CACT;CAEA,MAAM,cAAc,OAAO,YAA2C,WAAyB;EAC7F,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,gCACA,EAAE,MAAM,EAAE,WAAW,EAAE,GACvB,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,kBAAkB,OACtB,YACA,aACA,UACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,IAAI,sDAAsD;GACtF,MAAM;IAAE;IAAY;GAAY;GAChC;EACF,CAAC;EAED,OAAO;CACT;CAEA,MAAM,kBAAkB,OACtB,UACA,aACA,UACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,gEACA;GACE,MAAM;IAAE;IAAU;GAAY;GAC9B;EACF,CACF;EAEA,OAAO;CACT;CAEA,MAAM,gBAAgB,OACpB,YACA,aACA,OACA,WACmC;EACnC,MAAM,EAAE,MAAM,aAAa,MAAM,OAAO,IACtC,qDACA;GACE,MAAM;IAAE;IAAY;GAAY;GAChC;EACF,GACA,EAAE,OAAO,CACX;EAOA,OAAO;GACL,UANe,SAAS,QAAQ,IAAI,WAAW,YAAY,KAAK;GAOhE,SAAS;GACT,UAPyB,8BACzB,SAAS,QAAQ,IAAI,WAAW,mBAAmB,CAMxB,CAAC,EAAE,YAAY;EAC5C;CACF;CAEA,MAAM,cAAc,OAAO,WAAyB;EAClD,MAAM,EAAE,SAAS,MAAM,cAAc,IAAI,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC;EAExE,OAAO;CACT;CAKA,OAAO;EACL;EACA,QAJa,gBAAgB,EAAE,YAAY,CAItC;EACL;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}

package/dist/queue/index.cjs

@@ -1,3 +1,3 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_queue = require("../queue-CBi9Q8vp.cjs");
+const require_queue = require("../queue-J2uXzXR9.cjs");
 exports.createQueueClient = require_queue.createQueueClient;

package/dist/queue/index.d.cts

@@ -1,6 +1,6 @@
-import { Br as CreateJobBody, Hr as KillJobPath, Jr as SearchGroupedJobsQuery, Qr as components, Vr as GetQueueJobPath, Wr as OrchestrationRerunPlanPath, Xr as SearchJobsQuery, qr as SearchAllJobsQuery } from "../index-D1mduJne.cjs";
-import "../index-qJ2IjXIM.cjs";
-import { n as ClientInitOptions } from "../types-Dryq2GGE.cjs";
+import { Hr as CreateJobBody, Kr as OrchestrationRerunPlanPath, Qr as SearchJobsQuery, Ur as GetQueueJobPath, Wr as KillJobPath, Xr as SearchGroupedJobsQuery, Yr as SearchAllJobsQuery, ei as components } from "../index-FrsEkDvX.cjs";
+import "../index-B7el5pJU.cjs";
+import { n as ClientInitOptions } from "../types-B4ezB18r.cjs";
 import { c as Poller } from "../index-24GpnYhn.cjs";
 declare const createQueueClient: ({ baseUrl, middlewares, auth }: ClientInitOptions) => {
   healthCheck: (signal?: AbortSignal) => Promise<unknown>;

package/dist/queue/index.d.ts

@@ -1,6 +1,6 @@
-import { Br as CreateJobBody, Hr as KillJobPath, Jr as SearchGroupedJobsQuery, Qr as components, Vr as GetQueueJobPath, Wr as OrchestrationRerunPlanPath, Xr as SearchJobsQuery, qr as SearchAllJobsQuery } from "../index-D1mduJne.js";
-import "../index-BsT_cBJd.js";
-import { n as ClientInitOptions } from "../types-DUpVIuk7.js";
+import { Hr as CreateJobBody, Kr as OrchestrationRerunPlanPath, Qr as SearchJobsQuery, Ur as GetQueueJobPath, Wr as KillJobPath, Xr as SearchGroupedJobsQuery, Yr as SearchAllJobsQuery, ei as components } from "../index-FrsEkDvX.js";
+import "../index-C7JcNvbd.js";
+import { n as ClientInitOptions } from "../types-Tg3euEN_.js";
 import { c as Poller } from "../index-BUZ-vTRt.js";
 declare const createQueueClient: ({ baseUrl, middlewares, auth }: ClientInitOptions) => {
   healthCheck: (signal?: AbortSignal) => Promise<unknown>;

package/dist/queue/index.js

@@ -1 +1 @@
-export { t as createQueueClient } from "../queue-BiIs9fX4.js";
+export { t as createQueueClient } from "../queue-6cywDnA-.js";

package/dist/queue/types.d.cts

@@ -1,2 +1,2 @@
-import { Br as CreateJobBody, Gr as QueueJob, Hr as KillJobPath, Jr as SearchGroupedJobsQuery, Kr as QueueJobStatus, Ur as OrchestrationRerunPlan, Vr as GetQueueJobPath, Wr as OrchestrationRerunPlanPath, Xr as SearchJobsQuery, Yr as SearchGroupedJobsResponse, Zr as SearchJobsResponse, qr as SearchAllJobsQuery } from "../index-D1mduJne.cjs";
+import { $r as SearchJobsResponse, Gr as OrchestrationRerunPlan, Hr as CreateJobBody, Jr as QueueJobStatus, Kr as OrchestrationRerunPlanPath, Qr as SearchJobsQuery, Ur as GetQueueJobPath, Wr as KillJobPath, Xr as SearchGroupedJobsQuery, Yr as SearchAllJobsQuery, Zr as SearchGroupedJobsResponse, qr as QueueJob } from "../index-FrsEkDvX.cjs";
 export { CreateJobBody, GetQueueJobPath, KillJobPath, OrchestrationRerunPlan, OrchestrationRerunPlanPath, QueueJob, QueueJobStatus, SearchAllJobsQuery, SearchGroupedJobsQuery, SearchGroupedJobsResponse, SearchJobsQuery, SearchJobsResponse };

package/dist/queue/types.d.ts

@@ -1,2 +1,2 @@
-import { Br as CreateJobBody, Gr as QueueJob, Hr as KillJobPath, Jr as SearchGroupedJobsQuery, Kr as QueueJobStatus, Ur as OrchestrationRerunPlan, Vr as GetQueueJobPath, Wr as OrchestrationRerunPlanPath, Xr as SearchJobsQuery, Yr as SearchGroupedJobsResponse, Zr as SearchJobsResponse, qr as SearchAllJobsQuery } from "../index-D1mduJne.js";
+import { $r as SearchJobsResponse, Gr as OrchestrationRerunPlan, Hr as CreateJobBody, Jr as QueueJobStatus, Kr as OrchestrationRerunPlanPath, Qr as SearchJobsQuery, Ur as GetQueueJobPath, Wr as KillJobPath, Xr as SearchGroupedJobsQuery, Yr as SearchAllJobsQuery, Zr as SearchGroupedJobsResponse, qr as QueueJob } from "../index-FrsEkDvX.js";
 export { CreateJobBody, GetQueueJobPath, KillJobPath, OrchestrationRerunPlan, OrchestrationRerunPlanPath, QueueJob, QueueJobStatus, SearchAllJobsQuery, SearchGroupedJobsQuery, SearchGroupedJobsResponse, SearchJobsQuery, SearchJobsResponse };

package/dist/{queue-BiIs9fX4.js → queue-6cywDnA-.js}

@@ -1,9 +1,9 @@
 import { t as QueueJobError } from "./QueueJobError-BJNUINXW.js";
-import { t as createGenericFetchClient } from "./createGenericFetchClient-wGxemBwR.js";
-import { t as createOpenapiFetchClient } from "./createOpenapiFetchClient-Bhz62MUE.js";
+import { t as createGenericFetchClient } from "./createGenericFetchClient-Ba4E4T-Y.js";
+import { t as createOpenapiFetchClient } from "./createOpenapiFetchClient-DSk29o_s.js";
 import { t as collectByOffset } from "./collectByOffset-i8Tdoze7.js";
-import { a as poller, n as jobStatus, t as transientRetries } from "./transientRetries-BKqmb85B.js";
-import { t as createAuthHeaders } from "./auth-DYfQMQCs.js";
+import { a as poller, n as jobStatus, t as transientRetries } from "./transientRetries-BORv47mH.js";
+import { t as createAuthHeaders } from "./auth-ZBQWCJPM.js";
 // src/clients/queue/poller.ts
 const DEFAULT_POLL_INTERVAL = 2e3;
 /**
@@ -103,4 +103,4 @@ const createQueueClient = ({ baseUrl, middlewares, auth }) => {
 };
 export { createQueueClient as t };
 
-//# sourceMappingURL=queue-BiIs9fX4.js.map
+//# sourceMappingURL=queue-6cywDnA-.js.map

package/dist/{queue-BiIs9fX4.js.map → queue-6cywDnA-.js.map}

@@ -1 +1 @@
-{"version":3,"file":"queue-BiIs9fX4.js","names":[],"sources":["../src/clients/queue/poller.ts","../src/clients/queue/queueClient.ts"],"sourcesContent":["import { jobStatus } from '../../domain/jobStatus';\nimport { QueueJobError } from '../../errors';\nimport { poller } from '../../utils/async';\nimport { transientRetries } from '../transientRetries';\n\nimport type { GetQueueJobPath, QueueJob } from './types';\n\nconst DEFAULT_POLL_INTERVAL = 2000;\n\ntype QueueJobPollOptions = {\n  interval?: number;\n  maxAttempts?: number;\n  timeout?: number;\n  signal?: AbortSignal;\n};\n\ntype GetJob = (jobId: GetQueueJobPath['jobId'], signal?: AbortSignal) => Promise<QueueJob>;\n\ntype JobRef = { id: GetQueueJobPath['jobId']; component: string };\n\n/**\n * Waits on a queue job.\n *\n * - `init` is the job, or the enqueue that yields it; the job names the poll in dev tools.\n * - `start()` resolves any terminal job; `wait()` throws `QueueJobError` for one that ended `error`.\n */\nexport const createJobPoller =\n  ({ getJob }: { getJob: GetJob }) =>\n  ({\n    init,\n    interval,\n    maxAttempts,\n    timeout,\n    signal,\n  }: QueueJobPollOptions & { init: JobRef | ((signal: AbortSignal) => Promise<JobRef>) }) => {\n    const retries = transientRetries();\n\n    return poller({\n      pollKey: ({ id, component }) => ['queue.job', { id, component }] as const,\n      init,\n      pollFn: async ({ init: { id }, signal: pollSignal }) => {\n        const job = await getJob(id, pollSignal);\n        retries.reset();\n        return job;\n      },\n      isRetryable: retries.isRetryable,\n      until: ({ data }) => jobStatus.queue.isTerminal(data.status),\n      failed: (job) => (job.status === 'error' ? new QueueJobError(job) : undefined),\n      backoff: { type: 'constant', interval: interval ?? DEFAULT_POLL_INTERVAL },\n      maxAttempts,\n      timeout,\n      signal,\n    });\n  };\n","import { createGenericFetchClient, createOpenapiFetchClient } from '../../fetchClient';\nimport { collectByOffset } from '../../utils';\nimport { createAuthHeaders } from '../auth';\nimport type { ClientInitOptions } from '../types';\n\nimport type { paths } from './__generated__/schema';\nimport { createJobPoller } from './poller';\nimport type {\n  CreateJobBody,\n  GetQueueJobPath,\n  KillJobPath,\n  OrchestrationRerunPlanPath,\n  SearchAllJobsQuery,\n  SearchGroupedJobsQuery,\n  SearchJobsQuery,\n} from './types';\n\nexport const createQueueClient = ({ baseUrl, middlewares, auth }: ClientInitOptions) => {\n  const client = createOpenapiFetchClient<paths>({\n    baseUrl,\n    middlewares,\n    headers: createAuthHeaders(auth),\n  });\n  const genericClient = createGenericFetchClient({ baseUrl, middlewares });\n\n  const getJob = async (jobId: GetQueueJobPath['jobId'], signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/jobs/{jobId}',\n      {\n        path: { jobId },\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const searchJobs = async (query?: SearchJobsQuery, signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/search/jobs',\n      {\n        query,\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const searchAllJobs = (query: SearchAllJobsQuery, signal?: AbortSignal) => {\n    return collectByOffset({\n      pageSize: query.limit,\n      signal,\n      fetchFrom: (offset, pageSize, batchSignal) => {\n        return searchJobs({ ...query, limit: pageSize, offset }, batchSignal);\n      },\n    });\n  };\n\n  const createJob = async (body: CreateJobBody, signal?: AbortSignal) => {\n    const { data } = await client.post('/jobs', { body }, { signal });\n\n    return data;\n  };\n\n  const killJob = async (jobId: KillJobPath['jobId'], signal?: AbortSignal) => {\n    const { data } = await client.post(\n      '/jobs/{jobId}/kill',\n      {\n        path: { jobId },\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const searchGroupedJobs = async (query: SearchGroupedJobsQuery, signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/search/grouped-jobs',\n      {\n        query,\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const getOrchestrationRerunPlan = async (\n    jobId: OrchestrationRerunPlanPath['jobId'],\n    signal?: AbortSignal,\n  ) => {\n    const { data } = await client.get(\n      '/orchestration-rerun-plan/{jobId}',\n      {\n        path: { jobId },\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const healthCheck = async (signal?: AbortSignal) => {\n    const { data } = await genericClient.get('/health-check', {}, { signal });\n\n    return data;\n  };\n\n  // Every wait on a queue job is the same wait, so it is the client's rather than a caller's.\n  const poller = createJobPoller({ getJob });\n\n  return {\n    healthCheck,\n    poller,\n    getJob,\n    searchJobs,\n    searchAllJobs,\n    createJob,\n    killJob,\n    searchGroupedJobs,\n    getOrchestrationRerunPlan,\n  };\n};\n"],"mappings":";;;;;;;AAOA,MAAM,wBAAwB;;;;;;;AAmB9B,MAAa,mBACV,EAAE,cACF,EACC,MACA,UACA,aACA,SACA,aACyF;CACzF,MAAM,UAAU,iBAAiB;CAEjC,OAAO,OAAO;EACZ,UAAU,EAAE,IAAI,gBAAgB,CAAC,aAAa;GAAE;GAAI;EAAU,CAAC;EAC/D;EACA,QAAQ,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,iBAAiB;GACtD,MAAM,MAAM,MAAM,OAAO,IAAI,UAAU;GACvC,QAAQ,MAAM;GACd,OAAO;EACT;EACA,aAAa,QAAQ;EACrB,QAAQ,EAAE,WAAW,UAAU,MAAM,WAAW,KAAK,MAAM;EAC3D,SAAS,QAAS,IAAI,WAAW,UAAU,IAAI,cAAc,GAAG,IAAI,KAAA;EACpE,SAAS;GAAE,MAAM;GAAY,UAAU,YAAY;EAAsB;EACzE;EACA;EACA;CACF,CAAC;AACH;;ACpCF,MAAa,qBAAqB,EAAE,SAAS,aAAa,WAA8B;CACtF,MAAM,SAAS,yBAAgC;EAC7C;EACA;EACA,SAAS,kBAAkB,IAAI;CACjC,CAAC;CACD,MAAM,gBAAgB,yBAAyB;EAAE;EAAS;CAAY,CAAC;CAEvE,MAAM,SAAS,OAAO,OAAiC,WAAyB;EAC9E,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,iBACA,EACE,MAAM,EAAE,MAAM,EAChB,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,aAAa,OAAO,OAAyB,WAAyB;EAC1E,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,gBACA,EACE,MACF,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,iBAAiB,OAA2B,WAAyB;EACzE,OAAO,gBAAgB;GACrB,UAAU,MAAM;GAChB;GACA,YAAY,QAAQ,UAAU,gBAAgB;IAC5C,OAAO,WAAW;KAAE,GAAG;KAAO,OAAO;KAAU;IAAO,GAAG,WAAW;GACtE;EACF,CAAC;CACH;CAEA,MAAM,YAAY,OAAO,MAAqB,WAAyB;EACrE,MAAM,EAAE,SAAS,MAAM,OAAO,KAAK,SAAS,EAAE,KAAK,GAAG,EAAE,OAAO,CAAC;EAEhE,OAAO;CACT;CAEA,MAAM,UAAU,OAAO,OAA6B,WAAyB;EAC3E,MAAM,EAAE,SAAS,MAAM,OAAO,KAC5B,sBACA,EACE,MAAM,EAAE,MAAM,EAChB,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,oBAAoB,OAAO,OAA+B,WAAyB;EACvF,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,wBACA,EACE,MACF,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,4BAA4B,OAChC,OACA,WACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,qCACA,EACE,MAAM,EAAE,MAAM,EAChB,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,cAAc,OAAO,WAAyB;EAClD,MAAM,EAAE,SAAS,MAAM,cAAc,IAAI,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC;EAExE,OAAO;CACT;CAKA,OAAO;EACL;EACA,QAJa,gBAAgB,EAAE,OAAO,CAIjC;EACL;EACA;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}
+{"version":3,"file":"queue-6cywDnA-.js","names":[],"sources":["../src/clients/queue/poller.ts","../src/clients/queue/queueClient.ts"],"sourcesContent":["import { jobStatus } from '../../domain/jobStatus';\nimport { QueueJobError } from '../../errors';\nimport { poller } from '../../utils/async';\nimport { transientRetries } from '../transientRetries';\n\nimport type { GetQueueJobPath, QueueJob } from './types';\n\nconst DEFAULT_POLL_INTERVAL = 2000;\n\ntype QueueJobPollOptions = {\n  interval?: number;\n  maxAttempts?: number;\n  timeout?: number;\n  signal?: AbortSignal;\n};\n\ntype GetJob = (jobId: GetQueueJobPath['jobId'], signal?: AbortSignal) => Promise<QueueJob>;\n\ntype JobRef = { id: GetQueueJobPath['jobId']; component: string };\n\n/**\n * Waits on a queue job.\n *\n * - `init` is the job, or the enqueue that yields it; the job names the poll in dev tools.\n * - `start()` resolves any terminal job; `wait()` throws `QueueJobError` for one that ended `error`.\n */\nexport const createJobPoller =\n  ({ getJob }: { getJob: GetJob }) =>\n  ({\n    init,\n    interval,\n    maxAttempts,\n    timeout,\n    signal,\n  }: QueueJobPollOptions & { init: JobRef | ((signal: AbortSignal) => Promise<JobRef>) }) => {\n    const retries = transientRetries();\n\n    return poller({\n      pollKey: ({ id, component }) => ['queue.job', { id, component }] as const,\n      init,\n      pollFn: async ({ init: { id }, signal: pollSignal }) => {\n        const job = await getJob(id, pollSignal);\n        retries.reset();\n        return job;\n      },\n      isRetryable: retries.isRetryable,\n      until: ({ data }) => jobStatus.queue.isTerminal(data.status),\n      failed: (job) => (job.status === 'error' ? new QueueJobError(job) : undefined),\n      backoff: { type: 'constant', interval: interval ?? DEFAULT_POLL_INTERVAL },\n      maxAttempts,\n      timeout,\n      signal,\n    });\n  };\n","import { createGenericFetchClient, createOpenapiFetchClient } from '../../fetchClient';\nimport { collectByOffset } from '../../utils';\nimport { createAuthHeaders } from '../auth';\nimport type { ClientInitOptions } from '../types';\n\nimport type { paths } from './__generated__/schema';\nimport { createJobPoller } from './poller';\nimport type {\n  CreateJobBody,\n  GetQueueJobPath,\n  KillJobPath,\n  OrchestrationRerunPlanPath,\n  SearchAllJobsQuery,\n  SearchGroupedJobsQuery,\n  SearchJobsQuery,\n} from './types';\n\nexport const createQueueClient = ({ baseUrl, middlewares, auth }: ClientInitOptions) => {\n  const client = createOpenapiFetchClient<paths>({\n    baseUrl,\n    middlewares,\n    headers: createAuthHeaders(auth),\n  });\n  const genericClient = createGenericFetchClient({ baseUrl, middlewares });\n\n  const getJob = async (jobId: GetQueueJobPath['jobId'], signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/jobs/{jobId}',\n      {\n        path: { jobId },\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const searchJobs = async (query?: SearchJobsQuery, signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/search/jobs',\n      {\n        query,\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const searchAllJobs = (query: SearchAllJobsQuery, signal?: AbortSignal) => {\n    return collectByOffset({\n      pageSize: query.limit,\n      signal,\n      fetchFrom: (offset, pageSize, batchSignal) => {\n        return searchJobs({ ...query, limit: pageSize, offset }, batchSignal);\n      },\n    });\n  };\n\n  const createJob = async (body: CreateJobBody, signal?: AbortSignal) => {\n    const { data } = await client.post('/jobs', { body }, { signal });\n\n    return data;\n  };\n\n  const killJob = async (jobId: KillJobPath['jobId'], signal?: AbortSignal) => {\n    const { data } = await client.post(\n      '/jobs/{jobId}/kill',\n      {\n        path: { jobId },\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const searchGroupedJobs = async (query: SearchGroupedJobsQuery, signal?: AbortSignal) => {\n    const { data } = await client.get(\n      '/search/grouped-jobs',\n      {\n        query,\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const getOrchestrationRerunPlan = async (\n    jobId: OrchestrationRerunPlanPath['jobId'],\n    signal?: AbortSignal,\n  ) => {\n    const { data } = await client.get(\n      '/orchestration-rerun-plan/{jobId}',\n      {\n        path: { jobId },\n      },\n      { signal },\n    );\n\n    return data;\n  };\n\n  const healthCheck = async (signal?: AbortSignal) => {\n    const { data } = await genericClient.get('/health-check', {}, { signal });\n\n    return data;\n  };\n\n  // Every wait on a queue job is the same wait, so it is the client's rather than a caller's.\n  const poller = createJobPoller({ getJob });\n\n  return {\n    healthCheck,\n    poller,\n    getJob,\n    searchJobs,\n    searchAllJobs,\n    createJob,\n    killJob,\n    searchGroupedJobs,\n    getOrchestrationRerunPlan,\n  };\n};\n"],"mappings":";;;;;;;AAOA,MAAM,wBAAwB;;;;;;;AAmB9B,MAAa,mBACV,EAAE,cACF,EACC,MACA,UACA,aACA,SACA,aACyF;CACzF,MAAM,UAAU,iBAAiB;CAEjC,OAAO,OAAO;EACZ,UAAU,EAAE,IAAI,gBAAgB,CAAC,aAAa;GAAE;GAAI;EAAU,CAAC;EAC/D;EACA,QAAQ,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,iBAAiB;GACtD,MAAM,MAAM,MAAM,OAAO,IAAI,UAAU;GACvC,QAAQ,MAAM;GACd,OAAO;EACT;EACA,aAAa,QAAQ;EACrB,QAAQ,EAAE,WAAW,UAAU,MAAM,WAAW,KAAK,MAAM;EAC3D,SAAS,QAAS,IAAI,WAAW,UAAU,IAAI,cAAc,GAAG,IAAI,KAAA;EACpE,SAAS;GAAE,MAAM;GAAY,UAAU,YAAY;EAAsB;EACzE;EACA;EACA;CACF,CAAC;AACH;;ACpCF,MAAa,qBAAqB,EAAE,SAAS,aAAa,WAA8B;CACtF,MAAM,SAAS,yBAAgC;EAC7C;EACA;EACA,SAAS,kBAAkB,IAAI;CACjC,CAAC;CACD,MAAM,gBAAgB,yBAAyB;EAAE;EAAS;CAAY,CAAC;CAEvE,MAAM,SAAS,OAAO,OAAiC,WAAyB;EAC9E,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,iBACA,EACE,MAAM,EAAE,MAAM,EAChB,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,aAAa,OAAO,OAAyB,WAAyB;EAC1E,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,gBACA,EACE,MACF,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,iBAAiB,OAA2B,WAAyB;EACzE,OAAO,gBAAgB;GACrB,UAAU,MAAM;GAChB;GACA,YAAY,QAAQ,UAAU,gBAAgB;IAC5C,OAAO,WAAW;KAAE,GAAG;KAAO,OAAO;KAAU;IAAO,GAAG,WAAW;GACtE;EACF,CAAC;CACH;CAEA,MAAM,YAAY,OAAO,MAAqB,WAAyB;EACrE,MAAM,EAAE,SAAS,MAAM,OAAO,KAAK,SAAS,EAAE,KAAK,GAAG,EAAE,OAAO,CAAC;EAEhE,OAAO;CACT;CAEA,MAAM,UAAU,OAAO,OAA6B,WAAyB;EAC3E,MAAM,EAAE,SAAS,MAAM,OAAO,KAC5B,sBACA,EACE,MAAM,EAAE,MAAM,EAChB,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,oBAAoB,OAAO,OAA+B,WAAyB;EACvF,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,wBACA,EACE,MACF,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,4BAA4B,OAChC,OACA,WACG;EACH,MAAM,EAAE,SAAS,MAAM,OAAO,IAC5B,qCACA,EACE,MAAM,EAAE,MAAM,EAChB,GACA,EAAE,OAAO,CACX;EAEA,OAAO;CACT;CAEA,MAAM,cAAc,OAAO,WAAyB;EAClD,MAAM,EAAE,SAAS,MAAM,cAAc,IAAI,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC;EAExE,OAAO;CACT;CAKA,OAAO;EACL;EACA,QAJa,gBAAgB,EAAE,OAAO,CAIjC;EACL;EACA;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}