// Copyright © 2020 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 templates

import (
	"text/template"

	"github.com/spf13/cobra"
)

// CommandGroup is for grouping together commands
type CommandGroup struct {

	// Title for command group shown in help/usage messages
	Header string

	// List of commands for this group
	Commands []*cobra.Command
}

type CommandGroups []CommandGroup

// AddTo adds all commands from this group slice to the given command
func (g CommandGroups) AddTo(cmd *cobra.Command) {
	for _, group := range g {
		for _, sub := range group.Commands {
			cmd.AddCommand(sub)
		}
	}
}

// SetRootUsage sets our own help and usage function messages to the root command
func (g CommandGroups) SetRootUsage(rootCmd *cobra.Command, extraTemplateFunctions *template.FuncMap) {
	engine := newTemplateEngine(rootCmd, g, extraTemplateFunctions)
	setHelpFlagsToSubCommands(rootCmd)
	rootCmd.SetUsageFunc(engine.usageFunc())
	rootCmd.SetHelpFunc(engine.helpFunc())
}

func setHelpFlagsToSubCommands(parent *cobra.Command) {
	for _, cmd := range parent.Commands() {
		if cmd.HasSubCommands() {
			setHelpFlagsToSubCommands(cmd)
		}
		cmd.DisableFlagsInUseLine = true
	}
}