# frozen_string_literal: true

#
# Copyright (C) 2024 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.
#

class Loaders::OverrideAssignmentLoader < GraphQL::Batch::Loader
  def initialize(current_user)
    super()
    @current_user = current_user
  end

  def perform(assignments)
    # Bulk preload all override data to prevent N+1 queries
    DatesOverridable.preload_override_data_for_objects(assignments)

    # Conditionally preload assignment override students if needed
    if AssignmentOverrideApplicator.should_preload_override_students?(assignments, @current_user, "graphql_override_assignment_loader")
      AssignmentOverrideApplicator.preload_assignment_override_students(assignments, @current_user)
    end

    # Now call overridden_for on each assignment - this will use preloaded data
    assignments.each { |assignment| fulfill(assignment, assignment.overridden_for(@current_user)) }
  end
end
