package output import ( "fmt" "github.com/fatih/color" ) const ( MaxContentLines = 50 SnipMark = "----- cut at 50 entries -----\n" ) var ( DidSnip 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) { // 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) } // And print our content. fmt.Print(content) } // 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) return content + "\n" + printer.Sprint(SnipMark) }