semi-design/packages/semi-ui/tree/_story/tree.stories.js

import React, {useRef, useState} from 'react';
import {storiesOf} from '@storybook/react';
import Tree from '../index';
import AutoSizer from '../autoSizer';
import {Button, ButtonGroup, Input, Popover, Toast} from '../../index';
import BigTree from './BigData';
import testData from './data';
import {cloneDeep, difference, isEqual,} from 'lodash-es';
import {IconEdit, IconMapPin, IconMore} from '@douyinfe/semi-icons';

const TreeNode = Tree.TreeNode;

const stories = storiesOf('Tree', module);

const treeChildren = [
    {
        label: '北京',
        value: 'beijing',
        key: 'beijing',
    },
    {
        label: '上海',
        value: 'shanghai',
        key: 'shanghai',
    },
];

const treeData1 = [
    {
        label: '亚洲',
        value: 'yazhou',
        key: 'yazhou',
        children: [
            {
                label: '中国',
                value: 'zhongguo',
                key: 'zhongguo',
                disabled: true,
                children: treeChildren,
            },
            {
                label: '日本',
                value: 'riben',
                key: 'riben',
                children: [
                    {
                        label: '东京',
                        value: 'dongjing',
                        key: 'dongjing',
                    },
                    {
                        label: '大阪',
                        value: 'daban',
                        key: 'daban',
                    },
                ],
            },
        ],
    },
    {
        label: '北美洲',
        value: 'beimeizhou',
        key: 'beimeizhou',
        children: [
            {
                label: '美国',
                value: 'meiguo',
                key: 'meiguo',
            },
            {
                label: '加拿大',
                value: 'jianada',
                key: 'jianada',
            },
        ],
    },
];

const treeDataWithoutValue = [
    {
        label: '亚洲',
        key: 'yazhou',
        children: [
            {
                label: '中国',
                key: 'zhongguo',
                disabled: true,
                children: treeChildren,
            },
            {
                label: '日本',
                key: 'riben',
                children: [
                    {
                        label: '东京',
                        key: 'dongjing',
                    },
                    {
                        label: '大阪',
                        key: 'daban',
                    },
                ],
            },
        ],
    },
    {
        label: '北美洲',
        key: 'beimeizhou',
        children: [
            {
                label: '美国',
                key: 'meiguo',
            },
            {
                label: '加拿大',
                key: 'jianada',
            },
        ],
    },
];

const treeDataWithIcon = [
    {
        label: 'Asia',
        value: 'Asia',
        key: '0',
        icon: <IconMapPin style={{color: 'var(--semi-color-text-2)'}}/>,
        children: [
            {
                label: 'China',
                value: 'China',
                key: '0-0',
                icon: <IconMapPin style={{color: 'var(--semi-color-text-2)'}}/>,
            },
            {
                label: 'Japan',
                value: 'Japan',
                key: '0-1',
                icon: <IconMapPin style={{color: 'var(--semi-color-text-2)'}}/>,
            },
        ],
    },
];

let opts = {
    content: 'Hi, Bytedance dance dance',
    duration: 3,
};

const treeDataWithNode = [
    {
        label: (
            <span>
                <span style={{marginRight: 30}}>亚洲</span>
                <Button
                    style={{zIndex: 2}}
                    onClick={e => {
                        Toast.info(opts);
                        e.stopPropagation();
                    }}
                >
                    Display Toast
                </Button>
            </span>
        ),
        value: 'yazhou',
        key: 'yazhou',
        children: [
            {
                label: '中国',
                value: 'zhongguo',
                key: 'zhongguo',
                children: [
                    {
                        label: '北京',
                        value: 'beijing',
                        key: 'beijing',
                    },
                    {
                        label: '上海',
                        value: 'shanghai',
                        key: 'shanghai',
                    },
                ],
            },
            {
                label: '日本',
                value: 'riben',
                key: 'riben',
            },
        ],
    },
    {
        label: '北美洲',
        value: 'beimeizhou',
        key: 'beimeizhou',
        children: [
            {
                label: '美国',
                value: 'meiguo',
                key: 'meiguo',
            },
            {
                label: '加拿大',
                value: 'jianada',
                key: 'jianada',
            },
        ],
    },
];

const treeJson = {
    Node1: {
        Child1: '0-0-1',
        Child2: '0-0-2',
    },
    Node2: '0-1',
};

const benchmarkSet = size => {
    console.time('benchmarkSet');
    var set = new Set();
    for (var i = 0; i < size; i++) set.add(i);
    for (var i = 0; i < size; i++) set.has(i);
    console.timeEnd('benchmarkSet');
};

const benchmarkArr = size => {
    console.time('benchmarkArr');
    var arr = [];
    for (var i = 0; i < size; i++) arr.push(i);
    for (var i = 0; i < size; i++) arr.indexOf(i);
    console.timeEnd('benchmarkArr');
};

stories.add('bench mark', () => {
    const size = 100000;
    benchmarkSet(size);
    benchmarkArr(size);
    return <div>check console please</div>;
});

stories.add('simple tree', () => (
    <Tree
        treeData={treeData1}
        // onExpand={(e, { expanded, node }) => console.log('expand', e, expanded, node)}
        // onSelect={(e, bool, node) => console.log('select', e, bool, node)}
        // onChange={e => console.log('change', e)}
        onRightClick={(e, node) => console.log(e.currentTarget, node)}
        onDoubleClick={(e, node) => console.log(e, node)}
        motion={true}
    />
));

stories.add('multiple tree', () => (
    <Tree
        treeData={treeData1}
        multiple
        labelInValue
        // onExpand={(e, { expanded, node }) => console.log('expand', e, expanded, node)}
        // onSelect={(e, bool) => console.log('select', e, bool)}
        onRightClick={(e, node) => console.log(e, node)}
        onDoubleClick={(e, node) => console.log(e, node)}
        onChange={e => console.log('change', e)}
    />
));

stories.add('searchable tree', () => (
    <Tree treeData={treeData1} filterTreeNode treeNodeFilterProp="value" multiple searchStyle={{width: '300px'}}/>
));

stories.add('disabled tree', () => (
    <Tree treeData={treeData1} defaultValue={['dongjing', 'daban']} multiple disabled/>
));

stories.add('default tree', () => (
    <div>
        <Tree treeData={treeData1} defaultValue={['zhongguo']}/>
        <br/>
        <Tree
            treeData={treeData1}
            multiple
            defaultValue={['shanghai', 'dongjing', 'beijing', 'daban']}
            onChange={e => console.log(e)}
            onSelect={e => console.log(e)}
        />
    </div>
));

stories.add('expandAction', () => (
    <div>
        <Tree treeData={treeData1} defaultValue={['zhongguo']}/>
        <br/>
        <Tree
            treeData={treeData1}
            //multiple
            defaultValue={['shanghai', 'dongjing', 'beijing', 'daban']}
            expandAction="click"
            onDoubleClick={e => console.log(e.detail)}
        />
        <br/>
        <Tree
            treeData={treeData1}
            multiple
            defaultValue={['shanghai', 'dongjing', 'beijing', 'daban']}
            expandAction="doubleClick"
        />
    </div>
));

class Demo extends React.Component {
    constructor() {
        super();
        this.state = {
            value: 'shanghai',
        };
    }

    onChange(value) {
        this.setState({value});
    }

    render() {
        return (
            <Tree
                style={{width: 300}}
                dropdownStyle={{maxHeight: 400, overflow: 'auto'}}
                treeData={treeData1}
                value={this.state.value}
                placeholder="Please select"
                onChange={e => this.onChange(e)}
            />
        );
    }
}

stories.add('controlled Component single', () => <Demo/>);

class Demo2 extends React.Component {
    constructor() {
        super();
        this.state = {
            value: ['shanghai', 'beijing', 'zhongguo'],
        };
    }

    onChange(value) {
        console.log(value);
        this.setState({value});
    }

    render() {
        console.log(this.state.value);
        return (
            <Tree
                style={{width: 300}}
                multiple
                dropdownStyle={{maxHeight: 200, overflow: 'auto'}}
                treeData={treeData1}
                value={this.state.value}
                placeholder="Please select"
                onChange={e => this.onChange(e)}
            />
        );
    }
}

stories.add('controlled Component multiple', () => <Demo2/>);

stories.add('json tree', () => (
    <div>
        <Tree
            treeDataSimpleJson={treeJson}
            onChange={e => console.log('change', e)}
            onSelect={e => console.log('select', e)}
        />
        <br/>
        <Tree treeDataSimpleJson={treeJson} multiple onChange={e => console.log(e)} onSelect={e => console.log(e)}/>
    </div>
));

stories.add('icon tree', () => (
    <div>
        <Tree treeData={treeDataWithIcon}/>
        <br/>
        <Tree treeData={treeDataWithIcon} multiple blockNode/>
    </div>
));

stories.add('directory tree', () => <Tree treeData={treeData1} directory multiple blockNode/>);

const button = (
    <ButtonGroup size="small" theme="borderless">
        <Button
            onClick={e => {
                Toast.info(opts);
                e.stopPropagation();
            }}
        >
            提示
        </Button>
        <Button>点击</Button>
    </ButtonGroup>
);

const style = {
    display: 'flex',
    justifyContent: 'space-between',
    alignItems: 'center',
};

const treeDataWithNode2 = [
    {
        label: (
            <div style={style}>
                <span>亚洲</span>
                <ButtonGroup size="small" theme="borderless">
                    {button}
                </ButtonGroup>
            </div>
        ),
        value: 'yazhou',
        key: 'yazhou',
        children: [
            {
                label: (
                    <div style={style}>
                        <span>中国</span>
                        {button}
                    </div>
                ),
                value: 'zhongguo',
                key: 'zhongguo',
                children: [
                    {
                        label: (
                            <div style={style}>
                                <span>test</span>
                                {button}
                            </div>
                        ),
                        value: 'test',
                        key: 'test',
                    },
                ],
            },
            {
                label: (
                    <div style={style}>
                        <span>日本</span>
                        {button}
                    </div>
                ),
                value: 'riben',
                key: 'riben',
            },
        ],
    },
];

stories.add('tree label using node', () => (
    <div>
        <Tree treeData={treeDataWithNode2} blockNode defaultExpandAll/>
    </div>
));

const treeDataTest = [
    {
        value: '一级标签1',
        label: '一级标签1',
        id: 1,
        key: '1',
    },
];

class TagSideSheet2 extends React.Component {
    constructor() {
        super();
        this.state = {
            tagList: [],
            visibles: false,
        };
        this.onVisible = this.onVisible.bind(this);
        this.renderLabel = this.renderLabel.bind(this);
        this.transLabel = this.transLabel.bind(this);
    }

    componentDidMount() {
        let tagList = [...treeDataTest];
        this.setState({
            tagList,
        });
    }

    onVisible(visibles) {
        this.setState({
            visibles,
        });
    }

    renderLabel(item) {
        const {visibles} = this.state;
        console.log('rendering label', visibles);
        return (
            <Popover trigger="custom" position="bottomLeft" visible={visibles} content={'测试popover'}>
                <Button
                    icon={<IconEdit/>}
                    onClick={e => {
                        e.stopPropagation();
                        this.onVisible(!visibles);
                    }}
                >
                    {item.label}
                </Button>
            </Popover>
        );
    }

    transLabel(list) {
        // list = cloneDeep(list);
        list.forEach(item => {
            item.label = this.renderLabel(item);
            // item.key += Math.random().toString().slice(0, 5);
        });
        return list;
    }

    render() {
        const {tagList = []} = this.state;
        const transformedTags = this.transLabel(cloneDeep(tagList));
        console.log('transformedTags', transformedTags, treeDataTest);
        return <Tree treeData={transformedTags}/>;
    }
}

stories.add('tree label using popover', () => <TagSideSheet2/>);

stories.add('defaultExpandKeys tree', () => (
    <>
        <Tree treeData={treeData1} defaultExpandedKeys={['zhongguo', 'beimeizhou']} blockNode/>
        <Tree treeData={treeData1} defaultExpandAll blockNode/>
        <Tree treeData={treeData1} defaultExpandAll multiple blockNode/>
    </>
));

stories.add('labelInValue tree', () => (
    <>
        <Tree treeData={treeData1} labelInValue onChange={e => console.log(e)}/>
        <Tree treeData={treeData1} labelInValue onChange={e => console.log(e)} multiple/>
        <Tree treeData={treeDataWithIcon} labelInValue onChange={e => console.log(e)} multiple/>
    </>
));

class Test extends React.Component {
    constructor() {
        super();
        this.state = {
            loading: false,
        };
    }

    componentDidMount() {
        setTimeout(() => {
            console.log('set loading');
            this.setState({loading: true});
        }, 5000);
    }

    render() {
        const treeDataWithNode2 = [
            {
                label: (
                    <div style={style}>
                        <span>亚洲</span>
                        <ButtonGroup size="small" theme="borderless">
                            {button}
                        </ButtonGroup>
                    </div>
                ),
                value: 'yazhou',
                key: 'yazhou',
                children: [
                    {
                        label: (
                            <div style={style}>
                                <span>中国</span>
                                {button}
                            </div>
                        ),
                        value: 'zhongguo',
                        key: 'zhongguo',
                        children: [
                            {
                                label: (
                                    <div style={style}>
                                        <span>test</span>
                                        {button}
                                    </div>
                                ),
                                value: 'test',
                                key: 'test',
                            },
                        ],
                    },
                    {
                        label: (
                            <div style={style}>
                                <span>日本</span>
                                {button}
                            </div>
                        ),
                        value: 'riben',
                        key: 'riben',
                    },
                ],
            },
        ];
        return <Tree treeData={treeDataWithNode2}/>;
    }
}

stories.add('setState after 5s', () => <Test/>);

class DemoExpandedKeys extends React.Component {
    constructor() {
        super();
        this.state = {
            expand: ['zhongguo', 'beimeizhou'],
        };
    }

    onExpand(expand) {
        console.log(expand);
        this.setState({expand});
    }

    render() {
        return (
            <Tree
                style={{width: 300}}
                dropdownStyle={{maxHeight: 400, overflow: 'auto'}}
                treeData={treeData1}
                expandedKeys={this.state.expand}
                placeholder="Please select"
                onExpand={(e, rest) => this.onExpand(e, rest)}
            />
        );
    }
}

class DemoExpandeKeysMulti extends React.Component {
    constructor() {
        super();
        this.state = {
            expand: ['zhongguo'],
        };
    }

    onExpand(expand) {
        console.log(expand);
        this.setState({expand});
    }

    render() {
        return (
            <Tree
                style={{width: 300}}
                multiple
                dropdownStyle={{maxHeight: 400, overflow: 'auto'}}
                treeData={treeData1}
                expandedKeys={this.state.expand}
                placeholder="Please select"
                onExpand={e => this.onExpand(e)}
            />
        );
    }
}

stories.add('expandedKeys', () => (
    <>
        <DemoExpandedKeys/>
        <br/>
        <DemoExpandeKeysMulti/>
    </>
));

class DmExpandedKeys extends React.Component {
    constructor() {
        super();
        this.state = {
            treeData: [
                {
                    key: '0',
                    label: 'item-0',
                    value: '0',
                },
            ],
        };
        this.add = this.add.bind(this);
    }

    add() {
        let itemLength = Math.floor(Math.random() * 5) + 1;
        let treeData = new Array(itemLength).fill(0).map((v, i) => {
            let length = Math.floor(Math.random() * 3);
            let children = new Array(length).fill(0).map((cv, ci) => {
                let child = {
                    key: `${i}-${ci}`,
                    label: `Leaf-${i}-${ci}`,
                    value: `${i}-${ci}`,
                };
                return child;
            });
            let item = {
                key: `${i}`,
                label: `Item-${i}`,
                value: `${i}`,
                children,
            };
            return item;
        });
        this.setState({treeData});
    }

    render() {
        const {treeData} = this.state;
        return (
            <>
                <Tree treeData={this.state.treeData} expandedKeys={['0', '1']}/>
                <br/>
                <Button onClick={this.add}>动态改变数据</Button>
            </>
        );
    }
}

stories.add('dynamic expandKeys', () => (
    <>
        <DmExpandedKeys/>
    </>
));

class DmSelectedKeys extends React.Component {
    constructor() {
        super();
        this.state = {
            treeData: [
                {
                    key: '0',
                    label: 'item-0',
                    value: '0',
                },
            ],
        };
        this.add = this.add.bind(this);
    }

    add() {
        let itemLength = Math.floor(Math.random() * 5) + 1;
        let treeData = new Array(itemLength).fill(0).map((v, i) => {
            let length = Math.floor(Math.random() * 3);
            let children = new Array(length).fill(0).map((cv, ci) => {
                let child = {
                    key: `${i}-${ci}`,
                    label: `Leaf-${i}-${ci}`,
                    value: `${i}-${ci}`,
                };
                return child;
            });
            let item = {
                key: `${i}`,
                label: `Item-${i}`,
                value: `${i}`,
                children,
            };
            return item;
        });
        this.setState({treeData});
    }

    render() {
        const {treeData} = this.state;
        return (
            <>
                <Tree treeData={treeData} value={['0-0']} onChange={e => console.log(e)}/>
                <br/>
                <Button onClick={this.add}>动态改变数据</Button>
            </>
        );
    }
}

// the demo will not work anymore as value does not exist when the tree mounts
stories.add('dynamic selectedKey', () => (
    <>
        <DmSelectedKeys/>
    </>
));

stories.add('large amount of data', () => (
    <>
        <BigTree/>
    </>
));

stories.add('autosizer', () => (
    <div style={{
        boxSizing: 'border-box',
        height: 400,
        width: 200,
    }}>
        <div
            style={{
                boxSizing: 'border-box',
                height: '50%',
                width: 200,
            }}
        >
            <AutoSizer
                // defaultHeight={defaultHeight}
                // defaultWidth={defaultWidth}
            >
                {({height, width}) => (
                    <div
                        style={{
                            width: width,
                            height: height,
                        }}
                    >
                        {`width:${width}, height:${height}`}
                    </div>
                )}
            </AutoSizer>
        </div>
    </div>
));

const MotionCustomLabelDemo = () => {
    const treeData = [
        {
            label: '亚洲',
            value: 'Asia',
            key: '0',
            children: [
                {
                    label: '中国',
                    value: 'China',
                    key: '0-0',
                    children: [
                        {
                            label: '北京',
                            value: 'Beijing',
                            key: '0-0-0',
                        },
                        {
                            label: '上海',
                            value: 'Shanghai',
                            key: '0-0-1',
                        },
                    ],
                },
            ],
        },
        {
            label: '北美洲',
            value: 'North America',
            key: '1',
        },
    ];

    const [hoverKey, setHoverKey] = useState();
    const cusLabel = (list = []) => {
        const recusive = (list = []) => {
            if (!list.length) {
                return;
            }
            list.forEach(item => {
                const {type, label, key} = item;
                item.label = (
                    <div onMouseEnter={() => setHoverKey(key)}>
                        {label}
                        {hoverKey === key ? <IconMore/> : null}
                    </div>
                );
                recusive(item.children);
            });
        };
        recusive(list);
        return list;
    };
    return <Tree treeData={cusLabel(treeData)} defaultExpandAll/>;
};

stories.add('motion custom label', () => <MotionCustomLabelDemo/>);

const AutoParentDemo = () => {
    const [expandedKeys, setExpandedKeys] = useState(['beimeizhou']);
    const [selectedKeys, setSelectedKeys] = useState(['beimeizhou']);
    const [autoExpandParent, setAutoExpandParent] = useState(true);

    const onExpand = expandedKeys => {
        console.log('onExpand', expandedKeys);
        // if not set autoExpandParent to false, if children expanded, parent can not collapse.
        // or, you can remove all expanded children keys.
        setExpandedKeys(expandedKeys);
        setAutoExpandParent(false);
    };

    const onSelect = (selectedKeys, info) => {
        console.log('onSelect:', info);
        setSelectedKeys(selectedKeys);
    };

    return (
        <div>
            <Button
                onClick={() => {
                    setSelectedKeys(['riben']);
                    setExpandedKeys(['riben']);
                    setAutoExpandParent(true);
                }}
            >
                Update
            </Button>
            <Tree
                onExpand={onExpand}
                expandedKeys={expandedKeys}
                autoExpandParent={autoExpandParent}
                onChange={onSelect}
                value={selectedKeys}
                treeData={treeData1}
                multiple
            />
        </div>
    );
};

stories.add('autoExpandParent', () => (
    <>
        <AutoParentDemo/>
    </>
));

const findDescendantKeys = node => {
    let res = [node.key];
    const findChild = item => {
        if (!item) return;
        const {children} = item;

        if (children && children.length) {
            children.forEach(child => {
                res.push(child.key);
                findChild(child);
            });
        }
    };
    findChild(node);
    return res;
};

class DyTreeWithExpandControlled extends React.Component {
    constructor() {
        super();
        this.state = {
            treeData: [
                {
                    key: '0',
                    label: 'item-0',
                    value: '0',
                },
            ],
            expandedKeys: [],
            autoExpandParent: false,
            inputValue: '',
            collapsedKeys: new Set([]),
        };
    }

    componentDidUpdate(prevProps, prevState) {
        if (
            !isEqual(prevState.treeData, this.state.treeData) ||
            !isEqual(prevState.inputValue, this.state.inputValue)
        ) {
            const {treeData, inputValue, collapsedKeys} = this.state;
            let filteredKeys = [];

            const findFilteredKey = arr => {
                arr.forEach(item => {
                    if (item.label.indexOf(inputValue) > -1) {
                        filteredKeys.push(item.key);
                    }
                    if (item.children) {
                        findFilteredKey(item.children);
                    }
                });
            };
            findFilteredKey(treeData);
            const expanded = difference(filteredKeys, [...collapsedKeys]);
            this.setState({
                expandedKeys: expanded,
                autoExpandParent: true,
            });
        }
    }

    add = () => {
        let itemLength = Math.floor(Math.random() * 5) + 1;
        let treeData = new Array(itemLength).fill(0).map((v, i) => {
            let length = Math.floor(Math.random() * 3) + 1;
            let children = new Array(length).fill(0).map((cv, ci) => {
                let child = {
                    key: `${i}-${ci}`,
                    label: `Leaf-${i}-${ci}`,
                    value: `${i}-${ci}`,
                };
                return child;
            });
            let item = {
                key: `${i}`,
                label: `Item-${i}`,
                value: `${i}`,
                children,
            };
            return item;
        });
        this.setState({treeData});
    };

    search = val => {
        this.setState({inputValue: val});
    };

    onExpand = (keys, {expanded, node}) => {
        let collapsed = this.state.collapsedKeys;
        let desKeys = findDescendantKeys(node);
        if (!expanded) {
            desKeys.forEach(key => collapsed.add(key));
        } else {
            desKeys.forEach(key => collapsed.has(key) && collapsed.delete(key));
        }
        this.setState({
            expandedKeys: keys,
            autoExpandParent: false,
            collapsedKeys: collapsed,
        });
    };

    render() {
        const {treeData, expandedKeys, autoExpandParent} = this.state;
        return (
            <div>
                <Tree
                    treeData={treeData}
                    filterTreeNode
                    autoExpandParent={autoExpandParent}
                    expandedKeys={expandedKeys}
                    onSearch={this.search}
                    onExpand={this.onExpand}
                />
                <br/>
                <Button onClick={this.add} style={{margin: 20}}>
                    动态改变数据
                </Button>
            </div>
        );
    }
}

stories.add('dynamic treeData with searchValue and controlled expand', () => <DyTreeWithExpandControlled/>);

const CusSearchRender = () => {
    const [inputValue, setInputValue] = useState('');
    const ref = useRef();

    const setValue = value => {
        setInputValue(value);
        ref.current.search(value);
    };

    return (
        <Tree
            ref={ref}
            treeData={treeData1}
            filterTreeNode
            showFilteredOnly
            searchRender={({prefix, placeholder}) => (
                <Input
                    prefix={prefix}
                    placeholder={placeholder}
                    value={inputValue}
                    onChange={value => setValue(value)}
                />
            )}
        />
    );
};

stories.add('CusSearchRender', () => (
    <>
        <CusSearchRender/>
    </>
));

const RefSearch = () => {
    const ref = useRef();
    const treeData = [
        {
            label: 'Asia',
            value: 'Asia',
            key: '0',
            children: [
                {
                    label: 'China',
                    value: 'China',
                    key: '0-0',
                    children: [
                        {
                            label: 'Beijing',
                            value: 'Beijing',
                            key: '0-0-0',
                        },
                        {
                            label: 'Shanghai',
                            value: 'Shanghai',
                            key: '0-0-1',
                        },
                    ],
                },
                {
                    label: 'Japan',
                    value: 'Japan',
                    key: '0-1',
                    children: [
                        {
                            label: 'Osaka',
                            value: 'Osaka',
                            key: '0-1-0',
                        },
                    ],
                },
            ],
        },
        {
            label: 'North America',
            value: 'North America',
            key: '1',
            children: [
                {
                    label: 'United States',
                    value: 'United States',
                    key: '1-0',
                },
                {
                    label: 'Canada',
                    value: 'Canada',
                    key: '1-1',
                },
            ],
        },
    ];
    return (
        <div>
            <Input onChange={v => ref.current.search(v)}/>
            <Tree
                treeData={treeData}
                defaultValue="Shanghai"
                blockNode={false}
                ref={ref}
                filterTreeNode
                searchRender={() => null}
            />
        </div>
    );
};

stories.add('RefSearch', () => (
    <>
        <RefSearch/>
    </>
));

const initTreeDate = [
    {
        label: 'Expand to load',
        key: '0',
    },
    {
        label: 'Expand to load',
        key: '1',
    },
    {
        label: 'Tree Node',
        key: '2',
        isLeaf: true,
    },
];

function updateTreeData(list, key, children) {
    return list.map(node => {
        if (node.key === key) {
            return {...node, children};
        }
        if (node.children) {
            return {...node, children: updateTreeData(node.children, key, children)};
        }

        return node;
    });
}

const LoadingTreeDemo = () => {
    const [treeData, setTreeData] = useState(initTreeDate);

    function onLoadData({key, children}) {
        return new Promise(resolve => {
            if (children) {
                resolve();
                return;
            }

            setTimeout(() => {
                setTreeData(origin =>
                    updateTreeData(origin, key, [
                        {
                            label: 'Child Node',
                            key: `${key}-0`,
                        },
                        {
                            label: 'Child Node',
                            key: `${key}-1`,
                        },
                    ])
                );
                resolve();
            }, 1000);
        });
    }

    return <Tree loadData={onLoadData} treeData={cloneDeep(treeData)}/>;
};

stories.add('loading', () => (
    <>
        <LoadingTreeDemo/>
    </>
));

const LoadingWithSearch = () => {
    const [treeData, setTreeData] = useState(initTreeDate);

    function onLoadData({key, children}) {
        return new Promise(resolve => {
            if (children) {
                resolve();
                return;
            }

            setTimeout(() => {
                setTreeData(origin =>
                    updateTreeData(origin, key, [
                        {
                            label: 'Child Node',
                            key: `${key}-0`,
                        },
                        {
                            label: 'Child Node',
                            key: `${key}-1`,
                        },
                    ])
                );
                resolve();
            }, 1000);
        });
    }

    return <Tree loadData={onLoadData} treeData={cloneDeep(treeData)} filterTreeNode/>;
};

stories.add('Loading with search', () => (
    <>
        <LoadingWithSearch/>
    </>
));

const DisabledStrictly = () => {
    return (
        <>
            <span> disable shanghai(checked), China </span>
            <Tree
                treeData={[
                    {
                        label: 'Asia',
                        value: 'Asia',
                        key: '0',
                        children: [
                            {
                                label: 'China',
                                value: 'China',
                                key: '0-0',
                                disabled: true,
                                children: [
                                    {
                                        label: 'Beijing',
                                        value: 'Beijing',
                                        key: '0-0-0',
                                    },
                                    {
                                        label: 'Shanghai',
                                        value: 'Shanghai',
                                        key: '0-0-1',
                                        disabled: true,
                                    },
                                ],
                            },
                            {
                                label: 'Japan',
                                value: 'Japan',
                                key: '0-1',
                                children: [
                                    {
                                        label: 'Osaka',
                                        value: 'Osaka',
                                        key: '0-1-0'
                                    }
                                ]
                            },
                        ],
                    }
                ]}
                defaultValue='Shanghai'
                multiple
                defaultExpandAll
                disableStrictly
            />
            <br/>
            <span> disable shanghai(checked), beijing(checked) </span>
            <Tree
                treeData={[
                    {
                        label: 'Asia',
                        value: 'Asia',
                        key: '0',
                        children: [
                            {
                                label: 'China',
                                value: 'China',
                                key: '0-0',
                                children: [
                                    {
                                        label: 'Beijing',
                                        value: 'Beijing',
                                        key: '0-0-0',
                                        disabled: true,
                                    },
                                    {
                                        label: 'Shanghai',
                                        value: 'Shanghai',
                                        key: '0-0-1',
                                        disabled: true,
                                    },
                                ],
                            },
                            {
                                label: 'Japan',
                                value: 'Japan',
                                key: '0-1',
                                children: [
                                    {
                                        label: 'Osaka',
                                        value: 'Osaka',
                                        key: '0-1-0'
                                    }
                                ]
                            },
                        ],
                    }
                ]}
                defaultValue={['Shanghai', 'Beijing']}
                multiple
                defaultExpandAll
                disableStrictly
            />
            <span> disable shanghai(checked) </span>
            <Tree
                treeData={[
                    {
                        label: 'Asia',
                        value: 'Asia',
                        key: '0',
                        children: [
                            {
                                label: 'China',
                                value: 'China',
                                key: '0-0',
                                children: [
                                    {
                                        label: 'Beijing',
                                        value: 'Beijing',
                                        key: '0-0-0',
                                    },
                                    {
                                        label: 'Shanghai',
                                        value: 'Shanghai',
                                        key: '0-0-1',
                                        disabled: true,
                                    },
                                ],
                            },
                            {
                                label: 'Japan',
                                value: 'Japan',
                                key: '0-1',
                                children: [
                                    {
                                        label: 'Osaka',
                                        value: 'Osaka',
                                        key: '0-1-0'
                                    }
                                ]
                            },
                        ],
                    }
                ]}
                defaultValue={['Shanghai']}
                multiple
                defaultExpandAll
                disableStrictly
            />
            <br/>
            <span> disable shanghai </span>
            <Tree
                treeData={[
                    {
                        label: 'Asia',
                        value: 'Asia',
                        key: '0',
                        children: [
                            {
                                label: 'China',
                                value: 'China',
                                key: '0-0',
                                children: [
                                    {
                                        label: 'Beijing',
                                        value: 'Beijing',
                                        key: '0-0-0',
                                    },
                                    {
                                        label: 'Chengdu',
                                        value: 'Chengdu',
                                        key: '0-0-2',
                                    },
                                    {
                                        label: 'Shanghai',
                                        value: 'Shanghai',
                                        key: '0-0-1',
                                        disabled: true,
                                    },
                                ],
                            },
                            {
                                label: 'Japan',
                                value: 'Japan',
                                key: '0-1',
                                children: [
                                    {
                                        label: 'Osaka',
                                        value: 'Osaka',
                                        key: '0-1-0'
                                    }
                                ]
                            },
                        ],
                    }
                ]}
                multiple
                defaultExpandAll
                disableStrictly
            />
            <span> disable China(checked) - Shanghai </span>
            <Tree
                treeData={[
                    {
                        label: 'Asia',
                        value: 'Asia',
                        key: '0',
                        children: [
                            {
                                label: 'China',
                                value: 'China',
                                key: '0-0',
                                disabled: true,
                                children: [
                                    {
                                        label: 'Beijing',
                                        value: 'Beijing',
                                        key: '0-0-0',
                                    },
                                    {
                                        label: 'Shanghai',
                                        value: 'Shanghai',
                                        key: '0-0-1',
                                        disabled: true,
                                    },
                                ],
                            },
                            {
                                label: 'Japan',
                                value: 'Japan',
                                key: '0-1',
                                children: [
                                    {
                                        label: 'Osaka',
                                        value: 'Osaka',
                                        key: '0-1-0'
                                    }
                                ]
                            },
                        ],
                    }
                ]}
                defaultValue='China'
                multiple
                defaultExpandAll
                disableStrictly
            />
            <span> disable China </span>
            <Tree
                treeData={[
                    {
                        label: 'Asia',
                        value: 'Asia',
                        key: '0',
                        children: [
                            {
                                label: 'China',
                                value: 'China',
                                key: '0-0',
                                disabled: true,
                                children: [
                                    {
                                        label: 'Beijing',
                                        value: 'Beijing',
                                        key: '0-0-0',
                                    },
                                    {
                                        label: 'Shanghai',
                                        value: 'Shanghai',
                                        key: '0-0-1',
                                    },
                                ],
                            },
                            {
                                label: 'Japan',
                                value: 'Japan',
                                key: '0-1',
                                children: [
                                    {
                                        label: 'Osaka',
                                        value: 'Osaka',
                                        key: '0-1-0'
                                    }
                                ]
                            },
                        ],
                    }
                ]}
                multiple
                defaultExpandAll
                disableStrictly
            />
        </>
    )
}

stories.add('disableStrictly', () => <DisabledStrictly/>);

const ActionTree = () => {
    let initData = [{
        label: 'Asia',
        value: 'Asia',
        key: 'asia',
        children: [{
            label: 0,
            value: `${0}`,
            key: `${0}`,
        }, {
            label: 1,
            value: `${1}`,
            key: `${1}`,
        }, {
            label: 2,
            value: `${2}`,
            key: `${2}`,
        }, {
            label: 3,
            value: `${3}`,
            key: `${3}`,
        }, {
            label: 4,
            value: `${4}`,
            key: `${4}`,
        }]
    }];

    const [data, setData] = useState(initData);

    const remove = key => {
        let ind = data[0].children.findIndex(item => item.key === key);
        if (ind >= 0) {
            let items = cloneDeep(data);
            items[0].children.splice(ind, 1);
            setData(items);
        }
    }

    return (
        <Tree
            treeData={cloneDeep(data)}
            renderLabel={(label, data) => (<div>{label}<Button onClick={() => remove(data.key)}>remove</Button></div>)}
        />
    );
};

stories.add('Delete or Add Child Node', () => <ActionTree/>);


const MutipleHLTree = () => {
    const [selected, setSelected] = useState(new Set());
    const [selectedThroughParent, setSelectedThroughParent] = useState(new Set());

    const findDescendantKeys = (node) => {
        let res = [node.key];
        const findChild = item => {
            if (!item) return;
            const {children} = item;

            if (children && children.length) {
                children.forEach(child => {
                    res.push(child.key);
                    findChild(child);
                });
            }
        };
        findChild(node);
        return res;
    }

    const handleSelect = (key, bool, node) => {
        setSelected(new Set([key]));
        const descendantKeys = findDescendantKeys(node);
        setSelectedThroughParent(new Set(descendantKeys));
    }

    const renderLabel = ({
        className,
        data,
        onClick,
        expandIcon
    }) => {
        const {label, icon, key} = data;
        const isLeaf = !(data.children && data.children.length);
        const style = {
            backgroundColor: selected.has(key)
                ? 'rgba(var(--semi-blue-0), 1)'
                : selectedThroughParent.has(key)
                    ? 'rgba(var(--semi-blue-0), .5)' : 'transparent'
        }
        return (
            <li
                className={className}
                role="treenode"
                onClick={onClick}
                style={style}
            >
                {isLeaf ? null : expandIcon}
                {icon ? icon : null}
                <span>{label}</span>
            </li>
        )
    }

    const treeStyle = {
        width: 260,
        height: 420,
        border: '1px solid var(--semi-color-border)'
    }
    return (
        <Tree
            treeData={treeData1}
            renderFullLabel={renderLabel}
            onSelect={handleSelect}
            style={treeStyle}
        />
    )
}

stories.add('renderOuterLable', () => <MutipleHLTree/>)

stories.add('tree without value props', () => (
    <Tree
        treeData={treeDataWithoutValue}
        value="meiguo"
        defaultExpandAll
        onChange={(...args) => console.log(args)}
    />
));

const x = 3;
const y = 2;
const z = 1;
const gData = [];


const DnDTree = () => {
    
    const initialData = [
        {
            label: 'Asia',
            value: 'Asia',
            key: '0',
            children: [
                {
                    label: 'China',
                    value: 'China',
                    key: '0-0',
                    children: [
                        {
                            label: 'Beijing',
                            value: 'Beijing',
                            key: '0-0-0',
                            disabled: true,
                        },
                        {
                            label: 'Shanghai',
                            value: 'Shanghai',
                            key: '0-0-1',
                            disabled: true,
                        },
                    ],
                },
                {
                    label: 'Japan',
                    value: 'Japan',
                    key: '0-1',
                    children: [
                        {
                            label: 'Osaka',
                            value: 'Osaka',
                            key: '0-1-0'
                        }
                    ]
                },
            ],
        },
        {
            label: 'North America',
            value: 'North America',
            key: '1',
            children: [
                {
                    label: 'United States',
                    value: 'United States',
                    key: '1-0'
                },
                {
                    label: 'Canada',
                    value: 'Canada',
                    key: '1-1'
                }
            ]
        },
        {
            label: 'Europe',
            value: 'Europe',
            key: '2',
        }
    ];

    const [treeData, setTreeData] = useState(initialData);

    // const [expandedKeys, setExpandedKeys] = useState(['zhongguo']);

    function onDragEnter(info) {
        console.log(info);
        // if in controlled expandedKeys mode
        // setExpandedKeys(info.expandedKeys)
    }

    function onDrop(info) {
        const {dropToGap, node, dragNode} = info;
        const dropKey = node.key;
        const dragKey = dragNode.key;
        const dropPos = node.pos.split('-');
        const dropPosition = info.dropPosition - Number(dropPos[dropPos.length - 1]);
        const data = [...treeData];
        const loop = (data, key, callback) => {
            data.forEach((item, ind, arr) => {
                if (item.key === key) return callback(item, ind, arr);
                if (item.children) return loop(item.children, key, callback);
            })
        }

        let dragObj;
        loop(data, dragKey, (item, ind, arr) => {
            arr.splice(ind, 1);
            dragObj = item;
        })

        if (!dropToGap) {
            // inset into the dropPosition
            loop(data, dropKey, (item, ind, arr) => {
                item.children = item.children || [];
                item.children.push(dragObj)
            })
        } else if (dropPosition === 1 && node.children && node.expanded) {
            // has children && expanded and drop into the node bottom gap
            // insert to the top 这里我们添加在头部，可以是任意位置
            loop(data, dropKey, item => {
                item.children = item.children || [];
                item.children.unshift(dragObj)
            })
        } else {
            let dropNodeInd;
            let dropNodePosArr;
            loop(data, dropKey, (item, ind, arr) => {
                dropNodePosArr = arr;
                dropNodeInd = ind;
            })
            if (dropPosition === -1) {
                // insert to top
                dropNodePosArr.splice(dropNodeInd, 0, dragObj)
            } else {
                // insert to bottom
                dropNodePosArr.splice(dropNodeInd + 1, 0, dragObj)
            }
        }
        setTreeData(data)
    }

    return <Tree
        treeData={treeData}
        draggable
        //expandedKeys={expandedKeys}
        onDragEnter={onDragEnter}
        onDrop={onDrop}
    />;
};

stories.add('draggable Tree', () => <DnDTree/>)

const TestTree = () => {
    return (
        <Tree
            treeData={testData}
            // motion={false}
            style={{height: '100%'}}
            filterTreeNode
            expandAction="click"
            showFilteredOnly
        />
    );
};

stories.add('draggable', () => <TestTree/>);

stories.add('renderFullLabel with draggable', () => {
        const [selected, setSelected] = useState(new Set());
        const [selectedThroughParent, setSelectedThroughParent] = useState(new Set());
        const defaultTreeData = [
            {
                label: '黑色固定按钮',
                key: 'fix-btn-0'
            },
            {
                label: '模块',
                key: 'module-0',
                children: [
                    {
                        label: '可自由摆放的组件',
                        key: 'free-compo-0',
                    },
                    {
                        label: '分栏容器',
                        key: 'split-col-0',
                        children: [
                            {
                                label: '按钮组件',
                                key: 'btn-0'
                            },
                            {
                                label: '按钮组件',
                                key: 'btn-1'
                            }
                        ]
                    },
                ],
            },
            {
                label: '模块',
                key: 'module-1',
                children: [
                    {
                        label: '自定义组件',
                        key: 'cus-0'
                    }
                ]
            }
        ]
        const [treeData, setTreeData] = useState(defaultTreeData);

        const onDrop = (info) => {
            const {dropToGap, node, dragNode} = info;
            const dropKey = node.key;
            const dragKey = dragNode.key;
            const dropPos = node.pos.split('-');
            const dropPosition = info.dropPosition - Number(dropPos[dropPos.length - 1]);

            const data = [...treeData];
            const loop = (data, key, callback) => {
                data.forEach((item, ind, arr) => {
                    if (item.key === key) return callback(item, ind, arr);
                    if (item.children) return loop(item.children, key, callback);
                })
            }

            let dragObj;
            loop(data, dragKey, (item, ind, arr) => {
                arr.splice(ind, 1);
                dragObj = item;
            })

            if (!dropToGap) {
                loop(data, dropKey, (item, ind, arr) => {
                    item.children = item.children || [];
                    item.children.push(dragObj)
                })
            } else if (dropPosition === 1 && node.children && node.expanded) {
                loop(data, dropKey, item => {
                    item.children = item.children || [];
                    item.children.unshift(dragObj)
                })
            } else {
                let dropNodeInd;
                let dropNodePosArr;
                loop(data, dropKey, (item, ind, arr) => {
                    dropNodePosArr = arr;
                    dropNodeInd = ind;
                })
                if (dropPosition === -1) {
                    dropNodePosArr.splice(dropNodeInd, 0, dragObj)
                } else {
                    dropNodePosArr.splice(dropNodeInd + 1, 0, dragObj)
                }
            }
            setTreeData(data)
        }

        const findDescendantKeys = (node) => {
            let res = [node.key];
            const findChild = item => {
                if (!item) return;
                const {children} = item;

                if (children && children.length) {
                    children.forEach(child => {
                        res.push(child.key);
                        findChild(child);
                    });
                }
            };
            findChild(node);
            return res;
        }

        const handleSelect = (key, bool, node) => {
            setSelected(new Set([key]));
            const descendantKeys = findDescendantKeys(node);
            setSelectedThroughParent(new Set(descendantKeys));
        }

        const renderLabel = ({
            className,
            data,
            onClick,
            expandIcon
        }) => {
            const {label, icon, key} = data;
            const isLeaf = !(data.children && data.children.length);
            const style = {
                backgroundColor: selected.has(key)
                    ? 'rgba(var(--semi-blue-0), 1)'
                    : selectedThroughParent.has(key)
                        ? 'rgba(var(--semi-blue-0), .5)' : 'transparent'
            }
            return (
                <li
                    className={className}
                    role="treenode"
                    onClick={onClick}
                    style={style}
                >
                    {isLeaf ? <span style={{width: 24}}></span> : expandIcon}
                    {icon ? icon : null}
                    <span>{label}</span>
                </li>
            )
        }

        const treeStyle = {
            height: 420,
            border: '1px solid var(--semi-color-border)'
        }


        return <Tree
            treeData={treeData}
            draggable
            onDrop={onDrop}
            renderFullLabel={renderLabel}
            onSelect={handleSelect}
            style={treeStyle}
            defaultExpandAll
        />;
    }
);