linux-odyssey/packages/file-graph/fileNode.js

import path from 'path'
import File from './file.js'

export default class FileNode extends File {
  constructor(file) {
    super(file)
    this.children = []
  }

  addChild(fileInput) {
    const file = new File(fileInput)
    if (file.path === this.path) {
      throw new Error('File already exists in the tree')
    }

    const parentPath = path.dirname(file.path)
    if (parentPath === this.path) {
      // Direct child of this node
      this.children.push(new FileNode(file))
    } else {
      // Recursively call addChild on the correct child node
      const childNode = this.children.find((child) => child.contains(file))
      if (childNode) {
        childNode.addChild(file)
      } else {
        console.log(fileInput, this)
        throw new Error('Parent path not found in the tree')
      }
    }
  }

  toString() {
    const indent = '    '.repeat(this.path.split('/').length - 1)
    const childrenString = this.children
      .map((child) => child.toString())
      .join('')
    return `${indent}${this.name}${
      this.isDirectory() ? '/' : ''
    }\n${childrenString}`
  }
}