import React from 'react';

import {Progress} from '@gravity-ui/uikit';

import {defaultFormatProgressValues} from '../../utils/progress';
import {safeParseNumber} from '../../utils/utils';
import {getMemorySegmentColor} from '../MemoryViewer/utils';

import {ProgressContainer} from './ProgressContainer';
import i18n from './i18n';
import {
    MAX_PERCENTAGE,
    PROGRESS_SIZE,
    formatDisplayValues,
    formatProgressText,
} from './progressUtils';
import type {ProgressWrapperStackProps} from './types';

export function StackProgress({
    stack,
    totalCapacity,
    formatValues = defaultFormatProgressValues,
    className,
    width,
    size = PROGRESS_SIZE,
    withCapacityUsage = false,
}: ProgressWrapperStackProps) {
    const displaySegments = React.useMemo(() => {
        return stack.filter((segment) => !segment.isInfo && segment.value > 0);
    }, [stack]);

    const totalValue = React.useMemo(() => {
        return displaySegments.reduce((sum, segment) => sum + segment.value, 0);
    }, [displaySegments]);

    const numericTotalCapacity = React.useMemo(() => {
        return safeParseNumber(totalCapacity);
    }, [totalCapacity]);

    const maxValue = numericTotalCapacity || totalValue;

    const stackElements = React.useMemo(() => {
        return displaySegments.map((segment) => ({
            value: maxValue > 0 ? (segment.value / maxValue) * MAX_PERCENTAGE : 0,
            color: getMemorySegmentColor(segment.key),
            title: segment.label,
        }));
    }, [displaySegments, maxValue]);

    const [totalValueText, totalCapacityText] = React.useMemo(() => {
        return formatDisplayValues(totalValue, numericTotalCapacity || totalValue, formatValues);
    }, [formatValues, totalValue, numericTotalCapacity]);

    const displayText = React.useMemo(() => {
        return formatProgressText(totalValueText, totalCapacityText, numericTotalCapacity || 0);
    }, [totalValueText, totalCapacityText, numericTotalCapacity]);

    if (displaySegments.length === 0) {
        return <div className={className}>{i18n('alert_no-data')}</div>;
    }

    return (
        <ProgressContainer
            displayText={displayText}
            withCapacityUsage={withCapacityUsage}
            className={className}
            width={width}
        >
            <Progress value={MAX_PERCENTAGE} stack={stackElements} size={size} />
        </ProgressContainer>
    );
}