import {
    ArrowsRotateLeft,
    ArrowsRotateLeftSlash,
    Check,
    CirclePause,
    CircleQuestionFill,
    CircleStop,
} from '@gravity-ui/icons';

import {EVDiskDetailedReplicationStatus, EVDiskState} from '../../types/api/vdisk';
import {
    NOT_AVAILABLE_SEVERITY,
    NUMERIC_SEVERITY_TO_LABEL_VIEW,
    VDISK_STATE_SEVERITY_FOR_STATE_MODE,
} from '../../utils/disks/constants';
import {calculateStateIcon} from '../../utils/disks/iconCalculators';
import type {PreparedVDisk} from '../../utils/disks/types';
import {parseOptionalNonNegativeNumber} from '../../utils/utils';
import type {DiskStatusLabelData} from '../DiskStatus/DiskStatus';

import {vDiskStatusKeyset as i18n} from './i18n';

export function getVDiskStateLabel(data: PreparedVDisk): DiskStatusLabelData {
    const state = data.VDiskState;
    const severity = state
        ? (VDISK_STATE_SEVERITY_FOR_STATE_MODE[state] ?? NOT_AVAILABLE_SEVERITY)
        : NOT_AVAILABLE_SEVERITY;
    if (!state || severity === NOT_AVAILABLE_SEVERITY) {
        return {
            value: i18n('value_no-data'),
            theme: 'unknown',
            icon: CircleQuestionFill,
            tooltip: i18n('context_state-no-data'),
        };
    }

    const labels: Record<EVDiskState, Pick<DiskStatusLabelData, 'value' | 'tooltip'>> = {
        [EVDiskState.OK]: {
            value: i18n('value_ok'),
            tooltip: i18n('context_state-ok'),
        },
        [EVDiskState.Initial]: {
            value: i18n('value_initial'),
            tooltip: i18n('context_state-initial'),
        },
        [EVDiskState.SyncGuidRecovery]: {
            value: i18n('value_sync-guid-recovery'),
            tooltip: i18n('context_state-sync-guid-recovery'),
        },
        [EVDiskState.PDiskError]: {
            value: i18n('value_pdisk-error'),
            tooltip: i18n('context_state-pdisk-error'),
        },
        [EVDiskState.LocalRecoveryError]: {
            value: i18n('value_local-recovery-error'),
            tooltip: i18n('context_state-local-recovery-error'),
        },
        [EVDiskState.SyncGuidRecoveryError]: {
            value: i18n('value_sync-guid-recovery-error'),
            tooltip: i18n('context_state-sync-guid-recovery-error'),
        },
    };

    const labelView = NUMERIC_SEVERITY_TO_LABEL_VIEW[severity];

    return {
        ...labels[state],
        theme: labelView.theme,
        icon: state === EVDiskState.OK ? Check : calculateStateIcon(data),
        dangerHeavy: Boolean(labelView.dangerHeavy),
    };
}

export function getVDiskReplicationLabel(data: PreparedVDisk): DiskStatusLabelData | undefined {
    switch (data.DetailedReplicationStatus) {
        case EVDiskDetailedReplicationStatus.Replicated:
            return replicatedLabel();
        case EVDiskDetailedReplicationStatus.InProgress:
            return {
                title: i18n('label_replication'),
                value: i18n('value_in-progress'),
                theme: 'info',
                icon: ArrowsRotateLeft,
                tooltip: i18n('context_replication-in-progress'),
            };
        case EVDiskDetailedReplicationStatus.WaitingForToken:
            return {
                title: i18n('label_replication'),
                value: i18n('value_waiting-for-token'),
                theme: 'info',
                icon: CirclePause,
                tooltip: i18n('context_replication-waiting-for-token'),
            };
        case EVDiskDetailedReplicationStatus.PhantomsOnly:
            return {
                title: i18n('label_replication'),
                value: i18n('value_phantoms-only'),
                theme: 'info',
                icon: CircleStop,
                tooltip: i18n('context_replication-phantoms-only'),
            };
        default:
            if (data.DetailedReplicationStatus) {
                return undefined;
            }
            if (data.Replicated === true) {
                return replicatedLabel();
            }
            if (data.Replicated === false) {
                const progress = parseOptionalNonNegativeNumber(data.ReplicationProgress);
                const seconds = parseOptionalNonNegativeNumber(data.ReplicationSecondsRemaining);
                if ((progress !== undefined && progress <= 1) || seconds !== undefined) {
                    return {
                        title: i18n('label_replication'),
                        value: i18n('value_no-detailed-status'),
                        theme: 'info',
                        icon: ArrowsRotateLeft,
                        tooltip: i18n('context_not-replicated'),
                    };
                }
                return {
                    value: i18n('label_not-replicated'),
                    theme: 'normal',
                    icon: ArrowsRotateLeftSlash,
                    tooltip: i18n('context_not-replicated'),
                };
            }
            return undefined;
    }
}

function replicatedLabel(): DiskStatusLabelData {
    return {
        value: i18n('label_replicated'),
        theme: 'success',
        icon: Check,
        tooltip: i18n('context_replicated'),
    };
}