refactored WithXxx functions that set boolean options

pterm · Sep 19, 2022 · b17aa32 · b17aa32
1 parent 0ca1b0d
commit b17aa32
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/interactive_continue_printer.go b/interactive_continue_printer.go
@@ -10,6 +10,8 @@ import (
 	"atomicgo.dev/keyboard/keys"
 	"golang.org/x/text/cases"
 	"golang.org/x/text/language"
+
+	"github.com/pterm/pterm/internal"
 )
 
 var (
@@ -89,8 +91,8 @@ func (p InteractiveContinuePrinter) WithHandles(handles []string) *InteractiveCo
 
 // WithShowShortHandles will set ShowShortHandles to true
 // this makes the printer display the shorthand options instead their shorthand version.
-func (p InteractiveContinuePrinter) WithShowShortHandles() *InteractiveContinuePrinter {
-	p.ShowShortHandles = true
+func (p InteractiveContinuePrinter) WithShowShortHandles(b ...bool) *InteractiveContinuePrinter {
+	p.ShowShortHandles = internal.WithBoolean(b)
 	return &p
 }
 

diff --git a/panel_printer.go b/panel_printer.go
@@ -58,8 +58,7 @@ func (p PanelPrinter) WithBottomPadding(bottomPadding int) *PanelPrinter {
 
 // WithSameColumnWidth returns a new PanelPrinter with specific options.
 func (p PanelPrinter) WithSameColumnWidth(b ...bool) *PanelPrinter {
-	b2 := internal.WithBoolean(b)
-	p.SameColumnWidth = b2
+	p.SameColumnWidth = internal.WithBoolean(b)
 	return &p
 }