96a8d38be2
- Add party plugin registry with Register-before-Boot activation - Add backpack, compass, bonfire, and pact capability interfaces - Add examples/hello with two compiled plugin modules
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package bonfire
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// NewRoot builds the shared cobra root used by the summer tool and app binaries.
|
|
// It accepts command values rather than a party registry to avoid an import cycle.
|
|
func NewRoot(name string, commands []Command, out io.Writer) *cobra.Command {
|
|
if out == nil {
|
|
out = os.Stdout
|
|
}
|
|
root := &cobra.Command{
|
|
Use: name,
|
|
SilenceUsage: true,
|
|
SilenceErrors: true,
|
|
CompletionOptions: cobra.CompletionOptions{
|
|
DisableDefaultCmd: true,
|
|
},
|
|
}
|
|
root.SetOut(out)
|
|
root.SetErr(out)
|
|
for _, command := range commands {
|
|
root.AddCommand(wrap(command, out))
|
|
}
|
|
return root
|
|
}
|
|
|
|
func wrap(c Command, out io.Writer) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: c.Name,
|
|
Short: c.Description,
|
|
Args: cobra.ArbitraryArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if c.Run == nil {
|
|
return nil
|
|
}
|
|
writer := out
|
|
if cmd != nil {
|
|
if w := cmd.OutOrStdout(); w != nil {
|
|
writer = w
|
|
}
|
|
}
|
|
return c.Run(cmd.Context(), cobraInput{cmd: cmd, spec: c, args: args}, writerOutput{w: writer})
|
|
},
|
|
}
|
|
for _, flag := range c.Flags {
|
|
if flag.Shorthand != "" {
|
|
cmd.Flags().StringP(flag.Name, flag.Shorthand, flag.Default, flag.Description)
|
|
continue
|
|
}
|
|
cmd.Flags().String(flag.Name, flag.Default, flag.Description)
|
|
}
|
|
return cmd
|
|
}
|