Julia script to run processes with 2 second timeout

examples.txt

❯ ./timeout.jl ps
  PID TTY           TIME CMD
  697 ttys000    0:00.16 -zsh
 1132 ttys001    0:00.09 /bin/zsh --login
 3115 ttys002    0:00.09 /bin/zsh -l
 3632 ttys003    0:00.32 /bin/zsh -l
11816 ttys003    0:00.32 /Users/eph/.asdf/installs/julia/1.7.1/bin/julia --color=yes --startup-file=no -O0 --compile=min ./timeout.jl ps

❯ ./timeout.jl sleep 3 && echo "done"

❯ ./timeout.jl sleep 1 && echo "done"
done

timeout.jl

#!/bin/bash
#=
exec julia --color=yes --startup-file=no -O0 --compile=min "${BASH_SOURCE[0]}" "$@"
=#

function trykill(p, n=Base.SIGTERM)
    try
        kill(p, n)
        true
    catch e
        false
    end
end

out = PipeBuffer()
err = PipeBuffer()
p = run(pipeline(`$ARGS`, stdout=out, stderr=err), wait=false)
result = timedwait(() -> !process_running(p), 2) # 2 is the time in seconds to wait
if result == :timed_out
    success = trykill(p)
    success && exit(1)
    success = trykill(p, Base.SIGKILL)
    success && exit(1)
    error("Zombie! Could not kill process with pid $(getpid(p))")
else
    write(stdout, take!(out))
    write(stderr, take!(err))
    exit(0)
end