pancap/output/output.go

package output

import (
	"fmt"
	"github.com/fatih/color"
)

const (
	MaxContentLines = 50
)

// 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) {
	// 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
	cutCont := cutContent(content)
	fmt.Print(cutCont)
}

// 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.
				return content[:i+1] // +1 to include the last newline as well
			}
		}
	}

	// We are done before reaching the cut limit; return the whole content
	return content
}