65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/ian-howell/airshipctl/pkg/environment"
|
|
"github.com/ian-howell/airshipctl/pkg/log"
|
|
"github.com/ian-howell/airshipctl/pkg/plugin"
|
|
"github.com/ian-howell/airshipctl/pkg/util"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var settings environment.AirshipCTLSettings
|
|
|
|
const defaultPluginDir = "_plugins/bin"
|
|
|
|
// NewRootCmd creates the root `airshipctl` command. All other commands are
|
|
// subcommands branching from this one
|
|
func NewRootCmd(out io.Writer, args []string) (*cobra.Command, error) {
|
|
rootCmd := &cobra.Command{
|
|
Use: "airshipctl",
|
|
Short: "airshipctl is a unified entrypoint to various airship components",
|
|
}
|
|
rootCmd.SetOutput(out)
|
|
|
|
// Settings flags - This section should probably be moved to pkg/environment
|
|
rootCmd.PersistentFlags().BoolVar(&settings.Debug, "debug", false, "enable verbose output")
|
|
|
|
rootCmd.AddCommand(NewVersionCommand(out))
|
|
|
|
if _, err := os.Stat(defaultPluginDir); err == nil {
|
|
pluginFiles, err := util.ReadDir(defaultPluginDir)
|
|
if err != nil {
|
|
return nil, errors.New("could not read plugins: " + err.Error())
|
|
}
|
|
for _, pluginFile := range pluginFiles {
|
|
pathToPlugin := filepath.Join(defaultPluginDir, pluginFile.Name())
|
|
rootCmd.AddCommand(plugin.CreateCommandFromPlugin(pathToPlugin, out, args))
|
|
}
|
|
}
|
|
|
|
log.Init(&settings, out)
|
|
|
|
return rootCmd, nil
|
|
}
|
|
|
|
// Execute runs the base airshipctl command
|
|
func Execute(out io.Writer) {
|
|
rootCmd, err := NewRootCmd(out, os.Args[1:])
|
|
osExitIfError(out, err)
|
|
osExitIfError(out, rootCmd.Execute())
|
|
}
|
|
|
|
func osExitIfError(out io.Writer, err error) {
|
|
if err != nil {
|
|
fmt.Fprintln(out, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|