Simple Role based Authorization for Ruby on Rails

do_something.rb

# Roles Table
Role.create([{name:'admin'},{name:'staff'}])

# Users Roles Table
UserRole.create(user: User.find_by_username('corndog'), role: Role.find_by_name('admin'))

# User Model
class User < ApplicationRecord
  def role?(role)
    role.kind_of?(Array) ? role.map { |role| has_role?(role) }.any? : has_role?(role)
  end

  def has_role?(role)
    roles.any? { |r| r.name.underscore.to_sym == role }
  end
end

# Helper Method - Check One Role
current_user.role?(:admin)

# Helper Method - Check Two Roles
current_user.role?([:admin, :staff])