JavaScript组合模式_树形结构操作

class MenuItem {
  constructor(name) {
    this.name = name;
    this.children = [];
  }

  // 添加子节点
  add(child) {
    this.children.push(child);
    return this;
  }

  // 移除子节点
  remove(child) {
    const index = this.children.indexOf(child);
    if (index !== -1) {
      this.children.splice(index, 1);
    }
  }

  // 遍历并执行操作
  traverse(callback) {
    callback(this);
    this.children.forEach(child => child.traverse(callback));
  }

  // 获取结构信息
  toString() {
    return this.name;
  }
}

// 创建根菜单
const menu = new MenuItem("主菜单");

// 创建子菜单
const fileMenu = new MenuItem("文件");
const editMenu = new MenuItem("编辑");

// 添加子项
fileMenu.add(new MenuItem("新建"))
        .add(new MenuItem("打开"))
        .add(new MenuItem("保存"));

editMenu.add(new MenuItem("剪切"))
        .add(new MenuItem("复制"))
        .add(new MenuItem("粘贴"));

// 组合到主菜单
menu.add(fileMenu).add(editMenu);

// 遍历打印所有菜单项
menu.traverse(item => {
  console.log(item.toString());
});

MenuItem.prototype.findByName = function(name) {
  let result = null;
  this.traverse(node => {
    if (node.name === name) {
      result = node;
    }
  });
  return result;
};

// 支持路径追踪（需在 traverse 时传入路径）
MenuItem.prototype.getPath = function(targetName) {
  const path = [];
  const dfs = (node, currentPath) => {
    const newPath = [...currentPath, node.name];
    if (node.name === targetName) {
      path.push(...newPath);
      return true;
    }
    for (let child of node.children) {
      if (dfs(child, newPath)) return true;
    }
    return false;
  };
  dfs(this, []);
  return path.length ? path.join(' → ') : 'Not found';
};

console.log(menu.getPath("保存")); 
// 输出：主菜单 → 文件 → 保存