Starship.rs Prompt Wrapper

[jdugan6240]

The following implements support for the Starship cross-shell prompt:

-- Starship.rs support for Hilbish
-- Supports the following features:
-- * exit codes
-- * command duration (kind of - it usually works but is not 100% perfect)
-- * number of jobs
-- * continuation prompt (which some shells like fish don't even support!)
-- * anything else inherently handled by starship (git, username, etc)

local bait = require 'bait'
local hilbish = require 'hilbish'

local starship = {}

starship.execTime = {}

function starship.doPrompt(status, numJobs, numSecs)
        -- Since lua by default only has second-resolution with its os.time() function,
        -- we need to multiply by 1000 to get milliseconds elapsed.
        -- This, of course, means we don't actually get millisecond-level resolution, but whatever - we work with what we've got.
        local timeMs = numSecs * 1000

        local cmd = "starship prompt --status " .. status .. " --jobs " .. numJobs .. " --cmd-duration " .. timeMs
        local res = io.popen(cmd)
        hilbish.prompt(res:read("*all"))
end

function starship.doCommandExit(code, cmdStr)
        local curTime = os.time()
        local timeElapsed = 0

        if starship.execTime[cmdStr] ~= nil then
                timeElapsed = curTime - starship.execTime[cmdStr]
        end
        starship.execTime[cmdStr] = nil

        local jobs = hilbish.jobs.all()
        starship.doPrompt(code, #jobs, timeElapsed)
end

function starship.doCommandPreExec(input)
        starship.execTime[input] = os.time()
end

function starship.setup()
        local cmd = "starship prompt --continuation"
        local res = io.popen(cmd)
        hilbish.multiprompt(res:read("*all"))

        starship.doPrompt(0, 0, 0)

        bait.catch("command.preexec", function(input, _)
                starship.doCommandPreExec(input)
        end)

        bait.catch('command.exit', function(code, cmdStr)
                starship.doCommandExit(code, cmdStr)
        end)
end

This would be used by running the starship.setup() function.

I'm sure there's a better way to do this, but this seems to work pretty well for me.

[TorchedSammy]

nice! preferably someone would use promptua instead but that has some missing features so..

[jdugan6240]

Yeah, promptua would be good if one wanted to keep their prompt entirely Lua-based (although as you said, it could use some more features). This was only meant as an alternative 😊.

I mainly created this because I also use Starship to create the status line for my text editor, so reusing a piece of software I already had installed made sense.