low_type 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 76a73e683c2e961c41eef7d14d5f41ac3e7252f9c22cc2b4b807c6ede92b6a89
-  data.tar.gz: 0a264a10f5c070c2b08160e0c94d69cf403c86a2961a0b72eab069b5e47c33af
+  metadata.gz: f39ec6a2c5d0986b75bd52a8bec8033b38bd5ba094515994fa655e253b47a669
+  data.tar.gz: f7aa54e8ce075b121a8e8fed025ea5a60e13a151a6891cb6aaedbe63432f8769
 SHA512:
-  metadata.gz: cd2e3b8caf03f7575d765eb573b66035706f2cc82fecea20fb806902e94aae5a7040d3eac4afdb3888b49359a035903e5b714a4662dd1cc4b25b28d79d40797b
-  data.tar.gz: 56c33ae15714520170d9cc85df710ce20b5a50dd1162705341938129ab95f0c6556baec3cd19c3371523c3f3994ed549d024a17937c83688ec91879e8e36e6b0
+  metadata.gz: 021eec0aae1a09604d45f5911647000f1620a0d1482cd743ee5066ae660b7e425af246a4806b7ef50763bc1dad3587631eeeea5e428abb7cc20308dec043d9f1
+  data.tar.gz: a238797c74dbe741db04a104ebad2f081d197bd5cea38dde2ae4ee5cb4a6735de183d1722b08229691d063c14e2f8ffd2662957fb19573f840bb19bc01cc4626

data/lib/redefiner.rb

@@ -0,0 +1,84 @@
+require_relative 'param_proxy'
+require_relative 'parser'
+require_relative 'type_expression'
+
+module LowType
+  class Redefiner
+    class << self
+      def redefine_methods(method_nodes:, private_start_line:, klass:)
+        Module.new do
+          method_nodes.each do |method_node|
+            args = method_node.parameters.slice
+
+            proxy_method = eval("-> (#{args}) {}")
+            required_args, required_kwargs = Redefiner.required_args(proxy_method)
+
+            klass.low_params[method_node.name] = Redefiner.type_expressions_from_params(proxy_method, args, required_args, required_kwargs)
+            next if klass.low_params[method_node.name].empty?
+
+            define_method(method_node.name) do |*args, **kwargs|
+              klass.low_params[method_node.name].each do |param_proxy|
+                arg = param_proxy.position ? args[param_proxy.position] : kwargs[param_proxy.name]
+                arg = param_proxy.type_expression.default_value if arg.nil? && param_proxy.type_expression.default_value != :LOW_TYPE_UNDEFINED
+                param_proxy.type_expression.validate!(arg:, name: param_proxy.name)
+                param_proxy.position ? args[param_proxy.position] = arg : kwargs[param_proxy.name] = arg
+              end
+
+              super(*args, **kwargs)
+            end
+
+            if private_start_line && method_node.start_line > private_start_line
+              private method_node.name
+            end
+          end
+        end
+      end
+
+      def required_args(proxy_method)
+        required_args = []
+        required_kwargs = {}
+
+        proxy_method.parameters.each do |param|
+          param_type, param_name = param
+
+          case param_type
+          when :req
+            required_args << nil
+          when :keyreq
+            required_kwargs[param_name] = nil
+          end
+        end
+
+        [required_args, required_kwargs]
+      end
+
+      def type_expressions_from_params(proxy_method, args, required_args, required_kwargs)
+        typed_method = eval(
+          <<~RUBY
+            -> (#{args}) {
+              param_proxies = []
+
+              proxy_method.parameters.each_with_index do |param, position|
+                type, name = param
+                position = nil unless [:opt, :req, :rest].include?(type)
+
+                expression = binding.local_variable_get(name)
+
+                if expression.class == TypeExpression
+                  param_proxies << ParamProxy.new(type_expression: expression, name:, type:, position:)
+                elsif ::LowType.type?(expression)
+                  param_proxies << ParamProxy.new(type_expression: TypeExpression.new(type: expression), name:, type:, position:)
+                end
+              end
+
+              param_proxies
+            }
+          RUBY
+        )
+
+        # Call method with only its required args to evaluate type expressions (which are stored as default values).
+        typed_method.call(*required_args, **required_kwargs)
+      end
+    end
+  end
+end

data/lib/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LowType
-  VERSION = '0.2.1'
+  VERSION = '0.2.2'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: low_type
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - maedi
@@ -20,6 +20,7 @@ files:
 - lib/low_type.rb
 - lib/param_proxy.rb
 - lib/parser.rb
+- lib/redefiner.rb
 - lib/type_expression.rb
 - lib/version.rb
 homepage: https://codeberg.org/low_ruby/low_type