Fix lookup for PartialStruct constructors (#107)

Author: timholy

src/utils.jl

@@ -6,6 +6,7 @@ function checkname(fdef::Expr, name)
     fdef.head === :call || return false
     if fproto isa Expr
         fproto.head == :(::) && return last(fproto.args) == name
+        fproto.head == :curly && return fproto.args[1] === name
         # A metaprogramming-generated function
         fproto.head === :$ && return true   # uncheckable, let's assume all is well
         # Is the check below redundant?

test/runtests.jl

@@ -87,6 +87,13 @@ isdefined(Main, :Revise) ? Main.Revise.includet("script.jl") : include("script.j
     @test pkgfiles("ColorTypes") === nothing
     @test_throws ErrorException pkgfiles("NotAPkg")
 
+    # PartialStruct parametric constructors
+    m = @which LikeNamedTuple{(:a,)}((1,))
+    _, line = whereis(m)
+    def, linedef = definition(String, m)
+    @test line == linedef
+    @test occursin("LikeNamedTuple{names}", def)
+
     # Test a method marked as missing
     m = @which sum(1:5)
     CodeTracking.method_info[m.sig] = missing

test/script.jl

@@ -68,3 +68,11 @@ if isdefined(Base, Symbol("@assume_effects"))
 end
 
 has_semicolon1(x, y) = x + y;
+
+struct LikeNamedTuple{names,V}
+    vals::V
+end
+
+LikeNamedTuple() = LikeNamedTuple{(),Tuple{}}(())
+
+LikeNamedTuple{names}(args::Tuple) where {names} = LikeNamedTuple{names,typeof(args)}(args)