Function executeSnowflakeStatement

apps/web/src/lib/snowflake.ts:270–330

(params: {
  config: SnowflakeConfig;
  statement: string;
  bindings?: SnowflakeBinding[];
  timeoutSeconds?: number;
  signal?: AbortSignal;
})

 * column-declaration order.
 */
export async function executeSnowflakeStatement(params: {
  config: SnowflakeConfig;
  statement: string;
  bindings?: SnowflakeBinding[];
  timeoutSeconds?: number;
  signal?: AbortSignal;
}): Promise<SnowflakeRow[]> {
  const token = getOrBuildJwt(params.config);
  const requestId = crypto.randomUUID();

  const url = new URL(`https://${params.config.accountHost}/api/v2/statements`);
  url.searchParams.set('requestId', requestId);

  const body: Record<string, unknown> = {
    statement: params.statement,
    warehouse: params.config.warehouse,
    database: params.config.database,
    schema: params.config.schema,
    role: params.config.role,
  };

  if (params.bindings && params.bindings.length > 0) {
    body.bindings = Object.fromEntries(params.bindings.map((b, i) => [String(i + 1), b]));
  }

  if (params.timeoutSeconds !== undefined) {
    body.timeout = params.timeoutSeconds;
  }

  const response = await fetch(url, {
    method: 'POST',
    headers: { ...authHeaders(token), 'content-type': 'application/json' },
    body: JSON.stringify(body),
    signal: params.signal,
  });

  if (response.status === 200) {
    const payload = (await response.json()) as SnowflakeApiResponse;
    return parseAllRows(params.config, token, payload, params.signal);
  }

  if (response.status === 202) {
    const payload = (await response.json()) as SnowflakeApiResponse;
    if (!payload.statementStatusUrl) {
      throw new Error('Snowflake response missing statementStatusUrl');
    }
    const completed = await pollStatement(
      params.config,
      token,
      payload.statementStatusUrl,
      params.signal
    );
    if (completed.code === '090001' || Array.isArray(completed.data)) {
      return parseAllRows(params.config, token, completed, params.signal);
    }
    throw new Error(completed.message ?? 'Snowflake async query failed');
  }