Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bool flags in the paired --foo and --no-foo format #346

Merged
merged 7 commits into from
Aug 14, 2019
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion pkg/kn/core/root.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ import (
"github.com/knative/client/pkg/kn/commands/revision"
"github.com/knative/client/pkg/kn/commands/route"
"github.com/knative/client/pkg/kn/commands/service"
"github.com/knative/client/pkg/kn/flags"
homedir "github.com/mitchellh/go-homedir"
"github.com/spf13/cobra"
"github.com/spf13/viper"
Expand Down Expand Up @@ -115,6 +116,10 @@ func NewKnCommand(params ...commands.KnParams) *cobra.Command {

// Prevents Cobra from dealing with errors as we deal with them in main.go
SilenceErrors: true,

PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
return flags.ReconcileBoolFlags(cmd.Flags())
},
}
if p.Output != nil {
rootCmd.SetOutput(p.Output)
Expand All @@ -123,7 +128,7 @@ func NewKnCommand(params ...commands.KnParams) *cobra.Command {
// Persistent flags
rootCmd.PersistentFlags().StringVar(&commands.CfgFile, "config", "", "kn config file (default is $HOME/.kn/config.yaml)")
rootCmd.PersistentFlags().StringVar(&p.KubeCfgPath, "kubeconfig", "", "kubectl config file (default is $HOME/.kube/config)")
rootCmd.PersistentFlags().BoolVar(&p.LogHttp, "log-http", false, "log http traffic")
flags.AddBothBoolFlags(rootCmd.PersistentFlags(), &p.LogHttp, "log-http", "", false, "log http traffic")

plugin.AddPluginFlags(rootCmd)
plugin.BindPluginsFlagToViper(rootCmd)
Expand Down
104 changes: 104 additions & 0 deletions pkg/kn/flags/bool.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
// Copyright © 2019 The Knative Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package flags

import (
"fmt"
"strconv"
"strings"

"github.com/spf13/pflag"
)

var negPrefix = "no-"

// AddBothBoolFlags adds the given flag in both `--foo` and `--no-foo` variants.
// If you do this, make sure you call ReconcileBoolFlags later to catch errors and
// set the relationship between the flag values.
func AddBothBoolFlags(f *pflag.FlagSet, p *bool, name, short string, value bool, usage string) {

negativeName := negPrefix + name

f.BoolVarP(p, name, short, value, usage)
sixolet marked this conversation as resolved.
Show resolved Hide resolved
f.Bool(negativeName, !value, "do not "+usage)

if value {
sixolet marked this conversation as resolved.
Show resolved Hide resolved
err := f.MarkHidden(name)
rhuss marked this conversation as resolved.
Show resolved Hide resolved
if err != nil {
panic(err)
}
} else {
err := f.MarkHidden(negativeName)
if err != nil {
panic(err)
}
}
}

// ReconcileBoolFlags sets the value of the all the "--foo" flags based on
// "--no-foo" if provided, and returns an error if both were provided or an
// explicit value of false was provided to either (as that's confusing).
sixolet marked this conversation as resolved.
Show resolved Hide resolved
func ReconcileBoolFlags(f *pflag.FlagSet) error {
var err error
f.VisitAll(func(flag *pflag.Flag) {
// Return early from our comprehension
if err != nil {
return
sixolet marked this conversation as resolved.
Show resolved Hide resolved
}
// Walk the "no-" versions of the flags. Make sure we didn't set
// both, and set the positive value to the opposite of the "no-"
// value if it exists.
if strings.HasPrefix(flag.Name, "no-") {
sixolet marked this conversation as resolved.
Show resolved Hide resolved
sixolet marked this conversation as resolved.
Show resolved Hide resolved
positiveName := flag.Name[len(negPrefix):len(flag.Name)]
sixolet marked this conversation as resolved.
Show resolved Hide resolved
positive := f.Lookup(positiveName)
sixolet marked this conversation as resolved.
Show resolved Hide resolved
if flag.Changed {
if positive.Changed {
err = fmt.Errorf("only one of --%s and --%s may be specified",
flag.Name, positiveName)
return
}
var noValue bool
noValue, err = strconv.ParseBool(flag.Value.String())
if err != nil {
return
}
if !noValue {
err = fmt.Errorf("use --%s instead of providing false to --%s",
navidshaikh marked this conversation as resolved.
Show resolved Hide resolved
positiveName, flag.Name)
if err != nil {
sixolet marked this conversation as resolved.
Show resolved Hide resolved
return
}
}
err = positive.Value.Set(strconv.FormatBool(!noValue))
} else if positive.Changed {
// For the positive version, just check it wasn't set to the
// confusing "false" value.
var yesValue bool
yesValue, err = strconv.ParseBool(positive.Value.String())
sixolet marked this conversation as resolved.
Show resolved Hide resolved
if err != nil {
return
}
if !yesValue {
err = fmt.Errorf("use --%s instead of providing false to --%s",
flag.Name, positiveName)
if err != nil {
sixolet marked this conversation as resolved.
Show resolved Hide resolved
return
}
}
}
}
})
return err
}
72 changes: 72 additions & 0 deletions pkg/kn/flags/bool_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
// Copyright © 2019 The Knative Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package flags

import (
"testing"

"github.com/spf13/pflag"
"gotest.tools/assert"
"gotest.tools/assert/cmp"
)

type boolPairTestCase struct {
name string
defaultVal bool
flags []string
expectedResult bool
expectedErrText string
}

func TestBooleanPair(t *testing.T) {
cases := []*boolPairTestCase{
{"foo", true, []string{}, true, ""},
{"foo", true, []string{"--foo"}, true, ""},
{"foo", true, []string{"--no-foo"}, false, ""},
{"foo", false, []string{"--foo"}, true, ""},
{"foo", false, []string{}, false, ""},
{"foo", false, []string{"--no-foo"}, false, ""},
{"foo", true, []string{"--foo", "--no-foo"}, false, "only one of"},
rhuss marked this conversation as resolved.
Show resolved Hide resolved
{"foo", true, []string{"--no-foo", "--foo"}, false, "only one of"},
// Disallow confusing "false" value.
{"foo", true, []string{"--foo=false"}, false, "use --no-foo instead of providing false to --foo"},
{"foo", true, []string{"--no-foo=false"}, false, "use --foo instead of providing false to --no-foo"},

// Ensure tests still pass if positive sorts after no- alphabetically.
{"zoo", true, []string{}, true, ""},
{"zoo", true, []string{"--zoo"}, true, ""},
{"zoo", true, []string{"--no-zoo"}, false, ""},
{"zoo", false, []string{"--zoo"}, true, ""},
{"zoo", false, []string{}, false, ""},
{"zoo", false, []string{"--no-zoo"}, false, ""},
{"zoo", true, []string{"--zoo", "--no-zoo"}, false, "only one of"},
{"zoo", true, []string{"--no-zoo", "--zoo"}, false, "only one of"},
// Disallow confusing "false" value.
{"zoo", true, []string{"--zoo=false"}, false, "use --no-zoo instead of providing false to --zoo"},
{"zoo", true, []string{"--no-zoo=false"}, false, "use --zoo instead of providing false to --no-zoo"},
}
for _, c := range cases {
f := &pflag.FlagSet{}
var result bool
AddBothBoolFlags(f, &result, c.name, "", c.defaultVal, "set "+c.name)
f.Parse(c.flags)
err := ReconcileBoolFlags(f)
if c.expectedErrText != "" {
assert.ErrorContains(t, err, c.expectedErrText)
} else {
assert.Assert(t, cmp.Equal(result, c.expectedResult))
sixolet marked this conversation as resolved.
Show resolved Hide resolved
}
}
}