class TreeNode

# File lib/cli-tree.rb, line 13
def TreeNode.from_h(hash)
  if hash.is_a?(Hash)
    raw_children = hash.has_key?(:children) ? hash[:children] : []
    children = raw_children.map{|ch| TreeNode.from_h(ch)}
    TreeNode.new hash[:name], children
  else
    TreeNode.new hash
  end
end

# File lib/cli-tree.rb, line 23
def TreeNode.from_json(json)
  hash = JSON.parse json, symbolize_names: true
  TreeNode.from_h hash
end

# File lib/cli-tree.rb, line 8
def initialize(name, children = [])
  @name = name
  @children = children
end

# File lib/cli-tree.rb, line 58
def print(stream: STDOUT, prefix: '')
  stream.puts render.map{|line| "#{prefix}#{line}\n"}
end

# File lib/cli-tree.rb, line 39
def render
  lines = [@name]
  @children.each_with_index do |child, index|
    child_lines = child.render
    if index < @children.size - 1
      child_lines.each_with_index do |line, idx|
        prefix = (idx == 0) ? "├── " : "|   "
        lines << "#{prefix}#{line}"
      end
    else
      child_lines.each_with_index do |line, idx|
        prefix = (idx == 0) ? "└── " : "    "
        lines << "#{prefix}#{line}"
      end
    end
  end
  lines
end

# File lib/cli-tree.rb, line 28
def to_h
  {
    name: @name,
    children: @children.map{|node| node.to_h}
  }
end

# File lib/cli-tree.rb, line 35
def to_json(**kwargs)
  JSON.generate(to_h, **kwargs)
end