Compare commits
6 Commits
Author | SHA1 | Date | |
---|---|---|---|
9750ac16b3 | |||
7b274bfb98 | |||
e0eb8fd4c2 | |||
d368e85e33 | |||
49342818ec | |||
95f3bc48bd |
81
cliutils.nim
81
cliutils.nim
@ -1,4 +1,4 @@
|
||||
import docopt, json, osproc, posix, nre, streams, strtabs, unicode
|
||||
import docopt, json, osproc, posix, nre, streams, strtabs, tables, terminal, unicode
|
||||
import os except sleep
|
||||
import strutils except toUpper, toLower
|
||||
|
||||
@ -8,6 +8,8 @@ type
|
||||
docopt*: Table[string, Value]
|
||||
json*: JsonNode
|
||||
|
||||
TermColor = ForegroundColor or BackgroundColor
|
||||
|
||||
proc getVal*(cfg: CombinedConfig, key, default: string): string =
|
||||
let argKey = "--" & key
|
||||
let envKey = key.replace('-', '_').toUpper
|
||||
@ -15,7 +17,16 @@ proc getVal*(cfg: CombinedConfig, key, default: string): string =
|
||||
|
||||
if cfg.docopt[argKey]: return $cfg.docopt[argKey]
|
||||
elif existsEnv(envKey): return getEnv(envKey)
|
||||
elif cfg.json.hasKey(jsonKey): return cfg.json[jsonKey].getStr
|
||||
elif cfg.json.hasKey(jsonKey):
|
||||
let node = cfg.json[jsonKey]
|
||||
case node.kind
|
||||
of JString: return node.getStr
|
||||
of JInt: return $node.getInt
|
||||
of JFloat: return $node.getFloat
|
||||
of JBool: return $node.getBool
|
||||
of JNull: return ""
|
||||
of JObject: return $node
|
||||
of JArray: return $node
|
||||
else: return default
|
||||
|
||||
|
||||
@ -30,20 +41,20 @@ type HandleProcMsgCB* = proc (outMsg: TaintedString,
|
||||
errMsg: TaintedString, cmd: string): void
|
||||
|
||||
proc sendMsg*(h: HandleProcMsgCB, outMsg: TaintedString,
|
||||
errMsg: TaintedString = nil, cmd: string = ""): void =
|
||||
errMsg: TaintedString = "", cmd: string = ""): void =
|
||||
if h != nil: h(outMsg, errMsg, cmd)
|
||||
|
||||
proc makeProcMsgHandler*(outSink, errSink: File, prefixCmd: bool = true): HandleProcMsgCB =
|
||||
result = proc(outMsg, errMsg: TaintedString, cmd: string): void {.closure.} =
|
||||
let prefix = if cmd == nil or not prefixCmd: "" else: cmd & ": "
|
||||
if outMsg != nil: outSink.writeLine(prefix & outMsg)
|
||||
if errMsg != nil: errSink.writeLine(prefix & errMsg)
|
||||
let prefix = if cmd.len == 0 or not prefixCmd: "" else: cmd & ": "
|
||||
if outMsg.len > 0: outSink.writeLine(prefix & outMsg)
|
||||
if errMsg.len > 0: errSink.writeLine(prefix & errMsg)
|
||||
|
||||
proc makeProcMsgHandler*(outSink, errSink: Stream, prefixCmd: bool = true): HandleProcMsgCB =
|
||||
result = proc(outMsg, errMsg: TaintedString, cmd: string): void {.closure.} =
|
||||
let prefix = if cmd == nil or not prefixCmd: "" else: cmd & ": "
|
||||
if outMsg != nil: outSink.writeLine(prefix & outMsg)
|
||||
if errMsg != nil: errSink.writeLine(prefix & errMsg)
|
||||
let prefix = if cmd.len == 0 or not prefixCmd: "" else: cmd & ": "
|
||||
if outMsg.len > 0: outSink.writeLine(prefix & outMsg)
|
||||
if errMsg.len > 0: errSink.writeLine(prefix & errMsg)
|
||||
|
||||
proc combineProcMsgHandlers*(a, b: HandleProcMsgCB): HandleProcMsgCB =
|
||||
if a == nil: result = b
|
||||
@ -61,9 +72,9 @@ proc waitFor*(p: Process, msgCB: HandleProcMsgCB, procCmd: string = ""): int =
|
||||
var line = newStringOfCap(120).TaintedString
|
||||
while true:
|
||||
if pout.readLine(line):
|
||||
msgCB.sendMsg(line, nil, procCmd)
|
||||
msgCB.sendMsg(line, "", procCmd)
|
||||
elif perr.readLine(line):
|
||||
msgCB.sendMsg(nil, line, procCmd)
|
||||
msgCB.sendMsg("", line, procCmd)
|
||||
else:
|
||||
result = peekExitCode(p)
|
||||
if result != -1: break
|
||||
@ -82,15 +93,15 @@ proc execWithOutput*(command: string, workingDir:string = "",
|
||||
args: openArray[string] = [], env: StringTableRef = nil,
|
||||
options: set[ProcessOption] = {poUsePath},
|
||||
msgCB: HandleProcMsgCB = nil):
|
||||
tuple[output: TaintedString, error: TaintedString, exitCode: int] =
|
||||
tuple[output, error: TaintedString, exitCode: int] =
|
||||
|
||||
result = (TaintedString"", TaintedString"", -1)
|
||||
var outSeq, errSeq: seq[TaintedString]
|
||||
outSeq = @[]; errSeq = @[]
|
||||
var outputCollector = combineProcMsgHandlers(msgCB,
|
||||
proc(outMsg, errMsg: TaintedString, cmd: string): void {.closure.} =
|
||||
if outMsg != nil: outSeq.add(outMsg)
|
||||
if errMsg != nil: errSeq.add(errMsg))
|
||||
if outMsg.len > 0: outSeq.add(outMsg)
|
||||
if errMsg.len > 0: errSeq.add(errMsg))
|
||||
|
||||
result[2] = exec(command, workingDir, args, env, options, outputCollector)
|
||||
result[0] = outSeq.join("\n")
|
||||
@ -136,15 +147,15 @@ proc daemonize*(pidfile, si, so, se: string, daemonMain: proc(): void): Pid =
|
||||
flushFile(stdout)
|
||||
flushFile(stderr)
|
||||
|
||||
if not si.isNil and si != "":
|
||||
if si.len > 0:
|
||||
fi = open(si, fmRead)
|
||||
discard dup2(getFileHandle(fi), getFileHandle(stdin))
|
||||
|
||||
if not so.isNil and so != "":
|
||||
if so.len > 0:
|
||||
fo = open(so, fmAppend)
|
||||
discard dup2(getFileHandle(fo), getFileHandle(stdout))
|
||||
|
||||
if not se.isNil and so != "":
|
||||
if se.len > 0:
|
||||
fe = open(se, fmAppend)
|
||||
discard dup2(getFileHandle(fe), getFileHandle(stderr))
|
||||
|
||||
@ -162,3 +173,39 @@ proc daemonize*(pidfile, si, so, se: string, daemonMain: proc(): void): Pid =
|
||||
daemonMain()
|
||||
|
||||
return pid1
|
||||
|
||||
const termReset* = "\e[0;m"
|
||||
|
||||
proc termColor*(color: TermColor, bright, bold = false): string =
|
||||
var colorVal = ord(color)
|
||||
if bright: inc(colorVal, 60)
|
||||
return "\e[" & $colorVal & (if bold: ";1" else: "") & "m"
|
||||
|
||||
proc withColor*(str: string, color: TermColor, bright, bold = false): string =
|
||||
return termColor(color, bright, bold) & str
|
||||
|
||||
let STRIP_ANSI_REGEX = re"\x1B\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]"
|
||||
|
||||
proc stripAnsi*(str: string): string = return str.replace(STRIP_ANSI_REGEX, "")
|
||||
|
||||
proc queryParamsToCliArgs*(queryParams: Table[string, string]): seq[string] =
|
||||
result = @[]
|
||||
|
||||
for k,v in queryParams:
|
||||
# support ?arg1=val1&arg2=val2 -> cmd val1 val2
|
||||
if k.startsWith("arg"): result.add(v)
|
||||
|
||||
else :
|
||||
result[1].add("--" & k)
|
||||
if v != "true": result[1].add(v) # support things like ?verbose=true -> cmd --verbose
|
||||
|
||||
proc queryParamsToCliArgs*(queryParams: StringTableRef): seq[string] =
|
||||
result = @[]
|
||||
|
||||
for k,v in queryParams:
|
||||
# support ?arg1=val1&arg2=val2 -> cmd val1 val2
|
||||
if k.startsWith("arg"): result.add(v)
|
||||
|
||||
else :
|
||||
result[1].add("--" & k)
|
||||
if v != "true": result[1].add(v) # support things like ?verbose=true -> cmd --verbose
|
||||
|
@ -1,11 +1,11 @@
|
||||
# Package
|
||||
|
||||
version = "0.3.4"
|
||||
version = "0.6.1"
|
||||
author = "Jonathan Bernard"
|
||||
description = "Helper functions for writing command line interfaces."
|
||||
license = "MIT"
|
||||
|
||||
# Dependencies
|
||||
|
||||
requires @["nim >= 0.18.0", "docopt"]
|
||||
requires @["nim >= 0.19.0", "docopt >= 0.6.8"]
|
||||
|
||||
|
Reference in New Issue
Block a user