Implementation: allow to lookup implementations of class and module

Closes crystal-lang#4941
makenowjust · May 5, 2019 · fe0403b · fe0403b
1 parent 786f2ac
commit fe0403b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/compiler/crystal/tools/implementations.cr b/src/compiler/crystal/tools/implementations.cr
@@ -116,6 +116,15 @@ module Crystal
       end
     end
 
+    def visit(node : Path)
+      return contains_target(node) unless (loc = node.location) && (end_loc = node.end_location) && @target_location.between?(loc, end_loc)
+
+      target = node.target_const || node.type?.try &.instance_type
+      target.try &.locations.try &.each do |loc|
+        @locations << loc
+      end
+    end
+
     def visit(node)
       contains_target(node)
     end