75 lines
2.4 KiB
Ruby
75 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Namespaces
|
|
module Traversal
|
|
module Recursive
|
|
extend ActiveSupport::Concern
|
|
|
|
def root_ancestor
|
|
return self if parent.nil?
|
|
|
|
if persisted?
|
|
strong_memoize(:root_ancestor) do
|
|
self_and_ancestors.reorder(nil).find_by(parent_id: nil)
|
|
end
|
|
else
|
|
parent.root_ancestor
|
|
end
|
|
end
|
|
alias_method :recursive_root_ancestor, :root_ancestor
|
|
|
|
# Returns all ancestors, self, and descendants of the current namespace.
|
|
def self_and_hierarchy
|
|
object_hierarchy(self.class.where(id: id))
|
|
.all_objects
|
|
end
|
|
alias_method :recursive_self_and_hierarchy, :self_and_hierarchy
|
|
|
|
# Returns all the ancestors of the current namespaces.
|
|
def ancestors
|
|
return self.class.none unless parent_id
|
|
|
|
object_hierarchy(self.class.where(id: parent_id))
|
|
.base_and_ancestors
|
|
end
|
|
alias_method :recursive_ancestors, :ancestors
|
|
|
|
# returns all ancestors upto but excluding the given namespace
|
|
# when no namespace is given, all ancestors upto the top are returned
|
|
def ancestors_upto(top = nil, hierarchy_order: nil)
|
|
object_hierarchy(self.class.where(id: id))
|
|
.ancestors(upto: top, hierarchy_order: hierarchy_order)
|
|
end
|
|
|
|
def self_and_ancestors(hierarchy_order: nil)
|
|
return self.class.where(id: id) unless parent_id
|
|
|
|
object_hierarchy(self.class.where(id: id))
|
|
.base_and_ancestors(hierarchy_order: hierarchy_order)
|
|
end
|
|
alias_method :recursive_self_and_ancestors, :self_and_ancestors
|
|
|
|
# Returns all the descendants of the current namespace.
|
|
def descendants
|
|
object_hierarchy(self.class.where(parent_id: id))
|
|
.base_and_descendants
|
|
end
|
|
alias_method :recursive_descendants, :descendants
|
|
|
|
def self_and_descendants
|
|
object_hierarchy(self.class.where(id: id))
|
|
.base_and_descendants
|
|
end
|
|
alias_method :recursive_self_and_descendants, :self_and_descendants
|
|
|
|
def self_and_descendant_ids
|
|
self_and_descendants.select(:id)
|
|
end
|
|
alias_method :recursive_self_and_descendant_ids, :self_and_descendant_ids
|
|
|
|
def object_hierarchy(ancestors_base)
|
|
Gitlab::ObjectHierarchy.new(ancestors_base, options: { use_distinct: Feature.enabled?(:use_distinct_in_object_hierarchy, self) })
|
|
end
|
|
end
|
|
end
|
|
end
|