Handle abstract setters

Signed-off-by: Alexandre Terrasa <alexandre.terrasa@shopify.com>

Author: Morriar

lib/spoom/sorbet/translate/sorbet_sigs_to_rbs_comments.rb

@@ -63,7 +63,12 @@ def visit_def_node(node)
             @rewriter << Source::Replace.new(
               node.rparen_loc&.end_offset || node.name_loc.end_offset,
               node.location.end_offset - 1,
-              " = raise NotImplementedError, \"Abstract method called\"",
+              if node.name.end_with?("=")
+                indent = " " * node.location.start_column
+                "\n#{indent}  raise NotImplementedError, \"Abstract method called\"\n#{indent}end"
+              else
+                " = raise NotImplementedError, \"Abstract method called\""
+              end,
             )
           end
         end

test/spoom/sorbet/translate/sorbet_sigs_to_rbs_comments_test.rb

@@ -86,6 +86,13 @@ def bar; end
               sig { abstract.params(x: Integer).void }
               def baz(x)
               end
+
+              sig { abstract.params(x: Integer).void }
+              def foo=(x); end
+
+              sig { abstract.params(x: Integer).void }
+              def bar=(x)
+              end
             end
           RB
 
@@ -104,6 +111,18 @@ def bar = raise NotImplementedError, "Abstract method called"
               # @abstract
               #: (Integer x) -> void
               def baz(x) = raise NotImplementedError, "Abstract method called"
+
+              # @abstract
+              #: (Integer x) -> void
+              def foo=(x)
+                raise NotImplementedError, "Abstract method called"
+              end
+
+              # @abstract
+              #: (Integer x) -> void
+              def bar=(x)
+                raise NotImplementedError, "Abstract method called"
+              end
             end
           RBS
         end