import DataTable from '@gravity-ui/react-data-table';

import {
    getCapacityAlertColumn,
    getPDiskUsageColumn,
    getVDiskRawUsageColumn,
    getVDiskSlotUsageColumn,
} from '../../../../components/capacityMetricsColumns/columns';
import {
    getCpuColumn,
    getDataCenterColumn,
    getDiskSpaceUsageColumn,
    getHostColumn,
    getMemoryColumn,
    getMissingDisksColumn,
    getNodeIdColumn,
    getNodeNameColumn,
    getPileNameColumn,
    getPoolsColumn,
    getRAMColumn,
    getRackColumn,
    getTabletsColumn,
    getUptimeColumn,
    getVersionColumn,
} from '../../../../components/nodesColumns/columns';
import {
    NODES_COLUMNS_IDS,
    NODES_COLUMNS_TITLES,
    isSortableNodesColumn,
} from '../../../../components/nodesColumns/constants';
import type {NodesColumn} from '../../../../components/nodesColumns/types';
import {cn} from '../../../../utils/cn';
import {PDisks} from '../../PDisks/PDisks';

import type {GetStorageNodesColumnsParams} from './types';

import './StorageNodesColumns.scss';

const b = cn('ydb-storage-nodes-columns');

export const getPDisksColumn = ({
    viewContext,
    columnsSettings,
}: GetStorageNodesColumnsParams): NodesColumn => {
    return {
        name: NODES_COLUMNS_IDS.PDisks,
        header: NODES_COLUMNS_TITLES.PDisks,
        className: b('pdisks-column'),
        width: columnsSettings?.pDiskContainerWidth,
        render: ({row}) => {
            return (
                <PDisks
                    pDisks={row.PDisks}
                    vDisks={row.VDisks}
                    viewContext={viewContext}
                    pDiskWidth={columnsSettings?.pDiskWidth}
                    pDiskHeight={columnsSettings?.pDiskHeight}
                    expertMode={columnsSettings?.expertMode}
                />
            );
        },
        align: DataTable.LEFT,
        sortable: false,
        resizeable: false,
    };
};

export const getStorageNodesColumns = ({
    database,
    viewContext,
    columnsSettings,
}: GetStorageNodesColumnsParams): NodesColumn[] => {
    const columns: NodesColumn[] = [
        getNodeIdColumn(),
        getHostColumn({database}),
        getNodeNameColumn(),
        getDataCenterColumn(),
        getPileNameColumn(),
        getRackColumn(),
        getUptimeColumn(),
        getCpuColumn(),
        getPoolsColumn(),
        getRAMColumn(),
        getMemoryColumn(),
        getDiskSpaceUsageColumn(),
        getVersionColumn(),
        getMissingDisksColumn(),
        getPDisksColumn({viewContext, columnsSettings}),
        getTabletsColumn({database}),
        getPDiskUsageColumn(),
        getVDiskSlotUsageColumn(),
        getVDiskRawUsageColumn(),
        getCapacityAlertColumn(),
    ];

    const sortableColumns = columns.map((column) => ({
        ...column,
        sortable: isSortableNodesColumn(column.name),
    }));

    return sortableColumns;
};