container.go

package cli

import (
	"github.com/integrii/flaggy"
	"go.uber.org/zap"
)

type CommandContainer interface {
	AddCommand(Command)
	Flaggy() *flaggy.Subcommand
	Run(log *zap.Logger, opts *GlobalOptions) error
	AsCommand() Command
}

var _ Command = &cmdContainer{}

type cmdContainer struct {
	cmd  *flaggy.Subcommand
	cmds []Command
}

func NewCommandContainer(name, description string) CommandContainer {
	cmd := flaggy.NewSubcommand(name)
	cmd.Description = description
	return &cmdContainer{
		cmd: cmd,
	}
}

func (c *cmdContainer) AddCommand(cmd Command) {
	c.cmds = append(c.cmds, cmd)
	c.cmd.AttachSubcommand(cmd.Flaggy(), 1)
}

func (c *cmdContainer) Flaggy() *flaggy.Subcommand {
	return c.cmd
}

func (c *cmdContainer) Run(log *zap.Logger, opts *GlobalOptions) error {
	for _, cmd := range c.cmds {
		if cmd.Flaggy().Used {
			return cmd.Run(log, opts)
		}
	}
	flaggy.ShowHelpAndExit("No command specified")
	return nil
}

func (c *cmdContainer) AsCommand() Command {
	return c
}