cubesystems/releaf

module Releaf::Content::Builders
  module Tree
    def section_body
      tag(:div, class: body_classes) do
        tree
      end
    end

    def body_classes
      classes = [:body]
      classes << :empty if collection.size < 1
      classes
    end

    def sorted_tree
      sort_tree(build_tree)
    end

    def sort_tree(nodes)
      nodes.sort_by! { |item| item[:node].item_position }

      nodes.each do |node|
        sort_tree(node[:children]) if node[:children].present?
      end
    end

    def build_tree
      object_hash = collection.inject({}) do |result, node|
        result[node.id] = { node: node, children: [] }
        result
      end

      object_hash[nil] = { root: true, children: [] }

      object_hash.each_value do |node|
        next if node[:root]
        next if node[:node].parent_id && !object_hash[node[:node].parent_id]

        children = object_hash[node[:node].parent_id][:children]
        children << node
      end

      object_hash[nil][:children]
    end

    def tree
      tag(:div, class: "collection") do
        root_level
      end
    end

    def root_level
      return empty_body if collection.size < 1
      tree_level(sorted_tree, 1)
    end

    def empty_body
      tag(:div, class: "nothing-found") do
        t("Nothing found")
      end
    end

    def tree_level(list, level)
      tag(:ul, "data-level" => level) do
        list.collect do |resource|
          tree_resource(resource, level)
        end
      end
    end

    def tree_resource(resource, level)
      expanded = (layout_settings("content.tree.expanded.#{resource[:node].id}") == true)
      classes = []
      classes << 'collapsed' unless expanded
      classes << 'has-children' unless resource[:children].empty?

      tag(:li, class: classes, data: {level: level, id: resource[:node].id}) do
        tree_resource_blocks(resource, level, expanded)
      end
    end

    def tree_resource_blocks(resource, level, expanded)
      [tree_resource_collapser(resource, expanded),
       tree_resource_name(resource[:node]), tree_resource_children(resource, level)]
    end

    def tree_resource_collapser(resource, expanded)
      return if resource[:children].empty?
      tag(:div, class: "collapser-cell") do
        button(nil, (expanded ? 'chevron-down' : 'chevron-right'), class: %w(secondary collapser trigger), title: t(expanded ? "Collapse" : "Expand"))
      end
    end

    def tree_resource_children(resource, level)
      return if resource[:children].empty?
      tree_level(resource[:children], level + 1)
    end

    def tree_resource_name(resource)
      classes = ["node-cell"]
      classes << "active" if resource.active?

      tag(:div, class: classes) do
        tree_resource_name_button(resource)
      end
    end

    def tree_resource_name_button(resource)
      title = resource.content_id.present? ? "#{resource.content_type} ##{resource.content_id}" : resource.content_type
      tag(:a, class: "trigger", href: url_for(action: "edit", id: resource.id), title: title) do
        tag(:span, resource.name)
      end
    end
  end
end