fleet/frontend/components/Queries/QuerySidePanel/helpers.js

import { includes } from 'lodash';

const DEFAULT_NUM_COLUMNS_TO_DISPLAY = 5;
const ALL_PLATFORMS_AVAILABILITY = ['specs', 'utility'];

export const columnsToRender = (table, showAllColumns) => {
  if (showAllColumns) return table.columns;

  return table.columns.slice(0, DEFAULT_NUM_COLUMNS_TO_DISPLAY);
};

export const displayTypeForDataType = (dataType) => {
  switch (dataType) {
    case 'TEXT_TYPE':
      return 'text';
    case 'BIGINT_TYPE':
      return 'big int';
    case 'INTEGER_TYPE':
      return 'integer';
    default:
      return dataType;
  }
};

export const shouldShowAllColumns = (table) => {
  const { columns } = table;

  return columns.length <= DEFAULT_NUM_COLUMNS_TO_DISPLAY;
};

export const numAdditionalColumns = (table) => {
  const { columns } = table;

  return columns.length - DEFAULT_NUM_COLUMNS_TO_DISPLAY;
};

export const availability = (platform) => {
  if (!platform) return '';

  if (includes(ALL_PLATFORMS_AVAILABILITY, platform.toLowerCase())) {
    return 'All Platforms';
  }

  return platform;
};