diff --git a/bash_completions.go b/bash_completions.go index 8820ba8..59e90f9 100644 --- a/bash_completions.go +++ b/bash_completions.go @@ -225,7 +225,7 @@ __handle_command() fi c=$((c+1)) __debug "${FUNCNAME[0]}: looking for ${next_command}" - declare -F $next_command >/dev/null && $next_command + declare -F "$next_command" >/dev/null && $next_command } __handle_word() diff --git a/command.go b/command.go index 664bf5a..a5b47d0 100644 --- a/command.go +++ b/command.go @@ -113,7 +113,7 @@ type Command struct { flagErrorBuf *bytes.Buffer args []string // actual args parsed from flags - output *io.Writer // out writer if set in SetOutput(w) + output io.Writer // out writer if set in SetOutput(w) usageFunc func(*Command) error // Usage can be defined by application usageTemplate string // Can be defined by Application flagErrorFunc func(*Command, error) error @@ -141,7 +141,7 @@ func (c *Command) SetArgs(a []string) { // SetOutput sets the destination for usage and error messages. // If output is nil, os.Stderr is used. func (c *Command) SetOutput(output io.Writer) { - c.output = &output + c.output = output } // SetUsageFunc sets usage function. Usage can be defined by application. @@ -199,7 +199,7 @@ func (c *Command) OutOrStderr() io.Writer { func (c *Command) getOut(def io.Writer) io.Writer { if c.output != nil { - return *c.output + return c.output } if c.HasParent() { return c.parent.getOut(def) diff --git a/command_test.go b/command_test.go index 7947f6c..8b78ee3 100644 --- a/command_test.go +++ b/command_test.go @@ -191,6 +191,14 @@ func TestEnableCommandSortingIsDisabled(t *testing.T) { EnableCommandSorting = true } +func TestSetOutput(t *testing.T) { + cmd := &Command{} + cmd.SetOutput(nil) + if out := cmd.OutOrStdout(); out != os.Stdout { + t.Fatalf("expected setting output to nil to revert back to stdout, got %v", out) + } +} + func TestFlagErrorFunc(t *testing.T) { cmd := &Command{