#!/usr/bin/env python
# _*_coding:utf-8_*_

"""
@Time     : 2023/6/6 16:50
@Author   : ji hao ran
@File     : tree.py
@Project  : StreamlitAntdComponents
@Software : PyCharm
"""

from ..utils import *


def tree(
        items: List[Union[str, dict, TreeItem]] = None,
        index: Union[int, List[int]] = None,
        format_func: Union[Formatter, Callable] = None,
        label: str = None,
        description: str = None,
        icon: Union[str, BsIcon, AntIcon] = None,
        align: Align = 'start',
        size: Union[MantineSize, int] = 'sm',
        color: Union[MantineColor, str] = None,
        width: int = None,
        height: int = None,
        open_index: List[int] = None,
        open_all: bool = False,
        checkbox: bool = False,
        checkbox_strict: bool = False,
        show_line: bool = True,
        return_index: bool = False,
        on_change: Callable = None,
        args: Tuple[Any, ...] = None,
        kwargs: Dict[str, Any] = None,
        key=None
) -> List[Union[str, int]]:
    """antd design tree  https://ant.design/components/tree

    :param items: tree data
    :param index: default selected tree item index
    :param format_func: label formatter function,receive str and return str
    :param label: tree label,support str and markdown str
    :param description: tree description,support str and markdown str
    :param icon: tree item icon
    :param align: tree align
    :param size: tree size,support mantine size and int in px
    :param color: tree color,default streamlit primary color,support mantine color, hex and rgb color
    :param width: tree width
    :param height: tree height
    :param open_index: default opened indexes.if none,tree will open default index's parent nodes.
    :param open_all: open all items.priority[open_all>open_index]
    :param checkbox: show checkbox to allow multiple select
    :param checkbox_strict: parent item and children item are not associated
    :param show_line: show line
    :param return_index: if True,return tree item index,default return label
    :param on_change: item change callback
    :param args: callback args
    :param kwargs: callback kwargs
    :param key: component unique identifier
    :return: list of selected item label or index
    """
    if isinstance(index, list) and len(index) > 1 and not checkbox:
        raise ValueError(f'length of index ({len(index)}) should =1  when checkbox=False')
    # register callback
    register(key, on_change, args, kwargs)
    # parse items
    items, kv = ParseItems(items, format_func).multi()
    # parse index
    if index is None and checkbox:
        index = []
    if isinstance(index, int) and checkbox:
        index = [index]
    if isinstance(index, list) and not checkbox:
        index = index[0]
    # component params
    kw = update_kw(locals(), items=items, icon=parse_icon(icon))
    # component default
    default = get_default(index, return_index, kv)
    # pass component id and params to frontend
    return component(id=get_func_name(), kw=kw, default=default, key=key)