All files / src/gadgets/tree tree.js

100% Statements 26/26
100% Branches 10/10
100% Functions 6/6
100% Lines 26/26

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153                                                                                          21x 21x 43x         2x 1x 1x 1x 1x 1x             3x 3x 3x 2x 2x 2x 1x   1x   2x                               25x 25x   25x             51x 51x 51x 51x   51x                                                                                              
/* Copyright © 2016 Kuali, Inc. - All Rights Reserved
 * You may use and modify this code under the terms of the Kuali, Inc.
 * Pre-Release License Agreement. You may not distribute it.
 *
 * You should have received a copy of the Kuali, Inc. Pre-Release License
 * Agreement with this file. If not, please write to license@kuali.co.
 */
 
import classnames from 'classnames'
import Node from './node'
import NodeLabel from './node-label'
import { partial } from 'lodash'
import PropTypes from 'prop-types'
import React from 'react'
import styles from './style.css'
 
export default class Tree extends React.Component {
  static propTypes = {
    className: PropTypes.string,
    nodes: PropTypes.array,
    value: PropTypes.array,
    valueKey: PropTypes.string,
    labelKey: PropTypes.string,
    childKey: PropTypes.string,
    collapsible: PropTypes.bool,
    startCollapsed: PropTypes.bool,
    parentIcon: PropTypes.string,
    leafIcon: PropTypes.string,
    readOnly: PropTypes.bool
  }
 
  static defaultProps = {
    nodes: [],
    value: [],
    labelKey: 'label',
    valueKey: 'key',
    childKey: 'children',
    collapsible: true,
    startCollapsed: false,
    parentIcon: 'folder',
    leafIcon: 'insert_drive_file',
    readOnly: false
  }
 
  constructor (props) {
    super(props)
    this.state = {
      nodesCollapsed: props.nodes.map(node => props.startCollapsed)
    }
  }
 
  handleToggleNodeCollapse = (toggleIndex, e) => {
    if (this.props.collapsible) {
      e.stopPropagation()
      const newNodesCollapsed = [...this.state.nodesCollapsed]
      const collapsed = newNodesCollapsed[toggleIndex]
      newNodesCollapsed[toggleIndex] = !collapsed
      this.setState({
        nodesCollapsed: newNodesCollapsed
      })
    }
  }
 
  handleToggleNodeSelected = (nodeKey, e) => {
    e.stopPropagation()
    const { value, readOnly, onChange } = this.props
    if (!readOnly) {
      const newValue = [...value]
      const index = newValue.indexOf(nodeKey)
      if (index > -1) {
        newValue.splice(index, 1)
      } else {
        newValue.push(nodeKey)
      }
      onChange(newValue)
    }
  }
 
  render () {
    const {
      className,
      nodes,
      value,
      valueKey,
      labelKey,
      childKey,
      parentIcon,
      leafIcon,
      readOnly,
      ...rest
    } = this.props
    const { nodesCollapsed } = this.state
 
    return (
      <div className={className}>
        {nodes.map((node, i) => {
          const {
            [labelKey]: nodeLabel,
            [valueKey]: nodeKey,
            [childKey]: nodeChildren
          } = node
          const isSelected = value.includes(nodeKey)
          const selectedClass = isSelected ? styles.selected : ''
          const onNodeClick = partial(this.handleToggleNodeSelected, nodeKey)
 
          return (
            <div
              key={nodeKey}
              className={classnames({ [styles.parentNode]: !readOnly })}
            >
              {nodeChildren ? (
                <Node
                  nodeLabel={
                    <NodeLabel
                      label={nodeLabel}
                      iconName={parentIcon}
                      className={selectedClass}
                      onClick={onNodeClick}
                    />
                  }
                  collapsed={nodesCollapsed[i]}
                  onCollapseToggle={partial(this.handleToggleNodeCollapse, i)}
                >
                  <Tree
                    nodes={nodeChildren}
                    value={value}
                    parentIcon={parentIcon}
                    leafIcon={leafIcon}
                    readOnly={readOnly}
                    labelKey={labelKey}
                    valueKey={valueKey}
                    childKey={childKey}
                    {...rest}
                  />
                </Node>
              ) : (
                <div className={styles.item}>
                  <NodeLabel
                    label={nodeLabel}
                    iconName={leafIcon}
                    className={selectedClass}
                    onClick={onNodeClick}
                  />
                </div>
              )}
            </div>
          )
        })}
      </div>
    )
  }
}