Update #255

Author: olekukonko

config.go

@@ -8,10 +8,16 @@ import (
 	"reflect"                              // Reflection for type handling
 )
 
+type Control struct {
+	Hide tw.State
+}
+
 // Behavior defines table behavior settings that control features like auto-hiding columns and trimming spaces.
 type Behavior struct {
 	AutoHide  tw.State // Controls whether empty columns are automatically hidden (ignored in streaming mode)
 	TrimSpace tw.State // Controls whether leading/trailing spaces are trimmed from cell content
+	Header    Control
+	Footer    Control
 }
 
 // ColumnConfigBuilder is used to configure settings for a specific column across all table sections (header, row, footer).
@@ -908,18 +914,40 @@ func WithSymbols(symbols tw.Symbols) Option {
 	}
 }
 
-// WithTrimSpace enables or disables automatic trimming of leading/trailing spaces.
+// WithTrimSpace sets whether leading and trailing spaces are automatically trimmed.
 // Logs the change if debugging is enabled.
 func WithTrimSpace(state tw.State) Option {
 	return func(target *Table) {
 		target.config.Behavior.TrimSpace = state
 		if target.logger != nil {
-			target.logger.Debugf("Option: WithAutoHide applied to Table: %v", state)
+			target.logger.Debugf("Option: WithTrimSpace applied to Table: %v", state)
+		}
+	}
+}
+
+// WithHeaderControl sets the control behavior for the table header.
+// Logs the change if debugging is enabled.
+func WithHeaderControl(control Control) Option {
+	return func(target *Table) {
+		target.config.Behavior.Header = control
+		if target.logger != nil {
+			target.logger.Debugf("Option: WithHeaderControl applied to Table: %v", control) // Fixed 'state' to 'control'
+		}
+	}
+}
+
+// WithFooterControl sets the control behavior for the table footer.
+// Logs the change if debugging is enabled.
+func WithFooterControl(control Control) Option {
+	return func(target *Table) {
+		target.config.Behavior.Footer = control
+		if target.logger != nil {
+			target.logger.Debugf("Option: WithFooterControl applied to Table: %v", control) // Fixed log message and 'state' to 'control'
 		}
 	}
 }
 
-// WithAlignment helps to set default alignments
+// WithAlignment sets the default column alignment for the header, rows, and footer.
 func WithAlignment(alignment tw.Alignment) Option {
 	return func(target *Table) {
 		target.config.Header.ColumnAligns = alignment
@@ -928,7 +956,7 @@ func WithAlignment(alignment tw.Alignment) Option {
 	}
 }
 
-// WithPadding helps to set global padding
+// WithPadding sets the global padding for the header, rows, and footer.
 func WithPadding(padding tw.Padding) Option {
 	return func(target *Table) {
 		target.config.Header.Padding.Global = padding

tablewriter.go

@@ -266,8 +266,12 @@ func (t *Table) Debug() *bytes.Buffer {
 // In streaming mode, this processes and renders the header immediately.
 func (t *Table) Header(elements ...any) {
 	t.ensureInitialized()
-	t.logger.Debugf("Header() method called with raw variadic elements: %v (len %d). Streaming: %v, Started: %v",
-		elements, len(elements), t.config.Stream.Enable, t.hasPrinted)
+	t.logger.Debugf("Header() method called with raw variadic elements: %v (len %d). Streaming: %v, Started: %v", elements, len(elements), t.config.Stream.Enable, t.hasPrinted)
+
+	// just forget
+	if t.config.Behavior.Header.Hide.Enabled() {
+		return
+	}
 
 	if t.config.Stream.Enable && t.hasPrinted {
 		//  Streaming Path
@@ -300,14 +304,13 @@ func (t *Table) Header(elements ...any) {
 	preparedHeaderLines := t.prepareContent(headersAsStrings, t.config.Header)
 	t.headers = preparedHeaderLines // Store directly. Padding to t.maxColumns() will happen in prepareContexts.
 
-	t.logger.Debugf("Header set (batch mode), lines stored: %d. First line if exists: %v",
-		len(t.headers), func() []string {
-			if len(t.headers) > 0 {
-				return t.headers[0]
-			} else {
-				return nil
-			}
-		}())
+	t.logger.Debugf("Header set (batch mode), lines stored: %d. First line if exists: %v", len(t.headers), func() []string {
+		if len(t.headers) > 0 {
+			return t.headers[0]
+		} else {
+			return nil
+		}
+	}())
 }
 
 // Footer sets the table's footer content, padding to match column count.
@@ -318,8 +321,12 @@ func (t *Table) Header(elements ...any) {
 // In streaming mode, this processes and stores the footer for rendering by Close().
 func (t *Table) Footer(elements ...any) {
 	t.ensureInitialized()
-	t.logger.Debugf("Footer() method called with raw variadic elements: %v (len %d). Streaming: %v, Started: %v",
-		elements, len(elements), t.config.Stream.Enable, t.hasPrinted)
+	t.logger.Debugf("Footer() method called with raw variadic elements: %v (len %d). Streaming: %v, Started: %v", elements, len(elements), t.config.Stream.Enable, t.hasPrinted)
+
+	// just forget
+	if t.config.Behavior.Footer.Hide.Enabled() {
+		return
+	}
 
 	if t.config.Stream.Enable && t.hasPrinted {
 		//  Streaming Path

tests/extra_test.go

@@ -452,3 +452,61 @@ func TestSpaces(t *testing.T) {
 	})
 
 }
+
+func TestControl(t *testing.T) {
+	var buf bytes.Buffer
+	var data = [][]string{
+		{"No", "Age", "    City"},
+		{"    1", "25", "New York"},
+		{"2", "30", "x"},
+		{"       3", "28", "     Lagos"},
+	}
+	t.Run("Trim", func(t *testing.T) {
+		buf.Reset()
+		table := tablewriter.NewTable(&buf,
+			tablewriter.WithDebug(false),
+			tablewriter.WithTrimSpace(tw.On),
+			tablewriter.WithHeaderControl(tablewriter.Control{Hide: tw.On}),
+		)
+		table.Header(data[0])
+		table.Bulk(data[1:])
+		table.Render()
+
+		expected := `
+		┌───┬────┬──────────┐
+		│ 1 │ 25 │ New York │
+		│ 2 │ 30 │ x        │
+		│ 3 │ 28 │ Lagos    │
+		└───┴────┴──────────┘
+
+`
+		if !visualCheck(t, "UnicodeTableRendering", buf.String(), expected) {
+			t.Log(table.Debug())
+		}
+	})
+
+	t.Run("NoTrim", func(t *testing.T) {
+		buf.Reset()
+		table := tablewriter.NewTable(&buf,
+			tablewriter.WithTrimSpace(tw.On),
+			tablewriter.WithHeaderControl(tablewriter.Control{Hide: tw.Off}),
+		)
+		table.Header(data[0])
+		table.Bulk(data[1:])
+		table.Render()
+
+		expected := `
+	┌────┬─────┬──────────┐
+	│ NO │ AGE │   CITY   │
+	├────┼─────┼──────────┤
+	│ 1  │ 25  │ New York │
+	│ 2  │ 30  │ x        │
+	│ 3  │ 28  │ Lagos    │
+	└────┴─────┴──────────┘
+
+
+`
+		visualCheck(t, "UnicodeTableRendering", buf.String(), expected)
+	})
+
+}