def has_ancestry options = {}
raise Ancestry::AncestryException.new("Options for has_ancestry must be in a hash.") unless options.is_a? Hash
options.each do |key, value|
unless [:ancestry_column, :orphan_strategy, :cache_depth, :depth_cache_column, :touch].include? key
raise Ancestry::AncestryException.new("Unknown option for has_ancestry: #{key.inspect} => #{value.inspect}.")
end
end
cattr_accessor :ancestry_column
self.ancestry_column = options[:ancestry_column] || :ancestry
cattr_accessor :ancestry_base_class
self.ancestry_base_class = self
cattr_accessor :touch_ancestors
self.touch_ancestors = options[:touch] || false
include Ancestry::InstanceMethods
extend Ancestry::ClassMethods
extend Ancestry::MaterializedPath
cattr_reader :orphan_strategy
self.orphan_strategy = options[:orphan_strategy] || :destroy
validate :ancestry_exclude_self
scope :roots, lambda { where(root_conditions) }
scope :ancestors_of, lambda { |object| where(ancestor_conditions(object)) }
scope :children_of, lambda { |object| where(child_conditions(object)) }
scope :indirects_of, lambda { |object| where(indirect_conditions(object)) }
scope :descendants_of, lambda { |object| where(descendant_conditions(object)) }
scope :subtree_of, lambda { |object| where(subtree_conditions(object)) }
scope :siblings_of, lambda { |object| where(sibling_conditions(object)) }
scope :ordered_by_ancestry, Proc.new { |order|
if %w(mysql mysql2 sqlite sqlite3 postgresql).include?(connection.adapter_name.downcase) && ActiveRecord::VERSION::MAJOR >= 5
reorder(
Arel::Nodes::Ascending.new(Arel::Nodes::NamedFunction.new('COALESCE', [arel_table[ancestry_column], Arel.sql("''")])),
order
)
else
reorder(Arel.sql("(CASE WHEN #{connection.quote_table_name(table_name)}.#{connection.quote_column_name(ancestry_column)} IS NULL THEN 0 ELSE 1 END), #{connection.quote_table_name(table_name)}.#{connection.quote_column_name(ancestry_column)}"), order)
end
}
scope :ordered_by_ancestry_and, Proc.new { |order| ordered_by_ancestry(order) }
scope :path_of, lambda { |object| to_node(object).path }
before_save :update_descendants_with_new_ancestry
before_destroy :apply_orphan_strategy
if options[:cache_depth]
self.cattr_accessor :depth_cache_column
self.depth_cache_column = options[:depth_cache_column] || :ancestry_depth
before_validation :cache_depth
before_save :cache_depth
validates_numericality_of depth_cache_column, :greater_than_or_equal_to => 0, :only_integer => true, :allow_nil => false
end
{:before_depth => '<', :to_depth => '<=', :at_depth => '=', :from_depth => '>=', :after_depth => '>'}.each do |scope_name, operator|
scope scope_name, lambda { |depth|
raise Ancestry::AncestryException.new("Named scope '#{scope_name}' is only available when depth caching is enabled.") unless options[:cache_depth]
where("#{depth_cache_column} #{operator} ?", depth)
}
end
after_touch :touch_ancestors_callback
after_destroy :touch_ancestors_callback
if ActiveRecord::VERSION::STRING >= '5.1.0'
after_save :touch_ancestors_callback, if: :saved_changes?
else
after_save :touch_ancestors_callback, if: :changed?
end
end
def acts_as_tree(*args)
return super if defined?(super)
has_ancestry(*args)
end
end
end