add run options to ui client (#15)

This commit is contained in:
Jeff Raymakers
2025-06-25 12:09:40 -07:00
committed by GitHub
parent 2a8db10c6a
commit 8ec47ccec9
3 changed files with 61 additions and 19 deletions

View File

@@ -1,19 +1,25 @@
import { DuckDBUIRunOptions } from '../../client/types/DuckDBUIRunOptions.js';
import { toBase64 } from '../../util/functions/toBase64.js';
export interface DuckDBUIHttpRequestHeaderOptions {
description?: string;
export interface DuckDBUIHttpRequestHeaderOptions extends DuckDBUIRunOptions {
connectionName?: string;
databaseName?: string;
parameters?: unknown[];
}
export function makeDuckDBUIHttpRequestHeaders({
description,
connectionName,
databaseName,
schemaName,
errorsAsJson,
parameters,
resultChunkLimit,
resultDatabaseName,
resultSchemaName,
resultTableName,
resultTableChunkLimit,
}: DuckDBUIHttpRequestHeaderOptions): Headers {
const headers = new Headers();
// We base64 encode some values because they can contain characters invalid in an HTTP header.
if (description) {
headers.append('X-DuckDB-UI-Request-Description', description);
}
@@ -21,13 +27,14 @@ export function makeDuckDBUIHttpRequestHeaders({
headers.append('X-DuckDB-UI-Connection-Name', connectionName);
}
if (databaseName) {
// base64 encode the value because it can contain characters invalid in an HTTP header
headers.append('X-DuckDB-UI-Database-Name', toBase64(databaseName));
}
if (schemaName) {
headers.append('X-DuckDB-UI-Schema-Name', toBase64(schemaName));
}
if (parameters) {
headers.append('X-DuckDB-UI-Parameter-Count', String(parameters.length));
for (let i = 0; i < parameters.length; i++) {
// base64 encode the value because it can contain characters invalid in an HTTP header
// TODO: support non-string parameters?
headers.append(
`X-DuckDB-UI-Parameter-Value-${i}`,
@@ -35,5 +42,32 @@ export function makeDuckDBUIHttpRequestHeaders({
);
}
}
if (resultChunkLimit !== undefined) {
headers.append('X-DuckDB-UI-Result-Chunk-Limit', String(resultChunkLimit));
}
if (resultDatabaseName) {
headers.append(
'X-DuckDB-UI-Result-Database-Name',
toBase64(resultDatabaseName),
);
}
if (resultSchemaName) {
headers.append(
'X-DuckDB-UI-Result-Schema-Name',
toBase64(resultSchemaName),
);
}
if (resultTableName) {
headers.append('X-DuckDB-UI-Result-Table-Name', toBase64(resultTableName));
}
if (resultTableChunkLimit !== undefined) {
headers.append(
'X-DuckDB-UI-Result-Table-Chunk-Limit',
String(resultTableChunkLimit),
);
}
if (errorsAsJson) {
headers.append('X-DuckDB-UI-Errors-As-JSON', 'true');
}
return headers;
}