jsreport-studio/src/components/EntityTree/useCollapsed.js

jsreport templates editor and designer

import { useState, useCallback } from 'react'
import { checkIsGroupNode, checkIsGroupEntityNode } from './utils'

export default function useCollapsed ({
  listRef,
  getEntityById,
  getEntityByShortid
}) {
  const [collapsedNodes, setCollapsed] = useState({})

  const isNodeCollapsed = useCallback((node) => {
    if (checkIsGroupNode(node) && !checkIsGroupEntityNode(node)) {
      return collapsedNodes[node.id] == null ? false : collapsedNodes[node.id] === true
    }

    return collapsedNodes[node.id] == null ? true : collapsedNodes[node.id] === true
  }, [collapsedNodes])

  const toogleNodeCollapse = useCallback((node, forceState) => {
    const nodesToProcess = Array.isArray(node) ? node : [node]
    let newState

    if (nodesToProcess.length === 0) {
      return
    }

    newState = nodesToProcess.reduce((acu, nodeObj) => {
      let newCollapseState

      if (forceState != null) {
        newCollapseState = forceState === true
      } else {
        newCollapseState = !isNodeCollapsed(nodeObj)
      }

      acu[nodeObj.id] = newCollapseState

      return acu
    }, {})

    setCollapsed((prev) => ({
      ...prev,
      ...newState
    }))
  }, [isNodeCollapsed])

  const collapseHandler = useCallback((idOrShortid, state, options = {}) => {
    const { parents, self = true } = options
    const toCollapse = []
    let entity

    if (idOrShortid._id) {
      entity = getEntityById(idOrShortid._id, false)
    } else if (idOrShortid.shortid) {
      entity = getEntityByShortid(idOrShortid.shortid, false)
    }

    if (!entity) {
      return
    }

    if (entity.__entitySet === 'folders' && self === true) {
      toCollapse.push(entity)
    }

    if (parents === true) {
      let currentEntity = entity

      while (currentEntity != null) {
        if (currentEntity.folder != null) {
          currentEntity = getEntityByShortid(currentEntity.folder.shortid, false)

          if (currentEntity != null) {
            toCollapse.unshift(currentEntity)
          }
        } else {
          currentEntity = null
        }
      }
    }

    toogleNodeCollapse(toCollapse.map((folder) => {
      return listRef.current.entityNodesById[folder._id]
    }), state)
  }, [listRef, getEntityById, getEntityByShortid, toogleNodeCollapse])

  return {
    isNodeCollapsed,
    toogleNodeCollapse,
    collapseHandler
  }
}