pundit.rb in stationed-0.1.0

- old
+ new

@@ -1,18 +1,27 @@
 module Stationed
   module Generators
     module Plugins
       module Pundit
+        def self.prepended(base)
+          base.class_option :pundit,
+            type: :boolean,
+            default: true,
+            desc: 'Include and configure Pundit for authorization'
+        end
+
         def finish_template
+          return super unless options[:pundit]
           gem 'pundit'
           copy_file 'pundit.rb', 'spec/support/pundit.rb'
           copy_file 'authorization.rb', 'app/controllers/concerns/authorization.rb'
           inject_into_class 'app/controllers/application_controller.rb', 'ApplicationController', "\n  include Authorization\n\n"
           super
         end
 
         def run_bundle
           super
+          return unless options[:pundit]
           generate 'pundit:install'
         end
       end
     end
   end