rbi 0.2.4 → 0.3.0

data/lib/rbi/rewriters/sort_nodes.rb

@@ -4,9 +4,8 @@
 module RBI
   module Rewriters
     class SortNodes < Visitor
-      extend T::Sig
-
-      sig { override.params(node: T.nilable(Node)).void }
+      # @override
+      #: (Node? node) -> void
       def visit(node)
         sort_node_names!(node) if node
 
@@ -41,7 +40,7 @@ module RBI
 
       private
 
-      sig { params(node: Node).returns(Integer) }
+      #: (Node node) -> Integer
       def node_rank(node)
         case node
         when Group                then group_rank(node.kind)
@@ -69,7 +68,7 @@ module RBI
         end
       end
 
-      sig { params(kind: Group::Kind).returns(Integer) }
+      #: (Group::Kind kind) -> Integer
       def group_rank(kind)
         case kind
         when Group::Kind::Mixins              then 0
@@ -90,7 +89,7 @@ module RBI
         end
       end
 
-      sig { params(node: Node).returns(T.nilable(String)) }
+      #: (Node node) -> String?
       def node_name(node)
         case node
         when Module, Class, Struct, Const, Method, Helper, RequiresAncestor
@@ -102,7 +101,7 @@ module RBI
         end
       end
 
-      sig { params(node: Node).void }
+      #: (Node node) -> void
       def sort_node_names!(node)
         case node
         when Attr
@@ -113,9 +112,7 @@ module RBI
   end
 
   class Tree
-    extend T::Sig
-
-    sig { void }
+    #: -> void
     def sort_nodes!
       visitor = Rewriters::SortNodes.new
       visitor.visit(self)

data/lib/rbi/rewriters/translate_rbs_sigs.rb

@@ -0,0 +1,87 @@
+# typed: strict
+# frozen_string_literal: true
+
+module RBI
+  module Rewriters
+    # Translate all RBS signature comments to Sorbet RBI signatures
+    class TranslateRBSSigs < Visitor
+      class Error < RBI::Error; end
+
+      # @override
+      #: (Node? node) -> void
+      def visit(node)
+        return unless node
+
+        case node
+        when Tree
+          visit_all(node.nodes)
+        when AttrAccessor, AttrReader, AttrWriter
+          rbs_comments = extract_rbs_comments(node)
+          rbs_comments.each do |comment|
+            node.sigs << translate_rbs_attr_type(node, comment)
+          end
+        when Method
+          rbs_comments = extract_rbs_comments(node)
+          rbs_comments.each do |comment|
+            node.sigs << translate_rbs_method_type(node, comment)
+          end
+        end
+      end
+
+      private
+
+      #: (Method | Attr) -> Array[RBSComment]
+      def extract_rbs_comments(node)
+        comments = node.comments.dup
+        node.comments.clear
+
+        rbs_sigs = T.let([], T::Array[RBSComment])
+
+        comments.each do |comment|
+          case comment
+          when RBSComment
+            rbs_sigs << comment
+          else
+            node.comments << comment
+          end
+        end
+
+        rbs_sigs
+      end
+
+      #: (Method, RBSComment) -> Sig
+      def translate_rbs_method_type(node, comment)
+        method_type = ::RBS::Parser.parse_method_type(comment.text)
+        translator = RBS::MethodTypeTranslator.new(node)
+        translator.visit(method_type)
+        translator.result
+      end
+
+      #: (Attr, RBSComment) -> Sig
+      def translate_rbs_attr_type(node, comment)
+        attr_type = ::RBS::Parser.parse_type(comment.text)
+        sig = Sig.new
+
+        if node.is_a?(AttrWriter)
+          if node.names.size != 1
+            raise Error, "AttrWriter must have exactly one name"
+          end
+
+          name = T.must(node.names.first)
+          sig.params << SigParam.new(name.to_s, RBS::TypeTranslator.translate(attr_type))
+        end
+
+        sig.return_type = RBS::TypeTranslator.translate(attr_type)
+        sig
+      end
+    end
+  end
+
+  class Tree
+    #: -> void
+    def translate_rbs_sigs!
+      visitor = Rewriters::TranslateRBSSigs.new
+      visitor.visit(self)
+    end
+  end
+end