replicate REPL pkg mode for cells starting with ] (JuliaLang#715)

Author: stevengj

src/execute_request.jl

@@ -3,6 +3,9 @@
 # returning results.
 
 import Base.Libc: flush_cstdio
+if VERSION ≥ v"0.7"
+    import Pkg
+end
 
 const text_plain = MIME("text/plain")
 const image_svg = MIME("image/svg+xml")
@@ -164,6 +167,14 @@ function execute_request(socket, msg)
                    m -> string(replace(m, r"^\s*;" => "Base.repl_cmd(`"),
                                "`, ", stdout_name, ")"))
 
+    if VERSION ≥ v"0.7"
+        # "] ..." cells are interpreted as pkg shell commands
+        if occursin(r"^\].*$", code)
+            code = "IJulia.Pkg.REPLMode.do_cmd(IJulia.minirepl[], \"" *
+                escape_string(code[2:end]) * "\"; do_rethrow=true)"
+        end
+    end
+
     # a cell beginning with "? ..." is interpreted as a help request
     hcode = replace(code, r"^\s*\?" => "")
 

src/init.jl

@@ -51,6 +51,20 @@ function qtconsole()
     end
 end
 
+# similar to Pkg.REPLMode.MiniREPL, a minimal REPL-like emulator
+# for use with Pkg.do_cmd.  We have to roll our own to
+# make sure it uses the redirected stdout, and because
+# we don't have terminal support.
+if VERSION ≥ v"0.7"
+    import REPL
+    struct MiniREPL <: REPL.AbstractREPL
+        display::TextDisplay
+    end
+    REPL.REPLDisplay(repl::MiniREPL) = repl.display
+    const minirepl = Ref{MiniREPL}()
+end
+
+
 function init(args)
     inited && error("IJulia is already running")
     if length(args) > 0
@@ -118,6 +132,9 @@ function init(args)
         redirect_stderr(IJuliaStdio(stderr,"stderr"))
     end
     redirect_stdin(IJuliaStdio(stdin,"stdin"))
+    if VERSION ≥ v"0.7"
+        minirepl[] = MiniREPL(TextDisplay(stdout))
+    end
 
     if isdefined(Base, :CoreLogging)
         logger = Base.CoreLogging.SimpleLogger(Base.stderr)