rbi 0.0.11 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 727196c54429957687b3624d4b5a895e526973b045689a7689edb9cd73c97ca9
-  data.tar.gz: 115a6efabd78950aa980d0b03e3b9910f474ad03cdf04f7a87b56ed38bdc30b7
+  metadata.gz: ff552cac7ef282e9673502662c239c6e60e370d4d9e6d137db16ed03bd3c5a32
+  data.tar.gz: 6ca736d5854621dde27bafffd8ad4b393d2c60d5f7b9e4eba8ab938e7f9123f4
 SHA512:
-  metadata.gz: 7357ca0bb3e8700a95d1ea6e285e710f3545a9016e36bc0a64fe0682da80328d5c645b4d6cd3d14bba14e7ccd6b9134c225944ce27625de6a26a995f5283d173
-  data.tar.gz: 71559ec6d855d46d3ac7bb074dbfd24fa024aaa6563dd241a7c286b66d138bf18f6c6e688ee33df99bdc240cbd6f683ebc068cb5c98b7e2b37dde6d4c9debca8
+  metadata.gz: 9540175cf17bd99421f4c0e2d81dae2a6b79a0ba3528409c1aeb0fbc4923eb01b2b9a5d6d28a17c1b7d0ac4a4b06f10775ecabc460f54ad39fa5099256554730
+  data.tar.gz: 75a8657c94d508c5b53793be7b53fe6570850c6088cd67fc75c85b5d42555824b248d6f2c7a62527e8e50555276eb4e32ba33bb406c948d66a1f0a11f7f16014

data/lib/rbi/formatter.rb

@@ -0,0 +1,66 @@
+# typed: strict
+# frozen_string_literal: true
+
+module RBI
+  class Formatter
+    extend T::Sig
+
+    sig { returns(T::Boolean) }
+    attr_accessor :add_sig_templates, :group_nodes, :nest_singleton_methods, :nest_non_public_methods, :sort_nodes
+
+    sig { returns(T.nilable(Integer)) }
+    attr_accessor :max_line_length
+
+    sig do
+      params(
+        add_sig_templates: T::Boolean,
+        group_nodes: T::Boolean,
+        max_line_length: T.nilable(Integer),
+        nest_singleton_methods: T::Boolean,
+        nest_non_public_methods: T::Boolean,
+        sort_nodes: T::Boolean
+      ).void
+    end
+    def initialize(
+      add_sig_templates: false,
+      group_nodes: false,
+      max_line_length: nil,
+      nest_singleton_methods: false,
+      nest_non_public_methods: false,
+      sort_nodes: false
+    )
+      @add_sig_templates = add_sig_templates
+      @group_nodes = group_nodes
+      @max_line_length = max_line_length
+      @nest_singleton_methods = nest_singleton_methods
+      @nest_non_public_methods = nest_non_public_methods
+      @sort_nodes = sort_nodes
+    end
+
+    sig { params(file: RBI::File).returns(String) }
+    def print_file(file)
+      format_file(file)
+      file.string(max_line_length: @max_line_length)
+    end
+
+    sig { params(tree: RBI::Tree).returns(String) }
+    def print_tree(tree)
+      format_tree(tree)
+      tree.string(max_line_length: @max_line_length)
+    end
+
+    sig { params(file: RBI::File).void }
+    def format_file(file)
+      format_tree(file.root)
+    end
+
+    sig { params(tree: RBI::Tree).void }
+    def format_tree(tree)
+      tree.add_sig_templates! if @add_sig_templates
+      tree.nest_singleton_methods! if @nest_singleton_methods
+      tree.nest_non_public_methods! if @nest_non_public_methods
+      tree.group_nodes! if @group_nodes
+      tree.sort_nodes! if @sort_nodes
+    end
+  end
+end

data/lib/rbi/model.rb

@@ -1026,7 +1026,7 @@ module RBI
     attr_accessor :return_type
 
     sig { returns(T::Boolean) }
-    attr_accessor :is_abstract, :is_override, :is_overridable
+    attr_accessor :is_abstract, :is_override, :is_overridable, :is_final
 
     sig { returns(T::Array[String]) }
     attr_reader :type_params
@@ -1041,6 +1041,7 @@ module RBI
         is_abstract: T::Boolean,
         is_override: T::Boolean,
         is_overridable: T::Boolean,
+        is_final: T::Boolean,
         type_params: T::Array[String],
         checked: T.nilable(Symbol),
         loc: T.nilable(Loc),
@@ -1053,6 +1054,7 @@ module RBI
       is_abstract: false,
       is_override: false,
       is_overridable: false,
+      is_final: false,
       type_params: [],
       checked: nil,
       loc: nil,
@@ -1064,6 +1066,7 @@ module RBI
       @is_abstract = is_abstract
       @is_override = is_override
       @is_overridable = is_overridable
+      @is_final = is_final
       @type_params = type_params
       @checked = checked
       block&.call(self)
@@ -1078,7 +1081,7 @@ module RBI
     def ==(other)
       return false unless other.is_a?(Sig)
       params == other.params && return_type == other.return_type && is_abstract == other.is_abstract &&
-        is_override == other.is_override && is_overridable == other.is_overridable &&
+        is_override == other.is_override && is_overridable == other.is_overridable && is_final == other.is_final &&
         type_params == other.type_params && checked == other.checked
     end
   end

data/lib/rbi/parser.rb

@@ -289,7 +289,14 @@ module RBI
         names = node.children[2..-1].map { |child| parse_name(child) }
         MixesInClassMethods.new(*names, loc: loc, comments: comments)
       when :public, :protected, :private
-        visibility = Visibility.new(method_name, loc: loc)
+        visibility = case method_name
+        when :protected
+          Protected.new(loc: loc)
+        when :private
+          Private.new(loc: loc)
+        else
+          Public.new(loc: loc)
+        end
         nested_node = node.children[2]
         case nested_node&.type
         when :def, :defs
@@ -561,7 +568,7 @@ module RBI
     sig { params(node: AST::Node).returns(Sig) }
     def self.build(node)
       v = SigBuilder.new
-      v.visit_all(node.children[2..-1])
+      v.visit_all(node.children)
       v.current
     end
 
@@ -588,6 +595,9 @@ module RBI
       visit(node.children[0]) if node.children[0]
       name = node.children[1]
       case name
+      when :sig
+        arg = node.children[2]
+        @current.is_final = arg && arg.children[0] == :final
       when :abstract
         @current.is_abstract = true
       when :override

data/lib/rbi/printer.rb

@@ -200,7 +200,7 @@ module RBI
       end
 
       lines.each do |line|
-        text = line.strip
+        text = line.rstrip
         v.printt("#")
         v.print(" #{text}") unless text.empty?
         v.printn
@@ -455,7 +455,7 @@ module RBI
 
     sig { returns(T::Array[String]) }
     def comments_lines
-      comments.flat_map { |comment| comment.text.lines.map(&:strip) }
+      comments.flat_map { |comment| comment.text.lines.map(&:rstrip) }
     end
   end
 
@@ -670,7 +670,9 @@ module RBI
 
     sig { params(v: Printer).void }
     def print_as_line(v)
-      v.printt("sig { ")
+      v.printt("sig")
+      v.print("(:final)") if is_final
+      v.print(" { ")
       sig_modifiers.each do |modifier|
         v.print("#{modifier}.")
       end
@@ -758,7 +760,7 @@ module RBI
 
     sig { returns(T::Array[String]) }
     def comments_lines
-      comments.flat_map { |comment| comment.text.lines.map(&:strip) }
+      comments.flat_map { |comment| comment.text.lines.map(&:rstrip) }
     end
   end
 

data/lib/rbi/rewriters/group_nodes.rb

@@ -47,6 +47,8 @@ module RBI
     sig { returns(Group::Kind) }
     def group_kind
       case self
+      when Group
+        kind
       when Include, Extend
         Group::Kind::Mixins
       when RequiresAncestor

data/lib/rbi/version.rb

@@ -2,5 +2,5 @@
 # frozen_string_literal: true
 
 module RBI
-  VERSION = "0.0.11"
+  VERSION = "0.0.12"
 end

data/lib/rbi.rb

@@ -23,4 +23,5 @@ require "rbi/rewriters/remove_known_definitions"
 require "rbi/rewriters/sort_nodes"
 require "rbi/parser"
 require "rbi/printer"
+require "rbi/formatter"
 require "rbi/version"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbi
 version: !ruby/object:Gem::Version
-  version: 0.0.11
+  version: 0.0.12
 platform: ruby
 authors:
 - Alexandre Terrasa
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-01-27 00:00:00.000000000 Z
+date: 2022-02-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ast
@@ -77,6 +77,7 @@ files:
 - README.md
 - Rakefile
 - lib/rbi.rb
+- lib/rbi/formatter.rb
 - lib/rbi/index.rb
 - lib/rbi/loc.rb
 - lib/rbi/model.rb