signatures_at support for Base and stdlibs

Author: timholy

src/CodeTracking.jl

@@ -35,6 +35,8 @@ const method_lookup_callback = Ref{Any}(nothing)
 #        a list of method-signatures defined by that expression.
 const expressions_callback = Ref{Any}(nothing)
 
+const juliabase = joinpath("julia", "base")
+const juliastdlib = joinpath("julia", "stdlib", "v$(VERSION.major).$(VERSION.minor)")
 
 ### Public API
 
@@ -112,6 +114,18 @@ Return the signatures of all methods whose definition spans the specified locati
 Returns `nothing` if there are no methods at that location.
 """
 function signatures_at(filename::AbstractString, line::Integer)
+    if occursin(juliabase, filename)
+        rpath = postpath(filename, juliabase)
+        id = PkgId(Base)
+        return signatures_at(id, rpath, line)
+    elseif occursin(juliastdlib, filename)
+        rpath = postpath(filename, juliastdlib)
+        spath = splitpath(rpath)
+        libname = spath[1]
+        project = Base.active_project()
+        id = PkgId(Base.project_deps_get(project, libname), libname)
+        return signatures_at(id, joinpath(spath[2:end]...), line)
+    end
     for (id, pkgfls) in _pkgfiles
         if startswith(filename, basedir(pkgfls)) || id.name == "Main"
             bdir = basedir(pkgfls)

src/utils.jl

@@ -81,3 +81,11 @@ function maybe_fixup_stdlib_path(path)
     end
     return path
 end
+
+function postpath(filename, pre)
+    idx = findfirst(pre, filename)
+    idx === nothing && error(pre, " not found in ", filename)
+    post = filename[first(idx) + length(pre) : end]
+    post[1:1] == Base.Filesystem.path_separator && return post[2:end]
+    return post
+end

test/runtests.jl

@@ -75,3 +75,17 @@ include("script.jl")
     file, line = whereis(lin, m)
     @test endswith(file, String(lin.file))
 end
+
+@testset "With Revise" begin
+    if isdefined(Main, :Revise)
+        m = @which gcd(10, 20)
+        sigs = signatures_at(Base.find_source_file(String(m.file)), m.line)
+        @test !isempty(sigs)
+
+        m = first(methods(edit))
+        sigs = signatures_at(String(m.file), m.line)
+        @test !isempty(sigs)
+        sigs = signatures_at(Base.find_source_file(String(m.file)), m.line)
+        @test !isempty(sigs)
+    end
+end