From 6247409ad80911018f0e6d2af83bb4dc6e23d123 Mon Sep 17 00:00:00 2001 From: Alexander Dahmen Date: Mon, 9 Dec 2024 13:10:47 +0100 Subject: [PATCH] Onboard IaaS server action commands Supported commands: start, stop, reboot, deallocate, resize, console, log, rescue, unrescue Signed-off-by: Alexander Dahmen --- docs/stackit_beta_server.md | 9 + docs/stackit_beta_server_console.md | 42 ++++ docs/stackit_beta_server_deallocate.md | 39 +++ docs/stackit_beta_server_log.md | 46 ++++ docs/stackit_beta_server_reboot.md | 43 ++++ docs/stackit_beta_server_rescue.md | 40 ++++ docs/stackit_beta_server_resize.md | 40 ++++ docs/stackit_beta_server_start.md | 39 +++ docs/stackit_beta_server_stop.md | 39 +++ docs/stackit_beta_server_unrescue.md | 39 +++ internal/cmd/beta/server/console/console.go | 140 +++++++++++ .../cmd/beta/server/console/console_test.go | 208 ++++++++++++++++ .../cmd/beta/server/deallocate/deallocate.go | 127 ++++++++++ .../beta/server/deallocate/deallocate_test.go | 208 ++++++++++++++++ internal/cmd/beta/server/log/log.go | 165 +++++++++++++ internal/cmd/beta/server/log/log_test.go | 222 +++++++++++++++++ internal/cmd/beta/server/reboot/reboot.go | 131 ++++++++++ .../cmd/beta/server/reboot/reboot_test.go | 219 +++++++++++++++++ internal/cmd/beta/server/rescue/rescue.go | 144 +++++++++++ .../cmd/beta/server/rescue/rescue_test.go | 225 ++++++++++++++++++ internal/cmd/beta/server/resize/resize.go | 144 +++++++++++ .../cmd/beta/server/resize/resize_test.go | 224 +++++++++++++++++ internal/cmd/beta/server/server.go | 18 ++ internal/cmd/beta/server/start/start.go | 119 +++++++++ internal/cmd/beta/server/start/start_test.go | 208 ++++++++++++++++ internal/cmd/beta/server/stop/stop.go | 127 ++++++++++ internal/cmd/beta/server/stop/stop_test.go | 208 ++++++++++++++++ internal/cmd/beta/server/unrescue/unrescue.go | 127 ++++++++++ .../cmd/beta/server/unrescue/unrescue_test.go | 208 ++++++++++++++++ 29 files changed, 3548 insertions(+) create mode 100644 docs/stackit_beta_server_console.md create mode 100644 docs/stackit_beta_server_deallocate.md create mode 100644 docs/stackit_beta_server_log.md create mode 100644 docs/stackit_beta_server_reboot.md create mode 100644 docs/stackit_beta_server_rescue.md create mode 100644 docs/stackit_beta_server_resize.md create mode 100644 docs/stackit_beta_server_start.md create mode 100644 docs/stackit_beta_server_stop.md create mode 100644 docs/stackit_beta_server_unrescue.md create mode 100644 internal/cmd/beta/server/console/console.go create mode 100644 internal/cmd/beta/server/console/console_test.go create mode 100644 internal/cmd/beta/server/deallocate/deallocate.go create mode 100644 internal/cmd/beta/server/deallocate/deallocate_test.go create mode 100644 internal/cmd/beta/server/log/log.go create mode 100644 internal/cmd/beta/server/log/log_test.go create mode 100644 internal/cmd/beta/server/reboot/reboot.go create mode 100644 internal/cmd/beta/server/reboot/reboot_test.go create mode 100644 internal/cmd/beta/server/rescue/rescue.go create mode 100644 internal/cmd/beta/server/rescue/rescue_test.go create mode 100644 internal/cmd/beta/server/resize/resize.go create mode 100644 internal/cmd/beta/server/resize/resize_test.go create mode 100644 internal/cmd/beta/server/start/start.go create mode 100644 internal/cmd/beta/server/start/start_test.go create mode 100644 internal/cmd/beta/server/stop/stop.go create mode 100644 internal/cmd/beta/server/stop/stop_test.go create mode 100644 internal/cmd/beta/server/unrescue/unrescue.go create mode 100644 internal/cmd/beta/server/unrescue/unrescue_test.go diff --git a/docs/stackit_beta_server.md b/docs/stackit_beta_server.md index b8f078aad..6ec25fc7a 100644 --- a/docs/stackit_beta_server.md +++ b/docs/stackit_beta_server.md @@ -31,13 +31,22 @@ stackit beta server [flags] * [stackit beta](./stackit_beta.md) - Contains beta STACKIT CLI commands * [stackit beta server backup](./stackit_beta_server_backup.md) - Provides functionality for server backups * [stackit beta server command](./stackit_beta_server_command.md) - Provides functionality for Server Command +* [stackit beta server console](./stackit_beta_server_console.md) - Gets a URL for server remote console * [stackit beta server create](./stackit_beta_server_create.md) - Creates a server +* [stackit beta server deallocate](./stackit_beta_server_deallocate.md) - Deallocates an existing server * [stackit beta server delete](./stackit_beta_server_delete.md) - Deletes a server * [stackit beta server describe](./stackit_beta_server_describe.md) - Shows details of a server * [stackit beta server list](./stackit_beta_server_list.md) - Lists all servers of a project +* [stackit beta server log](./stackit_beta_server_log.md) - Gets server console log * [stackit beta server network-interface](./stackit_beta_server_network-interface.md) - Allows attaching/detaching network interfaces to servers * [stackit beta server public-ip](./stackit_beta_server_public-ip.md) - Allows attaching/detaching public IPs to servers +* [stackit beta server reboot](./stackit_beta_server_reboot.md) - Reboots a server +* [stackit beta server rescue](./stackit_beta_server_rescue.md) - Rescues an existing server +* [stackit beta server resize](./stackit_beta_server_resize.md) - Resizes the server to the given machine type * [stackit beta server service-account](./stackit_beta_server_service-account.md) - Allows attaching/detaching service accounts to servers +* [stackit beta server start](./stackit_beta_server_start.md) - Starts an existing server or allocates the server if deallocated +* [stackit beta server stop](./stackit_beta_server_stop.md) - Stops an existing server +* [stackit beta server unrescue](./stackit_beta_server_unrescue.md) - Unrescues an existing server * [stackit beta server update](./stackit_beta_server_update.md) - Updates a server * [stackit beta server volume](./stackit_beta_server_volume.md) - Provides functionality for server volumes diff --git a/docs/stackit_beta_server_console.md b/docs/stackit_beta_server_console.md new file mode 100644 index 000000000..5fccb3d5b --- /dev/null +++ b/docs/stackit_beta_server_console.md @@ -0,0 +1,42 @@ +## stackit beta server console + +Gets a URL for server remote console + +### Synopsis + +Gets a URL for server remote console. + +``` +stackit beta server console [flags] +``` + +### Examples + +``` + Get a URL for the server remote console with server ID "xxx" + $ stackit beta server console xxx + + Get a URL for the server remote console with server ID "xxx" in JSON format + $ stackit beta server console xxx --output-format json +``` + +### Options + +``` + -h, --help Help for "stackit beta server console" +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_deallocate.md b/docs/stackit_beta_server_deallocate.md new file mode 100644 index 000000000..96323db32 --- /dev/null +++ b/docs/stackit_beta_server_deallocate.md @@ -0,0 +1,39 @@ +## stackit beta server deallocate + +Deallocates an existing server + +### Synopsis + +Deallocates an existing server. + +``` +stackit beta server deallocate [flags] +``` + +### Examples + +``` + Deallocate an existing server with ID "xxx" + $ stackit beta server deallocate xxx +``` + +### Options + +``` + -h, --help Help for "stackit beta server deallocate" +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_log.md b/docs/stackit_beta_server_log.md new file mode 100644 index 000000000..b9cad0a9c --- /dev/null +++ b/docs/stackit_beta_server_log.md @@ -0,0 +1,46 @@ +## stackit beta server log + +Gets server console log + +### Synopsis + +Gets server console log. + +``` +stackit beta server log [flags] +``` + +### Examples + +``` + Get server console log for the server with ID "xxx" + $ stackit beta server log xxx + + Get server console log for the server with ID "xxx" and limit output lines to 1000 + $ stackit beta server log xxx --length 1000 + + Get server console log for the server with ID "xxx" in JSON format + $ stackit beta server log xxx --output-format json +``` + +### Options + +``` + -h, --help Help for "stackit beta server log" + --length int Maximum number of lines to list (default 2000) +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_reboot.md b/docs/stackit_beta_server_reboot.md new file mode 100644 index 000000000..a100c8a0d --- /dev/null +++ b/docs/stackit_beta_server_reboot.md @@ -0,0 +1,43 @@ +## stackit beta server reboot + +Reboots a server + +### Synopsis + +Reboots a server. + +``` +stackit beta server reboot [flags] +``` + +### Examples + +``` + Perform a soft reboot of a server with ID "xxx" + $ stackit beta server reboot xxx + + Perform a hard reboot of a server with ID "xxx" + $ stackit beta server reboot xxx --hard +``` + +### Options + +``` + -b, --hard Performs a hard reboot. (default false) + -h, --help Help for "stackit beta server reboot" +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_rescue.md b/docs/stackit_beta_server_rescue.md new file mode 100644 index 000000000..2064a6ad6 --- /dev/null +++ b/docs/stackit_beta_server_rescue.md @@ -0,0 +1,40 @@ +## stackit beta server rescue + +Rescues an existing server + +### Synopsis + +Rescues an existing server. + +``` +stackit beta server rescue [flags] +``` + +### Examples + +``` + Rescue an existing server with ID "xxx" using image with ID "yyy" as boot volume + $ stackit beta server rescue xxx --image-id yyy +``` + +### Options + +``` + -h, --help Help for "stackit beta server rescue" + --image-id string The image ID to be used for a temporary boot volume. +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_resize.md b/docs/stackit_beta_server_resize.md new file mode 100644 index 000000000..4ef3e6715 --- /dev/null +++ b/docs/stackit_beta_server_resize.md @@ -0,0 +1,40 @@ +## stackit beta server resize + +Resizes the server to the given machine type + +### Synopsis + +Resizes the server to the given machine type. + +``` +stackit beta server resize [flags] +``` + +### Examples + +``` + Resize a server with ID "xxx" to machine type "yyy" + $ stackit beta server resize xxx --machine-type yyy +``` + +### Options + +``` + -h, --help Help for "stackit beta server resize" + --machine-type string Name of the type of the machine for the server. Possible values are documented in https://docs.stackit.cloud/stackit/en/virtual-machine-flavors-75137231.html +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_start.md b/docs/stackit_beta_server_start.md new file mode 100644 index 000000000..7a811400c --- /dev/null +++ b/docs/stackit_beta_server_start.md @@ -0,0 +1,39 @@ +## stackit beta server start + +Starts an existing server or allocates the server if deallocated + +### Synopsis + +Starts an existing server or allocates the server if deallocated. + +``` +stackit beta server start [flags] +``` + +### Examples + +``` + Start an existing server with ID "xxx" + $ stackit beta server start xxx +``` + +### Options + +``` + -h, --help Help for "stackit beta server start" +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_stop.md b/docs/stackit_beta_server_stop.md new file mode 100644 index 000000000..1521146b0 --- /dev/null +++ b/docs/stackit_beta_server_stop.md @@ -0,0 +1,39 @@ +## stackit beta server stop + +Stops an existing server + +### Synopsis + +Stops an existing server. + +``` +stackit beta server stop [flags] +``` + +### Examples + +``` + Stop an existing server with ID "xxx" + $ stackit beta server stop xxx +``` + +### Options + +``` + -h, --help Help for "stackit beta server stop" +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/docs/stackit_beta_server_unrescue.md b/docs/stackit_beta_server_unrescue.md new file mode 100644 index 000000000..06020bdb6 --- /dev/null +++ b/docs/stackit_beta_server_unrescue.md @@ -0,0 +1,39 @@ +## stackit beta server unrescue + +Unrescues an existing server + +### Synopsis + +Unrescues an existing server. + +``` +stackit beta server unrescue [flags] +``` + +### Examples + +``` + Unrescue an existing server with ID "xxx" + $ stackit beta server unrescue xxx +``` + +### Options + +``` + -h, --help Help for "stackit beta server unrescue" +``` + +### Options inherited from parent commands + +``` + -y, --assume-yes If set, skips all confirmation prompts + --async If set, runs the command asynchronously + -o, --output-format string Output format, one of ["json" "pretty" "none" "yaml"] + -p, --project-id string Project ID + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit beta server](./stackit_beta_server.md) - Provides functionality for servers + diff --git a/internal/cmd/beta/server/console/console.go b/internal/cmd/beta/server/console/console.go new file mode 100644 index 000000000..9bc862cdf --- /dev/null +++ b/internal/cmd/beta/server/console/console.go @@ -0,0 +1,140 @@ +package console + +import ( + "context" + "encoding/json" + "fmt" + "net/url" + + "github.com/goccy/go-yaml" + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "console", + Short: "Gets a URL for server remote console", + Long: "Gets a URL for server remote console.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Get a URL for the server remote console with server ID "xxx"`, + "$ stackit beta server console xxx", + ), + examples.NewExample( + `Get a URL for the server remote console with server ID "xxx" in JSON format`, + "$ stackit beta server console xxx --output-format json", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + // Call API + req := buildRequest(ctx, model, apiClient) + resp, err := req.Execute() + if err != nil { + return fmt.Errorf("server console: %w", err) + } + + return outputResult(p, model, serverLabel, resp) + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiGetServerConsoleRequest { + return apiClient.GetServerConsole(ctx, model.ProjectId, model.ServerId) +} + +func outputResult(p *print.Printer, model *inputModel, serverLabel string, serverUrl *iaas.ServerConsoleUrl) error { + outputFormat := model.OutputFormat + + switch outputFormat { + case print.JSONOutputFormat: + details, err := json.MarshalIndent(serverUrl, "", " ") + if err != nil { + return fmt.Errorf("marshal url: %w", err) + } + p.Outputln(string(details)) + + return nil + case print.YAMLOutputFormat: + details, err := yaml.MarshalWithOptions(serverUrl, yaml.IndentSequence(true)) + if err != nil { + return fmt.Errorf("marshal url: %w", err) + } + p.Outputln(string(details)) + + return nil + default: + // unescape url in order to get rid of e.g. %40 + unescapedURL, err := url.PathUnescape(*serverUrl.GetUrl()) + if err != nil { + return fmt.Errorf("unescape url: %w", err) + } + + p.Outputf("Remote console URL %q for server %q\n", unescapedURL, serverLabel) + + return nil + } +} diff --git a/internal/cmd/beta/server/console/console_test.go b/internal/cmd/beta/server/console/console_test.go new file mode 100644 index 000000000..1dfc48384 --- /dev/null +++ b/internal/cmd/beta/server/console/console_test.go @@ -0,0 +1,208 @@ +package console + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiGetServerConsoleRequest)) iaas.ApiGetServerConsoleRequest { + request := testClient.GetServerConsole(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiGetServerConsoleRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/deallocate/deallocate.go b/internal/cmd/beta/server/deallocate/deallocate.go new file mode 100644 index 000000000..6396f2e0b --- /dev/null +++ b/internal/cmd/beta/server/deallocate/deallocate.go @@ -0,0 +1,127 @@ +package deallocate + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/spinner" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + "github.com/stackitcloud/stackit-sdk-go/services/iaas/wait" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "deallocate", + Short: "Deallocates an existing server", + Long: "Deallocates an existing server.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Deallocate an existing server with ID "xxx"`, + "$ stackit beta server deallocate xxx", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to deallocate server %q?", serverLabel) + err = p.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server deallocate: %w", err) + } + + // Wait for async operation, if async mode not enabled + if !model.Async { + s := spinner.New(p) + s.Start("Deallocating server") + _, err = wait.DeallocateServerWaitHandler(ctx, apiClient, model.ProjectId, model.ServerId).WaitWithContext(ctx) + if err != nil { + return fmt.Errorf("wait for server deallocating: %w", err) + } + s.Stop() + } + + operationState := "Deallocated" + if model.Async { + operationState = "Triggered deallocation of" + } + p.Info("%s server %q\n", operationState, serverLabel) + + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiDeallocateServerRequest { + return apiClient.DeallocateServer(ctx, model.ProjectId, model.ServerId) +} diff --git a/internal/cmd/beta/server/deallocate/deallocate_test.go b/internal/cmd/beta/server/deallocate/deallocate_test.go new file mode 100644 index 000000000..6ededf983 --- /dev/null +++ b/internal/cmd/beta/server/deallocate/deallocate_test.go @@ -0,0 +1,208 @@ +package deallocate + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiDeallocateServerRequest)) iaas.ApiDeallocateServerRequest { + request := testClient.DeallocateServer(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiDeallocateServerRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/log/log.go b/internal/cmd/beta/server/log/log.go new file mode 100644 index 000000000..7cfffcb84 --- /dev/null +++ b/internal/cmd/beta/server/log/log.go @@ -0,0 +1,165 @@ +package log + +import ( + "context" + "encoding/json" + "fmt" + "strings" + + "github.com/goccy/go-yaml" + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/flags" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" + + lengthLimitFlag = "length" + defaultLengthLimit = 2000 // lines +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string + Length *int64 +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "log", + Short: "Gets server console log", + Long: "Gets server console log.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Get server console log for the server with ID "xxx"`, + "$ stackit beta server log xxx", + ), + examples.NewExample( + `Get server console log for the server with ID "xxx" and limit output lines to 1000`, + "$ stackit beta server log xxx --length 1000", + ), + examples.NewExample( + `Get server console log for the server with ID "xxx" in JSON format`, + "$ stackit beta server log xxx --output-format json", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + // Call API + req := buildRequest(ctx, model, apiClient) + resp, err := req.Execute() + if err != nil { + return fmt.Errorf("server log: %w", err) + } + + log := *resp.GetOutput() + lines := strings.Split(log, "\n") + + if len(lines) > int(*model.Length) { + // Truncate output and show most recent logs + start := len(lines) - int(*model.Length) + return outputResult(p, model, serverLabel, strings.Join(lines[start:], "\n")) + } + + return outputResult(p, model, serverLabel, log) + }, + } + configureFlags(cmd) + return cmd +} + +func configureFlags(cmd *cobra.Command) { + cmd.Flags().Int64(lengthLimitFlag, defaultLengthLimit, "Maximum number of lines to list") +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + length := flags.FlagWithDefaultToInt64Value(p, cmd, lengthLimitFlag) + if length < 0 { + return nil, &errors.FlagValidationError{ + Flag: lengthLimitFlag, + Details: "must not be negative", + } + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + Length: utils.Ptr(length), + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiGetServerLogRequest { + return apiClient.GetServerLog(ctx, model.ProjectId, model.ServerId) +} + +func outputResult(p *print.Printer, model *inputModel, serverLabel, log string) error { + outputFormat := model.OutputFormat + + switch outputFormat { + case print.JSONOutputFormat: + details, err := json.MarshalIndent(log, "", " ") + if err != nil { + return fmt.Errorf("marshal url: %w", err) + } + p.Outputln(string(details)) + + return nil + case print.YAMLOutputFormat: + details, err := yaml.MarshalWithOptions(log, yaml.IndentSequence(true)) + if err != nil { + return fmt.Errorf("marshal url: %w", err) + } + p.Outputln(string(details)) + + return nil + default: + p.Outputf("Log for server %q\n%s", serverLabel, log) + return nil + } +} diff --git a/internal/cmd/beta/server/log/log_test.go b/internal/cmd/beta/server/log/log_test.go new file mode 100644 index 000000000..5e10870d6 --- /dev/null +++ b/internal/cmd/beta/server/log/log_test.go @@ -0,0 +1,222 @@ +package log + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + lengthLimitFlag: "3000", + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + Length: utils.Ptr(int64(3000)), + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiGetServerLogRequest)) iaas.ApiGetServerLogRequest { + request := testClient.GetServerLog(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "optional length missing (test default value)", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + delete(flagValues, lengthLimitFlag) + }), + isValid: true, + expectedModel: fixtureInputModel(func(model *inputModel) { + model.Length = utils.Ptr(int64(2000)) + }), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiGetServerLogRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/reboot/reboot.go b/internal/cmd/beta/server/reboot/reboot.go new file mode 100644 index 000000000..6983e7531 --- /dev/null +++ b/internal/cmd/beta/server/reboot/reboot.go @@ -0,0 +1,131 @@ +package reboot + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/flags" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" + + hardRebootFlag = "hard" + defaultHardReboot = false + hardRebootAction = "hard" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string + HardReboot bool +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "reboot", + Short: "Reboots a server", + Long: "Reboots a server.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Perform a soft reboot of a server with ID "xxx"`, + "$ stackit beta server reboot xxx", + ), + examples.NewExample( + `Perform a hard reboot of a server with ID "xxx"`, + "$ stackit beta server reboot xxx --hard", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to reboot server %q?", serverLabel) + err = p.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server reboot: %w", err) + } + + p.Info("Server %q rebooted\n", serverLabel) + + return nil + }, + } + configureFlags(cmd) + return cmd +} + +func configureFlags(cmd *cobra.Command) { + cmd.Flags().BoolP(hardRebootFlag, "b", defaultHardReboot, "Performs a hard reboot. (default false)") +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + HardReboot: flags.FlagToBoolValue(p, cmd, hardRebootFlag), + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiRebootServerRequest { + req := apiClient.RebootServer(ctx, model.ProjectId, model.ServerId) + // if hard reboot is set the action must be set (soft is default) + if model.HardReboot { + req = req.Action(hardRebootAction) + } + return req +} diff --git a/internal/cmd/beta/server/reboot/reboot_test.go b/internal/cmd/beta/server/reboot/reboot_test.go new file mode 100644 index 000000000..c4444fa85 --- /dev/null +++ b/internal/cmd/beta/server/reboot/reboot_test.go @@ -0,0 +1,219 @@ +package reboot + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + hardRebootFlag: "false", + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + HardReboot: false, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiRebootServerRequest)) iaas.ApiRebootServerRequest { + request := testClient.RebootServer(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiRebootServerRequest + }{ + { + description: "base (soft reboot)", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + { + description: "hard reboot is set", + model: fixtureInputModel(func(model *inputModel) { + model.HardReboot = true + }), + expectedRequest: fixtureRequest(func(request *iaas.ApiRebootServerRequest) { + *request = request.Action("hard") + }), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/rescue/rescue.go b/internal/cmd/beta/server/rescue/rescue.go new file mode 100644 index 000000000..b658f4ace --- /dev/null +++ b/internal/cmd/beta/server/rescue/rescue.go @@ -0,0 +1,144 @@ +package rescue + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/flags" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/spinner" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + "github.com/stackitcloud/stackit-sdk-go/services/iaas/wait" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" + + imageIdFlag = "image-id" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string + ImageId *string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "rescue", + Short: "Rescues an existing server", + Long: "Rescues an existing server.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Rescue an existing server with ID "xxx" using image with ID "yyy" as boot volume`, + "$ stackit beta server rescue xxx --image-id yyy", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to rescue server %q?", serverLabel) + err = p.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server rescue: %w", err) + } + + // Wait for async operation, if async mode not enabled + if !model.Async { + s := spinner.New(p) + s.Start("Rescuing server") + _, err = wait.RescueServerWaitHandler(ctx, apiClient, model.ProjectId, model.ServerId).WaitWithContext(ctx) + if err != nil { + return fmt.Errorf("wait for server rescuing: %w", err) + } + s.Stop() + } + + operationState := "Rescued" + if model.Async { + operationState = "Triggered rescue of" + } + p.Info("%s server %q. Image %q is used as temporary boot image\n", operationState, serverLabel, *model.ImageId) + + return nil + }, + } + configureFlags(cmd) + return cmd +} + +func configureFlags(cmd *cobra.Command) { + cmd.Flags().Var(flags.UUIDFlag(), imageIdFlag, "The image ID to be used for a temporary boot volume.") + + err := flags.MarkFlagsRequired(cmd, imageIdFlag) + cobra.CheckErr(err) +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + ImageId: flags.FlagToStringPointer(p, cmd, imageIdFlag), + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiRescueServerRequest { + req := apiClient.RescueServer(ctx, model.ProjectId, model.ServerId) + payload := iaas.RescueServerPayload{ + Image: model.ImageId, + } + return req.RescueServerPayload(payload) +} diff --git a/internal/cmd/beta/server/rescue/rescue_test.go b/internal/cmd/beta/server/rescue/rescue_test.go new file mode 100644 index 000000000..9d17daf78 --- /dev/null +++ b/internal/cmd/beta/server/rescue/rescue_test.go @@ -0,0 +1,225 @@ +package rescue + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() +var testImageId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + imageIdFlag: testImageId, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + ImageId: utils.Ptr(testImageId), + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiRescueServerRequest)) iaas.ApiRescueServerRequest { + request := testClient.RescueServer(testCtx, testProjectId, testServerId) + request = request.RescueServerPayload(iaas.RescueServerPayload{ + Image: utils.Ptr(testImageId), + }) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "required image id flag missing", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + delete(flagValues, imageIdFlag) + }), + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + delete(flagValues, projectIdFlag) + }), + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiRescueServerRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/resize/resize.go b/internal/cmd/beta/server/resize/resize.go new file mode 100644 index 000000000..95b5b3e00 --- /dev/null +++ b/internal/cmd/beta/server/resize/resize.go @@ -0,0 +1,144 @@ +package resize + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/flags" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/spinner" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + "github.com/stackitcloud/stackit-sdk-go/services/iaas/wait" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" + + machineTypeFlag = "machine-type" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string + MachineType *string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "resize", + Short: "Resizes the server to the given machine type", + Long: "Resizes the server to the given machine type.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Resize a server with ID "xxx" to machine type "yyy"`, + "$ stackit beta server resize xxx --machine-type yyy", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to resize server %q to machine type %q?", serverLabel, *model.MachineType) + err = p.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server resize: %w", err) + } + + // Wait for async operation, if async mode not enabled + if !model.Async { + s := spinner.New(p) + s.Start("Resizing server") + _, err = wait.ResizeServerWaitHandler(ctx, apiClient, model.ProjectId, model.ServerId).WaitWithContext(ctx) + if err != nil { + return fmt.Errorf("wait for server resizing: %w", err) + } + s.Stop() + } + + operationState := "Resized" + if model.Async { + operationState = "Triggered resize of" + } + p.Info("%s server %q\n", operationState, serverLabel) + + return nil + }, + } + configureFlags(cmd) + return cmd +} + +func configureFlags(cmd *cobra.Command) { + cmd.Flags().String(machineTypeFlag, "", "Name of the type of the machine for the server. Possible values are documented in https://docs.stackit.cloud/stackit/en/virtual-machine-flavors-75137231.html") + + err := flags.MarkFlagsRequired(cmd, machineTypeFlag) + cobra.CheckErr(err) +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + MachineType: flags.FlagToStringPointer(p, cmd, machineTypeFlag), + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiResizeServerRequest { + req := apiClient.ResizeServer(ctx, model.ProjectId, model.ServerId) + payload := iaas.ResizeServerPayload{ + MachineType: model.MachineType, + } + return req.ResizeServerPayload(payload) +} diff --git a/internal/cmd/beta/server/resize/resize_test.go b/internal/cmd/beta/server/resize/resize_test.go new file mode 100644 index 000000000..a93d8fdc1 --- /dev/null +++ b/internal/cmd/beta/server/resize/resize_test.go @@ -0,0 +1,224 @@ +package resize + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + machineTypeFlag: "t1.2", + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + MachineType: utils.Ptr("t1.2"), + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiResizeServerRequest)) iaas.ApiResizeServerRequest { + request := testClient.ResizeServer(testCtx, testProjectId, testServerId) + request = request.ResizeServerPayload(iaas.ResizeServerPayload{ + MachineType: utils.Ptr("t1.2"), + }) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "required machine type flag missing", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + delete(flagValues, machineTypeFlag) + }), + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + delete(flagValues, projectIdFlag) + }), + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiResizeServerRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/server.go b/internal/cmd/beta/server/server.go index 9b96c1322..6d75b27d5 100644 --- a/internal/cmd/beta/server/server.go +++ b/internal/cmd/beta/server/server.go @@ -3,13 +3,22 @@ package server import ( "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/backup" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/command" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/console" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/create" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/deallocate" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/delete" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/describe" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/list" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/log" networkinterface "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/network-interface" publicip "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/public-ip" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/reboot" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/rescue" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/resize" serviceaccount "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/service-account" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/start" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/stop" + "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/unrescue" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/update" "github.com/stackitcloud/stackit-cli/internal/cmd/beta/server/volume" @@ -44,4 +53,13 @@ func addSubcommands(cmd *cobra.Command, p *print.Printer) { cmd.AddCommand(update.NewCmd(p)) cmd.AddCommand(volume.NewCmd(p)) cmd.AddCommand(networkinterface.NewCmd(p)) + cmd.AddCommand(console.NewCmd(p)) + cmd.AddCommand(log.NewCmd(p)) + cmd.AddCommand(start.NewCmd(p)) + cmd.AddCommand(stop.NewCmd(p)) + cmd.AddCommand(reboot.NewCmd(p)) + cmd.AddCommand(deallocate.NewCmd(p)) + cmd.AddCommand(resize.NewCmd(p)) + cmd.AddCommand(rescue.NewCmd(p)) + cmd.AddCommand(unrescue.NewCmd(p)) } diff --git a/internal/cmd/beta/server/start/start.go b/internal/cmd/beta/server/start/start.go new file mode 100644 index 000000000..764d3c5d9 --- /dev/null +++ b/internal/cmd/beta/server/start/start.go @@ -0,0 +1,119 @@ +package start + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/spinner" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + "github.com/stackitcloud/stackit-sdk-go/services/iaas/wait" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "start", + Short: "Starts an existing server or allocates the server if deallocated", + Long: "Starts an existing server or allocates the server if deallocated.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Start an existing server with ID "xxx"`, + "$ stackit beta server start xxx", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server start: %w", err) + } + + // Wait for async operation, if async mode not enabled + if !model.Async { + s := spinner.New(p) + s.Start("Starting server") + _, err = wait.StartServerWaitHandler(ctx, apiClient, model.ProjectId, model.ServerId).WaitWithContext(ctx) + if err != nil { + return fmt.Errorf("wait for server starting: %w", err) + } + s.Stop() + } + + operationState := "Started" + if model.Async { + operationState = "Triggered start of" + } + p.Info("%s server %q\n", operationState, serverLabel) + + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiStartServerRequest { + return apiClient.StartServer(ctx, model.ProjectId, model.ServerId) +} diff --git a/internal/cmd/beta/server/start/start_test.go b/internal/cmd/beta/server/start/start_test.go new file mode 100644 index 000000000..f90b6048d --- /dev/null +++ b/internal/cmd/beta/server/start/start_test.go @@ -0,0 +1,208 @@ +package start + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiStartServerRequest)) iaas.ApiStartServerRequest { + request := testClient.StartServer(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiStartServerRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/stop/stop.go b/internal/cmd/beta/server/stop/stop.go new file mode 100644 index 000000000..a8fc651e6 --- /dev/null +++ b/internal/cmd/beta/server/stop/stop.go @@ -0,0 +1,127 @@ +package stop + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/spinner" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + "github.com/stackitcloud/stackit-sdk-go/services/iaas/wait" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "stop", + Short: "Stops an existing server", + Long: "Stops an existing server.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Stop an existing server with ID "xxx"`, + "$ stackit beta server stop xxx", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to stop server %q?", serverLabel) + err = p.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server stop: %w", err) + } + + // Wait for async operation, if async mode not enabled + if !model.Async { + s := spinner.New(p) + s.Start("Stopping server") + _, err = wait.StopServerWaitHandler(ctx, apiClient, model.ProjectId, model.ServerId).WaitWithContext(ctx) + if err != nil { + return fmt.Errorf("wait for server stopping: %w", err) + } + s.Stop() + } + + operationState := "Stopped" + if model.Async { + operationState = "Triggered stop of" + } + p.Info("%s server %q\n", operationState, serverLabel) + + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiStopServerRequest { + return apiClient.StopServer(ctx, model.ProjectId, model.ServerId) +} diff --git a/internal/cmd/beta/server/stop/stop_test.go b/internal/cmd/beta/server/stop/stop_test.go new file mode 100644 index 000000000..a4e889181 --- /dev/null +++ b/internal/cmd/beta/server/stop/stop_test.go @@ -0,0 +1,208 @@ +package stop + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiStopServerRequest)) iaas.ApiStopServerRequest { + request := testClient.StopServer(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiStopServerRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} diff --git a/internal/cmd/beta/server/unrescue/unrescue.go b/internal/cmd/beta/server/unrescue/unrescue.go new file mode 100644 index 000000000..c41d9f0c7 --- /dev/null +++ b/internal/cmd/beta/server/unrescue/unrescue.go @@ -0,0 +1,127 @@ +package unrescue + +import ( + "context" + "fmt" + + "github.com/stackitcloud/stackit-cli/internal/pkg/args" + "github.com/stackitcloud/stackit-cli/internal/pkg/errors" + "github.com/stackitcloud/stackit-cli/internal/pkg/examples" + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/client" + iaasUtils "github.com/stackitcloud/stackit-cli/internal/pkg/services/iaas/utils" + "github.com/stackitcloud/stackit-cli/internal/pkg/spinner" + "github.com/stackitcloud/stackit-cli/internal/pkg/utils" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" + "github.com/stackitcloud/stackit-sdk-go/services/iaas/wait" + + "github.com/spf13/cobra" +) + +const ( + serverIdArg = "SERVER_ID" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ServerId string +} + +func NewCmd(p *print.Printer) *cobra.Command { + cmd := &cobra.Command{ + Use: "unrescue", + Short: "Unrescues an existing server", + Long: "Unrescues an existing server.", + Args: args.SingleArg(serverIdArg, utils.ValidateUUID), + Example: examples.Build( + examples.NewExample( + `Unrescue an existing server with ID "xxx"`, + "$ stackit beta server unrescue xxx", + ), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(p, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(p) + if err != nil { + return err + } + + serverLabel, err := iaasUtils.GetServerName(ctx, apiClient, model.ProjectId, model.ServerId) + if err != nil { + p.Debug(print.ErrorLevel, "get server name: %v", err) + serverLabel = model.ServerId + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to unrescue server %q?", serverLabel) + err = p.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + err = req.Execute() + if err != nil { + return fmt.Errorf("server unrescue: %w", err) + } + + // Wait for async operation, if async mode not enabled + if !model.Async { + s := spinner.New(p) + s.Start("Unrescuing server") + _, err = wait.UnrescueServerWaitHandler(ctx, apiClient, model.ProjectId, model.ServerId).WaitWithContext(ctx) + if err != nil { + return fmt.Errorf("wait for server unrescuing: %w", err) + } + s.Stop() + } + + operationState := "Unrescued" + if model.Async { + operationState = "Triggered unrescue of" + } + p.Info("%s server %q\n", operationState, serverLabel) + + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + serverId := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ServerId: serverId, + } + + if p.IsVerbosityDebug() { + modelStr, err := print.BuildDebugStrFromInputModel(model) + if err != nil { + p.Debug(print.ErrorLevel, "convert model to string for debugging: %v", err) + } else { + p.Debug(print.DebugLevel, "parsed input values: %s", modelStr) + } + } + + return &model, nil +} + +func buildRequest(ctx context.Context, model *inputModel, apiClient *iaas.APIClient) iaas.ApiUnrescueServerRequest { + return apiClient.UnrescueServer(ctx, model.ProjectId, model.ServerId) +} diff --git a/internal/cmd/beta/server/unrescue/unrescue_test.go b/internal/cmd/beta/server/unrescue/unrescue_test.go new file mode 100644 index 000000000..82f75a370 --- /dev/null +++ b/internal/cmd/beta/server/unrescue/unrescue_test.go @@ -0,0 +1,208 @@ +package unrescue + +import ( + "context" + "testing" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/stackitcloud/stackit-sdk-go/services/iaas" +) + +var projectIdFlag = globalflags.ProjectIdFlag + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &iaas.APIClient{} +var testProjectId = uuid.NewString() +var testServerId = uuid.NewString() + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testServerId, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + projectIdFlag: testProjectId, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + Verbosity: globalflags.VerbosityDefault, + ProjectId: testProjectId, + }, + ServerId: testServerId, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *iaas.ApiUnrescueServerRequest)) iaas.ApiUnrescueServerRequest { + request := testClient.UnrescueServer(testCtx, testProjectId, testServerId) + for _, mod := range mods { + mod(&request) + } + return request +} + +func TestParseInput(t *testing.T) { + tests := []struct { + description string + argValues []string + flagValues map[string]string + isValid bool + expectedModel *inputModel + }{ + { + description: "base", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(), + isValid: true, + expectedModel: fixtureInputModel(), + }, + { + description: "no values", + argValues: []string{}, + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id missing", + argValues: fixtureArgValues(), + flagValues: map[string]string{}, + isValid: false, + }, + { + description: "project id invalid 1", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "" + }), + isValid: false, + }, + { + description: "project id invalid 2", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(flagValues map[string]string) { + flagValues[projectIdFlag] = "invalid-uuid" + }), + isValid: false, + }, + { + description: "server id invalid 1", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + { + description: "server id invalid 2", + argValues: fixtureArgValues(func(argValues []string) { + argValues[0] = "invalid-uuid" + }), + flagValues: fixtureFlagValues(), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := NewCmd(p) + err := globalflags.Configure(cmd.Flags()) + if err != nil { + t.Fatalf("configure global flags: %v", err) + } + + for flag, value := range tt.flagValues { + err := cmd.Flags().Set(flag, value) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("setting flag --%s=%s: %v", flag, value, err) + } + } + + err = cmd.ValidateArgs(tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating args: %v", err) + } + + err = cmd.ValidateRequiredFlags() + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error validating flags: %v", err) + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing input: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest iaas.ApiUnrescueServerRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + request := buildRequest(testCtx, tt.model, testClient) + + diff := cmp.Diff(request, tt.expectedRequest, + cmp.AllowUnexported(tt.expectedRequest), + cmpopts.EquateComparable(testCtx), + ) + if diff != "" { + t.Fatalf("Data does not match: %s", diff) + } + }) + } +}