107 lines
2.8 KiB
Nim
107 lines
2.8 KiB
Nim
import std/[httpclient, json, logging, smtp, strutils, times]
|
|
import docopt, zero_functional
|
|
import cliutils/[queue_logger, config]
|
|
import stock_watcherpkg/cliconstants
|
|
|
|
type
|
|
Config = object
|
|
cfg: CombinedConfig
|
|
apiKey: string
|
|
apiRoot: string
|
|
notifyAddresses: seq[string]
|
|
smtpHostname: string
|
|
smtpPassword: string
|
|
smtpPort: int
|
|
smtpUsername: string
|
|
|
|
proc sendMessages(
|
|
exitStatus: int,
|
|
cfg: Config,
|
|
messages: seq[QueuedLogMessage],
|
|
dryRun = false
|
|
) {.raises: [].} =
|
|
|
|
try:
|
|
var subjStatus: string
|
|
if dryRun:
|
|
subjStatus = "dry run"
|
|
elif exitStatus == QuitSuccess:
|
|
subjStatus = "success"
|
|
else:
|
|
subjStatus = "FAILED"
|
|
|
|
let dateStr = now().format("yyyy-MM-dd")
|
|
let msg = createMessage(
|
|
"Stock Watcher ($#) -- $#" % [subjStatus, dateStr],
|
|
MESSAGE_TEMPLATE % [ messages.join("\p") ],
|
|
cfg.notifyAddresses)
|
|
|
|
let smtpConn = newSmtp(useSsl = true)
|
|
smtpConn.connect(cfg.smtpHostname, Port cfg.smtpPort)
|
|
smtpConn.auth(cfg.smtpUsername, cfg.smtpPassword)
|
|
smtpConn.sendmail("operations@probatem.com", cfg.notifyAddresses, $msg)
|
|
|
|
except CatchableError:
|
|
try:
|
|
let ex = getCurrentException()
|
|
warn "Failed to send notification email: " & ex.msg
|
|
debug ex.getStackTrace
|
|
except CatchableError: discard
|
|
|
|
proc loadConfig(args: Table[string, Value]): Config =
|
|
let filepath = findConfigFile("stock_watcher")
|
|
|
|
let cfg = initCombinedConfig(filepath, args)
|
|
|
|
result = Config(
|
|
cfg: cfg,
|
|
apiKey: cfg.getVal("api-key"),
|
|
apiRoot: cfg.getVal("api-root"),
|
|
notifyAddresses: cfg.getJson("notify-addresses").getElems --> map(it.getStr),
|
|
smtpHostname: cfg.getVal("smtp-hostname"),
|
|
smtpPassword: cfg.getVal("smtp-password"),
|
|
smtpPort: parseInt(cfg.getVal("smtp-port")),
|
|
smtpUsername: cfg.getVal("smtp-username"))
|
|
|
|
when isMainModule:
|
|
var exitStatus = QuitSuccess
|
|
var cfg: Config
|
|
let args = docopt(USAGE, version = VERSION)
|
|
|
|
let queuedLogger = newQueueLogger(
|
|
levelThreshold = lvlInfo,
|
|
fmtStr = "$levelname -")
|
|
|
|
let consoleLogger = newConsoleLogger(
|
|
levelThreshold = lvlInfo,
|
|
fmtStr = "stock_watcher - $levelname: ")
|
|
|
|
try:
|
|
if args["--debug"]:
|
|
queuedLogger.levelThreshold = lvlDebug
|
|
consoleLogger.levelThreshold = lvlDebug
|
|
|
|
if args["--help"]:
|
|
stderr.writeLine(USAGE)
|
|
quit()
|
|
|
|
logging.addHandler(consoleLogger)
|
|
logging.addHandler(queuedLogger)
|
|
|
|
let cfg = loadConfig(args)
|
|
let http = newHttpClient()
|
|
http.headers = newHttpHeaders({
|
|
"Content-Type": "application/json",
|
|
"Authorization": "Bearer " & cfg.apiKey
|
|
})
|
|
|
|
except CatchableError:
|
|
let ex = getCurrentException()
|
|
fatal ex.msg
|
|
debug ex.getStackTrace
|
|
exitStatus = QuitFailure
|
|
|
|
finally:
|
|
sendMessages(exitStatus, cfg, queuedLogger.messages, args["--dry-run"])
|
|
quit(exitStatus)
|