mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
d1ace13713
* Remove `--config` and `--master-secret` falgs * Add `--workspace` flag * In CLI, only work on files with paths created from `cli/internal/cmd` * Properly print values for GCP on IAM create when not directly updating the config --------- Signed-off-by: Daniel Weiße <dw@edgeless.systems>
117 lines
3.1 KiB
Go
117 lines
3.1 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
/*
|
|
Package cmd is the entrypoint of the Constellation CLI.
|
|
|
|
Business logic of the CLI shall be implemented in the internal/cmd package.
|
|
*/
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"github.com/edgelesssys/constellation/v2/cli/internal/cmd"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// Execute starts the CLI.
|
|
func Execute() error {
|
|
cobra.EnableCommandSorting = false
|
|
rootCmd := NewRootCmd()
|
|
ctx, cancel := signalContext(context.Background(), os.Interrupt)
|
|
defer cancel()
|
|
return rootCmd.ExecuteContext(ctx)
|
|
}
|
|
|
|
// NewRootCmd creates the root command.
|
|
func NewRootCmd() *cobra.Command {
|
|
rootCmd := &cobra.Command{
|
|
Use: "constellation",
|
|
Short: "Manage your Constellation cluster",
|
|
Long: "Manage your Constellation cluster.",
|
|
PersistentPreRunE: preRunRoot,
|
|
}
|
|
|
|
// Set output of cmd.Print to stdout. (By default, it's stderr.)
|
|
rootCmd.SetOut(os.Stdout)
|
|
|
|
rootCmd.PersistentFlags().StringP("workspace", "C", "", "path to the Constellation workspace")
|
|
rootCmd.PersistentFlags().Bool("debug", false, "enable debug logging")
|
|
rootCmd.PersistentFlags().Bool("force", false, "disable version compatibility checks - might result in corrupted clusters")
|
|
rootCmd.PersistentFlags().String("tf-log", "NONE", "Terraform log level")
|
|
|
|
must(rootCmd.MarkPersistentFlagDirname("workspace"))
|
|
|
|
rootCmd.AddCommand(cmd.NewConfigCmd())
|
|
rootCmd.AddCommand(cmd.NewCreateCmd())
|
|
rootCmd.AddCommand(cmd.NewInitCmd())
|
|
rootCmd.AddCommand(cmd.NewMiniCmd())
|
|
rootCmd.AddCommand(cmd.NewStatusCmd())
|
|
rootCmd.AddCommand(cmd.NewVerifyCmd())
|
|
rootCmd.AddCommand(cmd.NewUpgradeCmd())
|
|
rootCmd.AddCommand(cmd.NewRecoverCmd())
|
|
rootCmd.AddCommand(cmd.NewTerminateCmd())
|
|
rootCmd.AddCommand(cmd.NewIAMCmd())
|
|
rootCmd.AddCommand(cmd.NewVersionCmd())
|
|
|
|
return rootCmd
|
|
}
|
|
|
|
// signalContext returns a context that is canceled on the handed signal.
|
|
// The signal isn't watched after its first occurrence. Call the cancel
|
|
// function to ensure the internal goroutine is stopped and the signal isn't
|
|
// watched any longer.
|
|
func signalContext(ctx context.Context, sig os.Signal) (context.Context, context.CancelFunc) {
|
|
sigCtx, stop := signal.NotifyContext(ctx, sig)
|
|
done := make(chan struct{}, 1)
|
|
stopDone := make(chan struct{}, 1)
|
|
|
|
go func() {
|
|
defer func() { stopDone <- struct{}{} }()
|
|
defer stop()
|
|
select {
|
|
case <-sigCtx.Done():
|
|
fmt.Println(" Signal caught. Press ctrl+c again to terminate the program immediately.")
|
|
case <-done:
|
|
}
|
|
}()
|
|
|
|
cancelFunc := func() {
|
|
done <- struct{}{}
|
|
<-stopDone
|
|
}
|
|
|
|
return sigCtx, cancelFunc
|
|
}
|
|
|
|
func preRunRoot(cmd *cobra.Command, _ []string) error {
|
|
cmd.SilenceUsage = true
|
|
|
|
workspace, err := cmd.Flags().GetString("workspace")
|
|
if err != nil {
|
|
return fmt.Errorf("getting workspace flag: %w", err)
|
|
}
|
|
|
|
// Change to workspace directory if set.
|
|
if workspace != "" {
|
|
if err := os.Chdir(workspace); err != nil {
|
|
return fmt.Errorf("changing from current directory to workspace %q: %w", workspace, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func must(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|