rake_task: Pass signature_root_dir only if set

Author: tk0miya

lib/rbs_rails/rake_task.rb

@@ -10,15 +10,15 @@ class RakeTask < Rake::TaskLib
 
     attr_accessor :ignore_model_if #: _Filter | nil
     attr_accessor :name #: Symbol
-    attr_writer :signature_root_dir #: Pathname
+    attr_writer :signature_root_dir #: Pathname?
 
     # @rbs name: ::Symbol
     # @rbs &block: (RbsRails::RakeTask) -> void
     def initialize(name = :rbs_rails, &block) #: void
       super()
 
       @name = name
-      @signature_root_dir = Rails.root / 'sig/rbs_rails'
+      @signature_root_dir = nil
 
       block.call(self) if block
 
@@ -30,7 +30,11 @@ def initialize(name = :rbs_rails, &block) #: void
     def def_all #: void
       desc 'Run all tasks of rbs_rails'
       task :"#{name}:all" do
-        sh "rbs_rails", "all", "--signature-root-dir=#{signature_root_dir}"
+        if signature_root_dir
+          sh "rbs_rails", "all", "--signature-root-dir=#{signature_root_dir}"
+        else
+          sh "rbs_rails", "all"
+        end
       end
     end
 
@@ -39,19 +43,29 @@ def def_generate_rbs_for_models #: void
       task :"#{name}:generate_rbs_for_models" do
         warn "ignore_model_if is deprecated." if ignore_model_if
 
-        sh "rbs_rails", "models", "--signature-root-dir=#{signature_root_dir}"
+        if signature_root_dir
+          sh "rbs_rails", "models", "--signature-root-dir=#{signature_root_dir}"
+        else
+          sh "rbs_rails", "models"
+        end
       end
     end
 
     def def_generate_rbs_for_path_helpers #: void
       desc 'Generate RBS files for path helpers'
       task :"#{name}:generate_rbs_for_path_helpers" do
-        sh "rbs_rails", "path_helpers", "--signature-root-dir=#{signature_root_dir}"
+        if signature_root_dir
+          sh "rbs_rails", "path_helpers", "--signature-root-dir=#{signature_root_dir}"
+        else
+          sh "rbs_rails", "path_helpers"
+        end
       end
     end
 
-    private def signature_root_dir #: Pathname
-      Pathname(@signature_root_dir)
+    private def signature_root_dir #: Pathname?
+      if path = @signature_root_dir
+        Pathname(path)
+      end
     end
   end
 end

sig/rbs_rails/rake_task.rbs

@@ -10,7 +10,7 @@ module RbsRails
 
     attr_accessor name: Symbol
 
-    attr_writer signature_root_dir: Pathname
+    attr_writer signature_root_dir: Pathname?
 
     # @rbs name: ::Symbol
     # @rbs &block: (RbsRails::RakeTask) -> void
@@ -22,6 +22,6 @@ module RbsRails
 
     def def_generate_rbs_for_path_helpers: () -> void
 
-    private def signature_root_dir: () -> Pathname
+    private def signature_root_dir: () -> Pathname?
   end
 end