rbi 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 502c584fe293212d2f14f1a6b8d76457645afe9a4bc6355cd32c2179bb9a4e68
-  data.tar.gz: 4ed5b4fd62249dec73277ab31fc07595f1cdcfacdec5d6f13de9951ccffecc0c
+  metadata.gz: 88fce4a776d172e793f815ac8eee6c6958a140f0657532eb2d792f4caf6d18e1
+  data.tar.gz: ea01e4a38874e2e9045e8ea528ca595fef2c1da4178e527642cc81078cc31de0
 SHA512:
-  metadata.gz: 84333e72029b10d6a9ee7fc2a2392e08d6e9fc2d0871deec2a35e6e1042fde559efba0de2df008b91fcecfff2ba1b1b414275fb6a29c0a2cc52db62b08ae9915
-  data.tar.gz: 1162d80c92024e93af674629594c726bd2b5b965bebc63150f6384f488d0a24dcf7c52f2c56552aecec2010d26352b43fc88e3191cbfc2a042ea8ea64eca7e00
+  metadata.gz: 948f130aa780426a010b991355d2ea3c45bb175b714b97b19094ac52a369cf45be637589e590fa72a3fbebcfe96e58f77979f40e05c90702d60da9ffa90cc8f9
+  data.tar.gz: bdae9d96ba12457444d6f1073c006b242ca2e683a6414698eee0bfa1ae9da164cb2b5b25337d637d46c55bc989de0a00b4b990d40bdd3fb5d5a79053be677621

data/lib/rbi/rewriters/filter_versions.rb

@@ -0,0 +1,123 @@
+# typed: strict
+# frozen_string_literal: true
+
+module RBI
+  module Rewriters
+    # Take a gem version and filter out all RBI that is not relevant to that version based on @version annotations
+    # in comments. As an example:
+    #
+    # ~~~rb
+    # tree = Parser.parse_string(<<~RBI)
+    #   class Foo
+    #     # @version > 0.3.0
+    #     def bar
+    #     end
+    #
+    #     # @version <= 0.3.0
+    #     def bar(arg1)
+    #     end
+    #   end
+    # RBI
+    #
+    # Rewriters::FilterVersions.filter(tree, Gem::Version.new("0.3.1"))
+    #
+    # assert_equal(<<~RBI, tree.string)
+    #   class Foo
+    #     # @version > 0.3.0
+    #     def bar
+    #     end
+    #   end
+    # RBI
+    # ~~~
+    #
+    # Supported operators:
+    # - equals `=`
+    # - not equals `!=`
+    # - greater than `>`
+    # - greater than or equal to `>=`
+    # - less than `<`
+    # - less than or equal to `<=`
+    # - pessimistic or twiddle-wakka`~>`
+    #
+    # And/or logic:
+    # - "And" logic: put multiple versions on the same line
+    #   - e.g. `@version > 0.3.0, <1.0.0` means version must be greater than 0.3.0 AND less than 1.0.0
+    # - "Or" logic: put multiple versions on subsequent lines
+    #   - e.g. the following means version must be less than 0.3.0 OR greater than 1.0.0
+    #       ```
+    #       # @version < 0.3.0
+    #       # @version > 1.0.0
+    #       ```
+    # Prerelease versions:
+    # - Prerelease versions are considered less than their non-prerelease counterparts
+    #   - e.g. `0.4.0-prerelease` is less than `0.4.0`
+    #
+    # RBI with no versions:
+    # - RBI with no version annotations are automatically counted towards ALL versions
+    class FilterVersions < Visitor
+      extend T::Sig
+
+      VERSION_PREFIX = "version "
+
+      class << self
+        extend T::Sig
+
+        sig { params(tree: Tree, version: Gem::Version).void }
+        def filter(tree, version)
+          v = new(version)
+          v.visit(tree)
+        end
+      end
+
+      sig { params(version: Gem::Version).void }
+      def initialize(version)
+        super()
+        @version = version
+      end
+
+      sig { override.params(node: T.nilable(Node)).void }
+      def visit(node)
+        return unless node
+
+        unless node.satisfies_version?(@version)
+          node.detach
+          return
+        end
+
+        visit_all(node.nodes.dup) if node.is_a?(Tree)
+      end
+    end
+  end
+
+  class Node
+    sig { params(version: Gem::Version).returns(T::Boolean) }
+    def satisfies_version?(version)
+      return true unless is_a?(NodeWithComments)
+
+      requirements = version_requirements
+      requirements.empty? || requirements.any? { |req| req.satisfied_by?(version) }
+    end
+  end
+
+  class NodeWithComments
+    sig { returns(T::Array[Gem::Requirement]) }
+    def version_requirements
+      annotations.select do |annotation|
+        annotation.start_with?(Rewriters::FilterVersions::VERSION_PREFIX)
+      end.map do |annotation|
+        versions = annotation.delete_prefix(Rewriters::FilterVersions::VERSION_PREFIX).split(/, */)
+        Gem::Requirement.new(versions)
+      end
+    end
+  end
+
+  class Tree
+    extend T::Sig
+
+    sig { params(version: Gem::Version).void }
+    def filter_versions!(version)
+      visitor = Rewriters::FilterVersions.new(version)
+      visitor.visit(self)
+    end
+  end
+end

data/lib/rbi/version.rb

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

data/lib/rbi.rb

@@ -10,6 +10,7 @@ require "rbi/index"
 require "rbi/rewriters/add_sig_templates"
 require "rbi/rewriters/annotate"
 require "rbi/rewriters/deannotate"
+require "rbi/rewriters/filter_versions"
 require "rbi/rewriters/merge_trees"
 require "rbi/rewriters/nest_singleton_methods"
 require "rbi/rewriters/nest_non_public_methods"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbi
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - Alexandre Terrasa
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-02-06 00:00:00.000000000 Z
+date: 2024-02-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: prism
@@ -19,7 +19,7 @@ dependencies:
         version: 0.18.0
     - - "<"
       - !ruby/object:Gem::Version
-        version: '0.22'
+        version: '0.25'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -29,7 +29,7 @@ dependencies:
         version: 0.18.0
     - - "<"
       - !ruby/object:Gem::Version
-        version: '0.22'
+        version: '0.25'
 - !ruby/object:Gem::Dependency
   name: sorbet-runtime
   requirement: !ruby/object:Gem::Requirement
@@ -64,6 +64,7 @@ files:
 - lib/rbi/rewriters/add_sig_templates.rb
 - lib/rbi/rewriters/annotate.rb
 - lib/rbi/rewriters/deannotate.rb
+- lib/rbi/rewriters/filter_versions.rb
 - lib/rbi/rewriters/group_nodes.rb
 - lib/rbi/rewriters/merge_trees.rb
 - lib/rbi/rewriters/nest_non_public_methods.rb
@@ -92,7 +93,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.5
+rubygems_version: 3.5.6
 signing_key: 
 specification_version: 4
 summary: RBI generation framework