# frozen_string_literal: true

require "mongoid"

Mongoid.configure do |config|
  config.clients.default = { hosts: [ENV.fetch("MONGO_HOST", "127.0.0.1:27017")], database: "mongoid_nested_includes_repro" }
end
Mongo::Logger.logger.level = Logger::WARN

class FindCounter
  attr_reader :finds

  def initialize = @finds= Hash.new(0)
  def started(event) = (finds[event.command["find"]] += 1 if event.command_name == "find")
  def succeeded(_) = nil
  def failed(_) = nil
end

class Author
  include Mongoid::Document
end

class Gallery
  include Mongoid::Document
  has_many :plain_photos, class_name: "Photo", inverse_of: :gallery
  has_many :rooted_photos, class_name: "::Photo", inverse_of: :gallery
end

class Photo
  include Mongoid::Document
  belongs_to :gallery
  belongs_to :author
end

module Shop
  class Order
    include Mongoid::Document
    has_many :items, class_name: "Item", inverse_of: :order
  end

  class Item
    include Mongoid::Document
    belongs_to :order, class_name: "Order", inverse_of: :items
    belongs_to :author, class_name: "::Author"
  end
end

Mongoid.purge!
3.times do
  gallery = Gallery.create!
  order = Shop::Order.create!
  Photo.create!(gallery: gallery, author: Author.create!)
  Shop::Item.create!(order: order, author: Author.create!)
end

def author_finds(label)
  counter = FindCounter.new
  subscriber = Mongoid.default_client.subscribe(Mongo::Monitoring::COMMAND, counter)
  yield
  Mongoid.default_client.unsubscribe(Mongo::Monitoring::COMMAND, subscriber)
  puts format("%-62s authors finds: %d", label, counter.finds["authors"])
end

puts "mongoid #{Mongoid::VERSION}, driver #{Mongo::VERSION}, ruby #{RUBY_VERSION}"
puts "rooted_photos class_name=#{Gallery.relations['rooted_photos'].class_name.inspect} klass=#{Gallery.relations['rooted_photos'].klass}"
puts "items class_name=#{Shop::Order.relations['items'].class_name.inspect} klass=#{Shop::Order.relations['items'].klass}"
author_finds('Gallery.includes(plain_photos: :author)  [class_name "Photo"]') { Gallery.includes(plain_photos: :author).each { |g| g.plain_photos.each(&:author) } }
author_finds('Gallery.includes(rooted_photos: :author) [class_name "::Photo"]') { Gallery.includes(rooted_photos: :author).each { |g| g.rooted_photos.each(&:author) } }
author_finds('Shop::Order.includes(items: :author)     [class_name "Item"]') { Shop::Order.includes(items: :author).each { |o| o.items.each(&:author) } }
Mongoid.default_client.database.drop

assoc_map = associations.group_by(&:inverse_class_name)
# ...
queue << assoc.class_name

queue << (assoc.polymorphic? ? assoc.class_name : assoc.klass.name)