removeEmptyFloder

[jdz321]

let Arr = [
  { name: "1", type: "floder", children: [{ name: "1", type: "file" }] },
  {
    name: "2",
    type: "floder",
    children: [
      {
        name: "1",
        type: "floder",
        children: [
          {
            name: "1",
            type: "floder",
            children: [{ name: "1", type: "floder", children: [] }],
          },
        ],
      },
    ],
  },
  {
    name: "3",
    type: "floder",
    children: [
      {
        name: "1",
        type: "floder",
        children: [{ name: "1", type: "floder", children: [] }],
      },
    ],
  },
];

function removeEmptyFloder(arr) {
  const emptyFloder = []
  const root = {
    children: arr,
  }

  function walkTree(tree = [], parent) {
    for (let node of tree) {
      node.parent = parent
      if (node.children && node.children.length) {
        walkTree(node.children, node)
      }
      // mark empty floder
      if (node.type === 'floder') {
        if (!node.children || node.children.length === 0) {
          emptyFloder.push(node)
        }
      }
    }
  }

  walkTree(arr, root)

  function removeSelf(node) {
    if (node.parent) {
      node.parent.children = node.parent.children.filter(item => item !== node)
      if (node.parent.children.length === 0) {
        removeSelf(node.parent)
      }
    }
  }

  emptyFloder.forEach(removeSelf)

  return JSON.parse(JSON.stringify(root.children, (key, value) => key === 'parent' ? undefined : value))
}

[jdz321]

function walkTree(tree = [], parent, emptyFloder) {
  for (let node of tree) {
    const shadowNode = {
      node,
      parent,
      children: [],
    }
    parent.children.push(shadowNode)
    if (node.children && node.children.length) {
      walkTree(node.children, shadowNode, emptyFloder)
    }
    // mark empty floder
    if (node.type === 'floder') {
      if (!node.children || node.children.length === 0) {
        emptyFloder.push(shadowNode)
      }
    }
  }
  return emptyFloder
}

function removeEmptyFloder(arr) {
  const shadowRoot = {
    children: [],
    node: { children: arr },
  }

  const emptyFloder = walkTree(arr, shadowRoot, [])

  emptyFloder.forEach((shadowNode) => {
    let { node, parent } = shadowNode
    while (parent) {
      const parentNode = parent.node
      parentNode.children = parentNode.children.filter(item => item !== node)
      if (parentNode.children.length === 0) {
        parent = parent.parent
        node = parent.node
      } else {
        break
      }
    }
  })

  return shadowRoot.node.children
}

[jdz321]

interface Node {
  name?: string;
  type?: 'floder' | 'file';
  children?: Node[];
}

function removeEmptyFloderInternal(node: Node) {
  if (node.type === 'floder') {
    node.children = (node.children || []).map(removeEmptyFloderInternal).filter((child) => {
      return child.type !== 'floder' || (child.children || []).length !== 0
    })
  }
  return node
}

function removeEmptyFloder(arr: Node[]) {
  return removeEmptyFloderInternal({ children: arr, type: 'floder' }).children
}