import {ActionTooltip, ArrowToggle, Button, Flex} from '@gravity-ui/uikit';
import type {CellContext, ColumnDef, Row} from '@tanstack/react-table';

import {ColumnHeader} from '../../../../components/Table/Table';
import {formatTimestamp} from '../../../../utils/dataFormatters/dataFormatters';

import {tabletInfoKeyset} from './i18n';
import {b} from './shared';
import type {TabletStorageItem} from './types';

function metricsCell(
    info: CellContext<TabletStorageItem, unknown>,
    formatter?: (value: string | number) => string | number,
) {
    const value = info.getValue<string | number>();
    const formattedValue = typeof formatter === 'function' ? formatter(value) : value;

    return <div className={b('metrics-cell')}>{formattedValue}</div>;
}

interface GroupIdCellProps<TData> {
    row: Row<TData>;
    name?: string;
    hasExpand?: boolean;
}

function GroupIdCell<TData>({row, name, hasExpand}: GroupIdCellProps<TData>) {
    const isExpandable = row.getCanExpand();
    const isExpanded = row.getIsExpanded();
    const toggleTitle = isExpanded
        ? tabletInfoKeyset('action_collapse')
        : tabletInfoKeyset('action_expand');

    return (
        <Flex gap={1} alignItems="flex-start" className={b('name-wrapper')}>
            {isExpandable && (
                <ActionTooltip title={toggleTitle}>
                    <Button
                        view="flat"
                        size="xs"
                        onClick={row.getToggleExpandedHandler()}
                        aria-label={toggleTitle}
                    >
                        <Button.Icon>
                            <ArrowToggle direction={isExpanded ? 'bottom' : 'right'} size={14} />
                        </Button.Icon>
                    </Button>
                </ActionTooltip>
            )}
            <div className={b('name-content', {'no-control': hasExpand && !isExpandable})}>
                {name}
            </div>
        </Flex>
    );
}

export function getColumns(hasExpand?: boolean) {
    const columns: ColumnDef<TabletStorageItem>[] = [
        {
            accessorKey: 'channelIndex',
            header: () => <ColumnHeader>{tabletInfoKeyset('label_channel-index')}</ColumnHeader>,
            size: 50,
            cell: metricsCell,
            meta: {align: 'right'},
        },
        {
            accessorKey: 'storagePoolName',
            header: () => <ColumnHeader>{tabletInfoKeyset('label_storage-pool')}</ColumnHeader>,
            size: 200,
            cell: metricsCell,
        },
        {
            accessorKey: 'GroupID',
            header: () => (
                <ColumnHeader className={hasExpand ? b('with-padding') : undefined}>
                    {tabletInfoKeyset('label_group-id')}
                </ColumnHeader>
            ),
            size: 100,
            cell: (info) => (
                <GroupIdCell row={info.row} name={info.getValue<string>()} hasExpand={hasExpand} />
            ),
        },
        {
            accessorKey: 'FromGeneration',
            header: () => <ColumnHeader>{tabletInfoKeyset('label_generation')}</ColumnHeader>,
            size: 100,
            cell: metricsCell,
            meta: {align: 'right'},
        },
        {
            accessorKey: 'Timestamp',
            header: () => <ColumnHeader>{tabletInfoKeyset('label_timestamp')}</ColumnHeader>,
            size: 200,
            cell: (info) => metricsCell(info, formatTimestamp),
            meta: {align: 'right'},
        },
    ];
    return columns;
}