web-console/src/views/load-data-view/parse-time-table/parse-time-table.tsx

/*
 * 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 classNames from 'classnames';
import React from 'react';
import type { RowRenderProps } from 'react-table';
import ReactTable from 'react-table';

import { TableCell, TableCellUnparseable } from '../../../components';
import type { IngestionSpec, TimestampSpec } from '../../../druid-models';
import {
  getTimestampDetailFromSpec,
  getTimestampSpecColumnFromSpec,
  possibleDruidFormatForValues,
  TIME_COLUMN,
} from '../../../druid-models';
import {
  DEFAULT_TABLE_CLASS_NAME,
  STANDARD_TABLE_PAGE_SIZE,
  STANDARD_TABLE_PAGE_SIZE_OPTIONS,
} from '../../../react-table';
import { caseInsensitiveContains, filterMap } from '../../../utils';
import type { SampleEntry, SampleResponse } from '../../../utils/sampler';
import { getHeaderNamesFromSampleResponse } from '../../../utils/sampler';

import './parse-time-table.scss';

export function parseTimeTableSelectedColumnName(
  sampleResponse: SampleResponse,
  timestampSpec: TimestampSpec | undefined,
): string | undefined {
  if (!timestampSpec) return;
  const timestampColumn = timestampSpec.column;
  if (
    !timestampColumn ||
    !getHeaderNamesFromSampleResponse(sampleResponse).includes(timestampColumn)
  ) {
    return;
  }
  return timestampColumn;
}

export interface ParseTimeTableProps {
  sampleBundle: {
    sampleResponse: SampleResponse;
    spec: Partial<IngestionSpec>;
  };
  columnFilter: string;
  possibleTimestampColumnsOnly: boolean;
  selectedColumnName: string | undefined;
  onTimestampColumnSelect: (newTimestampSpec: TimestampSpec) => void;
}

export const ParseTimeTable = React.memo(function ParseTimeTable(props: ParseTimeTableProps) {
  const {
    sampleBundle,
    columnFilter,
    possibleTimestampColumnsOnly,
    selectedColumnName,
    onTimestampColumnSelect,
  } = props;
  const { sampleResponse, spec } = sampleBundle;
  const timestampSpecColumn = getTimestampSpecColumnFromSpec(spec);
  const timestampDetail = getTimestampDetailFromSpec(spec);

  return (
    <ReactTable
      className={classNames('parse-time-table', DEFAULT_TABLE_CLASS_NAME)}
      data={sampleResponse.data}
      sortable={false}
      defaultPageSize={STANDARD_TABLE_PAGE_SIZE}
      pageSizeOptions={STANDARD_TABLE_PAGE_SIZE_OPTIONS}
      showPagination={sampleResponse.data.length > STANDARD_TABLE_PAGE_SIZE}
      columns={filterMap(getHeaderNamesFromSampleResponse(sampleResponse), (columnName, i) => {
        const isTimestamp = columnName === TIME_COLUMN;
        if (!isTimestamp && !caseInsensitiveContains(columnName, columnFilter)) return;
        const used = timestampSpecColumn === columnName;
        const possibleFormat = isTimestamp
          ? null
          : possibleDruidFormatForValues(
              filterMap(sampleResponse.data, d => (d.parsed ? d.parsed[columnName] : undefined)),
            );
        if (possibleTimestampColumnsOnly && !isTimestamp && !possibleFormat) return;

        const columnClassName = classNames({
          timestamp: isTimestamp,
          used,
          selected: selectedColumnName === columnName,
        });
        return {
          Header: (
            <div
              className={classNames({ clickable: !isTimestamp })}
              onClick={
                isTimestamp
                  ? undefined
                  : () => {
                      onTimestampColumnSelect({
                        column: columnName,
                        format: possibleFormat || '!!! Could not auto detect a format !!!',
                      });
                    }
              }
            >
              <div className="column-name">{columnName}</div>
              <div className="column-detail">
                {isTimestamp ? timestampDetail : possibleFormat || ''}
                &nbsp;
              </div>
            </div>
          ),
          headerClassName: columnClassName,
          className: columnClassName,
          id: String(i),
          accessor: (row: SampleEntry) => (row.parsed ? row.parsed[columnName] : null),
          Cell: function ParseTimeTableCell(row: RowRenderProps) {
            if (columnName === '__error__') {
              return <TableCell value={row.original.error} />;
            }
            if (row.original.unparseable) {
              return <TableCellUnparseable timestamp={isTimestamp} />;
            }
            return <TableCell value={isTimestamp ? new Date(Number(row.value)) : row.value} />;
          },
          width: isTimestamp ? 200 : 140,
          resizable: !isTimestamp,
        };
      })}
    />
  );
});