Function createAutoBoardChartValue

src/components/board/board-auto-value.ts:12–118

(
  prev: ChartValue,
  newResult: DatabaseResultSet,
  sql: string,
  theme: string
)

import { generateGradientColors } from "../chart/echart-options-builder";

export function createAutoBoardChartValue(
  prev: ChartValue,
  newResult: DatabaseResultSet,
  sql: string,
  theme: string
): ChartValue {
  return produce(prev, (draft) => {
    draft.params.layers[0].sql = sql;
    if (newResult.headers?.length > 0) {
      const columns = newResult.headers.map((header) => header.name);

      if (newResult.rows.length > 0) {
        const xAxisKeys: string[] = [];
        const yAxisKeys: string[] = [];
        const firstRecord = newResult.rows[0];
        const columnType = getColumnType(firstRecord);

        // remove the previous xAxisKey and yAxisKeys

        // study each column value and try to guess the type
        if (columnType) {
          for (const column of columns) {
            if (columnType[column] === "number") {
              yAxisKeys.push(column);
            } else {
              xAxisKeys.push(column);
            }
          }
          if (xAxisKeys.length === 0 && yAxisKeys.length > 1) {
            xAxisKeys.push(yAxisKeys.shift() as string);
          }
        } else {
          // if there is no column type, we will just use the first column as xAxisKey
          for (let i = 0; i < newResult.headers.length; i++) {
            if (i === 0) {
              xAxisKeys.push(newResult.headers[i].name);
            } else {
              yAxisKeys.push(newResult.headers[i].name);
            }
          }
        }

        if (draft.params.options.xAxisKey === "") {
          draft.params.options.xAxisKey =
            xAxisKeys.length > 0 ? xAxisKeys[0] : "";
        }

        if (draft.params.options.yAxisKeys.length === 0) {
          draft.params.options.yAxisKeys = yAxisKeys;
        }

        const suggestedChartTypes = suggestChartType(
          xAxisKeys,
          yAxisKeys,
          newResult.rows
        );

        draft.suggestedChartType = suggestedChartTypes;