feat(registry): add configurable timeout for tag deletion

Author: Mia-Cross

core/testing.go

@@ -674,6 +674,13 @@ func ExecBeforeCmdArgs(args []string) BeforeFunc {
 	}
 }
 
+// ExecBeforeCmdWithResult executes the given command and returns its result.
+func ExecBeforeCmdWithResult(ctx *BeforeFuncCtx, cmd string) any {
+	args := cmdToArgs(ctx.Meta, cmd)
+	ctx.Logger.Debugf("ExecBeforeCmd: args=%s\n", args)
+	return ctx.ExecuteCmd(args)
+}
+
 // ExecAfterCmd executes the given before command.
 func ExecAfterCmd(cmd string) AfterFunc {
 	return func(ctx *AfterFuncCtx) error {

internal/namespaces/registry/v1/custom.go

@@ -27,6 +27,7 @@ func GetCommands() *core.Commands {
 		registryInstallDockerHelperCommand(),
 	))
 
+	cmds.MustFind("registry", "tag", "delete").Override(tagDeleteBuilder)
 	cmds.MustFind("registry", "tag", "get").Override(tagGetBuilder)
 	cmds.MustFind("registry", "tag", "list").Override(tagListBuilder)
 

internal/namespaces/registry/v1/custom_tag.go

@@ -3,6 +3,8 @@ package registry
 import (
 	"context"
 	"fmt"
+	"reflect"
+	"time"
 
 	"github.com/fatih/color"
 	"github.com/scaleway/scaleway-cli/v2/core"
@@ -25,7 +27,7 @@ var (
 	}
 )
 
-type customTag struct {
+type CustomTag struct {
 	registry.Tag
 	FullName string
 }
@@ -59,7 +61,7 @@ func tagGetBuilder(c *core.Command) *core.Command {
 			return getTagResp, nil
 		}
 
-		res := customTag{
+		res := CustomTag{
 			Tag:      *tag,
 			FullName: fmt.Sprintf("%s/%s:%s", namespace.Endpoint, image.Name, tag.Name),
 		}
@@ -112,9 +114,9 @@ func tagListBuilder(c *core.Command) *core.Command {
 			return listTagResp, err
 		}
 
-		var customRes []customTag
+		var customRes []CustomTag
 		for _, tag := range listTagResp.([]*registry.Tag) {
-			customRes = append(customRes, customTag{
+			customRes = append(customRes, CustomTag{
 				Tag: *tag,
 				FullName: fmt.Sprintf("%s/%s:%s",
 					namespace.Endpoint,
@@ -129,3 +131,29 @@ func tagListBuilder(c *core.Command) *core.Command {
 
 	return c
 }
+
+type customTagDeleteArgs struct {
+	*registry.DeleteTagRequest
+	timeout time.Duration
+}
+
+func tagDeleteBuilder(c *core.Command) *core.Command {
+	c.ArgsType = reflect.TypeOf(customTagDeleteArgs{})
+	c.ArgSpecs.AddBefore("force", &core.ArgSpec{
+		Name:       "timeout",
+		Short:      "Maximum time to handle the request",
+		Required:   false,
+		Positional: false,
+	})
+
+	c.Interceptor = func(ctx context.Context, argsI any, runner core.CommandRunner) (any, error) {
+		args := argsI.(*customTagDeleteArgs)
+
+		ctxWithTimeout, cancel := context.WithTimeout(ctx, args.timeout)
+		defer cancel()
+
+		return runner(ctxWithTimeout, args.DeleteTagRequest)
+	}
+
+	return c
+}

internal/namespaces/registry/v1/custom_tag_test.go

@@ -0,0 +1,56 @@
+package registry_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/scaleway/scaleway-cli/v2/core"
+	"github.com/scaleway/scaleway-cli/v2/internal/namespaces/registry/v1"
+	"github.com/scaleway/scaleway-cli/v2/internal/testhelpers"
+)
+
+func Test_RegistryTagDelete(t *testing.T) {
+	registryNamespaceMetaKey := "RegistryNamespace"
+	helloWorldImage := "hello-world:latest"
+	helloWorldImageMetaKey := "HelloWorldImage"
+	tagIDMetaKey := "TagID"
+
+	t.Run("timeout-ok", core.Test(&core.TestConfig{
+		Commands: registry.GetCommands(),
+		BeforeFunc: core.BeforeFuncCombine(
+			core.ExecStoreBeforeCmd(
+				registryNamespaceMetaKey,
+				fmt.Sprintf("scw registry namespace create name=%s is-public=false",
+					core.GetRandomName("test-rg-tag-delete"),
+				),
+			),
+			core.BeforeFuncWhenUpdatingCassette(
+				core.BeforeFuncCombine(
+					core.ExecBeforeCmd("scw registry login"),
+					testhelpers.PushRegistryImage(helloWorldImage, registryNamespaceMetaKey),
+				),
+			),
+			testhelpers.StoreImageIdentifierInMeta(
+				registryNamespaceMetaKey,
+				helloWorldImage,
+				helloWorldImageMetaKey,
+			),
+			testhelpers.StoreTagIDInMeta(registryNamespaceMetaKey, helloWorldImage, tagIDMetaKey),
+		),
+		Cmd: fmt.Sprintf("scw registry tag delete {{ .%s }} timeout=1s", tagIDMetaKey),
+		Check: core.TestCheckCombine(
+			core.TestCheckGolden(),
+			core.TestCheckExitCode(0),
+		),
+		AfterFunc: func(ctx *core.AfterFuncCtx) error {
+			return core.ExecAfterCmd(
+				fmt.Sprintf(
+					"scw registry namespace delete {{ .%s.ID }}",
+					registryNamespaceMetaKey,
+				),
+			)(
+				ctx,
+			)
+		},
+	}))
+}