Module: Rigor::Inference::MethodDispatcher
- Defined in:
- lib/rigor/inference/method_dispatcher.rb,
lib/rigor/inference/method_dispatcher/cgi_folding.rb,
lib/rigor/inference/method_dispatcher/set_folding.rb,
lib/rigor/inference/method_dispatcher/uri_folding.rb,
lib/rigor/inference/method_dispatcher/call_context.rb,
lib/rigor/inference/method_dispatcher/data_folding.rb,
lib/rigor/inference/method_dispatcher/file_folding.rb,
lib/rigor/inference/method_dispatcher/math_folding.rb,
lib/rigor/inference/method_dispatcher/rbs_dispatch.rb,
lib/rigor/inference/method_dispatcher/time_folding.rb,
lib/rigor/inference/method_dispatcher/block_folding.rb,
lib/rigor/inference/method_dispatcher/method_folding.rb,
lib/rigor/inference/method_dispatcher/reduce_folding.rb,
lib/rigor/inference/method_dispatcher/regexp_folding.rb,
lib/rigor/inference/method_dispatcher/shape_dispatch.rb,
lib/rigor/inference/method_dispatcher/kernel_dispatch.rb,
lib/rigor/inference/method_dispatcher/constant_folding.rb,
lib/rigor/inference/method_dispatcher/iterator_dispatch.rb,
lib/rigor/inference/method_dispatcher/overload_selector.rb,
lib/rigor/inference/method_dispatcher/receiver_affinity.rb,
lib/rigor/inference/method_dispatcher/singleton_folding.rb,
lib/rigor/inference/method_dispatcher/array_to_h_folding.rb,
lib/rigor/inference/method_dispatcher/shellwords_folding.rb,
lib/rigor/inference/method_dispatcher/literal_string_folding.rb
Overview
Coordinates method dispatch for the inference engine.
Given ‘(receiver_type, method_name, arg_types, block_type, environment)`, the dispatcher returns the inferred result type or `nil` when no rule matches. `nil` is a deliberately blunt “I don’t know” signal: callers (today only ‘ExpressionTyper`) own the fail-soft fallback and decide whether to record a `FallbackTracer` event.
Tiers (in order):
-
ConstantFolding: executes the Ruby operation directly when the receiver and argument are ‘Constant` carriers and the method is on the curated whitelist. Slice 2.
-
ShapeDispatch: returns the precise element/value type for a curated catalogue of ‘Tuple`/`HashShape` element-access methods (`first`, `last`, `[]` with a static integer/key, `fetch`, `dig`, `size`/`length`/`count`). Slice 5 phase 2.
-
RbsDispatch: looks up the receiver’s class in the RBS environment carried by the scope and translates the method’s return type into a Rigor::Type. Slice 4.
‘ShapeDispatch` deliberately runs above RbsDispatch so the precise per-position/per-key answer wins over the projected `Array#[]`/`Hash#fetch` answer; it falls through (`nil`) when the call cannot be proved against the static shape, in which case the projection answer from RbsDispatch applies.
The dispatcher’s public signature reserves space for ‘block_type:` and ADR-2 plugin extensions (later slices), so call sites added now do not have to be rewritten when those tiers arrive.
Defined Under Namespace
Modules: ArrayToHFolding, BlockFolding, CGIFolding, ConstantFolding, DataFolding, FileFolding, IteratorDispatch, KernelDispatch, LiteralStringFolding, MathFolding, MethodFolding, OverloadSelector, RbsDispatch, ReceiverAffinity, ReduceFolding, RegexpFolding, SetFolding, ShapeDispatch, ShellwordsFolding, SingletonFolding, TimeFolding, URIFolding Classes: CallContext
Class Method Summary collapse
- .array_new_fill(type) ⇒ Object
- .array_new_lift(class_name, arg_types) ⇒ Object
- .array_new_size(type) ⇒ Object
-
.boundary_cross_class_name(receiver_type, environment, rbs_result) ⇒ Object
Composite preflight for #record_boundary_cross_if_applicable.
- .budget_silence_result(class_name, index, _environment) ⇒ Object
-
.class_new_lift(class_name, arg_types) ⇒ Object
‘Class.new` and `Class.new(Parent)` create a brand-new anonymous class.
-
.collect_gated_contributions(index, relevant, name, call_node, scope, receiver_type) ⇒ Object
The post-gate walk, in registry order — the same order the ungated walk used.
-
.collect_plugin_contributions(registry, call_node, scope, receiver_type) ⇒ Object
Collects every plugin’s flow / dynamic-return contribution for one call site.
- .constant_constructor_lift(class_name, arg_types) ⇒ Object
- .constant_metaclass(value) ⇒ Object
- .date_new_lift(class_name, arg_types) ⇒ Object
- .dep_source_class_name(receiver_type) ⇒ Object
-
.dependency_source_return_type(contribution) ⇒ Object
Maps a DependencySourceInference::Walker::CatalogEntry to the Type the dispatcher returns at the call site.
-
.discovered_method_lookup(receiver_type) ⇒ Object
Resolves the ‘(class_name, kind)` pair scope_indexer keys its `discovered_methods` table on.
-
.dispatch(receiver_type:, method_name:, arg_types:, block_type: nil, environment: nil, call_node: nil, scope: nil) ⇒ Rigor::Type?
Inferred result type, or ‘nil` for “no rule”.
- .dispatch_precise_tiers(context) ⇒ Object
-
.expected_block_param_types(receiver_type:, method_name:, arg_types:, environment: nil) ⇒ Array<Rigor::Type>
Returns the positional block parameter types declared by the receiving method’s selected RBS overload, translated into ‘Rigor::Type`.
-
.explicit_non_self_receiver?(call_node) ⇒ Boolean
True when the call node carries an explicit receiver that is not the literal ‘self`.
- .hash_new_lift(class_name, arg_types) ⇒ Object
- .meta_class(receiver_type) ⇒ Object
- .meta_new(receiver_type, arg_types = []) ⇒ Object
-
.plugin_owns_receiver?(class_name, environment) ⇒ Boolean
ADR-52 WD1 — the per-dispatch plugins × owns_receivers × ‘class_ordering` walk moved into the compiled contribution table: the union is built once per registry (almost always empty → O(1) false) and per-class verdicts memoise per run.
-
.promote_synthetic_match(matches, method_name, arg_types, block_type, environment) ⇒ Object
First non-nil promotion wins.
-
.promote_via_origin_module(synthetic, method_name, arg_types, block_type, environment) ⇒ Object
Slice 6a-TierB.
- .promote_via_return_type(synthetic, environment) ⇒ Object
-
.range_new_excl(excl_type) ⇒ Object
Resolves the optional ‘exclude_end` argument for `range_new_lift`.
- .range_new_lift(class_name, arg_types) ⇒ Object
- .rbs_display_for(rbs_result) ⇒ Object
- .rbs_result_untyped?(rbs_result) ⇒ Boolean
-
.record_boundary_cross_if_applicable(receiver_type, method_name, rbs_result, environment) ⇒ Object
ADR-10 slice 5c — record a ‘dynamic.dependency-source.boundary-cross` event when RBS dispatch resolves a call AND the receiver class belongs to a `mode: :full` opt-in gem whose Walker also catalogued the same `(class_name, method_name)`.
- .regexp_new_lift(class_name, arg_types) ⇒ Object
-
.resolve(receiver_type:, method_name:, arg_types:, block_type: nil, environment: nil, call_node: nil, scope: nil) ⇒ Object
rubocop:disable Metrics/MethodLength, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity.
- .set_new_lift(class_name, arg_types) ⇒ Object
- .static_refinement_class_for(receiver_type) ⇒ Object
-
.static_refinement_owner_for(receiver_type, candidates) ⇒ Object
Picks the most specific override owner the receiver honours.
-
.struct_new_lift(class_name, arg_types) ⇒ Object
‘Struct.new(:a, :b)` synthesises an anonymous Struct subclass (a class object), not a Struct instance — so the chained idiom `Struct.new(:a, :b).new(1, 2)` must resolve `.new` again on a class-like carrier.
- .synthetic_method_class_name(receiver_type) ⇒ Object
- .try_dependency_source(receiver_type, method_name, environment) ⇒ Object
-
.try_discovered_method(receiver_type, method_name, scope) ⇒ Object
v0.1.3 — discovered-method dispatch tier.
-
.try_hkt_builtin_return(receiver_type, method_name, arg_types, environment) ⇒ Object
ADR-2 § “Flow Contribution Bundle” / v0.1.1 Track 2 slice 7; ADR-52 WD3 — consults each loaded plugin’s gated ‘dynamic_return` rules, wraps the contributed types as `FlowContribution` bundles, merges them through `FlowContribution::Merger`, and returns the merged `return_type` slot (or nil when no plugin contributed a return type).
-
.try_meta_introspection(receiver_type, method_name, arg_types = []) ⇒ Object
Slice 7 phase 8 — meta-introspection shortcuts.
- .try_plugin_contribution(call_node, scope, receiver_type) ⇒ Object
-
.try_project_patched_method(receiver_type, method_name, environment) ⇒ Object
ADR-17 slice 2 — project-side patched-method tier.
-
.try_static_refinement(receiver_type, method_name, arg_types) ⇒ Object
Consults the Rigor-bundled static refinement table for a (owner-class, method-name, kind) entry.
-
.try_synthesized_stub_type(receiver_type, environment) ⇒ Object
ADR-5 robustness — returns ‘Dynamic` when the receiver is an instance or singleton of a type Rigor synthesized (a missing-namespace module or a referenced-type stub).
-
.try_synthetic_method(receiver_type, method_name, arg_types, block_type, environment) ⇒ Object
ADR-10 slice 2b-ii.
- .try_user_class_fallback(receiver_type, environment, call_node, context) ⇒ Object
- .user_class_fallback_receiver(receiver_type, environment) ⇒ Object
Class Method Details
.array_new_fill(type) ⇒ Object
1091 1092 1093 1094 1095 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1091 def array_new_fill(type) return Type::Combinator.constant_of(nil) if type.nil? type end |
.array_new_lift(class_name, arg_types) ⇒ Object
1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1074 def array_new_lift(class_name, arg_types) return nil unless class_name == "Array" return nil if arg_types.empty? || arg_types.size > 2 size = array_new_size(arg_types.first) return nil if size.nil? || size.negative? || size > ARRAY_NEW_TUPLE_LIMIT fill = array_new_fill(arg_types[1]) Type::Combinator.tuple_of(*Array.new(size, fill)) end |
.array_new_size(type) ⇒ Object
1085 1086 1087 1088 1089 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1085 def array_new_size(type) return nil unless type.is_a?(Type::Constant) && type.value.is_a?(Integer) type.value end |
.boundary_cross_class_name(receiver_type, environment, rbs_result) ⇒ Object
Composite preflight for #record_boundary_cross_if_applicable. Returns the receiver class name only when every prerequisite for emitting the diagnostic is satisfied (environment carries an index + reporter, receiver is a nominal carrier, RBS-side result is not the trivial ‘Dynamic` envelope). Returns `nil` to short-circuit otherwise.
645 646 647 648 649 650 651 652 653 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 645 def boundary_cross_class_name(receiver_type, environment, rbs_result) return nil if environment.nil? return nil if environment.dependency_source_index.nil? return nil if environment.dependency_source_index.empty? return nil if environment.boundary_cross_reporter.nil? return nil if rbs_result_untyped?(rbs_result) dep_source_class_name(receiver_type) end |
.budget_silence_result(class_name, index, _environment) ⇒ Object
665 666 667 668 669 670 671 672 673 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 665 def budget_silence_result(class_name, index, _environment) return nil unless index.budget_overrun_strategy == :dependency_silence owning_gem = index.gem_for(class_name) return nil if owning_gem.nil? return nil unless index.budget_exceeded.include?(owning_gem) Type::Combinator.untyped end |
.class_new_lift(class_name, arg_types) ⇒ Object
‘Class.new` and `Class.new(Parent)` create a brand-new anonymous class. Statically that class is representable as the parent’s singleton type — its singleton-method surface is the parent’s (plus whatever the block defines, which we do not statically track here), so ‘Singleton` lets downstream `klass.some_class_method` resolve. No parent →`singleton(Object)`. Anything else (dynamic parent, more than one positional, …) falls back to `Nominal` via the surrounding `meta_new` tail.
1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1020 def class_new_lift(class_name, arg_types) return nil unless class_name == "Class" return Type::Combinator.singleton_of("Object") if arg_types.empty? return nil unless arg_types.size == 1 parent = arg_types.first return parent if parent.is_a?(Type::Singleton) nil end |
.collect_gated_contributions(index, relevant, name, call_node, scope, receiver_type) ⇒ Object
The post-gate walk, in registry order — the same order the ungated walk used.
741 742 743 744 745 746 747 748 749 750 751 752 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 741 def collect_gated_contributions(index, relevant, name, call_node, scope, receiver_type) result = nil relevant.each do |plugin| next unless index.dynamic_candidate_for?(plugin, name) dynamic = plugin.dynamic_return_type(call_node: call_node, scope: scope, receiver_type: receiver_type) (result ||= []) << FlowContribution.new(return_type: dynamic) if dynamic rescue StandardError next end result || EMPTY_CONTRIBUTIONS end |
.collect_plugin_contributions(registry, call_node, scope, receiver_type) ⇒ Object
Collects every plugin’s flow / dynamic-return contribution for one call site. Two prunings keep this off the hot path on plugin-heavy projects (it was the #1 allocation site and a top CPU cost):
-
Only the plugins that structurally implement a per-call path are visited — ‘registry.contribution_index.for_method_dispatch` is the registry-ordered subset declaring a `dynamic_return`. Iterating the subset in registry order, and gating each path by membership, yields the exact same contributions in the same order as visiting every plugin would (a skipped plugin’s call returns nil/[] anyway). The receiver-class ancestry match still happens per dispatch inside ‘dynamic_return_type`.
-
Contributions accumulate lazily — allocate only when one actually appears, and share a frozen empty array otherwise. The caller treats the result as read-only (‘.empty?` / `Merger.merge`).
-
ADR-52 WD1 — method-name gates compiled at registry build. The global gate makes the common “no plugin cares about this call” case a single Set probe; the per-plugin gate skips a plugin whose ‘dynamic_return` rules are all `methods:`-gated on other names. A pruned consultation could only have returned nil, so contribution order and content are unchanged.
721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 721 def collect_plugin_contributions(registry, call_node, scope, receiver_type) index = registry.contribution_index relevant = index.for_method_dispatch return EMPTY_CONTRIBUTIONS if relevant.empty? # `call_node` is not always a CallNode — the `&:symbol` block # path dispatches with the `Prism::BlockArgumentNode` itself # (`ExpressionTyper#symbol_block_return_type`). A bare `.name` # here raised, and the raise was silently absorbed by # `block_return_type_for`'s rescue, nil-ing the block type and # flipping `select(&:p)`-style calls onto their no-block # Enumerator overloads (caught by the GitLab corpus gate). name = call_node.respond_to?(:name) ? call_node.name : nil return EMPTY_CONTRIBUTIONS unless index.dispatch_candidate?(name) collect_gated_contributions(index, relevant, name, call_node, scope, receiver_type) end |
.constant_constructor_lift(class_name, arg_types) ⇒ Object
1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1052 def constant_constructor_lift(class_name, arg_types) builder = CONSTANT_CONSTRUCTORS[class_name] return nil if builder.nil? return nil unless arg_types.size == 1 arg = arg_types.first return nil unless arg.is_a?(Type::Constant) && arg.value.is_a?(String) result = builder.call(arg.value) Type::Combinator.constant_of(result) rescue StandardError nil end |
.constant_metaclass(value) ⇒ Object
1245 1246 1247 1248 1249 1250 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1245 def (value) CONSTANT_METACLASSES.each do |klass, name| return Type::Combinator.singleton_of(name) if value.is_a?(klass) end nil end |
.date_new_lift(class_name, arg_types) ⇒ Object
1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1223 def date_new_lift(class_name, arg_types) return nil unless DATE_NEW_CLASSES.include?(class_name) return nil unless arg_types.size.between?(1, 8) return nil unless arg_types.all?(Type::Constant) values = arg_types.map(&:value) return nil unless values.all? { |v| v.is_a?(Integer) || v.is_a?(Rational) || v.is_a?(String) } klass = class_name == "Date" ? Date : DateTime Type::Combinator.constant_of(klass.new(*values)) rescue StandardError nil end |
.dep_source_class_name(receiver_type) ⇒ Object
686 687 688 689 690 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 686 def dep_source_class_name(receiver_type) case receiver_type when Type::Nominal, Type::Singleton then receiver_type.class_name end end |
.dependency_source_return_type(contribution) ⇒ Object
633 634 635 636 637 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 633 def dependency_source_return_type(contribution) return Type::Combinator.untyped if contribution.return_type.nil? Type::Combinator.dynamic(contribution.return_type) end |
.discovered_method_lookup(receiver_type) ⇒ Object
296 297 298 299 300 301 302 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 296 def discovered_method_lookup(receiver_type) case receiver_type when Type::Nominal then [receiver_type.class_name, :instance] when Type::Singleton then [receiver_type.class_name, :singleton] else [nil, nil] end end |
.dispatch(receiver_type:, method_name:, arg_types:, block_type: nil, environment: nil, call_node: nil, scope: nil) ⇒ Rigor::Type?
Returns inferred result type, or ‘nil` for “no rule”.
79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 79 def dispatch(receiver_type:, method_name:, arg_types:, block_type: nil, environment: nil, call_node: nil, scope: nil) result = resolve( receiver_type: receiver_type, method_name: method_name, arg_types: arg_types, block_type: block_type, environment: environment, call_node: call_node, scope: scope ) # `rigor trace` — record the dispatch outcome (resolved type, or # the fail-soft `nil` the caller widens to `Dynamic[Top]`). if FlowTracer.active? FlowTracer.dispatch( receiver: receiver_type, method_name: method_name, args: arg_types, result: result, location: call_node&.location ) end result end |
.dispatch_precise_tiers(context) ⇒ Object
808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 808 def dispatch_precise_tiers(context) # ADR-48 — Data value folding runs ahead of meta-introspection: # `meta_new` intercepts every `Singleton[*].new` (returning # `Nominal`), which would mask a `Data` class's precise instance. # The tier only fires on Data receivers (a `Data.define`, a # `DataClass`/`DataInstance`, or a `Singleton` with a recorded # member layout), so it never shadows meta's Array/Set/Range lifts. data_result = DataFolding.try_dispatch(context) return data_result if data_result = (context.receiver, context.method_name, context.args) return if PRECISE_TIERS_HEAD.each do |tier| result = tier.try_dispatch(context) return result if result end receiver = context.receiver if receiver.is_a?(Type::Singleton) && (folder = STDLIB_SINGLETON_FOLDERS[receiver.class_name]) result = folder.try_dispatch(context) return result if result end PRECISE_TIERS_TAIL.each do |tier| result = tier.try_dispatch(context) return result if result end nil end |
.expected_block_param_types(receiver_type:, method_name:, arg_types:, environment: nil) ⇒ Array<Rigor::Type>
Returns the positional block parameter types declared by the receiving method’s selected RBS overload, translated into ‘Rigor::Type`. Used by the StatementEvaluator’s CallNode handler to bind block parameter names before evaluating the block body.
The probe is best-effort: it returns an empty array whenever the receiver, environment, method definition, or selected overload does not provide statically declared block parameter types. Callers MUST treat the empty array as “no information”; the binder falls back to ‘Dynamic` for every parameter slot in that case.
1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1270 def expected_block_param_types(receiver_type:, method_name:, arg_types:, environment: nil) return [] if receiver_type.nil? context = CallContext.build( receiver: receiver_type, method_name: method_name, args: arg_types, environment: environment ) iterator_result = IteratorDispatch.block_param_types(context) return iterator_result if iterator_result RbsDispatch.block_param_types(context) end |
.explicit_non_self_receiver?(call_node) ⇒ Boolean
True when the call node carries an explicit receiver that is not the literal ‘self`. Such a call cannot legally dispatch to a private method, so the user-class fallback must skip private signatures rather than return a confidently-wrong type. Returns false for implicit-self calls and `self.`-receiver calls (both may legally reach a private method in modern Ruby), and false when no `call_node` is supplied (internal dispatcher callers).
880 881 882 883 884 885 886 887 888 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 880 def explicit_non_self_receiver?(call_node) return false if call_node.nil? return false unless call_node.respond_to?(:receiver) receiver = call_node.receiver return false if receiver.nil? !receiver.is_a?(Prism::SelfNode) end |
.hash_new_lift(class_name, arg_types) ⇒ Object
‘Hash.new(default)` — lifts the default value’s type into the Hash’s value parameter so a subsequent ‘h` read surfaces the default type rather than `Dynamic`. The common counter idiom `h = Hash.new(0); h += 1` then types the read as `Integer`. The key parameter is left `untyped` (the default carrier imposes no key constraint), so reads of any key resolve through the value parameter. A value-pinned `Constant` default (`0`) is widened to its nominal (`Integer`): the hash’s values mutate over its lifetime, so pinning the parameter to the literal would be unsound for the aggregate.
Only the single-argument default form folds. The zero-arg (‘Hash.new`) and the block form (`Hash.new { |h, k| … }`) keep the bare `Nominal` answer — the block’s value type is not available at this ‘:new` dispatch site, and conservatively leaving the read as today’s behaviour is precision-additive.
1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1113 def hash_new_lift(class_name, arg_types) return nil unless class_name == "Hash" return nil unless arg_types.size == 1 default = arg_types.first return nil if default.nil? value = Type::Combinator.widen_value_pinned(default) Type::Combinator.nominal_of("Hash", type_args: [Type::Combinator.untyped, value]) end |
.meta_class(receiver_type) ⇒ Object
930 931 932 933 934 935 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 930 def (receiver_type) case receiver_type when Type::Nominal then Type::Combinator.singleton_of(receiver_type.class_name) when Type::Constant then (receiver_type.value) end end |
.meta_new(receiver_type, arg_types = []) ⇒ Object
‘Singleton.new` returns `Nominal` (a fresh instance), regardless of whether Foo is in RBS. This short-circuits the Class.new generic-`instance` plumbing for user classes, so a discovered-class `ScanAccumulator.new` types as `Nominal` rather than `Class`.
v0.0.7 — for the curated set of immutable scalar-shaped classes that ‘Type::Constant::SCALAR_CLASSES` accepts (today: `Pathname`), `.new(Constant<…>)` lifts to a `Constant<…>` carrier so downstream method calls fold through the standard catalog tier.
949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 949 def (receiver_type, arg_types = []) return nil unless receiver_type.is_a?(Type::Singleton) constant_lift = constant_constructor_lift(receiver_type.class_name, arg_types) return constant_lift if constant_lift array_lift = array_new_lift(receiver_type.class_name, arg_types) return array_lift if array_lift range_lift = range_new_lift(receiver_type.class_name, arg_types) return range_lift if range_lift set_lift = set_new_lift(receiver_type.class_name, arg_types) return set_lift if set_lift hash_lift = hash_new_lift(receiver_type.class_name, arg_types) return hash_lift if hash_lift regexp_lift = regexp_new_lift(receiver_type.class_name, arg_types) return regexp_lift if regexp_lift date_lift = date_new_lift(receiver_type.class_name, arg_types) return date_lift if date_lift struct_new_lift = struct_new_lift(receiver_type.class_name, arg_types) return struct_new_lift if struct_new_lift class_new_lift = class_new_lift(receiver_type.class_name, arg_types) return class_new_lift if class_new_lift Type::Combinator.nominal_of(receiver_type.class_name) end |
.plugin_owns_receiver?(class_name, environment) ⇒ Boolean
ADR-52 WD1 — the per-dispatch plugins × owns_receivers בclass_ordering` walk moved into the compiled contribution table: the union is built once per registry (almost always empty → O(1) false) and per-class verdicts memoise per run.
679 680 681 682 683 684 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 679 def plugin_owns_receiver?(class_name, environment) registry = environment&.plugin_registry return false if registry.nil? || registry.empty? registry.contribution_index.owns_receiver?(class_name, environment) end |
.promote_synthetic_match(matches, method_name, arg_types, block_type, environment) ⇒ Object
First non-nil promotion wins. Tier B (origin_module) and Tier C (return_type nominal lookup) are tried in the same registration-order pass per WD11 first-wins —the slice-3b scanner sets ‘origin_module` for Tier B entries and leaves it absent for Tier C, so the two paths self-route per match.
482 483 484 485 486 487 488 489 490 491 492 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 482 def promote_synthetic_match(matches, method_name, arg_types, block_type, environment) return nil if environment.nil? matches.each do |synthetic| promoted = promote_via_origin_module(synthetic, method_name, arg_types, block_type, environment) || promote_via_return_type(synthetic, environment) return promoted if promoted end nil end |
.promote_via_origin_module(synthetic, method_name, arg_types, block_type, environment) ⇒ Object
Slice 6a-TierB. For Tier B emissions (origin_module recorded in provenance), redispatch the call on the included module’s ‘Nominal` type via `RbsDispatch`. Returns nil when the SyntheticMethod is not a Tier B entry or when the origin_module is not in the RBS env.
499 500 501 502 503 504 505 506 507 508 509 510 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 499 def promote_via_origin_module(synthetic, method_name, arg_types, block_type, environment) module_name = synthetic.provenance[:origin_module] return nil unless module_name module_type = Type::Combinator.nominal_of(module_name) RbsDispatch.try_dispatch( CallContext.build( receiver: module_type, method_name: method_name, args: arg_types, environment: environment, block_type: block_type ) ) end |
.promote_via_return_type(synthetic, environment) ⇒ Object
522 523 524 525 526 527 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 522 def promote_via_return_type(synthetic, environment) return_type = synthetic.return_type return nil if return_type.nil? || TIER_C_PLACEHOLDER_RETURNS.include?(return_type) environment.nominal_for_name(return_type) end |
.range_new_excl(excl_type) ⇒ Object
Resolves the optional ‘exclude_end` argument for `range_new_lift`. Returns `false` (no arg), `true`/`false` (Constant arg), or `nil` to signal “decline” (wrong type / wrong value class).
1158 1159 1160 1161 1162 1163 1164 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1158 def range_new_excl(excl_type) case excl_type when nil then false when Type::Constant excl_type.value if [true, false].include?(excl_type.value) end end |
.range_new_lift(class_name, arg_types) ⇒ Object
‘Range.new(b, e)` / `Range.new(b, e, excl)` — folds to `Constant` when both endpoints are `Constant` or both are `Constant`, and the optional third argument is a `Constant`. Nil endpoints (beginless / endless ranges) are not folded because the useful instance methods (`to_a`, `first`, `last`) are not defined for them; the RBS tier answers `Nominal` for those forms.
1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1131 def range_new_lift(class_name, arg_types) return nil unless class_name == "Range" return nil if arg_types.size < 2 || arg_types.size > 3 b_type = arg_types[0] e_type = arg_types[1] return nil unless b_type.is_a?(Type::Constant) && e_type.is_a?(Type::Constant) b_val = b_type.value e_val = e_type.value # Only fold homogeneous Integer or String endpoint pairs. return nil unless b_val.instance_of?(e_val.class) return nil unless b_val.is_a?(Integer) || b_val.is_a?(String) excl = range_new_excl(arg_types[2]) return nil if excl.nil? Type::Combinator.constant_of(Range.new(b_val, e_val, excl)) rescue StandardError nil end |
.rbs_display_for(rbs_result) ⇒ Object
659 660 661 662 663 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 659 def rbs_display_for(rbs_result) return "untyped" if rbs_result.nil? rbs_result.respond_to?(:describe) ? rbs_result.describe : rbs_result.inspect end |
.rbs_result_untyped?(rbs_result) ⇒ Boolean
655 656 657 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 655 def rbs_result_untyped?(rbs_result) rbs_result.is_a?(Type::Dynamic) && rbs_result.static_facet.is_a?(Type::Top) end |
.record_boundary_cross_if_applicable(receiver_type, method_name, rbs_result, environment) ⇒ Object
ADR-10 slice 5c — record a ‘dynamic.dependency-source.boundary-cross` event when RBS dispatch resolves a call AND the receiver class belongs to a `mode: :full` opt-in gem whose Walker also catalogued the same `(class_name, method_name)`. The dispatcher still returns the RBS answer (per ADR-10’s tier order: authoritative-source wins), but the reporter accumulates the crossing for end-of-run audit diagnostics.
Five honest fall-throughs keep the gate narrow:
-
environment / index / reporter missing — slice 5c needs all three.
-
receiver has no nominal class name (Dynamic-only carriers) — nothing to look up.
-
receiver class doesn’t belong to a ‘mode: :full` gem — the user didn’t opt this gem into the distinct dispatch path.
-
the gem-source catalog has no entry for the method —only RBS knows about it; nothing to cross.
-
the RBS-side result is itself ‘Dynamic` — the “agreement” is trivially `untyped ≈ untyped`, no meaningful divergence to flag.
613 614 615 616 617 618 619 620 621 622 623 624 625 626 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 613 def record_boundary_cross_if_applicable(receiver_type, method_name, rbs_result, environment) class_name = boundary_cross_class_name(receiver_type, environment, rbs_result) return if class_name.nil? index = environment.dependency_source_index return unless index.full_mode?(class_name) return unless index.contribution_for(class_name: class_name, method_name: method_name) environment.boundary_cross_reporter.record( class_name: class_name, method_name: method_name, gem_name: index.gem_for(class_name), rbs_display: rbs_display_for(rbs_result) ) end |
.regexp_new_lift(class_name, arg_types) ⇒ Object
1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1190 def regexp_new_lift(class_name, arg_types) return nil unless class_name == "Regexp" return nil if arg_types.empty? || arg_types.size > 2 pattern_arg = arg_types.first return nil unless pattern_arg.is_a?(Type::Constant) && pattern_arg.value.is_a?(String) opts = if arg_types.size == 2 opt_type = arg_types[1] return nil unless opt_type.is_a?(Type::Constant) opt_type.value else 0 end Type::Combinator.constant_of(Regexp.new(pattern_arg.value, opts)) rescue StandardError nil end |
.resolve(receiver_type:, method_name:, arg_types:, block_type: nil, environment: nil, call_node: nil, scope: nil) ⇒ Object
rubocop:disable Metrics/MethodLength, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity
98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 98 def resolve(receiver_type:, method_name:, arg_types:, # rubocop:disable Metrics/MethodLength, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity block_type: nil, environment: nil, call_node: nil, scope: nil) return nil if receiver_type.nil? # Build the call context once and thread it — unchanged — # through every tier (`_DispatchTier#try_dispatch`). The # dispatcher's own private fallback tiers still read the # positional locals below; only the tier modules consume the # context object. context = CallContext.build( receiver: receiver_type, method_name: method_name, args: arg_types, block_type: block_type, environment: environment, call_node: call_node, scope: scope ) bound_method_result = MethodFolding.try_backward(context) return bound_method_result if bound_method_result precise = dispatch_precise_tiers(context) return precise if precise # v0.1.1 Track 2 slice 7 — plugin return-type contribution # tier. Sits ahead of `RbsDispatch` so a plugin that # understands a domain-specific dispatch (e.g. an # `ActiveRecord::Base.find` returning `Nominal[<resolved # model>]`) wins over the RBS-projected envelope. Only # consults the registry when both `call_node` and `scope` # are supplied — the dispatcher's own internal callers # (per-element block fold, etc.) skip this tier. plugin_result = try_plugin_contribution(call_node, scope, receiver_type) return plugin_result if plugin_result # ADR-20 slice 3 — Rigor-bundled HKT-builtin return- # type tier. Sits ABOVE `RbsDispatch.try_dispatch` so # the handful of stdlib methods whose upstream RBS # signature is `untyped` but whose runtime shape Rigor # models via a Lightweight HKT (`json::value`, # eventually `dry_monads::result`, …) get the reduced # type instead of `Dynamic[Top]`. The table that # populates this tier lives in # `Rigor::Builtins::HktBuiltins::METHOD_RETURN_OVERRIDES`; # plugin-supplied per-method overrides are out of # scope for slice 3 and continue to flow through the # `try_plugin_contribution` tier above. hkt_builtin_result = try_hkt_builtin_return(receiver_type, method_name, arg_types, environment) return hkt_builtin_result if hkt_builtin_result # Rigor-bundled static refinement tier. Sits between HKT # and RBS so stdlib methods whose upstream RBS is broader # than the documented behaviour (e.g. `Kernel#__dir__` # declared `() -> String?` when the documented return is # `non-empty-string | nil`) get the tightened type # without modifying the vendored `ruby/rbs` submodule. # The override table lives in # `Rigor::Builtins::StaticReturnRefinements::OVERRIDES`. static_refinement = try_static_refinement(receiver_type, method_name, arg_types) return static_refinement if static_refinement rbs_result = RbsDispatch.try_dispatch(context) if rbs_result record_boundary_cross_if_applicable(receiver_type, method_name, rbs_result, environment) return rbs_result end # ADR-16 Tier B / Tier C — synthetic-method tier. Sits # BELOW RBS dispatch (per WD13: user-authored RBS overrides # substrate synthesis) and ABOVE the dependency-source # inference tier so a plugin's declared emit table beats # the generic gem-source fallback for the same class. Slice # 6a-TierB (origin_module dispatch) lands precise return # types for Tier B emissions; Tier C emissions still return # `Dynamic[T]` at this tier (slice 6b is the Tier C # promotion via ADR-13's resolver chain). synthetic_result = try_synthetic_method( receiver_type, method_name, arg_types, block_type, environment ) return synthetic_result if synthetic_result # ADR-17 slice 2 — project-side patched-method tier. # Sits BELOW the substrate / plugin tiers and ABOVE # dependency-source inference per ADR-17 § "Inference # contract". When the user's `pre_eval:` list named a # file that re-opens a class (e.g., # `lib/core_ext/string_extensions.rb` declaring # `class String; def to_url; end; end`), the pre-pass # populated `ProjectPatchedMethods` with the `(class, # method, kind)` triple; this tier surfaces it as # `Dynamic[top]` so the patched call resolves # cross-file without `call.undefined-method`. patched_result = try_project_patched_method(receiver_type, method_name, environment) return patched_result if patched_result # ADR-10 slice 2b-ii — dependency-source inference tier. # Sits BELOW RBS dispatch (RBS / RBS::Inline / generated # stubs / plugin contracts always win) and ABOVE the # user-class fallback so a method defined in an opt-in # gem stops emitting `call.undefined-method` even when # no signature contract resolves. Returns # `Dynamic[top]` — slice 2b-ii deliberately stops at the # dynamic-origin envelope; per-method return-type # precision is queued for a later slice. dep_source_result = try_dependency_source(receiver_type, method_name, environment) return dep_source_result if dep_source_result # v0.1.3 — discovered-method dispatch tier. When the # receiver class has no RBS BUT scope_indexer recorded # `def method_name` for that class (or singleton), the # call dispatches to `Dynamic[top]` rather than falling # through to the user-class fallback. Sits below RBS / # dependency-source so authoritative signatures still win. # The scope-indexer-built table records every project-side # `def`, `define_method`, and `alias_method`; the # `discovered_method?` consult here closes the # fail-soft-event hot spot on implicit-self calls # (`sibling_private(...)`) inside `lib/rigor/`'s own # internals (analyser private helpers don't have RBS). discovered_result = try_discovered_method(receiver_type, method_name, scope) return discovered_result if discovered_result # ADR-5 robustness — synthesized-stub-type tier. When the # receiver is a type Rigor invented to make an otherwise- # unbuildable project signature resolve (a missing-namespace # module, or a stub for a referenced-but-undeclared type like # an unavailable `DRb::DRbServer`), the stub carries no methods, # so an unresolved call against it would otherwise mis-fire # `call.undefined-method`. Resolve it to `Dynamic[Top]` instead # — the same no-false-positive contract as the dependency- # source tier. Sits below every real resolution tier so a # genuine signature always wins. stub_result = try_synthesized_stub_type(receiver_type, environment) return stub_result if stub_result # Slice 7 phase 10 — user-class ancestor fallback. When # the receiver is `Nominal[T]` or `Singleton[T]` for a # class not in the RBS environment (typically a # user-defined class), retry the dispatch against the # implicit ancestor: `Nominal[Object]` for instance # receivers and `Singleton[Object]` for singleton # receivers. This resolves Kernel intrinsics # (`require`, `raise`, `puts`, ...) and Module/Class # introspection (`attr_reader`, `private`, ...) on # user classes without requiring the user to author # their own RBS. try_user_class_fallback(receiver_type, environment, call_node, context) end |
.set_new_lift(class_name, arg_types) ⇒ Object
‘Set.new` / `Set.new(tuple_of_constants)` — folds to `Constant` when zero arguments are given or the single argument is a `Tuple` whose every element is a `Constant`. Mirrors `SetFolding#fold_new` but lives here so the `:new` path in `try_meta_introspection` can reach it before the RBS tier answers `Nominal`.
1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1171 def set_new_lift(class_name, arg_types) return nil unless class_name == "Set" return Type::Combinator.constant_of(::Set.new) if arg_types.empty? return nil if arg_types.size > 1 arg = arg_types.first return nil unless arg.is_a?(Type::Tuple) return nil unless arg.elements.all?(Type::Constant) values = arg.elements.map(&:value) Type::Combinator.constant_of(::Set.new(values)) rescue StandardError nil end |
.static_refinement_class_for(receiver_type) ⇒ Object
409 410 411 412 413 414 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 409 def static_refinement_class_for(receiver_type) case receiver_type when Type::Singleton, Type::Nominal receiver_type.class_name end end |
.static_refinement_owner_for(receiver_type, candidates) ⇒ Object
Picks the most specific override owner the receiver honours. For Kernel-owned overrides the receiver simply needs to be a real-class Nominal / Singleton (i.e. not BasicObject and not a Dynamic / Constant / shape carrier — those carriers go through their own narrower tiers).
400 401 402 403 404 405 406 407 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 400 def static_refinement_owner_for(receiver_type, candidates) receiver_class = static_refinement_class_for(receiver_type) return nil unless receiver_class return "Kernel" if candidates.include?("Kernel") && receiver_class != "BasicObject" candidates.find { |owner| owner == receiver_class } end |
.struct_new_lift(class_name, arg_types) ⇒ Object
‘Struct.new(:a, :b)` synthesises an anonymous Struct subclass (a class object), not a Struct instance — so the chained idiom `Struct.new(:a, :b).new(1, 2)` must resolve `.new` again on a class-like carrier. The constant-bound form (`S = Struct.new(:a); S.new(1)`) already records `Singleton` via `ScopeIndexer#record_meta_new_constant?`; this lift gives the chained (anonymous) position the same class-like carrier so the trailing `.new` dispatches instead of firing a spurious `undefined method ’new’ for Struct`.
The disambiguation mirrors ‘ScopeIndexer#struct_new_call?`: a call whose positionals are all `Constant<Symbol>` literals is a member-list class definition → `Singleton`. The following `AnonStruct.new(1, 2)` carries non-symbol args, so it falls through this gate to `Nominal` (a fresh instance) via the `meta_new` tail. ADR-48 deferred full Struct value folding (member-reader precision) on mutability grounds; this is the narrower `.new`-dispatch-only fix and contributes no member layout, so `instance.a` stays at its RBS/Dynamic type.
1001 1002 1003 1004 1005 1006 1007 1008 1009 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 1001 def struct_new_lift(class_name, arg_types) return nil unless class_name == "Struct" positional = arg_types.grep_v(Type::HashShape) return nil if positional.empty? return nil unless positional.all? { |t| t.is_a?(Type::Constant) && t.value.is_a?(Symbol) } Type::Combinator.singleton_of("Struct") end |
.synthetic_method_class_name(receiver_type) ⇒ Object
529 530 531 532 533 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 529 def synthetic_method_class_name(receiver_type) case receiver_type when Type::Nominal, Type::Singleton then receiver_type.class_name end end |
.try_dependency_source(receiver_type, method_name, environment) ⇒ Object
559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 559 def try_dependency_source(receiver_type, method_name, environment) index = environment&.dependency_source_index return nil if index.nil? || index.empty? class_name = dep_source_class_name(receiver_type) return nil if class_name.nil? # ADR-10 5a — per-receiver plugin veto. When a # registered plugin declares `manifest(owns_receivers: # [<class>])` AND the call's receiver IS that class # (or a subclass), decline and let plugins handle the # call. Plugins that own a receiver are the # authoritative source for that type; gem-source # inference must not contribute behind their backs. return nil if plugin_owns_receiver?(class_name, environment) contribution = index.contribution_for(class_name: class_name, method_name: method_name) return dependency_source_return_type(contribution) if contribution # ADR-10 5b — β budget semantics. On a catalog miss, # if the receiver class belongs to a budget-exceeded # gem AND the user opted into `:dependency_silence`, # return `Dynamic[top]` rather than falling through to # the user-class fallback. The user-class fallback # would otherwise emit `call.undefined-method` for # methods Rigor's catalog couldn't reach because the # walker hit its cap. budget_silence_result(class_name, index, environment) end |
.try_discovered_method(receiver_type, method_name, scope) ⇒ Object
v0.1.3 — discovered-method dispatch tier. ‘scope` carries the `discovered_methods` table built once per program by `ScopeIndexer` (a `Hash[String, Hash[Symbol, :instance | :singleton]]`). When the receiver names a discovered class AND the requested method is recorded for that class’s appropriate kind, return ‘Type::Combinator.untyped` — the dispatcher cannot infer a more precise return type from the bare `def` shape, but the call site stops being a fail-soft hot spot.
Returns ‘nil` when scope / receiver class is unavailable, when the method is not in the discovered table, OR when `discovered_def_nodes` carries a re-typable body for the method (so the downstream `ExpressionTyper#try_user_method_inference` tier can re-type the body for a precise return type rather than collapsing to `Dynamic` here).
The tier does NOT gate on ‘rbs_class_known?`. RBS dispatch already had its turn upstream and returned `nil` (otherwise we wouldn’t be here). When RBS knows the class but the particular method is missing from the sig — common for internal helpers and for auto-generated stubs that emit ‘class X` without enumerating every method — falling through to the user-class fallback would mistakenly fire `call.undefined-method`. Honoring the discovered table here keeps the sibling-private call resolution working under partial RBS coverage.
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 273 def try_discovered_method(receiver_type, method_name, scope) return nil if scope.nil? class_name, kind = discovered_method_lookup(receiver_type) return nil if class_name.nil? return nil unless scope.discovered_method?(class_name, method_name, kind) # Decline when a re-typable body is recorded for the method, so the # downstream `ExpressionTyper` inference tier can fold a precise # return instead of collapsing to `Dynamic[top]` here — instance # bodies via `user_def_for`, singleton bodies (`def self.x` / # `module_function`) via `singleton_def_for` (module-singleton # call resolution, ADR-57 follow-up). return nil if kind == :instance && scope.user_def_for(class_name, method_name) return nil if kind == :singleton && scope.singleton_def_for(class_name, method_name) Type::Combinator.untyped end |
.try_hkt_builtin_return(receiver_type, method_name, arg_types, environment) ⇒ Object
ADR-2 § “Flow Contribution Bundle” / v0.1.1 Track 2 slice 7; ADR-52 WD3 — consults each loaded plugin’s gated ‘dynamic_return` rules, wraps the contributed types as `FlowContribution` bundles, merges them through `FlowContribution::Merger`, and returns the merged `return_type` slot (or nil when no plugin contributed a return type).
Plugins whose hook raises have their contribution silently dropped for this call so the dispatch chain keeps moving — the run-level diagnostic envelope (per ADR-2 § “Plugin Trust and I/O Policy”) is owned by ‘Analysis::Runner#plugin_emitted_diagnostics`. ADR-20 slice 3 — looks up the receiver / method pair in Builtins::HktBuiltins::METHOD_RETURN_OVERRIDES and returns the reduced HKT type. Only fires when the receiver is a Type::Singleton (the `JSON.parse` shape) and the registry-backed reduction succeeds; returns `nil` otherwise so the dispatcher falls through to RBS.
349 350 351 352 353 354 355 356 357 358 359 360 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 349 def try_hkt_builtin_return(receiver_type, method_name, arg_types, environment) return nil if environment.nil? return nil unless receiver_type.is_a?(Type::Singleton) Rigor::Builtins::HktBuiltins.method_return_override( class_name: receiver_type.class_name, method_name: method_name, kind: :singleton, arg_types: arg_types, hkt_registry: environment.hkt_registry ) end |
.try_meta_introspection(receiver_type, method_name, arg_types = []) ⇒ Object
Slice 7 phase 8 — meta-introspection shortcuts. The default ‘Object#class` RBS return type is `Class`, but for a receiver of known nominal identity we can do better: `instance_of(Foo).class` is `Singleton` (the class object itself), which downstream dispatch uses to resolve `self.class.some_class_method`. The same logic answers `Foo.class` as `Singleton` (deliberate; calling `.class` on a class object yields `Class`, the metaclass). We also special-case `is_a?`- adjacent calls and the trivial `instance_of?(self)` later as the rule catalogue grows; for now only `class` is handled.
923 924 925 926 927 928 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 923 def (receiver_type, method_name, arg_types = []) case method_name when :class then (receiver_type) when :new then (receiver_type, arg_types) end end |
.try_plugin_contribution(call_node, scope, receiver_type) ⇒ Object
416 417 418 419 420 421 422 423 424 425 426 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 416 def try_plugin_contribution(call_node, scope, receiver_type) return nil if call_node.nil? || scope.nil? registry = scope.environment&.plugin_registry return nil if registry.nil? || registry.empty? contributions = collect_plugin_contributions(registry, call_node, scope, receiver_type) return nil if contributions.empty? FlowContribution::Merger.merge(contributions).return_type end |
.try_project_patched_method(receiver_type, method_name, environment) ⇒ Object
ADR-17 slice 2 — project-side patched-method tier. Slice 3a uses the registry’s heuristic-extracted ‘return_type` (populated via the same `Analysis::DependencySourceInference::ReturnTypeHeuristic` the ADR-10 walker uses): a `def to_url; “hello”; end` patched onto `String` now resolves `s.to_url` to `Dynamic[Nominal]` instead of the pre-3a `Dynamic`. Falls back to `Dynamic` when the heuristic declined (non-literal tail expression).
544 545 546 547 548 549 550 551 552 553 554 555 556 557 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 544 def try_project_patched_method(receiver_type, method_name, environment) registry = environment&.project_patched_methods return nil if registry.nil? || registry.empty? class_name = synthetic_method_class_name(receiver_type) return nil if class_name.nil? kind = receiver_type.is_a?(Type::Singleton) ? :singleton : :instance entry = registry.lookup(class_name: class_name, method_name: method_name, kind: kind) return nil if entry.nil? return Type::Combinator.untyped if entry.return_type.nil? Type::Combinator.dynamic(entry.return_type) end |
.try_static_refinement(receiver_type, method_name, arg_types) ⇒ Object
Consults the Rigor-bundled static refinement table for a (owner-class, method-name, kind) entry. Kernel methods are mixed into every non-BasicObject class, so an implicit-self ‘__dir__` call (receiver_type = Nominal) is matched by looking up Kernel as the owner. Explicit `Kernel.__dir__` (receiver_type = Singleton) and instance-side calls (receiver_type = Nominal) share the `:both` row.
The receiver-side ancestor check is intentionally cheap: any non-BasicObject Nominal / Singleton matches every Kernel-owned override. BasicObject explicitly excludes Kernel and is therefore rejected. The narrow risk of a user-defined ‘def __dir__` shadowing Kernel’s method would also alter the runtime answer; users with that configuration opt out via a ‘signature_paths` overlay declaring their own return type.
379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 379 def try_static_refinement(receiver_type, method_name, arg_types) candidates = Rigor::Builtins::StaticReturnRefinements.owners_for(method_name) return nil if candidates.empty? owner = static_refinement_owner_for(receiver_type, candidates) return nil unless owner kind = receiver_type.is_a?(Type::Singleton) ? :singleton : :instance Rigor::Builtins::StaticReturnRefinements.lookup( owner_class_name: owner, method_name: method_name, kind: kind, arg_types: arg_types ) end |
.try_synthesized_stub_type(receiver_type, environment) ⇒ Object
ADR-5 robustness — returns ‘Dynamic` when the receiver is an instance or singleton of a type Rigor synthesized (a missing-namespace module or a referenced-type stub). The stub has no methods, so the call would otherwise reach the user-class fallback and surface `call.undefined-method`; the honest answer for a type Rigor invented is “unknown shape”, i.e. `Dynamic`. Returns nil (declines) for any real type.
311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 311 def try_synthesized_stub_type(receiver_type, environment) return nil if environment.nil? loader = environment.rbs_loader return nil if loader.nil? || !loader.respond_to?(:synthesized_type_names) names = loader.synthesized_type_names return nil if names.empty? class_name = case receiver_type when Type::Nominal, Type::Singleton then receiver_type.class_name.to_s.sub(/\A::/, "") end return nil unless class_name && names.include?(class_name) Type::Combinator.untyped end |
.try_synthetic_method(receiver_type, method_name, arg_types, block_type, environment) ⇒ Object
ADR-10 slice 2b-ii. Consults the per-run ‘Analysis::DependencySourceInference::Index` carried by the environment for `(class_name, method_name)` observations harvested from opt-in gems’ ‘roots:`. On a hit, returns `Combinator.untyped` so the call site carries the `Dynamic` provenance (per ADR-10’s “Inference contract”: gem-source-inferred shapes never publish as ground-truth ‘T`). Returns `nil` when the environment carries no index, the index has no entry, or the receiver has no nominal class to look up. ADR-16 synthetic-method tier. Slice 2b shipped the floor —a match short-circuits at the right precedence (above dep-source / discovered / user-class-fallback; below RBS) and returns `Dynamic`. Slice 6 (precision promotion):
-
Tier B path (slice 6a, ‘provenance` recorded by the slice-3b scanner): redispatch on `Nominal` via `RbsDispatch` so the module’s authored RBS return type wins. Devise’s ‘valid_password?` returns `bool`, not `Dynamic`.
-
Tier C path (slice 6b, plain ‘return_type:` string from the manifest’s emit table): look up ‘environment.nominal_for_name(return_type)` so `attribute :avatar, Types::String` emits a synthetic reader returning `Nominal` (when the class is in RBS). Unparameterised class names only — parameterised forms (`Array`, `Hash[K, V]`) and plugin-supplied utility-type names (`Pick<T, K>`) require routing through the full ADR-13 `Plugin::TypeNodeResolver` chain, which slice 6 does not yet wire in (the resolver chain is consulted only for `%arigor:v1:…` payloads as of ADR-13 slice 3).
459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 459 def try_synthetic_method(receiver_type, method_name, arg_types, block_type, environment) index = environment&.synthetic_method_index return nil if index.nil? || index.empty? class_name = synthetic_method_class_name(receiver_type) return nil if class_name.nil? matches = case receiver_type when Type::Singleton then index.lookup_singleton(class_name, method_name) else index.lookup_instance(class_name, method_name) end return nil if matches.empty? promoted = promote_synthetic_match(matches, method_name, arg_types, block_type, environment) promoted || Type::Combinator.untyped end |
.try_user_class_fallback(receiver_type, environment, call_node, context) ⇒ Object
839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 839 def try_user_class_fallback(receiver_type, environment, call_node, context) return nil if environment.nil? fallback_receiver = user_class_fallback_receiver(receiver_type, environment) return nil if fallback_receiver.nil? # Preserve the ORIGINAL receiver type as the `self` # substitution so `Kernel#dup: () -> self` and other # `self`-returning methods route through Object's RBS # while still returning the caller's type rather than # `Object`. Without this, `base = self.dup` inside a # `Bundler::URI::Generic` instance method types `base` # as `Object` because `Bundler::URI::Generic` is not in # RBS and the fallback's `self` resolves to Object. # # `public_only:` — when the call has an EXPLICIT, non-`self` # receiver (`Favourite.select(...)`), suppress the private # `Object`/`Kernel`/`Class` methods the fallback would # otherwise resolve. Ruby raises `NoMethodError` for a # private method called with an explicit receiver, so # resolving `Favourite.select` to the private `Kernel#select` # (`-> Array[String]`) is a confidently-wrong type. Implicit- # self / `self.`-receiver calls (`puts`, `raise`, `require`) # keep resolving — those are the fallback's intended targets. RbsDispatch.try_dispatch( context.with( receiver: fallback_receiver, self_type_override: receiver_type, public_only: explicit_non_self_receiver?(call_node), call_node: nil, scope: nil ) ) end |
.user_class_fallback_receiver(receiver_type, environment) ⇒ Object
890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 |
# File 'lib/rigor/inference/method_dispatcher.rb', line 890 def user_class_fallback_receiver(receiver_type, environment) case receiver_type when Type::Nominal # Modules: even when RBS knows the module, an instance # method on a mixin-only module (e.g. `PP::ObjectMixin`) # observes Kernel / Object methods through every concrete # includer's ancestor chain. Route through the # `Nominal[Object]` fallback so `self.inspect` / # `self.respond_to?` / `self.class` resolve cleanly when # the module itself does not declare them. known = Rigor::Reflection.rbs_class_known?(receiver_type.class_name, environment: environment) return environment.nominal_for_name("Object") if !known || environment.rbs_module?(receiver_type.class_name) nil when Type::Singleton return nil if Rigor::Reflection.rbs_class_known?(receiver_type.class_name, environment: environment) environment.singleton_for_name("Class") end end |