make the REPL specific code into an extension

Co-authored-by: Mark Kittisopikult

Author: KristofferC

Project.toml

@@ -9,13 +9,13 @@ version = "1.11.0"
 Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 Downloads = "f43a241f-c20a-4ad4-852c-f6b1247861c6"
+Example = "7876af07-990d-54b4-ab0e-23690620f79a"
 FileWatching = "7b1f6079-737a-58dc-b8bc-7a2ca5c1b5ee"
 LibGit2 = "76f85450-5226-5b5a-8eaa-529ad045b433"
 Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
 Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 Markdown = "d6f4376e-aef5-505a-96c1-9c027394607a"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
-REPL = "3fa0cd96-eef1-5676-8a61-b3b8758bbffb"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SHA = "ea8e919c-243c-51af-8825-aaa63cd721ce"
 Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
@@ -24,6 +24,12 @@ Tar = "a4e569a6-e804-4fa4-b0f3-eef7a1d5b13e"
 UUIDs = "cf7118a7-6976-5b1a-9a39-7adc72f591a4"
 p7zip_jll = "3f19e933-33d8-53b3-aaab-bd5110c3b7a0"
 
+[weakdeps]
+REPL = "3fa0cd96-eef1-5676-8a61-b3b8758bbffb"
+
+[extensions]
+REPLExt = "REPL"
+
 [compat]
 HistoricalStdlibVersions = "1.2"
 

ext/REPLExt/REPLExt.jl

@@ -0,0 +1,315 @@
+module REPLExt
+
+using Markdown, UUIDs, Dates
+
+import REPL
+import .REPL: LineEdit, REPLCompletions, TerminalMenus
+
+import Pkg
+import .Pkg: linewrap, pathrepr, compat, can_fancyprint, printpkgstyle
+using .Pkg: Types, Operations, API, Registry, Resolve, REPLMode
+
+using .REPLMode: Statement, CommandSpec, Command, prepare_cmd
+
+using .Types: Context, PkgError, pkgerror, EnvCache
+
+
+include("completions.jl")
+include("compat.jl")
+
+######################
+# REPL mode creation #
+######################
+
+struct PkgCompletionProvider <: LineEdit.CompletionProvider end
+
+function LineEdit.complete_line(c::PkgCompletionProvider, s)
+    partial = REPL.beforecursor(s.input_buffer)
+    full = LineEdit.input_string(s)
+    ret, range, should_complete = completions(full, lastindex(partial))
+    return ret, partial[range], should_complete
+end
+
+prev_project_file = nothing
+prev_project_timestamp = nothing
+prev_prefix = ""
+
+function projname(project_file::String)
+    project = try
+        Types.read_project(project_file)
+    catch
+        nothing
+    end
+    if project === nothing || project.name === nothing
+        name = basename(dirname(project_file))
+    else
+        name = project.name::String
+    end
+    for depot in Base.DEPOT_PATH
+        envdir = joinpath(depot, "environments")
+        if startswith(abspath(project_file), abspath(envdir))
+            return "@" * name
+        end
+    end
+    return name
+end
+
+function promptf()
+    global prev_project_timestamp, prev_prefix, prev_project_file
+    project_file = try
+        Types.find_project_file()
+    catch
+        nothing
+    end
+    prefix = ""
+    if project_file !== nothing
+        if prev_project_file == project_file && prev_project_timestamp == mtime(project_file)
+            prefix = prev_prefix
+        else
+            project_name = projname(project_file)
+            if project_name !== nothing
+                if textwidth(project_name) > 30
+                    project_name = first(project_name, 27) * "..."
+                end
+                prefix = "($(project_name)) "
+                prev_prefix = prefix
+                prev_project_timestamp = mtime(project_file)
+                prev_project_file = project_file
+            end
+        end
+    end
+    if Pkg.OFFLINE_MODE[]
+        prefix = "$(prefix)[offline] "
+    end
+    return "$(prefix)pkg> "
+end
+
+do_cmds(repl::REPL.AbstractREPL, input::String) = do_cmds(repl, prepare_cmd(input))
+
+function do_cmds(repl::REPL.AbstractREPL, commands::Vector{Command})
+    try
+        return REPLMode.do_cmds(commands, repl.t.out_stream)
+    catch err
+        if err isa PkgError || err isa Resolve.ResolverError
+            Base.display_error(repl.t.err_stream, ErrorException(sprint(showerror, err)), Ptr{Nothing}[])
+        else
+            Base.display_error(repl.t.err_stream, err, Base.catch_backtrace())
+        end
+    end
+end
+
+# Set up the repl Pkg REPLMode
+function create_mode(repl::REPL.AbstractREPL, main::LineEdit.Prompt)
+    pkg_mode = LineEdit.Prompt(promptf;
+        prompt_prefix = repl.options.hascolor ? Base.text_colors[:blue] : "",
+        prompt_suffix = "",
+        complete = PkgCompletionProvider(),
+        sticky = true)
+
+    pkg_mode.repl = repl
+    hp = main.hist
+    hp.mode_mapping[:pkg] = pkg_mode
+    pkg_mode.hist = hp
+
+    search_prompt, skeymap = LineEdit.setup_search_keymap(hp)
+    prefix_prompt, prefix_keymap = LineEdit.setup_prefix_keymap(hp, pkg_mode)
+
+    pkg_mode.on_done = (s, buf, ok) -> begin
+        ok || return REPL.transition(s, :abort)
+        input = String(take!(buf))
+        REPL.reset(repl)
+        do_cmds(repl, input)
+        REPL.prepare_next(repl)
+        REPL.reset_state(s)
+        s.current_mode.sticky || REPL.transition(s, main)
+    end
+
+    mk = REPL.mode_keymap(main)
+
+    shell_mode = nothing
+    for mode in repl.interface.modes
+        if mode isa LineEdit.Prompt
+            mode.prompt == "shell> " && (shell_mode = mode)
+        end
+    end
+
+    repl_keymap = Dict()
+    if shell_mode !== nothing
+        let shell_mode=shell_mode
+            repl_keymap[';'] = function (s,o...)
+                if isempty(s) || position(LineEdit.buffer(s)) == 0
+                    buf = copy(LineEdit.buffer(s))
+                    LineEdit.transition(s, shell_mode) do
+                        LineEdit.state(s, shell_mode).input_buffer = buf
+                    end
+                else
+                    LineEdit.edit_insert(s, ';')
+                end
+            end
+        end
+    end
+
+    b = Dict{Any,Any}[
+        skeymap, repl_keymap, mk, prefix_keymap, LineEdit.history_keymap,
+        LineEdit.default_keymap, LineEdit.escape_defaults
+    ]
+    pkg_mode.keymap_dict = LineEdit.keymap(b)
+    return pkg_mode
+end
+
+function repl_init(repl::REPL.AbstractREPL)
+    main_mode = repl.interface.modes[1]
+    pkg_mode = create_mode(repl, main_mode)
+    push!(repl.interface.modes, pkg_mode)
+    keymap = Dict{Any,Any}(
+        ']' => function (s,args...)
+            if isempty(s) || position(LineEdit.buffer(s)) == 0
+                buf = copy(LineEdit.buffer(s))
+                LineEdit.transition(s, pkg_mode) do
+                    LineEdit.state(s, pkg_mode).input_buffer = buf
+                end
+            else
+                LineEdit.edit_insert(s, ']')
+            end
+        end
+    )
+    main_mode.keymap_dict = LineEdit.keymap_merge(main_mode.keymap_dict, keymap)
+    return
+end
+
+
+function try_prompt_pkg_add(pkgs::Vector{Symbol})
+    ctx = try
+        Context()
+    catch
+        # Context() will error if there isn't an active project.
+        # If we can't even do that, exit early.
+        return false
+    end
+    if isempty(ctx.registries)
+        if !REG_WARNED[]
+            printstyled(ctx.io, " │ "; color=:green)
+            printstyled(ctx.io, "Attempted to find missing packages in package registries but no registries are installed.\n")
+            printstyled(ctx.io, " └ "; color=:green)
+            printstyled(ctx.io, "Use package mode to install a registry. `pkg> registry add` will install the default registries.\n\n")
+            REG_WARNED[] = true
+        end
+        return false
+    end
+    available_uuids = [Types.registered_uuids(ctx.registries, String(pkg)) for pkg in pkgs] # vector of vectors
+    filter!(u -> all(!isequal(Operations.JULIA_UUID), u), available_uuids) # "julia" is in General but not installable
+    isempty(available_uuids) && return false
+    available_pkgs = pkgs[isempty.(available_uuids) .== false]
+    isempty(available_pkgs) && return false
+    resp = try
+        plural1 = length(pkgs) == 1 ? "" : "s"
+        plural2 = length(available_pkgs) == 1 ? "a package" : "packages"
+        plural3 = length(available_pkgs) == 1 ? "is" : "are"
+        plural4 = length(available_pkgs) == 1 ? "" : "s"
+        missing_pkg_list = length(pkgs) == 1 ? String(pkgs[1]) : "[$(join(pkgs, ", "))]"
+        available_pkg_list = length(available_pkgs) == 1 ? String(available_pkgs[1]) : "[$(join(available_pkgs, ", "))]"
+        msg1 = "Package$(plural1) $(missing_pkg_list) not found, but $(plural2) named $(available_pkg_list) $(plural3) available from a registry."
+        for line in linewrap(msg1, io = ctx.io, padding = length(" │ "))
+            printstyled(ctx.io, " │ "; color=:green)
+            println(ctx.io, line)
+        end
+        printstyled(ctx.io, " │ "; color=:green)
+        println(ctx.io, "Install package$(plural4)?")
+        msg2 = string("add ", join(available_pkgs, ' '))
+        for (i, line) in pairs(linewrap(msg2; io = ctx.io, padding = length(string(" |   ", REPLMode.promptf()))))
+            printstyled(ctx.io, " │   "; color=:green)
+            if i == 1
+                printstyled(ctx.io, REPLMode.promptf(); color=:blue)
+            else
+                print(ctx.io, " "^length(REPLMode.promptf()))
+            end
+            println(ctx.io, line)
+        end
+        printstyled(ctx.io, " └ "; color=:green)
+        Base.prompt(stdin, ctx.io, "(y/n/o)", default = "y")
+    catch err
+        if err isa InterruptException # if ^C is entered
+            println(ctx.io)
+            return false
+        end
+        rethrow()
+    end
+    if isnothing(resp) # if ^D is entered
+        println(ctx.io)
+        return false
+    end
+    resp = strip(resp)
+    lower_resp = lowercase(resp)
+    if lower_resp in ["y", "yes"]
+        API.add(string.(available_pkgs))
+    elseif lower_resp in ["o"]
+        editable_envs = filter(v -> v != "@stdlib", LOAD_PATH)
+        option_list = String[]
+        keybindings = Char[]
+        shown_envs = String[]
+        # We use digits 1-9 as keybindings in the env selection menu
+        # That's why we can display at most 9 items in the menu
+        for i in 1:min(length(editable_envs), 9)
+            env = editable_envs[i]
+            expanded_env = Base.load_path_expand(env)
+
+            isnothing(expanded_env) && continue
+
+            n = length(option_list) + 1
+            push!(option_list, "$(n): $(pathrepr(expanded_env)) ($(env))")
+            push!(keybindings, only("$n"))
+            push!(shown_envs, expanded_env)
+        end
+        menu = TerminalMenus.RadioMenu(option_list, keybindings=keybindings, pagesize=length(option_list))
+        default = something(
+            # select the first non-default env by default, if possible
+            findfirst(!=(Base.active_project()), shown_envs),
+            1
+        )
+        print(ctx.io, "\e[1A\e[1G\e[0J") # go up one line, to the start, and clear it
+        printstyled(ctx.io, " └ "; color=:green)
+        choice = try
+            TerminalMenus.request("Select environment:", menu, cursor=default)
+        catch err
+            if err isa InterruptException # if ^C is entered
+                println(ctx.io)
+                return false
+            end
+            rethrow()
+        end
+        choice == -1 && return false
+        API.activate(shown_envs[choice]) do
+            API.add(string.(available_pkgs))
+        end
+    elseif (lower_resp in ["n"])
+        return false
+    else
+        println(ctx.io, "Selection not recognized")
+        return false
+    end
+    if length(available_pkgs) < length(pkgs)
+        return false # declare that some pkgs couldn't be installed
+    else
+        return true
+    end
+end
+
+
+
+function __init__()
+    if isdefined(Base, :active_repl)
+        repl_init(Base.active_repl)
+    else
+        atreplinit() do repl
+            if isinteractive() && repl isa REPL.LineEditREPL
+                isdefined(repl, :interface) || (repl.interface = REPL.setup_interface(repl))
+                repl_init(repl)
+            end
+        end
+    end
+    push!(empty!(REPL.install_packages_hooks), try_prompt_pkg_add)
+end
+
+include("precompile.jl")
+
+end