Provide fallback when mid_code_range method not exist

ruby · Apr 30, 2024 · d372cdd · d372cdd
1 parent 581a225
commit d372cdd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/typeprof/core/service.rb b/lib/typeprof/core/service.rb
@@ -176,7 +176,13 @@ def definitions(path, pos)
               site.resolve(genv, nil) do |me, _ty, _mid, _orig_ty|
                 next unless me
                 me.defs.each do |mdef|
-                  defs << [mdef.node.lenv.path, mdef.node.mid_code_range]
+                  code_range =
+                    case mdef.node
+                    when AST::DefNode then mdef.node.mid_code_range
+                    else mdef.node.code_range
+                    end
+
+                  defs << [mdef.node.lenv.path, code_range]
                 end
               end
             end