module Pundit::Authorization

def authorize(record, query = nil, policy_class: nil)
  query ||= "#{action_name}?"
  @_pundit_policy_authorized = true
  Pundit.authorize(pundit_user, record, query, policy_class: policy_class, cache: policies)
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)
  policies[record] ||= Pundit.policy!(pundit_user, 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_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!(pundit_user, scope)
end

def pundit_policy_scoped?
  !!@_pundit_policy_scoped
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