superset-frontend/plugins/plugin-chart-cartodiagram/src/plugin/buildQuery.ts

/**
 * 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 { QueryFormData, getChartBuildQueryRegistry } from '@superset-ui/core';

export default function buildQuery(formData: QueryFormData) {
  const {
    selected_chart: selectedChartString,
    geom_column: geometryColumn,
    extra_form_data: extraFormData,
  } = formData;
  const selectedChart = JSON.parse(selectedChartString);
  const vizType = selectedChart.viz_type;
  const chartFormData = JSON.parse(selectedChart.params);
  // Pass extra_form_data to chartFormData so that
  // dashboard filters will also be applied to the charts
  // on the map.
  chartFormData.extra_form_data = {
    ...chartFormData.extra_form_data,
    ...extraFormData,
  };

  // adapt groupby property to ensure geometry column always exists
  // and is always at first position
  let { groupby } = chartFormData;
  if (!groupby) {
    groupby = [];
  }
  // add geometry column at the first place
  groupby?.unshift(geometryColumn);
  chartFormData.groupby = groupby;

  // TODO: find way to import correct type "InclusiveLoaderResult"
  const buildQueryRegistry = getChartBuildQueryRegistry();
  const chartQueryBuilder = buildQueryRegistry.get(vizType) as any;

  const chartQuery = chartQueryBuilder(chartFormData);
  return chartQuery;
}