asb_cloud_front/src/components/WellTreeSelector.tsx

import { TreeSelect } from 'antd'
import { DefaultValueType } from 'rc-tree-select/lib/interface'
import { useState, useEffect, ReactNode } from 'react'
import { useHistory, useRouteMatch } from 'react-router-dom'

import { DepositService } from '../services/api'
import { ReactComponent as DepositIcon } from '../images/DepositIcon.svg'
import { ReactComponent as ClusterIcon } from '../images/ClusterIcon.svg'
import { DepositDto } from '../services/api'
import { RawDate } from '../utils'

import LoaderPortal from './LoaderPortal'
import { invokeWebApiWrapperAsync } from './factory'
import { WellIcon, WellIconState } from './icons'

import '../styles/wellTreeSelect.css'

export const getWellState = (idState?: number): WellIconState => idState === 1 ? 'active' : 'unknown'
export const checkIsWellOnline = (lastTelemetryDate?: RawDate): boolean => {
    if (!lastTelemetryDate || isNaN(+new Date(lastTelemetryDate))) return false
    return Date.now() - +new Date(lastTelemetryDate) < 600_000
}

export type TreeNodeData = {
    title?: string | null
    key?: string
    value?: DefaultValueType
    icon?: ReactNode
    children?: TreeNodeData[]
}

const getLabel = (wellsTree: any[], value?: string): string | undefined => {
    if (!value) return value
    const type = value.replaceAll('/', ' ').trim().split(' ')[0]
    let deposit: any, cluster: any, well: any
    switch (type) {
        case 'cluster':
            deposit = wellsTree.find((deposit) => (
                cluster = deposit.children?.find((cluster: any) => cluster.key === value)
            ))
            if (deposit && cluster)
                return `${deposit.title} / ${cluster.title}`
            break
        case 'well':
            deposit = wellsTree.find((deposit) => (
                cluster = deposit.children?.find((cluster: any) => (
                    well = cluster.children?.find((well: any) => well.key === value)
                ))
            ))
            if (deposit && cluster && well)
                return `${deposit.title} / ${cluster.title} / ${well.title}`
            break
        default: break
    }
    return value
}

export const WellTreeSelector = () => {
    const [wellsTree, setWellsTree] = useState<any[]>([]) // TODO: Исправить тип (необходимо разобраться с типом значения  rc-select)
    const [showLoader, setShowLoader] = useState<boolean>(false)
    const [value, setValue] = useState<string>()
    const history = useHistory()
    const routeMatch = useRouteMatch('/:route/:id')

    useEffect(() => {
        invokeWebApiWrapperAsync(
            async () => {
                const deposits: Array<DepositDto> = await DepositService.getDeposits()
                const wellsTree: TreeNodeData[] = deposits.map(deposit =>({
                    title: deposit.caption,
                    key: `/deposit/${deposit.id}`,
                    icon: <DepositIcon width={24} height={24}/>,
                    children: deposit.clusters?.map(cluster => ({
                        title: cluster.caption,
                        key: `/cluster/${cluster.id}`,
                        icon: <ClusterIcon width={24} height={24}/>,
                        children: cluster.wells?.map(well => ({
                            title: well.caption,
                            key: `/well/${well.id}`,
                            icon: <WellIcon
                                width={24}
                                height={24}
                                state={getWellState(well.idState)}
                                online={checkIsWellOnline(well.lastTelemetryDate)}
                            />
                        })),
                    })),
                }))
                setWellsTree(wellsTree)
                setValue(getLabel(wellsTree, routeMatch?.url))
            },
            setShowLoader,
            `Не удалось загрузить список скважин`
        )
    }, [])

    const onChange = (value: string): void => {
        if (wellsTree)
            setValue(getLabel(wellsTree, value))
    }

    const onSelect = (value: string): void => {
        if (value) history.push(value)
    }

    return (
        <LoaderPortal show={showLoader}>
            <TreeSelect
                treeIcon
                className={'header-tree-select'}
                bordered={false}
                dropdownMatchSelectWidth={false}
                placeholder={'Выберите месторождение'}
                treeData={wellsTree}
                treeDefaultExpandAll
                onSelect={onSelect}
                onChange={onChange}
                value={value}
            />
        </LoaderPortal>
    )
}

export default WellTreeSelector