feat: allow specifying which controller to override

Author: nicolastemciuc

lib/generators/devise/webauthn/controllers_generator.rb

@@ -16,10 +16,13 @@ class ControllersGenerator < Rails::Generators::Base
       source_root File.expand_path("templates/controllers", __dir__)
       argument :scope, required: true,
                        desc: "The scope to create controllers in, e.g. users, admins"
+      class_option :controllers, aliases: "-c", type: :array,
+        desc: "Select specific controllers to generate (#{CONTROLLERS.join(', ')})"
 
       def create_controllers
         @scope_prefix = scope.blank? ? "" : "#{scope.camelize}::"
-        CONTROLLERS.each do |name|
+        controllers = options[:controllers] || CONTROLLERS
+        controllers.each do |name|
           template "#{name}_controller.rb",
                    "app/controllers/#{scope}/#{name}_controller.rb"
         end

spec/generators/devise/webauthn/controllers_generator_spec.rb

@@ -34,4 +34,30 @@
       assert_file "app/controllers/users/two_factor_authentications_controller.rb", /Users::TwoFactorAuthenticationsController/
     end
   end
+
+  context "when using -c to select specific controllers" do
+    let(:generator_instance) do
+      generator(["users"], controllers: %w[passkeys two_factor_authentications])
+    end
+
+    it "generates only the selected controllers" do
+      invoke generator_instance
+
+      assert_file "app/controllers/users/passkeys_controller.rb", /Users::PasskeysController/
+      assert_file "app/controllers/users/two_factor_authentications_controller.rb",
+                  /Users::TwoFactorAuthenticationsController/
+
+      assert_no_file "app/controllers/users/second_factor_webauthn_credentials_controller.rb"
+    end
+  end
+
+  context "when -c includes an unknown controller" do
+    let(:generator_instance) do
+      generator(["users"], controllers: %w[passkeys not_a_real_controller])
+    end
+
+    it "raises an error" do
+      expect { invoke generator_instance }.to raise_error(Thor::Error)
+    end
+  end
 end