Contents

module Lookbook
  class TreeNode
    include Enumerable

    delegate_missing_to :content

    attr_accessor :path, :content
    attr_reader :children

    def initialize(path = nil, content = nil, position: 10000)
      @path = path.to_s
      @content = content
      @position = position
      @children = []
    end

    def id
      Utils.id(content_value(:id, path))
    end

    def name
      segments.last
    end

    def label
      content_value(:label, name.titleize)
    end

    def position
      content_value(:position, @position)
    end

    def type
      content_value(:type, :directory)
    end

    def depth
      path.split("/").size
    end

    def add_child(name, content = nil, position: 10000)
      children << TreeNode.new("#{path}/#{name}", content, position: position)
      sort_children
    end

    def has_child?(name)
      !!get_child(name)
    end

    def get_child(name)
      children.find { |child| child.name == name }
    end

    def content?
      content.present?
    end

    def each(&block)
      if block
        children.each do |child|
          yield child
        end
      else
        to_enum(:each)
      end
    end

    protected

    def content_value(method_name, fallback = nil)
      value = content.send(method_name) if content
      value || fallback
    end

    def sort_children
      @children.sort_by! { |child| [child.position, child.label] }
    end

    def segments
      path.split("/")
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
lookbook-1.4.0	lib/lookbook/support/tree_node.rb