pancap/output/printer.go

84 lines
2.0 KiB
Go

package output
import (
"fmt"
"github.com/fatih/color"
"strings"
)
const (
MaxContentLines = 50
SnipMark = "----- cut at 50 entries -----"
)
var (
DidSnip bool
DidAvoidEmptyBlock bool
)
// Prints a block of information with the given headline
// If content is empty, printing the headline is omitted.
// If the content is longer than MaxContentLines, content is cut.
func PrintBlock(headline string, content string) {
// Avoid printing empty blocks - at least if user didn't specify it otherwise
if len(content) == 0 && !*printEmptyBlocks {
// No content and we are not forced to print empty blocks, return
DidAvoidEmptyBlock = true
return
}
// Print a newline to add some space between blocks
fmt.Println("")
// Check if we need to print a headline
if len(content) > 0 {
// We have content, we can print the headline
headlineColor := color.New(color.FgRed, color.Bold)
headlineColor.Println(headline)
}
// Cut to MaxContentLines if required
if !(*fullOutput) {
// User states that they don't want to see the whole output - cut content.
content = cutContent(content)
}
// Print our content.
fmt.Print(content)
// Check if we need to add a newline character at the end
if !strings.HasSuffix(content, "\n") {
fmt.Print("\n")
}
}
// Cut content after MaxContentLines lines
func cutContent(content string) string {
numNewlines := 0
// iterate over every character
for i, c := range content {
// check if character is newline
if c == '\n' {
// it is, count occurrence
numNewlines++
// Check if we already hit our limit yet
if numNewlines == MaxContentLines {
// Found nth newline, return content up to this position and add a notice about it.
DidSnip = true
return addSnipMark(content[:i])
}
}
}
// We are done before reaching the cut limit; return the whole content
return content
}
// Adds a notice about the snipping process
func addSnipMark(content string) string {
printer := color.New(color.Bold, color.BgBlack)
return content + "\n" + printer.Sprint(SnipMark) + "\n"
}