module Pundit::Authorization

def authorize(record, query = nil, policy_class: nil)
  query ||= "#{action_name}?"
  @_pundit_policy_authorized = true
  pundit.authorize(record, query: query, policy_class: policy_class)
end

def permitted_attributes(record, action = action_name)
  policy = policy(record)
  method_name = if policy.respond_to?("permitted_attributes_for_#{action}")
    "permitted_attributes_for_#{action}"
  else
    "permitted_attributes"
  end
  pundit_params_for(record).permit(*policy.public_send(method_name))
end

def policies
  @_pundit_policies ||= {}
end

def policy(record)
  pundit.policy!(record)
end

def policy_scope(scope, policy_scope_class: nil)
  @_pundit_policy_scoped = true
  policy_scope_class ? policy_scope_class.new(pundit_user, scope).resolve : pundit_policy_scope(scope)
end

def policy_scopes
  @_pundit_policy_scopes ||= {}
end

def pundit
  @pundit ||= Pundit::Context.new(
    user: pundit_user,
    policy_cache: Pundit::CacheStore::LegacyStore.new(policies)
  )
end

def pundit_params_for(record)
  params.require(PolicyFinder.new(record).param_key)
end

def pundit_policy_authorized?
  !!@_pundit_policy_authorized
end

def pundit_policy_scope(scope)
  policy_scopes[scope] ||= pundit.policy_scope!(scope)
end

def pundit_policy_scoped?
  !!@_pundit_policy_scoped
end

def pundit_reset!
  @pundit = nil
  @_pundit_policies = nil
  @_pundit_policy_scopes = nil
  @_pundit_policy_authorized = nil
  @_pundit_policy_scoped = nil
end

def pundit_user
  current_user
end

def skip_authorization
  @_pundit_policy_authorized = :skipped
end

def skip_policy_scope
  @_pundit_policy_scoped = :skipped
end

def verify_authorized
  raise AuthorizationNotPerformedError, self.class unless pundit_policy_authorized?
end

def verify_policy_scoped
  raise PolicyScopingNotPerformedError, self.class unless pundit_policy_scoped?
end