feat(mnq) : add autocompletion for credentials and account (#3700)

Author: jremy42

internal/namespaces/mnq/v1beta1/custom.go

@@ -11,6 +11,16 @@ func GetCommands() *core.Commands {
 
 	human.RegisterMarshalerFunc(mnq.SnsInfoStatus(""), human.EnumMarshalFunc(mnqSqsInfoStatusMarshalSpecs))
 
+	cmds.MustFind("mnq", "nats", "get-account").Override(mnqNatsGetAccountBuilder)
+	cmds.MustFind("mnq", "nats", "list-credentials").Override(mnqNatsListCredentialsBuilder)
+
+	cmds.MustFind("mnq", "sqs", "list-credentials").Override(mnqSqsListCredentialsBuilder)
+	cmds.MustFind("mnq", "sqs", "get-credentials").Override(mnqSqsGetCredentialsBuilder)
+
+	cmds.MustFind("mnq", "sns", "list-credentials").Override(mnqSnsListCredentialsBuilder)
+
+	cmds.MustFind("mnq", "sns", "get-credentials").Override(mnqSnsGetCredentialsBuilder)
+
 	cmds.Merge(core.NewCommands(
 		createContextCommand(),
 	))

internal/namespaces/mnq/v1beta1/custom_nats.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"reflect"
+	"strings"
 
 	"github.com/scaleway/scaleway-cli/v2/internal/core"
 	mnq "github.com/scaleway/scaleway-sdk-go/api/mnq/v1beta1"
@@ -91,3 +92,47 @@ Credentials and context file are saved in your nats context folder with 0600 per
 		},
 	}
 }
+
+func mnqNatsGetAccountBuilder(c *core.Command) *core.Command {
+	c.ArgSpecs.GetByName("nats-account-id").AutoCompleteFunc = autocompleteNatsAccountID
+	return c
+}
+
+func mnqNatsListCredentialsBuilder(c *core.Command) *core.Command {
+	c.ArgSpecs.GetByName("nats-account-id").AutoCompleteFunc = autocompleteNatsAccountID
+	return c
+}
+
+var completeListNatsAccountIDCache *mnq.ListNatsAccountsResponse
+
+func autocompleteNatsAccountID(ctx context.Context, prefix string, request any) core.AutocompleteSuggestions {
+	region := scw.Region("")
+	switch req := request.(type) {
+	case *mnq.NatsAPIGetNatsAccountRequest:
+		region = req.Region
+	case *mnq.NatsAPIListNatsCredentialsRequest:
+		region = req.Region
+	}
+
+	suggestions := core.AutocompleteSuggestions(nil)
+
+	client := core.ExtractClient(ctx)
+	api := mnq.NewNatsAPI(client)
+
+	if completeListNatsAccountIDCache == nil {
+		res, err := api.ListNatsAccounts(&mnq.NatsAPIListNatsAccountsRequest{
+			Region: region,
+		})
+		if err != nil {
+			return nil
+		}
+		completeListNatsAccountIDCache = res
+	}
+
+	for _, natsAccountID := range completeListNatsAccountIDCache.NatsAccounts {
+		if strings.HasPrefix(natsAccountID.ID, prefix) {
+			suggestions = append(suggestions, natsAccountID.ID)
+		}
+	}
+	return suggestions
+}

internal/namespaces/mnq/v1beta1/custom_sns.go

@@ -0,0 +1,83 @@
+package mnq
+
+import (
+	"context"
+	"strings"
+
+	"github.com/scaleway/scaleway-cli/v2/internal/core"
+	mnq "github.com/scaleway/scaleway-sdk-go/api/mnq/v1beta1"
+)
+
+func mnqSnsListCredentialsBuilder(c *core.Command) *core.Command {
+	c.ArgSpecs.GetByName("project-id").AutoCompleteFunc = autocompleteSnsProjectID
+	return c
+}
+
+func mnqSnsGetCredentialsBuilder(c *core.Command) *core.Command {
+	c.ArgSpecs.GetByName("sns-credentials-id").AutoCompleteFunc = autocompleteSnsCredentialsID
+	return c
+}
+
+var completeSnsInfoCache *mnq.SnsInfo
+
+func autocompleteSnsProjectID(ctx context.Context, prefix string, request any) core.AutocompleteSuggestions {
+	_ = prefix
+	req := request.(*mnq.SnsAPIListSnsCredentialsRequest)
+	suggestions := core.AutocompleteSuggestions(nil)
+
+	client := core.ExtractClient(ctx)
+	api := mnq.NewSnsAPI(client)
+
+	if completeSnsInfoCache == nil {
+		res, err := api.GetSnsInfo(&mnq.SnsAPIGetSnsInfoRequest{
+			Region: req.Region,
+		})
+		if err != nil {
+			return nil
+		}
+		completeSnsInfoCache = res
+	}
+
+	suggestions = append(suggestions, completeSnsInfoCache.ProjectID)
+	return suggestions
+}
+
+var completeSnsListCredentials *mnq.ListSnsCredentialsResponse
+
+func autocompleteSnsCredentialsID(ctx context.Context, prefix string, request any) core.AutocompleteSuggestions {
+	req := request.(*mnq.SnsAPIGetSnsCredentialsRequest)
+
+	suggestions := core.AutocompleteSuggestions(nil)
+
+	if req.Region != "" {
+		client := core.ExtractClient(ctx)
+		api := mnq.NewSnsAPI(client)
+		if completeSnsInfoCache == nil {
+			res, err := api.GetSnsInfo(&mnq.SnsAPIGetSnsInfoRequest{
+				Region: req.Region,
+			})
+			if err != nil {
+				return nil
+			}
+			completeSnsInfoCache = res
+		}
+		if completeSnsListCredentials == nil {
+			res, err := api.ListSnsCredentials(&mnq.SnsAPIListSnsCredentialsRequest{
+				Region:    req.Region,
+				ProjectID: &completeSnsInfoCache.ProjectID,
+			})
+
+			if err != nil {
+				return nil
+			}
+			completeSnsListCredentials = res
+		}
+
+		for _, snsCredentials := range completeSnsListCredentials.SnsCredentials {
+			if strings.HasPrefix(snsCredentials.ID, prefix) {
+				suggestions = append(suggestions, snsCredentials.ID)
+			}
+		}
+	}
+	return suggestions
+}

internal/namespaces/mnq/v1beta1/custom_sqs.go

@@ -1,7 +1,11 @@
 package mnq
 
 import (
+	"context"
+	"strings"
+
 	"github.com/fatih/color"
+	"github.com/scaleway/scaleway-cli/v2/internal/core"
 	"github.com/scaleway/scaleway-cli/v2/internal/human"
 	mnq "github.com/scaleway/scaleway-sdk-go/api/mnq/v1beta1"
 )
@@ -13,3 +17,75 @@ var (
 		mnq.SqsInfoStatusDisabled:      &human.EnumMarshalSpec{Attribute: color.FgRed},
 	}
 )
+
+func mnqSqsListCredentialsBuilder(c *core.Command) *core.Command {
+	c.ArgSpecs.GetByName("project-id").AutoCompleteFunc = autocompleteSqsProjectID
+	return c
+}
+
+func mnqSqsGetCredentialsBuilder(c *core.Command) *core.Command {
+	c.ArgSpecs.GetByName("sqs-credentials-id").AutoCompleteFunc = autocompleteSqsCredentialsID
+	return c
+}
+
+var completeSqsInfoCache *mnq.SqsInfo
+
+func autocompleteSqsProjectID(ctx context.Context, prefix string, request any) core.AutocompleteSuggestions {
+	_ = prefix
+	req := request.(*mnq.SqsAPIListSqsCredentialsRequest)
+	suggestions := core.AutocompleteSuggestions(nil)
+	client := core.ExtractClient(ctx)
+	api := mnq.NewSqsAPI(client)
+	if completeSqsInfoCache == nil {
+		res, err := api.GetSqsInfo(&mnq.SqsAPIGetSqsInfoRequest{
+			Region: req.Region,
+		})
+		if err != nil {
+			return nil
+		}
+		completeSqsInfoCache = res
+	}
+
+	suggestions = append(suggestions, completeSqsInfoCache.ProjectID)
+	return suggestions
+}
+
+var completeSqsListCredentials *mnq.ListSqsCredentialsResponse
+
+func autocompleteSqsCredentialsID(ctx context.Context, prefix string, request any) core.AutocompleteSuggestions {
+	req := request.(*mnq.SqsAPIGetSqsCredentialsRequest)
+
+	suggestions := core.AutocompleteSuggestions(nil)
+
+	if req.Region != "" {
+		client := core.ExtractClient(ctx)
+		api := mnq.NewSqsAPI(client)
+		if completeSqsInfoCache == nil {
+			res, err := api.GetSqsInfo(&mnq.SqsAPIGetSqsInfoRequest{
+				Region: req.Region,
+			})
+			if err != nil {
+				return nil
+			}
+			completeSqsInfoCache = res
+		}
+		if completeSqsListCredentials != nil {
+			res, err := api.ListSqsCredentials(&mnq.SqsAPIListSqsCredentialsRequest{
+				Region:    req.Region,
+				ProjectID: &completeSqsInfoCache.ProjectID,
+			})
+
+			if err != nil {
+				return nil
+			}
+			completeSqsListCredentials = res
+		}
+
+		for _, sqsCredentials := range completeSqsListCredentials.SqsCredentials {
+			if strings.HasPrefix(sqsCredentials.ID, prefix) {
+				suggestions = append(suggestions, sqsCredentials.ID)
+			}
+		}
+	}
+	return suggestions
+}