class Kfc

# File plugins/kfc.rb, line 62
def self.build_tracking_fetch(fetch)
  error_count = 0
  tracking_fetch = lambda do |url, **fetch_opts|
    fetch.call(url, **fetch_opts)
  rescue *KfcFetch::TRANSIENT_ERRORS => e
    error_count += 1 if KfcFetch.transient_error?(e)
    raise
  end

  [tracking_fetch, -> { error_count }]
end

# File plugins/kfc.rb, line 20
def self.command_pattern
  /kfc(?:\s+(.+))?$/i
end

# File plugins/kfc.rb, line 78
def self.default_fetch(url, headers: nil)
  KfcApi.live_fetch(url)
end

# File plugins/kfc.rb, line 26
def self.dispatch_command(args_text, lookup: nil)
  parsed = KfcStores.parse_location(args_text)
  return { error: parsed[:error] } if parsed[:error]

  lookup ||= ->(location, **) { lookup_menu(location) }
  { reply: lookup.call(parsed[:location]) }
end

# File plugins/kfc.rb, line 34
def self.lookup_menu(location, **opts)
  fetch = opts[:fetch] || method(:default_fetch)
  sleep_fn = opts[:sleep_fn] || method(:sleep)
  max_rounds = opts.fetch(:max_rounds, LOOKUP_MAX_ROUNDS)
  round_delay = opts.fetch(:round_delay, LOOKUP_ROUND_DELAY)
  max_attempts = opts.fetch(:max_attempts, KfcFetch::DEFAULT_MAX_ATTEMPTS)
  api_opts = opts.dup
  %i[fetch sleep_fn max_rounds round_delay max_attempts].each { |key| api_opts.delete(key) }

  last_result = nil
  max_rounds.times do |round|
    tracking_fetch, transient_errors = build_tracking_fetch(fetch)
    last_result = KfcApi.lookup_menu(
      location,
      fetch: tracking_fetch,
      max_attempts: max_attempts,
      sleep_fn: sleep_fn,
      **api_opts
    )
    return last_result unless store_lookup_failed?(last_result)
    return last_result if transient_errors.call.zero?

    sleep_fn.call(round_delay) if round < max_rounds - 1
  end

  last_result
end

# File plugins/kfc.rb, line 74
def self.store_lookup_failed?(report)
  report.to_s.include?("Could not find a KFC store")
end

# File plugins/kfc.rb, line 82
def execute(m, args_text)
  result = self.class.dispatch_command(args_text)

  if result[:error]
    themed_flood_safe_reply(m, result[:error])
    return
  end

  themed_flood_safe_reply(m, result[:reply]) if result[:reply]
end