58 lines
2.0 KiB
Nim
58 lines
2.0 KiB
Nim
|
import terminal, times, strformat, strutils
|
||
|
|
import std/[dirs, paths]
|
||
|
|
import ../../common/[types, profile]
|
||
|
|
|
||
|
|
proc makeAgentLogDirectory*(cq: Conquest, agentId: string): bool =
|
||
|
|
try:
|
||
|
|
let cqDir = cq.profile.getString("conquest_directory")
|
||
|
|
createDir(cast[Path](fmt"{cqDir}/data/logs/{agentId}"))
|
||
|
|
return true
|
||
|
|
except OSError:
|
||
|
|
return false
|
||
|
|
|
||
|
|
proc log*(cq: Conquest, logEntry: string) =
|
||
|
|
if cq.interactAgent == nil:
|
||
|
|
return
|
||
|
|
|
||
|
|
let
|
||
|
|
date = now().format("dd-MM-yyyy")
|
||
|
|
timestamp = now().format("dd-MM-yyyy HH:mm:ss")
|
||
|
|
cqDir = cq.profile.getString("conquest_directory")
|
||
|
|
agentLogPath = fmt"{cqDir}/data/logs/{cq.interactAgent.agentId}/{date}.log"
|
||
|
|
|
||
|
|
# Write log entry to file
|
||
|
|
let file = open(agentLogPath, fmAppend)
|
||
|
|
file.writeLine(fmt"[{timestamp}] {logEntry}")
|
||
|
|
file.flushFile()
|
||
|
|
|
||
|
|
proc extractStrings(args: string): string =
|
||
|
|
if not args.startsWith("("):
|
||
|
|
return args
|
||
|
|
|
||
|
|
# Remove styling arguments, such as fgRed, styleBright, resetStyle, etc. by extracting only arguments that are quoted
|
||
|
|
var message: string
|
||
|
|
for str in args[1..^2].split(", "):
|
||
|
|
if str.startsWith("\""):
|
||
|
|
message &= str
|
||
|
|
return message.replace("\"", "")
|
||
|
|
|
||
|
|
template info*(cq: Conquest, args: varargs[untyped]) =
|
||
|
|
cq.writeLine(fgBlack, styleBright, "[*] ", resetStyle, args)
|
||
|
|
cq.log("[*] " & extractStrings($(args)))
|
||
|
|
|
||
|
|
template error*(cq: Conquest, args: varargs[untyped]) =
|
||
|
|
cq.writeLine(fgRed, styleBright, "[-] ", resetStyle, args)
|
||
|
|
cq.log("[-] " & extractStrings($(args)))
|
||
|
|
|
||
|
|
template warn*(cq: Conquest, args: varargs[untyped]) =
|
||
|
|
cq.writeLine(fgYellow, "[!] ", resetStyle, args)
|
||
|
|
cq.log("[!] " & extractStrings($(args)))
|
||
|
|
|
||
|
|
template success*(cq: Conquest, args: varargs[untyped]) =
|
||
|
|
cq.writeLine(fgGreen, "[+] ", resetStyle, args)
|
||
|
|
cq.log("[+] " & extractStrings($(args)))
|
||
|
|
|
||
|
|
template output*(cq: Conquest, args: varargs[untyped]) =
|
||
|
|
cq.writeLine(args)
|
||
|
|
cq.log("[>] " & extractStrings($(args)))
|