mirror of
https://github.com/apache/superset.git
synced 2026-07-26 08:32:42 +00:00
Simplifies repeated postFormData + mountExploreUrl + dashboard_page_id pattern used across 6+ files by providing a clean, type-safe async utility. Benefits: - DrillDetailModal: 15 lines → 5 lines (67% reduction) - Centralized URL generation logic for better maintainability - Type-safe options parameter for chartId, tabId, dashboardPageId - Comprehensive test coverage with 4 test scenarios - Better error handling and promise chain management Files refactored: - DrillDetailModal: Simplified async URL generation - DrillByModal: Clean promise chain with proper error handling - SqlLab ResultSet: Modern async/await pattern - Added comprehensive utility tests All component tests passing ✅ 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
136 lines
3.9 KiB
TypeScript
136 lines
3.9 KiB
TypeScript
/**
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
import { SupersetClient, JsonObject, JsonResponse } from '@superset-ui/core';
|
|
import { sanitizeFormData } from 'src/utils/sanitizeFormData';
|
|
import { URL_PARAMS } from 'src/constants';
|
|
import { mountExploreUrl } from './index';
|
|
|
|
type Payload = {
|
|
datasource_id: number;
|
|
datasource_type: string;
|
|
form_data: string;
|
|
chart_id?: number;
|
|
};
|
|
|
|
const assembleEndpoint = (key?: string, tabId?: string) => {
|
|
let endpoint = 'api/v1/explore/form_data';
|
|
|
|
if (key) endpoint = endpoint.concat(`/${key}`);
|
|
if (tabId) endpoint = endpoint.concat(`?tab_id=${tabId}`);
|
|
|
|
return endpoint;
|
|
};
|
|
|
|
const assemblePayload = (
|
|
datasourceId: number,
|
|
datasourceType: string,
|
|
formData: JsonObject,
|
|
chartId?: number,
|
|
): Payload => {
|
|
const payload: Payload = {
|
|
datasource_id: datasourceId,
|
|
datasource_type: datasourceType,
|
|
form_data: JSON.stringify(sanitizeFormData(formData)),
|
|
};
|
|
|
|
if (chartId) payload.chart_id = chartId;
|
|
|
|
return payload;
|
|
};
|
|
|
|
export const postFormData = (
|
|
datasourceId: number,
|
|
datasourceType: string,
|
|
formData: JsonObject,
|
|
chartId?: number,
|
|
tabId?: string,
|
|
): Promise<string> =>
|
|
SupersetClient.post({
|
|
endpoint: assembleEndpoint(undefined, tabId),
|
|
jsonPayload: assemblePayload(
|
|
datasourceId,
|
|
datasourceType,
|
|
formData,
|
|
chartId,
|
|
),
|
|
}).then((r: JsonResponse) => r.json.key);
|
|
|
|
export const putFormData = (
|
|
datasourceId: number,
|
|
datasourceType: string,
|
|
key: string,
|
|
formData: JsonObject,
|
|
chartId?: number,
|
|
tabId?: string,
|
|
): Promise<string> =>
|
|
SupersetClient.put({
|
|
endpoint: assembleEndpoint(key, tabId),
|
|
jsonPayload: assemblePayload(
|
|
datasourceId,
|
|
datasourceType,
|
|
formData,
|
|
chartId,
|
|
),
|
|
}).then((r: JsonResponse) => r.json.message);
|
|
|
|
/**
|
|
* Generate explore URL by posting formData to server and creating URL with key.
|
|
* This solves URL length limitations by storing complex formData server-side.
|
|
*
|
|
* @param datasourceId - The datasource ID
|
|
* @param datasourceType - The datasource type (typically 'table' or 'query')
|
|
* @param formData - The form data object to be posted
|
|
* @param options - Optional parameters
|
|
* @param options.chartId - Chart ID for the explore URL
|
|
* @param options.tabId - Tab ID for multi-tab environments
|
|
* @param options.dashboardPageId - Dashboard page ID to maintain context
|
|
* @returns Promise that resolves to the complete explore URL
|
|
*/
|
|
export const generateExploreUrl = async (
|
|
datasourceId: number,
|
|
datasourceType: string,
|
|
formData: JsonObject,
|
|
options?: {
|
|
chartId?: number;
|
|
tabId?: string;
|
|
dashboardPageId?: string;
|
|
},
|
|
): Promise<string> => {
|
|
// Post formData to server and get key
|
|
const key = await postFormData(
|
|
datasourceId,
|
|
datasourceType,
|
|
formData,
|
|
options?.chartId,
|
|
options?.tabId,
|
|
);
|
|
|
|
// Generate base explore URL with form_data_key
|
|
const baseUrl = mountExploreUrl(null, {
|
|
[URL_PARAMS.formDataKey.name]: key,
|
|
});
|
|
|
|
// Add dashboard_page_id if provided
|
|
const finalUrl = options?.dashboardPageId
|
|
? `${baseUrl}${baseUrl.includes('?') ? '&' : '?'}dashboard_page_id=${options.dashboardPageId}`
|
|
: baseUrl;
|
|
|
|
return finalUrl;
|
|
};
|