# frozen_string_literal: true

class FileTree
  def file_icon(file)
    if file.file_type.audio?
      'fa fa-file-audio-o'
    elsif file.file_type.image?
      'fa fa-file-image-o'
    elsif file.file_type.video?
      'fa fa-file-video-o'
    elsif file.read_only?
      'fa fa-lock'
    elsif file.file_type.executable?
      'fa fa-file-code-o'
    elsif file.file_type.renderable?
      'fa fa-file-text-o'
    else
      'fa fa-file-o'
    end
  end
  private :file_icon

  def folder_icon
    'fa fa-folder-o'
  end
  private :folder_icon

  def initialize(files = [])
    # Our tree needs a root node, but we won't display it.
    @root = Tree::TreeNode.new('ROOT')

    files.uniq(&:filepath).each do |file|
      parent = @root
      (file.path || '').split('/').each do |segment|
        node = parent.children.detect {|child| child.name == segment } || parent.add(Tree::TreeNode.new(segment))
        parent = node
      end
      parent.add(Tree::TreeNode.new(file.name_with_extension, file))
    end
  end

  def map_to_js_tree(node)
    {
      children: node.children.map {|child| map_to_js_tree(child) },
      icon: node_icon(node),
      id: node.content.try(:ancestor_id),
      state: {
        disabled: !node.is_leaf?,
        opened: !node.is_leaf?,
      },
      text: node.name,
    }
  end
  private :map_to_js_tree

  def node_icon(node)
    if node.is_leaf? && !node.is_root?
      file_icon(node.content)
    else
      folder_icon
    end
  end
  private :node_icon

  def to_js_tree
    {
      core: {
        data: @root.children.map {|child| map_to_js_tree(child) },
      },
    }.to_json
  end
end