2012-11-04 04:20:45 -05:00
# Pundit
2012-11-19 07:52:52 -05:00
[![Build Status ](https://secure.travis-ci.org/elabs/pundit.png?branch=master )](https://travis-ci.org/elabs/pundit)
2013-06-18 04:20:34 -04:00
[![Code Climate ](https://codeclimate.com/github/elabs/pundit.png )](https://codeclimate.com/github/elabs/pundit)
2012-11-19 07:52:52 -05:00
2012-11-19 07:36:35 -05:00
Pundit provides a set of helpers which guide you in leveraging regular Ruby
classes and object oriented design patterns to build a simple, robust and
scaleable authorization system.
2012-11-04 04:20:45 -05:00
## Installation
``` ruby
gem "pundit"
```
2012-11-19 07:13:37 -05:00
Include Pundit in your application controller:
``` ruby
class ApplicationController < ActionController::Base
include Pundit
protect_from_forgery
end
```
2012-11-19 05:04:18 -05:00
Optionally, you can run the generator, which will set up an application policy
2012-11-19 07:37:33 -05:00
with some useful defaults for you:
2012-11-19 05:04:18 -05:00
``` sh
rails g pundit:install
```
2013-08-16 15:27:49 -04:00
After generating your application policy, restart the Rails server so that Rails
can pick up any classes in the new `app/policies/` directory.
2012-11-04 04:20:45 -05:00
## Policies
Pundit is focused around the notion of policy classes. We suggest that you put
2014-01-30 16:08:01 -05:00
these classes in `app/policies` . This is a simple example that allows updating
a post if the user is an admin, or if the post is unpublished:
2012-11-04 04:20:45 -05:00
``` ruby
class PostPolicy
attr_reader :user, :post
def initialize(user, post)
@user = user
@post = post
end
2014-01-30 16:08:01 -05:00
def update?
2012-11-04 04:20:45 -05:00
user.admin? or not post.published?
end
end
```
As you can see, this is just a plain Ruby class. As a convenience, we can inherit
2014-01-30 14:10:48 -05:00
from Struct or use Struct.new to define the policy class:
2012-11-04 04:20:45 -05:00
``` ruby
2013-09-27 08:17:34 -04:00
PostPolicy = Struct.new(:user, :post) do
2014-01-30 16:08:01 -05:00
def update?
2012-11-04 04:20:45 -05:00
user.admin? or not post.published?
end
end
```
Pundit makes the following assumptions about this class:
- The class has the same name as some kind of model class, only suffixed
with the word "Policy".
- The first argument is a user. In your controller, Pundit will call the
`current_user` method to retrieve what to send into this argument
- The second argument is some kind of model object, whose authorization
you want to check. This does not need to be an ActiveRecord or even
an ActiveModel object, it can be anything really.
2014-01-30 16:08:01 -05:00
- The class implements some kind of query method, in this case `update?` .
2012-11-04 04:20:45 -05:00
Usually, this will map to the name of a particular controller action.
That's it really.
Supposing that you have an instance of class `Post` , Pundit now lets you do
this in your controller:
``` ruby
2014-01-30 16:08:01 -05:00
def update
@post = Post.find(params[:id])
2012-11-04 04:20:45 -05:00
authorize @post
2014-01-30 16:08:01 -05:00
if @post .update(post_params)
2012-11-04 04:20:45 -05:00
redirect_to @post
else
2014-01-30 16:08:01 -05:00
render :edit
2012-11-04 04:20:45 -05:00
end
end
```
The authorize method automatically infers that `Post` will have a matching
`PostPolicy` class, and instantiates this class, handing in the current user
and the given record. It then infers from the action name, that it should call
2014-01-30 16:08:01 -05:00
`update?` on this instance of the policy. In this case, you can imagine that
2012-11-04 04:20:45 -05:00
`authorize` would have done something like this:
``` ruby
2014-01-30 16:08:01 -05:00
raise "not authorized" unless PostPolicy.new(current_user, @post ).update?
2012-11-04 04:20:45 -05:00
```
2013-02-26 08:21:31 -05:00
You can pass a second argument to `authorize` if the name of the permission you
2012-11-19 07:06:48 -05:00
want to check doesn't match the action name. For example:
``` ruby
def publish
@post = Post.find(params[:id])
authorize @post , :update?
@post .publish!
redirect_to @post
end
```
2012-11-04 04:20:45 -05:00
You can easily get a hold of an instance of the policy through the `policy`
method in both the view and controller. This is especially useful for
conditionally showing links or buttons in the view:
``` erb
2014-01-30 16:08:01 -05:00
< % if policy(@post).update? %>
< %= link_to "Edit post", edit_post_path(@post) %>
2012-11-04 04:20:45 -05:00
< % end %>
```
## Ensuring policies are used
Pundit adds a method called `verify_authorized` to your controllers. This
method will raise an exception if `authorize` has not yet been called. You
should run this method in an `after_filter` to ensure that you haven't
forgotten to authorize the action. For example:
``` ruby
class ApplicationController < ActionController::Base
after_filter :verify_authorized, :except => :index
end
```
2014-01-29 10:57:55 -05:00
Likewise, Pundit also adds `verify_policy_scoped` to your controller. This
2013-04-18 01:05:24 -04:00
will raise an exception in the vein of `verify_authorized` . However it tracks
if `policy_scoped` is used instead of `authorize` . This is mostly useful for
controller actions like `index` which find collections with a scope and don't
authorize individual instances.
``` ruby
class ApplicationController < ActionController::Base
after_filter :verify_policy_scoped, :only => :index
end
```
2012-11-04 04:20:45 -05:00
## Scopes
Often, you will want to have some kind of view listing records which a
particular user has access to. When using Pundit, you are expected to
define a class called a policy scope. It can look something like this:
``` ruby
2013-09-27 08:17:34 -04:00
PostPolicy = Struct.new(:user, :post) do
2013-11-14 04:31:29 -05:00
self::Scope = Struct.new(:user, :scope) do
2012-11-04 04:20:45 -05:00
def resolve
if user.admin?
2013-10-16 20:11:45 -04:00
scope.all
2012-11-04 04:20:45 -05:00
else
scope.where(:published => true)
end
end
end
2014-01-30 16:08:01 -05:00
def update?
2012-11-04 04:20:45 -05:00
user.admin? or not post.published?
end
end
```
Pundit makes the following assumptions about this class:
- The class has the name `Scope` and is nested under the policy class.
- The first argument is a user. In your controller, Pundit will call the
`current_user` method to retrieve what to send into this argument.
- The second argument is a scope of some kind on which to perform some kind of
query. It will usually be an ActiveRecord class or a
`ActiveRecord::Relation` , but it could be something else entirely.
- Instances of this class respond to the method `resolve` , which should return
some kind of result which can be iterated over. For ActiveRecord classes,
this would usually be an `ActiveRecord::Relation` .
You can now use this class from your controller via the `policy_scope` method:
``` ruby
def index
@posts = policy_scope(Post)
end
```
Just as with your policy, this will automatically infer that you want to use
the `PostPolicy::Scope` class, it will instantiate this class and call
`resolve` on the instance. In this case it is a shortcut for doing:
``` ruby
def index
@posts = PostPolicy::Scope.new(current_user, Post).resolve
end
```
You can, and are encouraged to, use this method in views:
``` erb
< % policy_scope(@user.posts).each do |post| %>
2013-04-08 21:51:35 -04:00
< p > < % link_to post.title, post_path(post) %>< / p >
2012-11-04 04:20:45 -05:00
< % end %>
```
2013-01-08 03:46:20 -05:00
## Manually specifying policy classes
Sometimes you might want to explicitly declare which policy to use for a given
class, instead of letting Pundit infer it. This can be done like so:
``` ruby
class Post
def self.policy_class
2013-12-31 13:16:02 -05:00
PostablePolicy
2013-01-08 03:46:20 -05:00
end
end
```
2012-11-19 05:08:04 -05:00
## Just plain old Ruby
2012-11-04 04:20:45 -05:00
As you can see, Pundit doesn't do anything you couldn't have easily done
yourself. It's a very small library, it just provides a few neat helpers.
Together these give you the power of building a well structured, fully working
authorization system without using any special DSLs or funky syntax or
anything.
Remember that all of the policy and scope classes are just plain Ruby classes,
which means you can use the same mechanisms you always use to DRY things up.
Encapsulate a set of permissions into a module and include them in multiple
policies. Use `alias_method` to make some permissions behave the same as
others. Inherit from a base set of permissions. Use metaprogramming if you
2012-11-19 05:08:04 -05:00
really have to.
2012-11-04 04:20:45 -05:00
2012-11-19 05:05:20 -05:00
## Generator
Use the supplied generator to generate policies:
``` sh
rails g pundit:policy post
```
2012-11-04 04:20:45 -05:00
## Closed systems
In many applications, only logged in users are really able to do anything. If
you're building such a system, it can be kind of cumbersome to check that the
user in a policy isn't `nil` for every single permission.
We suggest that you define a filter that redirects unauthenticated users to the
login page. As a secondary defence, if you've defined an ApplicationPolicy, it
might be a good idea to raise an exception if somehow an unauthenticated user
got through. This way you can fail more gracefully.
``` ruby
2012-11-19 05:08:04 -05:00
class ApplicationPolicy
2012-11-04 04:20:45 -05:00
def initialize(user, record)
2012-12-03 08:59:11 -05:00
raise Pundit::NotAuthorizedError, "must be logged in" unless user
2012-11-19 05:08:04 -05:00
@user = user
@record = record
2012-11-04 04:20:45 -05:00
end
end
```
2012-11-19 05:04:18 -05:00
2013-08-16 15:27:49 -04:00
## Rescuing a denied Authorization in Rails
2013-06-24 12:47:33 -04:00
2013-11-11 12:30:05 -05:00
Pundit raises a `Pundit::NotAuthorizedError` you can [rescue_from ](http://guides.rubyonrails.org/action_controller_overview.html#rescue-from ) in your `ApplicationController` . You can customize the `user_not_authorized` method in every controller.
2013-06-24 12:47:33 -04:00
```ruby
class ApplicationController < ActionController::Base
protect_from_forgery
include Pundit
rescue_from Pundit::NotAuthorizedError, with: :user_not_authorized
private
def user_not_authorized
flash[:error] = "You are not authorized to perform this action."
redirect_to request.headers["Referer"] || root_path
end
end
2013-06-24 16:54:48 -04:00
```
2013-06-24 12:47:33 -04:00
2012-11-19 05:04:18 -05:00
## Manually retrieving policies and scopes
Sometimes you want to retrieve a policy for a record outside the controller or
view. For example when you delegate permissions from one policy to another.
You can easily retrieve policies and scopes like this:
``` ruby
Pundit.policy!(user, post)
Pundit.policy(user, post)
Pundit.policy_scope!(user, Post)
Pundit.policy_scope(user, Post)
```
The bang methods will raise an exception if the policy does not exist, whereas
those without the bang will return nil.
2014-01-29 10:57:55 -05:00
## Customize Pundit user
2013-07-12 23:42:34 -04:00
In some cases your controller might not have access to `current_user` , or your
2014-01-29 10:57:55 -05:00
`current_user` is not the method that should be invoked by Pundit. Simply
2013-07-12 23:42:34 -04:00
define a method in your controller called `pundit_user` .
```ruby
def pundit_user
User.find_by_other_means
end
```
2013-11-09 04:59:28 -05:00
## Strong parameters
2013-06-17 19:47:39 -04:00
2013-11-09 04:59:28 -05:00
In Rails 4 (or Rails 3.2 with the
[strong_parameters ](https://github.com/rails/strong_parameters ) gem),
mass-assignment protection is handled in the controller.
2014-01-09 13:16:40 -05:00
Pundit helps you permit different users to set different attributes. Don't
forget to provide your policy an instance of object or a class so correct
permissions could be loaded.
2013-06-17 19:47:39 -04:00
```ruby
2013-11-09 04:59:28 -05:00
# app/policies/post_policy.rb
class PostPolicy < ApplicationPolicy
2013-06-17 19:47:39 -04:00
def permitted_attributes
if user.admin? || user.owner_of?(post)
2013-11-09 04:59:28 -05:00
[:title, :body, :tag_list]
2013-06-17 19:47:39 -04:00
else
2013-11-09 04:59:28 -05:00
[:tag_list]
2013-06-17 19:47:39 -04:00
end
end
end
2013-11-09 04:59:28 -05:00
# app/controllers/posts_controller.rb
2013-06-17 19:47:39 -04:00
class PostsController < ApplicationController
def update
2013-11-09 04:59:28 -05:00
@post = Post.find(params[:id])
if @post .update(post_params)
redirect_to @post
else
render :edit
end
2013-06-17 19:47:39 -04:00
end
private
2013-11-09 04:59:28 -05:00
def post_params
2014-01-09 13:16:40 -05:00
params.require(:post).permit(*policy(@post || Post).permitted_attributes)
2013-06-17 19:47:39 -04:00
end
end
```
2013-03-27 09:10:12 -04:00
## RSpec
2013-10-24 21:30:55 -04:00
### Policy Specs
2013-03-27 09:10:12 -04:00
Pundit includes a mini-DSL for writing expressive tests for your policies in RSpec.
Require `pundit/rspec` in your `spec_helper.rb` :
``` ruby
require "pundit/rspec"
```
Then put your policy specs in `spec/policies` , and make them look somewhat like this:
``` ruby
describe PostPolicy do
subject { PostPolicy }
2014-01-30 16:08:01 -05:00
permissions :update? do
2013-03-27 09:10:12 -04:00
it "denies access if post is published" do
2013-12-26 12:25:56 -05:00
expect(subject).not_to permit(User.new(:admin => false), Post.new(:published => true))
2013-03-27 09:10:12 -04:00
end
it "grants access if post is published and user is an admin" do
2013-12-26 12:25:56 -05:00
expect(subject).to permit(User.new(:admin => true), Post.new(:published => true))
2013-03-27 09:10:12 -04:00
end
2013-04-12 15:48:27 -04:00
it "grants access if post is unpublished" do
2013-12-26 12:25:56 -05:00
expect(subject).to permit(User.new(:admin => false), Post.new(:published => false))
2013-03-27 09:10:12 -04:00
end
end
end
```
2013-06-18 06:57:46 -04:00
An alternative approach to Pundit policy specs is scoping them to a user context as outlined in this
2013-08-28 05:03:13 -04:00
[excellent post ](http://thunderboltlabs.com/blog/2013/03/27/testing-pundit-policies-with-rspec/ ).
2013-06-18 06:57:46 -04:00
2013-10-24 21:30:55 -04:00
### View Specs
When writing view specs, you'll notice that the policy helper is not available
and views under test that use it will fail. Thankfully, it's very easy to stub
out the policy to have it return whatever is appropriate for the spec.
``` ruby
describe "users/show" do
before(:each) do
user = assign(:user, build_stubbed(:user))
controller.stub(:current_user).and_return user
end
2013-11-09 04:59:28 -05:00
2013-10-24 21:30:55 -04:00
it "renders the destroy action" do
allow(view).to receive(:policy).and_return double(edit?: false, destroy?: true)
2013-11-09 04:59:28 -05:00
2013-10-24 21:30:55 -04:00
render
expect(rendered).to match 'Destroy'
end
end
```
This technique enables easy unit testing of tricky conditionaly view logic
based on what is or is not authorized.
# External Resources
- [Migrating to Pundit from CanCan ](http://blog.carbonfive.com/2013/10/21/migrating-to-pundit-from-cancan/ )
- [Testing Pundit Policies with RSpec ](http://thunderboltlabs.com/blog/2013/03/27/testing-pundit-policies-with-rspec/ )
2012-11-19 05:04:18 -05:00
# License
Licensed under the MIT license, see the separate LICENSE.txt file.