module Ecoportal
  module API
    class GraphQL
      module Base
        class LocationStructure < Ecoportal::API::GraphQL::Base::Model
          extend Ecoportal::API::GraphQL::Helpers::LocationsTree

          passkey :id
          passthrough :name, :weight
          passboolean :archived, :published, :visitorManagementEnabled
          embeds_many :nodes, klass: Ecoportal::API::GraphQL::Base::LocationNode, order_key: :weight
          passthrough :draft

          def treeify
            self.class.treeify(nodes)
          end
        end
      end
    end
  end
end