module Eco
  module API
    module Common
      module ClassHelpers

        # Creates an class and instance object methods with name `name` to resolve `klass` name
        def class_resolver(name, klass)
          define_singleton_method(name) { resolve_class(klass) }
          define_method(name) { self.class.resolve_class(klass) }
        end

        # With given a `klass` name it resolves to an actual `Class`
        # @return [Class] the class that was being searched by name `klass`.
        def resolve_class(klass, exception: true)
          @resolved ||= {}
          @resolved[klass] ||=
            case klass
              when Class
                klass
              when String
                begin
                  Kernel.const_get(klass)
                rescue NameError => e
                  raise if exception
                end
              when Symbol
                resolve_class(self.send(klass))
              else
                raise "Unknown class: #{klass}" if exception
            end
        end

        # Helper to normalize `key` into a correct `ruby` **constant name**
        # @param key [String, Symbol] to be normalized
        # @return [String] a correct constant name
        def to_constant(key)
          str_name = key.to_s.strip.split(/[\-\_ ]/i).compact.map do |str|
            str.slice(0).upcase + str.slice(1..-1).downcase
          end.join("")
        end

        # If the class for `name` exists, it returns it. Otherwise it generates it.
        # @param name [String, Symbol] the name of the new class
        # @param inherits [Class] the parent class to _inherit_ from
        # @param parent_space [String] parent namespace of the generated class, if not given: `self`
        # @yield [child_class] configure the new class
        # @yieldparam child_class [Class] the new class
        # @return [Class] the new generated class
        def new_class(name, inherits:, parent_space: nil)
          name            = name.to_sym.freeze
          class_name      = to_constant(name)
          parent_space    = parent_space ? resolve_class(parent_space) : self
          full_class_name = "#{parent_space}::#{class_name}"

          unless target_class = resolve_class(full_class_name, exception: false)
            target_class = Class.new(inherits)
            parent_space.const_set class_name, target_class
          end

          target_class.tap do |klass|
            yield(klass) if block_given?
          end
        end

        # Finds all child classes of the current class.
        # @param parent_class [Class] the parent class we want to find children of.
        # @param direct [Boolean] it will only include direct child classes.
        # @param scope [nil, Array] to only look for descendants among the ones in `scope`.
        # @return [Arrary<Class>] the child classes in hierarchy order.
        def descendants(parent_class: self, direct: false, scope: nil)
          scope ||= ObjectSpace.each_object(::Class)
          return [] if scope.empty?
          scope.select do |klass|
            klass < parent_class
          end.sort do |k1, k2|
            next -1 if k2 < k1
            next  1 if k1 < k2
            0
          end.tap do |siblings|
            if direct
              siblings.reject! do |si|
                siblings.any? {|s| si < s}
              end
            end
          end
        end

        # @param parent_class [Class] the parent class we want to find children of.
        # @param direct [Boolean] it will only include direct child classes.
        # @return [Boolean] `true` if the current class has child classes, and `false` otherwise.
        def descendants?(parent_class: self, direct: false)
          descendants(parent_class: parent_class, direct: direct).length > 0
        end

      end
    end
  end
end