rbi 0.0.1 → 0.0.5

data/lib/rbi/rewriters/add_sig_templates.rb

@@ -0,0 +1,71 @@
+# typed: strict
+# frozen_string_literal: true
+
+module RBI
+  module Rewriters
+    class AddSigTemplates < Visitor
+      extend T::Sig
+
+      sig { params(with_todo_comment: T::Boolean).void }
+      def initialize(with_todo_comment: true)
+        super()
+        @with_todo_comment = with_todo_comment
+      end
+
+      sig { override.params(node: T.nilable(Node)).void }
+      def visit(node)
+        case node
+        when Tree
+          visit_all(node.nodes)
+        when Attr
+          add_attr_sig(node)
+        when Method
+          add_method_sig(node)
+        end
+      end
+
+      private
+
+      sig { params(attr: Attr).void }
+      def add_attr_sig(attr)
+        return unless attr.sigs.empty?
+        return if attr.names.size > 1
+
+        params = []
+        params << SigParam.new(attr.names.first.to_s, "T.untyped") if attr.is_a?(AttrWriter)
+
+        attr.sigs << Sig.new(
+          params: params,
+          return_type: "T.untyped"
+        )
+        add_todo_comment(attr)
+      end
+
+      sig { params(method: Method).void }
+      def add_method_sig(method)
+        return unless method.sigs.empty?
+
+        method.sigs << Sig.new(
+          params: method.params.map { |param| SigParam.new(param.name, "T.untyped") },
+          return_type: "T.untyped"
+        )
+        add_todo_comment(method)
+      end
+
+      sig { params(node: NodeWithComments).void }
+      def add_todo_comment(node)
+        node.comments << Comment.new("TODO: fill in signature with appropriate type information") if @with_todo_comment
+      end
+    end
+  end
+
+  class Tree
+    extend T::Sig
+
+    sig { params(with_todo_comment: T::Boolean).void }
+    def add_sig_templates!(with_todo_comment: true)
+      visitor = Rewriters::AddSigTemplates.new(with_todo_comment: with_todo_comment)
+      visitor.visit(self)
+    end
+  end
+end

data/lib/rbi/rewriters/group_nodes.rb

@@ -0,0 +1,104 @@
+# typed: strict
+# frozen_string_literal: true
+
+module RBI
+  module Rewriters
+    class GroupNodes < Visitor
+      extend T::Sig
+
+      sig { override.params(node: T.nilable(Node)).void }
+      def visit(node)
+        return unless node
+
+        case node
+        when Tree
+          kinds = node.nodes.map(&:group_kind)
+          kinds.compact!
+          kinds.uniq!
+
+          groups = {}
+          kinds.each { |kind| groups[kind] = Group.new(kind) }
+
+          node.nodes.dup.each do |child|
+            visit(child)
+            child.detach
+            groups[child.group_kind] << child
+          end
+
+          groups.each { |_, group| node << group }
+        end
+      end
+    end
+  end
+
+  class Tree
+    extend T::Sig
+
+    sig { void }
+    def group_nodes!
+      visitor = Rewriters::GroupNodes.new
+      visitor.visit(self)
+    end
+  end
+
+  class Node
+    extend T::Sig
+
+    sig { returns(Group::Kind) }
+    def group_kind
+      case self
+      when Include, Extend
+        Group::Kind::Mixins
+      when Helper
+        Group::Kind::Helpers
+      when TypeMember
+        Group::Kind::TypeMembers
+      when MixesInClassMethods
+        Group::Kind::MixesInClassMethods
+      when TStructField
+        Group::Kind::TStructFields
+      when TEnumBlock
+        Group::Kind::TEnums
+      when VisibilityGroup
+        Group::Kind::Methods
+      when Method
+        if name == "initialize"
+          Group::Kind::Inits
+        else
+          Group::Kind::Methods
+        end
+      when Scope, Const
+        Group::Kind::Consts
+      else
+        raise "Unknown group for #{self}"
+      end
+    end
+  end
+
+  class Group < Tree
+    extend T::Sig
+
+    sig { returns(Kind) }
+    attr_reader :kind
+
+    sig { params(kind: Kind).void }
+    def initialize(kind)
+      super()
+      @kind = kind
+    end
+
+    class Kind < T::Enum
+      enums do
+        Mixins              = new
+        Helpers             = new
+        TypeMembers         = new
+        MixesInClassMethods = new
+        TStructFields       = new
+        TEnums              = new
+        Inits               = new
+        Methods             = new
+        Consts              = new
+      end
+    end
+  end
+end