make snoop reusable

Author: SimonDanisch

.gitignore

@@ -2,3 +2,5 @@
 *.jl.*.cov
 *.jl.mem
 deps.jl
+*.csv
+snoopy.jl

src/PackageCompiler.jl

@@ -5,70 +5,63 @@ using SnoopCompile
 
 include("build_sysimg.jl")
 
-function snoop(path, compilationfile)
+function snoop(path, compilationfile, reuse)
     cd(@__DIR__)
-    mktempdir() do tmpfolder
-        csv = joinpath(homedir(), "test.csv") #joinpath(tmpfolder, "precompile.csv")
-        # Snoop compiler can't handle the path as a variable, so we just create a file
-        open(joinpath(tmpfolder, "snoopy.jl"), "w") do io
+    csv = "precompile.csv"
+    # Snoop compiler can't handle the path as a variable, so we just create a file
+    if !reuse
+        open(joinpath("snoopy.jl"), "w") do io
             println(io, "include(\"$(escape_string(path))\")")
         end
-        cd(tmpfolder)
         SnoopCompile.@snoop csv begin
             include("snoopy.jl")
         end
-        data = SnoopCompile.read(csv)
-        pc = SnoopCompile.parcel(reverse!(data[2]))
-        delims = r"([\{\} \n\(\),])_([\{\} \n\(\),])"
-        tmp_mod = eval(:(module $(gensym()) end))
-        open(compilationfile, "w") do io
-            for (k, v) in pc
-                k == :unknown && continue
-                println(k)
+    end
+    data = SnoopCompile.read(csv)
+    pc = SnoopCompile.parcel(reverse!(data[2]))
+    delims = r"([\{\} \n\(\),])_([\{\} \n\(\),])"
+    tmp_mod = eval(:(module $(gensym()) end))
+    open(compilationfile, "w") do io
+        for (k, v) in pc
+            k == :unknown && continue
+            try
+                eval(tmp_mod, :(using $k))
+                println(io, "using $k")
+                info("using $k")
+            catch e
+                println("Module not found: $k")
+            end
+        end
+        for (k, v) in pc
+            for ln in v
+                # replace `_` for free parameters, which print out a warning otherwise
+                ln = replace(ln, delims, s"\1XXX\2")
+                # only print out valid lines
+                # TODO figure out why some precompile statements have undefined free variables in there
                 try
-                    eval(tmp_mod, :(using $k))
-                    println(io, "using $k")
+                    eval(tmp_mod, parse(ln))
+                    println(io, ln)
                 catch e
-                    warn(e)
-                end
-            end
-            for (k, v) in pc
-                for ln in v
-                    # replace `_` for free parameters, which print out a warning otherwise
-                    ln = replace(ln, delims, s"\1XXX\2")
-                    # only print out valid lines
-                    # TODO figure out why some precompile statements have undefined free variables in there
-                    try
-                        eval(tmp_mod, parse(ln))
-                        println(io, ln)
-                    catch e
-                        warn(e)
-                    end
+                    warn("Not emitted: ", ln)
                 end
             end
         end
     end
-    cd(@__DIR__)
-end
-
-
-function compile(file; force = false)
-    build_sysimg(default_sysimg_path(), "native", file, force = force)
 end
 
-function revert(file; force = false)
+function revert()
     build_sysimg(force = true)
 end
 
-function compile_package(package, force = false)
+function compile_package(package, force = false, reuse = false)
     realpath = if ispath(package)
         normpath(abspath(package))
     else
         Pkg.dir(package)
     end
     testroot = joinpath(realpath, "test")
     precompile_file = joinpath(testroot, "precompile.jl")
-    snoop(joinpath(testroot, "runtests.jl"), precompile_file)
+    snoop(joinpath(testroot, "runtests.jl"), precompile_file, reuse)
     build_sysimg(default_sysimg_path(), "native", precompile_file, force = force)
 end