module GraphQL::StaticValidation::DirectivesAreInValidLocations

def assert_includes_location(directive_defn, directive_ast, required_location)
  if !directive_defn.locations.include?(required_location)
    location_name = LOCATION_MESSAGE_NAMES[required_location]
    allowed_location_names = directive_defn.locations.map { |loc| LOCATION_MESSAGE_NAMES[loc] }
    add_error(GraphQL::StaticValidation::DirectivesAreInValidLocationsError.new(
      "'@#{directive_defn.graphql_name}' can't be applied to #{location_name} (allowed: #{allowed_location_names.join(", ")})",
      nodes: directive_ast,
      target: location_name,
      name: directive_defn.graphql_name
    ))
  end
end