JavaScript - Как динамически создавать вложенный массив объектов?

Question

Мне предоставляется строка B1-FEOL-SPUTTER-0015, строка будет разделена на - (тире), а затем использована для создания вложенного массива объектов.Это оригинальное дерево:

[{
  "text": "B1",
  "nodes": [{
    "text": "FEOL",
    "nodes": [{
      "text": "SPUTTER"
    }, {
      "text": "COATING"
    }, {
      "text": "EXPOSING"
    }, {
      "text": "DEVELOP"
    }, {
      "text": "PLATING"
    }, {
      "text": "BOND"
    }]
  }, {
    "text": "BEOL",
    "nodes": [{
      "text": "GRINDING"
    }, {
      "text": "BALLDROP"
    }, {
      "text": "PROBING"
    }, {
      "text": "BACKCOATING"
    }, {
      "text": "MARKING"
    }, {
      "text": "SORTING"
    }, {
      "text": "TUG"
    }]
  }]
}, {
  "text": "B2",
  "nodes": [{
    "text": "FEOL",
    "nodes": [{
      "text": "SPUTTER"
    }, {
      "text": "COATING"
    }, {
      "text": "EXPOSING"
    }, {
      "text": "DEVELOP"
    }, {
      "text": "PLATING"
    }, {
      "text": "BOND"
    }]
  }, {
    "text": "BEOL",
    "nodes": [{
      "text": "GRINDING"
    }, {
      "text": "BALLDROP"
    }, {
      "text": "PROBING"
    }, {
      "text": "BACKCOATING"
    }, {
      "text": "MARKING"
    }, {
      "text": "SORTING"
    }, {
      "text": "TUG"
    }]
  }]
}]

Я пытался использовать рекурсивные функции, вводя новые значения в .forEach(manipulateTree).Однако я не смог заставить его создать новые узлы в массиве, что должно быть сделано в операторе else.

import fs from 'fs'
import path from 'path'
import util from 'util'

// This is avaliable in the link below, leading to pastebin
let tree = require('./server/configs/tree.json')

// The values I used to test:

// This works
// const newBranch = 'B1-FEOL-SPUTTER-0015'

// This doesn't
const newBranch = 'B3-ASDF-DSDF987SDF7-0015'

const locations = newBranch.split('-')

let nodes = {}

let iteration = -1
const manipulateTree = branch => {
  iteration++
  // Found node
  if (branch.text === locations[iteration]) {
    // Check if the iteration is second last in the branch array
    if (iteration === locations.length - 2) {
      branch.nodes.push({
        'text': locations[iteration + 1],
        'location': locations.join('-')
      })
    } else {
      // If not then the recursive function will continue operation
      if (branch.nodes && branch.nodes.length > 0) {
        const found = branch.nodes.find(node => node.text === locations[iteration + 1])
        if (found) {
          branch.nodes.forEach(manipulateTree)
        }
      }
    }
  } else {
    console.log('No such node')
    if (iteration === locations.length - 1) {
      branch = {
        'text': locations[iteration],
        'location': locations.join('-')
      }
    } else {
      branch = {
        'text': locations[iteration],
        'nodes': []
      }
    }
  }
  return branch
}

if (tree && tree.length > 0) {
  tree.forEach(manipulateTree)
}

console.log(util.inspect(tree, false, null, true))

Конечный результат, который мне нужен, таков:

[{
  "text": "B1",
  "nodes": [{
    "text": "FEOL",
    "nodes": [{
      "text": "SPUTTER"
    }, {
      "text": "COATING"
    }, {
      "text": "EXPOSING"
    }, {
      "text": "DEVELOP"
    }, {
      "text": "PLATING"
    }, {
      "text": "BOND"
    }]
  }, {
    "text": "BEOL",
    "nodes": [{
      "text": "GRINDING"
    }, {
      "text": "BALLDROP"
    }, {
      "text": "PROBING"
    }, {
      "text": "BACKCOATING"
    }, {
      "text": "MARKING"
    }, {
      "text": "SORTING"
    }, {
      "text": "TUG"
    }]
  }]
}, {
  "text": "B2",
  "nodes": [{
    "text": "FEOL",
    "nodes": [{
      "text": "SPUTTER"
    }, {
      "text": "COATING"
    }, {
      "text": "EXPOSING"
    }, {
      "text": "DEVELOP"
    }, {
      "text": "PLATING"
    }, {
      "text": "BOND"
    }]
  }, {
    "text": "BEOL",
    "nodes": [{
      "text": "GRINDING"
    }, {
      "text": "BALLDROP"
    }, {
      "text": "PROBING"
    }, {
      "text": "BACKCOATING"
    }, {
      "text": "MARKING"
    }, {
      "text": "SORTING"
    }, {
      "text": "TUG"
    }]
  }]
}, { // This hear is the end result I want if there is no such node from the original tree
  "text": "B3",
  "nodes": [{
    "text": "ASDF",
    "nodes": [{
      "text": "DSDF987SDF7",
      "nodes": [{
        "text": "0015",
        "location": "B3-ASDF-DSDF987SDF7-0015"
      }]
    }]
  }]
}]

Answer 1

Учитывая древовидную структуру и путь, вам просто нужно перебрать путь и отслеживать, где вы находитесь. Если вы оказались в каком-то месте, которое еще не существует, добавьте его и продолжайте. Вы можете отслеживать индекс, чтобы знать, когда вы нажали последний элемент, и должны добавить массив location вместо nodes:

let tree = [{"text": "B1","nodes": [{"text": "FEOL","nodes": [{"text": "SPUTTER"}, {"text": "COATING"}, {"text": "EXPOSING"}, {"text": "DEVELOP"}, {"text": "PLATING"}, {"text": "BOND"}]}, {"text": "BEOL","nodes": [{"text": "GRINDING"}, {"text": "BALLDROP"}, {"text": "PROBING"}, {"text": "BACKCOATING"}, {"text": "MARKING"}, {"text": "SORTING"}, {"text": "TUG"}]}]}, {"text": "B2","nodes": [{"text": "FEOL","nodes": [{"text": "SPUTTER"}, {"text": "COATING"}, {"text": "EXPOSING"}, {"text": "DEVELOP"}, {"text": "PLATING"}, {"text": "BOND"}]}, {"text": "BEOL","nodes": [{"text": "GRINDING"}, {"text": "BALLDROP"}, {"text": "PROBING"}, {"text": "BACKCOATING"}, {"text": "MARKING"}, {"text": "SORTING"}, {"text": "TUG"}]}]}]

const newBranch = 'B3-ASDF-DSDF987SDF7-0015'

let components = newBranch.split('-') 

let nodes = tree                                    // nodes is the current array
components.forEach((text, i) => {
    let current = nodes.find(i => i.text == text )  // find matching item
    if (!current) {
        current = i == components.length - 1        // if no matching item add it 
          ? {text, location: newBranch }
          : {text, nodes:[] }
        nodes.push(current)
    }
    else if (!current.nodes) current.nodes = []          
    nodes = current.nodes  
})

console.log(tree)