module Pagy::GearboxExtra

def assign_last
  return super if !@vars[:gearbox_extra] || @vars[:limit_extra]
  gears = @vars[:gearbox_limit]
  # This algorithm is thousands of times faster than the one in the geared_pagination gem
  @last = (if count > (sum = gears.sum)
             [((count - sum).to_f / gears.last).ceil, 1].max + gears.count
           else
             pages     = 0
             remainder = count
             while remainder.positive?
               pages     += 1
               remainder -= gears[pages - 1]
             end
             [pages, 1].max
           end)
  @last = @vars[:max_pages] if @vars[:max_pages] && @last > @vars[:max_pages]
end

def assign_limit
  return super if !@vars[:gearbox_extra] || @vars[:limit_extra]
  gears = @vars[:gearbox_limit]
  raise VariableError.new(self, :gearbox_limit, 'to be an Array of positives', gears) \
        unless gears.is_a?(Array) && gears.all? { |num| num.positive? rescue false } # rubocop:disable Style/RescueModifier
  @limit = gears[@page - 1] || gears.last
end

def assign_offset
  return super if !@vars[:gearbox_extra] || @vars[:limit_extra]
  gears   = @vars[:gearbox_limit]
  @offset = if @page <= gears.count
              gears[0, @page - 1].sum
            else
              gears.sum + (gears.last * (@page - gears.count - 1))
            end + @outset
end