123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- /*
- Copyright 2016 The Kubernetes 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 templates
- import (
- "github.com/spf13/cobra"
- )
- type CommandGroup struct {
- Message string
- Commands []*cobra.Command
- }
- type CommandGroups []CommandGroup
- func (g CommandGroups) Add(c *cobra.Command) {
- for _, group := range g {
- c.AddCommand(group.Commands...)
- }
- }
- func (g CommandGroups) Has(c *cobra.Command) bool {
- for _, group := range g {
- for _, command := range group.Commands {
- if command == c {
- return true
- }
- }
- }
- return false
- }
- func AddAdditionalCommands(g CommandGroups, message string, cmds []*cobra.Command) CommandGroups {
- group := CommandGroup{Message: message}
- for _, c := range cmds {
- // Don't show commands that have no short description
- if !g.Has(c) && len(c.Short) != 0 {
- group.Commands = append(group.Commands, c)
- }
- }
- if len(group.Commands) == 0 {
- return g
- }
- return append(g, group)
- }
|