Initial version: dump file contents or exec command.
This commit is contained in:
commit
e958983e3e
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
log_happy
|
||||||
|
nimcache/
|
239
log_happy.nim
Normal file
239
log_happy.nim
Normal file
@ -0,0 +1,239 @@
|
|||||||
|
## Log Happy
|
||||||
|
## =========
|
||||||
|
##
|
||||||
|
## Little tool to extract expected information from log streams.
|
||||||
|
import json, logging, ncurses, os, osproc, streams, strutils, threadpool
|
||||||
|
import nre except toSeq
|
||||||
|
|
||||||
|
import private/ncurses_ext
|
||||||
|
|
||||||
|
from posix import signal
|
||||||
|
|
||||||
|
type
|
||||||
|
Expectation* = ref object
|
||||||
|
label: string
|
||||||
|
pattern: Regex
|
||||||
|
expected, found: bool
|
||||||
|
|
||||||
|
let expPattern = re"^-([eE])([^;]+);(.+)$"
|
||||||
|
|
||||||
|
var msgChannel: Channel[string]
|
||||||
|
|
||||||
|
proc exitErr(msg: string): void =
|
||||||
|
stderr.writeLine "log_happy: " & msg
|
||||||
|
quit(QuitFailure)
|
||||||
|
|
||||||
|
proc readStream(stream: Stream): void =
|
||||||
|
var line = TaintedString""
|
||||||
|
try:
|
||||||
|
while stream.readLine(line): msgChannel.send(line)
|
||||||
|
except: discard ""
|
||||||
|
|
||||||
|
when isMainModule:
|
||||||
|
var cmdProc: Process
|
||||||
|
try:
|
||||||
|
let usage = """
|
||||||
|
Usage:
|
||||||
|
log_happy [options]
|
||||||
|
log_happy <filename> [options]
|
||||||
|
log_happy [options] -- <cmd>
|
||||||
|
|
||||||
|
Options:
|
||||||
|
|
||||||
|
-e<label>;<pattern> Add something to expect to see in the log stream.
|
||||||
|
-E<label>;<patterh> Add something to expect not to see in the log stream.
|
||||||
|
-d<def-file> Specify a JSON definitions blob.
|
||||||
|
-i<in-file> Read JSON definitions from <in-file>.
|
||||||
|
-f Similar to tail, do not stop when the end of file
|
||||||
|
is reached but wait for additional modifications
|
||||||
|
to the file. -f is ignored when the input is STDIN.
|
||||||
|
|
||||||
|
Expectation definitions take the format LABEL;REGEX where the LABEL is the text
|
||||||
|
label log_happy will use to describe the event, and REGEX is the regular
|
||||||
|
expression log_happy will use to identify the event.
|
||||||
|
|
||||||
|
Expectations JSON definitions follow this format:
|
||||||
|
{
|
||||||
|
"expected": {
|
||||||
|
"<label>": "<regex>",
|
||||||
|
"<label>": "<regex>"
|
||||||
|
},
|
||||||
|
"unexpected": {
|
||||||
|
"<label>": "<regex>",
|
||||||
|
"<label>": "<regex>"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
|
||||||
|
let args = commandLineParams()
|
||||||
|
|
||||||
|
var expectations: seq[Expectation] = @[]
|
||||||
|
var inStream: Stream
|
||||||
|
var haveStream = false
|
||||||
|
var follow = false
|
||||||
|
var cmd = "NOCMD"
|
||||||
|
|
||||||
|
if args.len == 0:
|
||||||
|
echo usage
|
||||||
|
quit(QuitSuccess)
|
||||||
|
|
||||||
|
for arg in args:
|
||||||
|
if cmd != "NOCMD": cmd &= " " & arg
|
||||||
|
|
||||||
|
elif arg.startsWith("-d"):
|
||||||
|
let filename = arg[2..^1]
|
||||||
|
try:
|
||||||
|
let defs = parseFile(filename)
|
||||||
|
|
||||||
|
if defs.hasKey("expected"):
|
||||||
|
for item in defs["expected"].pairs():
|
||||||
|
expectations.add(Expectation(
|
||||||
|
label: item.key,
|
||||||
|
pattern: re(item.val.getStr()),
|
||||||
|
expected: true,
|
||||||
|
found: false))
|
||||||
|
|
||||||
|
if defs.hasKey("unexpected"):
|
||||||
|
for item in defs["unexpected"].pairs():
|
||||||
|
expectations.add(Expectation(
|
||||||
|
label: item.key,
|
||||||
|
pattern: re(item.val.getStr()),
|
||||||
|
expected: false,
|
||||||
|
found: false))
|
||||||
|
except:
|
||||||
|
exitErr "could not parse definitions file (" &
|
||||||
|
filename & "):\n\t" & getCurrentExceptionMsg()
|
||||||
|
|
||||||
|
elif arg.startsWith("-i"):
|
||||||
|
let filename = arg[2..^1]
|
||||||
|
try:
|
||||||
|
if not existsFile(filename):
|
||||||
|
exitErr "no such file (" & filename & ")"
|
||||||
|
|
||||||
|
inStream = newFileStream(filename)
|
||||||
|
haveStream = true
|
||||||
|
except:
|
||||||
|
exitErr "could not open file (" & filename & "):\t\n" &
|
||||||
|
getCurrentExceptionMsg()
|
||||||
|
|
||||||
|
elif arg.match(expPattern).isSome:
|
||||||
|
var m = arg.match(expPattern).get().captures()
|
||||||
|
expectations.add(Expectation(
|
||||||
|
label: m[1],
|
||||||
|
pattern: re(m[2]),
|
||||||
|
expected: m[0] == "e",
|
||||||
|
found: false))
|
||||||
|
|
||||||
|
elif arg == "-f": follow = true
|
||||||
|
elif arg == "--": cmd = ""
|
||||||
|
else: exitErr "unrecognized argument: " & arg
|
||||||
|
|
||||||
|
if cmd == "NOCMD" and not haveStream:
|
||||||
|
exitErr "no input file or command to execute."
|
||||||
|
|
||||||
|
if not haveStream and cmd != "NOCMD":
|
||||||
|
cmdProc = startProcess(cmd, "", [], nil, {poStdErrToStdOut, poEvalCommand, poUsePath})
|
||||||
|
inStream = cmdProc.outputStream
|
||||||
|
|
||||||
|
open(msgChannel)
|
||||||
|
spawn readStream(inStream)
|
||||||
|
|
||||||
|
# Init ncurses
|
||||||
|
let stdscr = initscr()
|
||||||
|
var height, width: int
|
||||||
|
getmaxyx(stdscr, height, width)
|
||||||
|
|
||||||
|
startColor()
|
||||||
|
noecho()
|
||||||
|
cbreak()
|
||||||
|
keypad(stdscr, true)
|
||||||
|
clear()
|
||||||
|
nonl()
|
||||||
|
|
||||||
|
init_pair(1, GREEN, BLACK)
|
||||||
|
init_pair(2, RED, BLACK)
|
||||||
|
|
||||||
|
let dispHeight = max(expectations.len + 1, 2)
|
||||||
|
let logwin = newwin(height - dispHeight, width, 0, 0)
|
||||||
|
let dispwin = newwin(dispHeight, width, height - dispHeight, 0)
|
||||||
|
dispwin.wborder(' ', ' ', '_', ' ', '_', '_', ' ', ' ')
|
||||||
|
dispwin.wrefresh()
|
||||||
|
|
||||||
|
logwin.scrollok(true)
|
||||||
|
logwin.nodelay(true)
|
||||||
|
|
||||||
|
# init run loop
|
||||||
|
var stop = false
|
||||||
|
var firstPass = true
|
||||||
|
var drawDisp = false
|
||||||
|
|
||||||
|
while not stop:
|
||||||
|
|
||||||
|
var lineMsg = msgChannel.tryRecv()
|
||||||
|
|
||||||
|
if lineMsg.dataAvailable:
|
||||||
|
let line = lineMsg.msg
|
||||||
|
for expect in expectations:
|
||||||
|
if not expect.found:
|
||||||
|
if line.find(expect.pattern).isSome:
|
||||||
|
expect.found = true
|
||||||
|
drawDisp = true
|
||||||
|
|
||||||
|
if drawDisp or firstPass:
|
||||||
|
dispwin.wmove(1, 0)
|
||||||
|
for expect in expectations:
|
||||||
|
let text =
|
||||||
|
if expect.found: " FOUND "
|
||||||
|
else: " MISSING "
|
||||||
|
|
||||||
|
if expect.expected == expect.found:
|
||||||
|
dispwin.wattron(COLOR_PAIR(1))
|
||||||
|
dispwin.wprintw(expect.label & text & "\n")
|
||||||
|
dispwin.wattroff(COLOR_PAIR(1))
|
||||||
|
|
||||||
|
else:
|
||||||
|
dispwin.wattron(COLOR_PAIR(2))
|
||||||
|
dispwin.wprintw(expect.label & text & "\n")
|
||||||
|
dispwin.wattroff(COLOR_PAIR(2))
|
||||||
|
|
||||||
|
dispwin.wrefresh()
|
||||||
|
firstPass = false
|
||||||
|
|
||||||
|
logwin.wprintw("\n" & line)
|
||||||
|
logwin.wrefresh()
|
||||||
|
|
||||||
|
let ch = logwin.wgetch()
|
||||||
|
|
||||||
|
# Stop on CTRL-D
|
||||||
|
if ch == 4:
|
||||||
|
stop = true
|
||||||
|
if not cmdProc.isNil and cmdProc.running(): cmdProc.terminate()
|
||||||
|
break
|
||||||
|
|
||||||
|
# Reset on CTRL-R
|
||||||
|
elif ch == 18:
|
||||||
|
for expect in expectations: expect.found = false
|
||||||
|
drawDisp = true
|
||||||
|
|
||||||
|
sleep 1
|
||||||
|
|
||||||
|
# echo $args
|
||||||
|
# echo "\n\n\n"
|
||||||
|
#
|
||||||
|
# for line in lines fin:
|
||||||
|
# stdout.cursorUp(1)
|
||||||
|
# stdout.eraseLine()
|
||||||
|
# stdout.cursorUp(1)
|
||||||
|
# stdout.eraseLine()
|
||||||
|
#
|
||||||
|
# stdout.writeLine line
|
||||||
|
|
||||||
|
# Scan for patterns
|
||||||
|
# Print PASS/FAIL when patterns are seen.
|
||||||
|
except:
|
||||||
|
exitErr getCurrentExceptionMsg()
|
||||||
|
|
||||||
|
finally:
|
||||||
|
if not cmdProc.isNil and cmdProc.running(): cmdProc.kill()
|
||||||
|
close(msgChannel)
|
||||||
|
endwin()
|
1
log_happy.nim.cfg
Normal file
1
log_happy.nim.cfg
Normal file
@ -0,0 +1 @@
|
|||||||
|
--threads:on
|
12
log_happy.nimble
Normal file
12
log_happy.nimble
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
# Package
|
||||||
|
|
||||||
|
version = "0.1.0"
|
||||||
|
author = "Jonathan Bernard"
|
||||||
|
description = "Scan logs for regex-defined events and report on what was found."
|
||||||
|
license = "MIT"
|
||||||
|
bin = @["log_happy"]
|
||||||
|
|
||||||
|
# Dependencies
|
||||||
|
|
||||||
|
requires @["nim >= 0.16.1", "docopt >= 0.6.4", "ncurses"]
|
||||||
|
|
39
private/ncurses_ext.nim
Normal file
39
private/ncurses_ext.nim
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
import ncurses
|
||||||
|
export ncurses
|
||||||
|
|
||||||
|
proc keypad*(win: ptr window, enable: bool): int {.cdecl, discardable, importc: "keypad", dynlib: libncurses.}
|
||||||
|
proc scrollok*(win: ptr window, enable: bool): int {.cdecl, discardable, importc: "scrollok", dynlib: libncurses.}
|
||||||
|
proc nodelay*(win: ptr window, enable: bool): int {.cdecl, discardable, importc: "nodelay", dynlib: libncurses.}
|
||||||
|
proc nonl*(): int {.cdecl, discardable, importc: "nonl", dynlib: libncurses.}
|
||||||
|
|
||||||
|
proc newwin*(num_rows, num_cols, begin_x, begin_y: int): ptr window {.cdecl, discardable, importc: "newwin", dynlib: libncurses.}
|
||||||
|
proc delwin*(win: ptr window): int {.cdecl, discardable, importc: "delwin", dynlib: libncurses.}
|
||||||
|
#proc newwin(num_rows, num_cols, begin_x, begin_y: int): ptr window {.cdecl, discardable, importc: "newwin", dynlib: libncurses.}
|
||||||
|
|
||||||
|
#proc wgetch*(win: ptr window): int {.cdecl, discardable, importc: "wgetch", dynlib: libncurses.}
|
||||||
|
proc wrefresh*(win: ptr window): int {.cdecl, discardable, importc: "wrefresh", dynlib: libncurses.}
|
||||||
|
proc wclear*(win: ptr window): int {.cdecl, discardable, importc: "wclear", dynlib: libncurses.}
|
||||||
|
proc mvwhline*(win: ptr window, rol, col: int, chType: char, n: int): int {.cdecl, discardable, importc: "whline", dynlib: libncurses.}
|
||||||
|
proc wmove*(win: ptr window, rol, col: int): int {.cdecl, discardable, importc: "wmove", dynlib: libncurses.}
|
||||||
|
proc wprintw*(win: ptr window, str: cstring): int {.cdecl, discardable, importc: "wprintw", dynlib: libncurses.}
|
||||||
|
#proc mvwprintw*(win: ptr window, row, col: int, str: cstring): int {.cdecl, discardable, importc: "mvwprintw", dynlib: libncurses.}
|
||||||
|
|
||||||
|
proc box*(win: ptr window, vert_border, horiz_border: char): int {.cdecl, discardable, importc: "box", dynlib: libncurses.}
|
||||||
|
proc wborder*(win: ptr window, ls,rs,ts,bs,tl,tr,bl,br: char): int {.cdecl, discardable, importc: "wborder", dynlib: libncurses.}
|
||||||
|
|
||||||
|
proc init_pair*(n, f, b: int): int {.cdecl, discardable, importc: "init_pair", dynlib: libncurses.}
|
||||||
|
proc COLOR_PAIR*(n: int): int {.cdecl, discardable, importc: "COLOR_PAIR", dynlib: libncurses.}
|
||||||
|
|
||||||
|
proc wattron*(win: ptr window, attrs: int): int {.cdecl, discardable, importc: "wattron", dynlib: libncurses.}
|
||||||
|
proc wattroff*(win: ptr window, attrs: int): int {.cdecl, discardable, importc: "wattron", dynlib: libncurses.}
|
||||||
|
|
||||||
|
const BLACK* = 0
|
||||||
|
const RED* = 1
|
||||||
|
const GREEN* = 2
|
||||||
|
const YELLOW* = 3
|
||||||
|
const BLUE* = 4
|
||||||
|
const MAGENTA* = 5
|
||||||
|
const CYAN* = 6
|
||||||
|
const WHITE* = 7
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user