From af1e39802b3ed5736c1f22578d9f73e72ee4edf8 Mon Sep 17 00:00:00 2001 From: Mauritz Uphoff Date: Thu, 17 Jul 2025 17:10:04 +0200 Subject: [PATCH] feat(ske): add commands to trigger hibernate, wakeup maintenance, and reconcile --- docs/stackit_ske.md | 4 + docs/stackit_ske_cluster.md | 2 +- docs/stackit_ske_cluster_describe.md | 4 +- docs/stackit_ske_hibernate.md | 40 ++++ docs/stackit_ske_maintenance.md | 40 ++++ docs/stackit_ske_reconcile.md | 40 ++++ docs/stackit_ske_wakeup.md | 40 ++++ internal/cmd/ske/cluster/describe/describe.go | 4 +- .../cmd/ske/cluster/hibernate/hibernate.go | 100 ++++++++++ .../ske/cluster/hibernate/hibernate_test.go | 183 +++++++++++++++++ .../ske/cluster/maintenance/maintenance.go | 100 ++++++++++ .../cluster/maintenance/maintenance_test.go | 184 ++++++++++++++++++ .../cmd/ske/cluster/reconcile/reconcile.go | 93 +++++++++ .../ske/cluster/reconcile/reconcile_test.go | 184 ++++++++++++++++++ internal/cmd/ske/cluster/wakeup/wakeup.go | 95 +++++++++ .../cmd/ske/cluster/wakeup/wakeup_test.go | 183 +++++++++++++++++ internal/cmd/ske/ske.go | 14 +- 17 files changed, 1302 insertions(+), 8 deletions(-) create mode 100644 docs/stackit_ske_hibernate.md create mode 100644 docs/stackit_ske_maintenance.md create mode 100644 docs/stackit_ske_reconcile.md create mode 100644 docs/stackit_ske_wakeup.md create mode 100644 internal/cmd/ske/cluster/hibernate/hibernate.go create mode 100644 internal/cmd/ske/cluster/hibernate/hibernate_test.go create mode 100644 internal/cmd/ske/cluster/maintenance/maintenance.go create mode 100644 internal/cmd/ske/cluster/maintenance/maintenance_test.go create mode 100644 internal/cmd/ske/cluster/reconcile/reconcile.go create mode 100644 internal/cmd/ske/cluster/reconcile/reconcile_test.go create mode 100644 internal/cmd/ske/cluster/wakeup/wakeup.go create mode 100644 internal/cmd/ske/cluster/wakeup/wakeup_test.go diff --git a/docs/stackit_ske.md b/docs/stackit_ske.md index b6a307937..d7bad3402 100644 --- a/docs/stackit_ske.md +++ b/docs/stackit_ske.md @@ -35,6 +35,10 @@ stackit ske [flags] * [stackit ske describe](./stackit_ske_describe.md) - Shows overall details regarding SKE * [stackit ske disable](./stackit_ske_disable.md) - Disables SKE for a project * [stackit ske enable](./stackit_ske_enable.md) - Enables SKE for a project +* [stackit ske hibernate](./stackit_ske_hibernate.md) - Trigger hibernate for a SKE cluster * [stackit ske kubeconfig](./stackit_ske_kubeconfig.md) - Provides functionality for SKE kubeconfig +* [stackit ske maintenance](./stackit_ske_maintenance.md) - Trigger maintenance for a SKE cluster * [stackit ske options](./stackit_ske_options.md) - Lists SKE provider options +* [stackit ske reconcile](./stackit_ske_reconcile.md) - Trigger reconcile for a SKE cluster +* [stackit ske wakeup](./stackit_ske_wakeup.md) - Trigger wakeup from hibernation for a SKE cluster diff --git a/docs/stackit_ske_cluster.md b/docs/stackit_ske_cluster.md index 7df9ba39e..378f1720d 100644 --- a/docs/stackit_ske_cluster.md +++ b/docs/stackit_ske_cluster.md @@ -32,7 +32,7 @@ stackit ske cluster [flags] * [stackit ske](./stackit_ske.md) - Provides functionality for SKE * [stackit ske cluster create](./stackit_ske_cluster_create.md) - Creates an SKE cluster * [stackit ske cluster delete](./stackit_ske_cluster_delete.md) - Deletes a SKE cluster -* [stackit ske cluster describe](./stackit_ske_cluster_describe.md) - Shows details of a SKE cluster +* [stackit ske cluster describe](./stackit_ske_cluster_describe.md) - Shows details of a SKE cluster * [stackit ske cluster generate-payload](./stackit_ske_cluster_generate-payload.md) - Generates a payload to create/update SKE clusters * [stackit ske cluster list](./stackit_ske_cluster_list.md) - Lists all SKE clusters * [stackit ske cluster update](./stackit_ske_cluster_update.md) - Updates an SKE cluster diff --git a/docs/stackit_ske_cluster_describe.md b/docs/stackit_ske_cluster_describe.md index eb30860a9..e19f15585 100644 --- a/docs/stackit_ske_cluster_describe.md +++ b/docs/stackit_ske_cluster_describe.md @@ -1,10 +1,10 @@ ## stackit ske cluster describe -Shows details of a SKE cluster +Shows details of a SKE cluster ### Synopsis -Shows details of a STACKIT Kubernetes Engine (SKE) cluster. +Shows details of a STACKIT Kubernetes Engine (SKE) cluster. ``` stackit ske cluster describe CLUSTER_NAME [flags] diff --git a/docs/stackit_ske_hibernate.md b/docs/stackit_ske_hibernate.md new file mode 100644 index 000000000..3988018fe --- /dev/null +++ b/docs/stackit_ske_hibernate.md @@ -0,0 +1,40 @@ +## stackit ske hibernate + +Trigger hibernate for a SKE cluster + +### Synopsis + +Trigger hibernate for a STACKIT Kubernetes Engine (SKE) cluster. + +``` +stackit ske hibernate CLUSTER_NAME [flags] +``` + +### Examples + +``` + Trigger hibernate for a SKE cluster with name "my-cluster" + $ stackit ske cluster hibernate my-cluster +``` + +### Options + +``` + -h, --help Help for "stackit ske hibernate" +``` + +### 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 + --region string Target region for region-specific requests + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit ske](./stackit_ske.md) - Provides functionality for SKE + diff --git a/docs/stackit_ske_maintenance.md b/docs/stackit_ske_maintenance.md new file mode 100644 index 000000000..bc64686bb --- /dev/null +++ b/docs/stackit_ske_maintenance.md @@ -0,0 +1,40 @@ +## stackit ske maintenance + +Trigger maintenance for a SKE cluster + +### Synopsis + +Trigger maintenance for a STACKIT Kubernetes Engine (SKE) cluster. + +``` +stackit ske maintenance CLUSTER_NAME [flags] +``` + +### Examples + +``` + Trigger maintenance for a SKE cluster with name "my-cluster" + $ stackit ske cluster maintenance my-cluster +``` + +### Options + +``` + -h, --help Help for "stackit ske maintenance" +``` + +### 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 + --region string Target region for region-specific requests + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit ske](./stackit_ske.md) - Provides functionality for SKE + diff --git a/docs/stackit_ske_reconcile.md b/docs/stackit_ske_reconcile.md new file mode 100644 index 000000000..3edf67dde --- /dev/null +++ b/docs/stackit_ske_reconcile.md @@ -0,0 +1,40 @@ +## stackit ske reconcile + +Trigger reconcile for a SKE cluster + +### Synopsis + +Trigger reconcile for a STACKIT Kubernetes Engine (SKE) cluster. + +``` +stackit ske reconcile CLUSTER_NAME [flags] +``` + +### Examples + +``` + Trigger reconcile for a SKE cluster with name "my-cluster" + $ stackit ske cluster reconcile my-cluster +``` + +### Options + +``` + -h, --help Help for "stackit ske reconcile" +``` + +### 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 + --region string Target region for region-specific requests + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit ske](./stackit_ske.md) - Provides functionality for SKE + diff --git a/docs/stackit_ske_wakeup.md b/docs/stackit_ske_wakeup.md new file mode 100644 index 000000000..4f5fb5e31 --- /dev/null +++ b/docs/stackit_ske_wakeup.md @@ -0,0 +1,40 @@ +## stackit ske wakeup + +Trigger wakeup from hibernation for a SKE cluster + +### Synopsis + +Trigger wakeup from hibernation for a STACKIT Kubernetes Engine (SKE) cluster. + +``` +stackit ske wakeup CLUSTER_NAME [flags] +``` + +### Examples + +``` + Trigger wakeup from hibernation for a SKE cluster with name "my-cluster" + $ stackit ske cluster wakeup my-cluster +``` + +### Options + +``` + -h, --help Help for "stackit ske wakeup" +``` + +### 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 + --region string Target region for region-specific requests + --verbosity string Verbosity of the CLI, one of ["debug" "info" "warning" "error"] (default "info") +``` + +### SEE ALSO + +* [stackit ske](./stackit_ske.md) - Provides functionality for SKE + diff --git a/internal/cmd/ske/cluster/describe/describe.go b/internal/cmd/ske/cluster/describe/describe.go index 40bf4897f..ed7feccdb 100644 --- a/internal/cmd/ske/cluster/describe/describe.go +++ b/internal/cmd/ske/cluster/describe/describe.go @@ -31,8 +31,8 @@ type inputModel struct { func NewCmd(params *params.CmdParams) *cobra.Command { cmd := &cobra.Command{ Use: fmt.Sprintf("describe %s", clusterNameArg), - Short: "Shows details of a SKE cluster", - Long: "Shows details of a STACKIT Kubernetes Engine (SKE) cluster.", + Short: "Shows details of a SKE cluster", + Long: "Shows details of a STACKIT Kubernetes Engine (SKE) cluster.", Args: args.SingleArg(clusterNameArg, nil), Example: examples.Build( examples.NewExample( diff --git a/internal/cmd/ske/cluster/hibernate/hibernate.go b/internal/cmd/ske/cluster/hibernate/hibernate.go new file mode 100644 index 000000000..5e23f7614 --- /dev/null +++ b/internal/cmd/ske/cluster/hibernate/hibernate.go @@ -0,0 +1,100 @@ +package hibernate + +import ( + "context" + "fmt" + + "github.com/spf13/cobra" + "github.com/stackitcloud/stackit-cli/internal/cmd/params" + "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/ske/client" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +const ( + clusterNameArg = "CLUSTER_NAME" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ClusterName string +} + +func NewCmd(params *params.CmdParams) *cobra.Command { + cmd := &cobra.Command{ + Use: fmt.Sprintf("hibernate %s", clusterNameArg), + Short: "Trigger hibernate for a SKE cluster", + Long: "Trigger hibernate for a STACKIT Kubernetes Engine (SKE) cluster.", + Args: args.SingleArg(clusterNameArg, nil), + Example: examples.Build( + examples.NewExample( + `Trigger hibernate for a SKE cluster with name "my-cluster"`, + "$ stackit ske cluster hibernate my-cluster"), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(params.Printer, cmd, args) + if err != nil { + return err + } + // Configure API client + apiClient, err := client.ConfigureClient(params.Printer, params.CliVersion) + if err != nil { + return err + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to trigger hibernate for %q in project %q?", model.ClusterName, model.ProjectId) + err = params.Printer.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + _, err = req.Execute() + if err != nil { + return fmt.Errorf("hibernate SKE cluster: %w", err) + } + + params.Printer.Info("Hibernate got triggered for %q\n", model.ClusterName) + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + clusterName := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ClusterName: clusterName, + } + + 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 *ske.APIClient) ske.ApiTriggerHibernateRequest { + req := apiClient.TriggerHibernate(ctx, model.ProjectId, model.Region, model.ClusterName) + return req +} diff --git a/internal/cmd/ske/cluster/hibernate/hibernate_test.go b/internal/cmd/ske/cluster/hibernate/hibernate_test.go new file mode 100644 index 000000000..0be99a5a4 --- /dev/null +++ b/internal/cmd/ske/cluster/hibernate/hibernate_test.go @@ -0,0 +1,183 @@ +package hibernate + +import ( + "context" + "testing" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/spf13/cobra" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &ske.APIClient{} +var testProjectId = uuid.NewString() +var testRegion = "eu01" +var testClusterName = "my-cluster" + +func fixtureArgValues(mods ...func(argValues []string)) []string { + argValues := []string{ + testClusterName, + } + for _, mod := range mods { + mod(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(flagValues map[string]string)) map[string]string { + flagValues := map[string]string{ + globalflags.ProjectIdFlag: testProjectId, + globalflags.RegionFlag: testRegion, + } + for _, mod := range mods { + mod(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(model *inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + ProjectId: testProjectId, + Region: testRegion, + Verbosity: globalflags.VerbosityDefault, + }, + ClusterName: testClusterName, + } + for _, mod := range mods { + mod(model) + } + return model +} + +func fixtureRequest(mods ...func(request *ske.ApiTriggerHibernateRequest)) ske.ApiTriggerHibernateRequest { + request := testClient.TriggerHibernate(testCtx, testProjectId, testRegion, testClusterName) + 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: "missing project id", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + delete(fv, globalflags.ProjectIdFlag) + }), + isValid: false, + }, + { + description: "invalid project id - empty string", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "" + }), + isValid: false, + }, + { + description: "invalid uuid format", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "not-a-uuid" + }), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := &cobra.Command{} + 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) + } + } + + if len(tt.argValues) == 0 { + _, err := parseInput(p, cmd, tt.argValues) + if err == nil && !tt.isValid { + t.Fatalf("expected error due to missing args") + } + return + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("error parsing flags: %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:\n%s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest ske.ApiTriggerHibernateRequest + }{ + { + 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, + cmpopts.EquateComparable(testCtx), + cmp.AllowUnexported(tt.expectedRequest), + ) + if diff != "" { + t.Fatalf("request mismatch:\n%s", diff) + } + }) + } +} diff --git a/internal/cmd/ske/cluster/maintenance/maintenance.go b/internal/cmd/ske/cluster/maintenance/maintenance.go new file mode 100644 index 000000000..4a3696da2 --- /dev/null +++ b/internal/cmd/ske/cluster/maintenance/maintenance.go @@ -0,0 +1,100 @@ +package maintenance + +import ( + "context" + "fmt" + + "github.com/spf13/cobra" + "github.com/stackitcloud/stackit-cli/internal/cmd/params" + "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/ske/client" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +const ( + clusterNameArg = "CLUSTER_NAME" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ClusterName string +} + +func NewCmd(params *params.CmdParams) *cobra.Command { + cmd := &cobra.Command{ + Use: fmt.Sprintf("maintenance %s", clusterNameArg), + Short: "Trigger maintenance for a SKE cluster", + Long: "Trigger maintenance for a STACKIT Kubernetes Engine (SKE) cluster.", + Args: args.SingleArg(clusterNameArg, nil), + Example: examples.Build( + examples.NewExample( + `Trigger maintenance for a SKE cluster with name "my-cluster"`, + "$ stackit ske cluster maintenance my-cluster"), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(params.Printer, cmd, args) + if err != nil { + return err + } + // Configure API client + apiClient, err := client.ConfigureClient(params.Printer, params.CliVersion) + if err != nil { + return err + } + + if !model.AssumeYes { + prompt := fmt.Sprintf("Are you sure you want to trigger maintenance for %q in project %q?", model.ClusterName, model.ProjectId) + err = params.Printer.PromptForConfirmation(prompt) + if err != nil { + return err + } + } + + // Call API + req := buildRequest(ctx, model, apiClient) + _, err = req.Execute() + if err != nil { + return fmt.Errorf("trigger maintenance SKE cluster: %w", err) + } + + params.Printer.Info("Maintenance got triggered for %q\n", model.ClusterName) + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + clusterName := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ClusterName: clusterName, + } + + 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 *ske.APIClient) ske.ApiTriggerMaintenanceRequest { + req := apiClient.TriggerMaintenance(ctx, model.ProjectId, model.Region, model.ClusterName) + return req +} diff --git a/internal/cmd/ske/cluster/maintenance/maintenance_test.go b/internal/cmd/ske/cluster/maintenance/maintenance_test.go new file mode 100644 index 000000000..225f1282d --- /dev/null +++ b/internal/cmd/ske/cluster/maintenance/maintenance_test.go @@ -0,0 +1,184 @@ +package maintenance + +import ( + "context" + "testing" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/spf13/cobra" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &ske.APIClient{} +var testProjectId = uuid.NewString() +var testRegion = "eu01" +var testClusterName = "my-cluster" + +func fixtureArgValues(mods ...func([]string)) []string { + argValues := []string{ + testClusterName, + } + for _, m := range mods { + m(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(map[string]string)) map[string]string { + flagValues := map[string]string{ + globalflags.ProjectIdFlag: testProjectId, + globalflags.RegionFlag: testRegion, + } + for _, m := range mods { + m(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(*inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + ProjectId: testProjectId, + Region: testRegion, + Verbosity: globalflags.VerbosityDefault, + }, + ClusterName: testClusterName, + } + for _, m := range mods { + m(model) + } + return model +} + +func fixtureRequest(mods ...func(*ske.ApiTriggerMaintenanceRequest)) ske.ApiTriggerMaintenanceRequest { + request := testClient.TriggerMaintenance(testCtx, testProjectId, testRegion, testClusterName) + for _, m := range mods { + m(&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: "missing project id", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + delete(fv, globalflags.ProjectIdFlag) + }), + isValid: false, + }, + { + description: "invalid project id - empty string", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "" + }), + isValid: false, + }, + { + description: "invalid uuid format", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "not-a-uuid" + }), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := &cobra.Command{} + 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) + } + } + + if len(tt.argValues) == 0 { + _, err := parseInput(p, cmd, tt.argValues) + if err == nil && !tt.isValid { + t.Fatalf("expected error due to missing args") + } + return + } + + 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("input model mismatch:\n%s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest ske.ApiTriggerMaintenanceRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + got := buildRequest(testCtx, tt.model, testClient) + want := tt.expectedRequest + + diff := cmp.Diff(got, want, + cmpopts.EquateComparable(testCtx), + cmp.AllowUnexported(want), + ) + if diff != "" { + t.Fatalf("request mismatch:\n%s", diff) + } + }) + } +} diff --git a/internal/cmd/ske/cluster/reconcile/reconcile.go b/internal/cmd/ske/cluster/reconcile/reconcile.go new file mode 100644 index 000000000..9ccbe97f0 --- /dev/null +++ b/internal/cmd/ske/cluster/reconcile/reconcile.go @@ -0,0 +1,93 @@ +package reconcile + +import ( + "context" + "fmt" + + "github.com/spf13/cobra" + "github.com/stackitcloud/stackit-cli/internal/cmd/params" + "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/ske/client" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +const ( + clusterNameArg = "CLUSTER_NAME" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ClusterName string +} + +func NewCmd(params *params.CmdParams) *cobra.Command { + cmd := &cobra.Command{ + Use: fmt.Sprintf("reconcile %s", clusterNameArg), + Short: "Trigger reconcile for a SKE cluster", + Long: "Trigger reconcile for a STACKIT Kubernetes Engine (SKE) cluster.", + Args: args.SingleArg(clusterNameArg, nil), + Example: examples.Build( + examples.NewExample( + `Trigger reconcile for a SKE cluster with name "my-cluster"`, + "$ stackit ske cluster reconcile my-cluster"), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(params.Printer, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(params.Printer, params.CliVersion) + if err != nil { + return err + } + + // Call API + req := buildRequest(ctx, model, apiClient) + _, err = req.Execute() + if err != nil { + return fmt.Errorf("reconcile SKE cluster: %w", err) + } + + params.Printer.Info("Reconcile got triggered for %q\n", model.ClusterName) + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + clusterName := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ClusterName: clusterName, + } + + 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 *ske.APIClient) ske.ApiTriggerReconcileRequest { + req := apiClient.TriggerReconcile(ctx, model.ProjectId, model.Region, model.ClusterName) + return req +} diff --git a/internal/cmd/ske/cluster/reconcile/reconcile_test.go b/internal/cmd/ske/cluster/reconcile/reconcile_test.go new file mode 100644 index 000000000..9c073273c --- /dev/null +++ b/internal/cmd/ske/cluster/reconcile/reconcile_test.go @@ -0,0 +1,184 @@ +package reconcile + +import ( + "context" + "testing" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/spf13/cobra" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &ske.APIClient{} +var testProjectId = uuid.NewString() +var testRegion = "eu01" +var testClusterName = "my-cluster" + +func fixtureArgValues(mods ...func([]string)) []string { + argValues := []string{ + testClusterName, + } + for _, m := range mods { + m(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(map[string]string)) map[string]string { + flagValues := map[string]string{ + globalflags.ProjectIdFlag: testProjectId, + globalflags.RegionFlag: testRegion, + } + for _, m := range mods { + m(flagValues) + } + return flagValues +} + +func fixtureInputModel(mods ...func(*inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + ProjectId: testProjectId, + Region: testRegion, + Verbosity: globalflags.VerbosityDefault, + }, + ClusterName: testClusterName, + } + for _, m := range mods { + m(model) + } + return model +} + +func fixtureRequest(mods ...func(request *ske.ApiTriggerReconcileRequest)) ske.ApiTriggerHibernateRequest { + request := testClient.TriggerReconcile(testCtx, testProjectId, testRegion, testClusterName) + for _, m := range mods { + m(&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: "missing project id", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + delete(fv, globalflags.ProjectIdFlag) + }), + isValid: false, + }, + { + description: "invalid project id - empty string", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "" + }), + isValid: false, + }, + { + description: "invalid uuid format", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "not-a-uuid" + }), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := &cobra.Command{} + 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) + } + } + + if len(tt.argValues) == 0 { + _, err := parseInput(p, cmd, tt.argValues) + if err == nil && !tt.isValid { + t.Fatalf("expected error due to missing args") + } + return + } + + 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("input model mismatch:\n%s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest ske.ApiTriggerHibernateRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + got := buildRequest(testCtx, tt.model, testClient) + want := tt.expectedRequest + + diff := cmp.Diff(got, want, + cmpopts.EquateComparable(testCtx), + cmp.AllowUnexported(want), + ) + if diff != "" { + t.Fatalf("request mismatch:\n%s", diff) + } + }) + } +} diff --git a/internal/cmd/ske/cluster/wakeup/wakeup.go b/internal/cmd/ske/cluster/wakeup/wakeup.go new file mode 100644 index 000000000..5aa103e1a --- /dev/null +++ b/internal/cmd/ske/cluster/wakeup/wakeup.go @@ -0,0 +1,95 @@ +package wakeup + +import ( + "context" + "fmt" + + "github.com/spf13/cobra" + "github.com/stackitcloud/stackit-cli/internal/cmd/params" + "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/ske/client" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +const ( + clusterNameArg = "CLUSTER_NAME" +) + +type inputModel struct { + *globalflags.GlobalFlagModel + ClusterName string +} + +func NewCmd(params *params.CmdParams) *cobra.Command { + cmd := &cobra.Command{ + Use: fmt.Sprintf("wakeup %s", clusterNameArg), + Short: "Trigger wakeup from hibernation for a SKE cluster", + Long: "Trigger wakeup from hibernation for a STACKIT Kubernetes Engine (SKE) cluster.", + Args: args.SingleArg(clusterNameArg, nil), + Example: examples.Build( + examples.NewExample( + `Trigger wakeup from hibernation for a SKE cluster with name "my-cluster"`, + "$ stackit ske cluster wakeup my-cluster"), + ), + RunE: func(cmd *cobra.Command, args []string) error { + ctx := context.Background() + model, err := parseInput(params.Printer, cmd, args) + if err != nil { + return err + } + + // Configure API client + apiClient, err := client.ConfigureClient(params.Printer, params.CliVersion) + if err != nil { + return err + } + + // Call API + req := buildRequest(ctx, model, apiClient) + _, err = req.Execute() + if err != nil { + return fmt.Errorf("wakeup SKE cluster: %w", err) + } + + params.Printer.Info("Wakeup got triggered for %q\n", model.ClusterName) + return nil + }, + } + return cmd +} + +func parseInput(p *print.Printer, cmd *cobra.Command, inputArgs []string) (*inputModel, error) { + clusterName := inputArgs[0] + + globalFlags := globalflags.Parse(p, cmd) + if globalFlags.ProjectId == "" { + return nil, &errors.ProjectIdError{} + } + + model := inputModel{ + GlobalFlagModel: globalFlags, + ClusterName: clusterName, + } + + 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 +} + +// TODO: rewrite to Wakeup +func buildRequest(ctx context.Context, model *inputModel, apiClient *ske.APIClient) ske.ApiTriggerHibernateRequest { + // TODO: rewrite to Wakeup + req := apiClient.TriggerHibernate(ctx, model.ProjectId, model.Region, model.ClusterName) + return req +} diff --git a/internal/cmd/ske/cluster/wakeup/wakeup_test.go b/internal/cmd/ske/cluster/wakeup/wakeup_test.go new file mode 100644 index 000000000..9a07a0105 --- /dev/null +++ b/internal/cmd/ske/cluster/wakeup/wakeup_test.go @@ -0,0 +1,183 @@ +package wakeup + +import ( + "context" + "testing" + + "github.com/google/go-cmp/cmp" + "github.com/google/go-cmp/cmp/cmpopts" + "github.com/google/uuid" + "github.com/spf13/cobra" + + "github.com/stackitcloud/stackit-cli/internal/pkg/globalflags" + "github.com/stackitcloud/stackit-cli/internal/pkg/print" + "github.com/stackitcloud/stackit-sdk-go/services/ske" +) + +type testCtxKey struct{} + +var testCtx = context.WithValue(context.Background(), testCtxKey{}, "foo") +var testClient = &ske.APIClient{} +var testProjectId = uuid.NewString() +var testRegion = "eu01" +var testClusterName = "my-cluster" + +func fixtureArgValues(mods ...func([]string)) []string { + argValues := []string{testClusterName} + for _, m := range mods { + m(argValues) + } + return argValues +} + +func fixtureFlagValues(mods ...func(map[string]string)) map[string]string { + flags := map[string]string{ + globalflags.ProjectIdFlag: testProjectId, + globalflags.RegionFlag: testRegion, + } + for _, m := range mods { + m(flags) + } + return flags +} + +func fixtureInputModel(mods ...func(*inputModel)) *inputModel { + model := &inputModel{ + GlobalFlagModel: &globalflags.GlobalFlagModel{ + ProjectId: testProjectId, + Region: testRegion, + Verbosity: globalflags.VerbosityDefault, + }, + ClusterName: testClusterName, + } + for _, m := range mods { + m(model) + } + return model +} + +func fixtureRequest(mods ...func(*ske.ApiTriggerHibernateRequest)) ske.ApiTriggerHibernateRequest { + // TODO: rewrite to Wakeup + req := testClient.TriggerHibernate(testCtx, testProjectId, testRegion, testClusterName) + for _, m := range mods { + m(&req) + } + return req +} + +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: "missing project id", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + delete(fv, globalflags.ProjectIdFlag) + }), + isValid: false, + }, + { + description: "invalid project id - empty string", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "" + }), + isValid: false, + }, + { + description: "invalid uuid format", + argValues: fixtureArgValues(), + flagValues: fixtureFlagValues(func(fv map[string]string) { + fv[globalflags.ProjectIdFlag] = "not-a-uuid" + }), + isValid: false, + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + p := print.NewPrinter() + cmd := &cobra.Command{} + 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) + } + } + + if len(tt.argValues) == 0 { + _, err := parseInput(p, cmd, tt.argValues) + if err == nil && !tt.isValid { + t.Fatalf("expected failure due to missing args") + } + return + } + + model, err := parseInput(p, cmd, tt.argValues) + if err != nil { + if !tt.isValid { + return + } + t.Fatalf("unexpected error: %v", err) + } + + if !tt.isValid { + t.Fatalf("did not fail on invalid input") + } + + diff := cmp.Diff(model, tt.expectedModel) + if diff != "" { + t.Fatalf("input model mismatch:\n%s", diff) + } + }) + } +} + +func TestBuildRequest(t *testing.T) { + tests := []struct { + description string + model *inputModel + expectedRequest ske.ApiTriggerHibernateRequest + }{ + { + description: "base", + model: fixtureInputModel(), + expectedRequest: fixtureRequest(), + }, + } + + for _, tt := range tests { + t.Run(tt.description, func(t *testing.T) { + got := buildRequest(testCtx, tt.model, testClient) + want := tt.expectedRequest + + diff := cmp.Diff(got, want, + cmpopts.EquateComparable(testCtx), + cmp.AllowUnexported(want), + ) + if diff != "" { + t.Fatalf("request mismatch:\n%s", diff) + } + }) + } +} diff --git a/internal/cmd/ske/ske.go b/internal/cmd/ske/ske.go index 137165e06..3d2cde077 100644 --- a/internal/cmd/ske/ske.go +++ b/internal/cmd/ske/ske.go @@ -3,6 +3,10 @@ package ske import ( "github.com/stackitcloud/stackit-cli/internal/cmd/params" "github.com/stackitcloud/stackit-cli/internal/cmd/ske/cluster" + "github.com/stackitcloud/stackit-cli/internal/cmd/ske/cluster/hibernate" + "github.com/stackitcloud/stackit-cli/internal/cmd/ske/cluster/maintenance" + "github.com/stackitcloud/stackit-cli/internal/cmd/ske/cluster/reconcile" + "github.com/stackitcloud/stackit-cli/internal/cmd/ske/cluster/wakeup" "github.com/stackitcloud/stackit-cli/internal/cmd/ske/credentials" "github.com/stackitcloud/stackit-cli/internal/cmd/ske/describe" "github.com/stackitcloud/stackit-cli/internal/cmd/ske/disable" @@ -28,11 +32,15 @@ func NewCmd(params *params.CmdParams) *cobra.Command { } func addSubcommands(cmd *cobra.Command, params *params.CmdParams) { + cmd.AddCommand(cluster.NewCmd(params)) + cmd.AddCommand(credentials.NewCmd(params)) cmd.AddCommand(describe.NewCmd(params)) + cmd.AddCommand(disable.NewCmd(params)) cmd.AddCommand(enable.NewCmd(params)) + cmd.AddCommand(hibernate.NewCmd(params)) cmd.AddCommand(kubeconfig.NewCmd(params)) - cmd.AddCommand(disable.NewCmd(params)) - cmd.AddCommand(cluster.NewCmd(params)) - cmd.AddCommand(credentials.NewCmd(params)) + cmd.AddCommand(maintenance.NewCmd(params)) cmd.AddCommand(options.NewCmd(params)) + cmd.AddCommand(reconcile.NewCmd(params)) + cmd.AddCommand(wakeup.NewCmd(params)) }