36 lines
1.1 KiB
Nim
36 lines
1.1 KiB
Nim
import cliutils, docopt, linenoise, os, osproc, sequtils, strutils
|
|
|
|
proc writeOutput(msg, errMsg: TaintedString, cmd: string): void =
|
|
if not msg.isEmptyOrWhitespace: stdout.writeLine(msg)
|
|
if not errMsg.isEmptyOrWhitespace: stderr.writeLine(errMsg)
|
|
|
|
when isMainModule:
|
|
|
|
let usage = """
|
|
Usage:
|
|
cmd_shell [--prepend-args] <cmd> [-- <args>...]
|
|
"""
|
|
# Parse arguments
|
|
let args = docopt(usage, version = "1.1.0")
|
|
|
|
let cmdPrompt = $args["<cmd>"]
|
|
let cmd = '"' & cmdPrompt & '"'
|
|
|
|
let argsForCmd =
|
|
if args["<args>"]: @(args["<args>"]).mapIt("'" & $it & "'").join(" ")
|
|
else: ""
|
|
|
|
discard historySetMaxLen(
|
|
if existsEnv("HISTSIZE"): (cint)parseInt(getEnv("HISTSIZE"))
|
|
else: (cint)1000)
|
|
|
|
var line = readLine(cmdPrompt & "> ")
|
|
|
|
while line != nil:
|
|
if args["--prepend-args"]:
|
|
discard exec(cmd & " " & argsForCmd & " " & $line, "", [], nil, {poUsePath, poEvalCommand}, writeOutput)
|
|
else:
|
|
discard exec(cmd & " " & $line & " " & argsForCmd, "", [], nil, {poUsePath, poEvalCommand}, writeOutput)
|
|
historyAdd(line)
|
|
line = readLine(cmdPrompt & "> ")
|