add: delete sagemaker domain (#232)

cmd/delete/sagemaker.go

@@ -2,6 +2,7 @@ package delete
 
 import (
 	"github.com/awslabs/eksdemo/pkg/resource"
+	"github.com/awslabs/eksdemo/pkg/resource/sagemaker/domain"
 	"github.com/awslabs/eksdemo/pkg/resource/sagemaker/userprofile"
 	"github.com/spf13/cobra"
 )
@@ -27,6 +28,7 @@ func NewSageMakerCmd() *cobra.Command {
 
 func init() {
 	sagemaker = []func() *resource.Resource{
+		domain.New,
 		userprofile.New,
 	}
 }

pkg/aws/sagemaker.go

@@ -16,6 +16,14 @@ func NewSageMakerClient() *SageMakerClient {
 	return &SageMakerClient{sagemaker.NewFromConfig(GetConfig())}
 }
 
+func (c *SageMakerClient) DeleteDomain(domainID string) error {
+	_, err := c.Client.DeleteDomain(context.Background(), &sagemaker.DeleteDomainInput{
+		DomainId: aws.String(domainID),
+	})
+
+	return err
+}
+
 func (c *SageMakerClient) DeleteUserProfile(domainID, userProfileName string) error {
 	_, err := c.Client.DeleteUserProfile(context.Background(), &sagemaker.DeleteUserProfileInput{
 		DomainId:        aws.String(domainID),

pkg/cmd/error.go

@@ -19,3 +19,12 @@ type FlagRequiresFlagError struct {
 func (e *FlagRequiresFlagError) Error() string {
 	return fmt.Sprintf("%q flag requires %q flag", e.Flag1, e.Flag2)
 }
+
+type MustIncludeEitherArgumentOrFlag struct {
+	Arg  string
+	Flag string
+}
+
+func (e *MustIncludeEitherArgumentOrFlag) Error() string {
+	return fmt.Sprintf("must include either %q argument or %q flag", e.Arg, e.Flag)
+}

pkg/resource/sagemaker/domain/manager.go

@@ -0,0 +1,58 @@
+package domain
+
+import (
+	"errors"
+	"fmt"
+
+	awssdk "github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/awslabs/eksdemo/pkg/aws"
+	"github.com/awslabs/eksdemo/pkg/resource"
+)
+
+type Manager struct {
+	resource.CreateNotSupported
+	resource.DryRun
+	resource.UpdateNotSupported
+	sagemakerClient *aws.SageMakerClient
+	domainGetter    *Getter
+}
+
+func (m *Manager) Init() {
+	if m.sagemakerClient == nil {
+		m.sagemakerClient = aws.NewSageMakerClient()
+	}
+	m.domainGetter = NewGetter(m.sagemakerClient)
+}
+
+func (m *Manager) Delete(o resource.Options) error {
+	options, ok := o.(*Options)
+	if !ok {
+		return fmt.Errorf("internal error, unable to cast options to domain.Options")
+	}
+
+	domainID := options.DomainID
+	domainName := options.Name
+
+	if domainID == "" {
+		domain, err := m.domainGetter.GetDomainByName(domainName)
+
+		if err != nil {
+			var rnfe *resource.NotFoundByNameError
+			if errors.As(err, &rnfe) {
+				fmt.Printf("SageMaker Domain with name %q does not exist\n", domainName)
+				return nil
+			}
+			return err
+		}
+
+		domainID = awssdk.ToString(domain.DomainId)
+	}
+
+	err := m.sagemakerClient.DeleteDomain(domainID)
+	if err != nil {
+		return aws.FormatErrorAsMessageOnly(err)
+	}
+	fmt.Printf("SageMaker Domain Id %q deleted\n", domainID)
+
+	return nil
+}

pkg/resource/sagemaker/domain/options.go

@@ -9,15 +9,34 @@ import (
 type Options struct {
 	resource.CommonOptions
 
-	// Get
+	// Get, Delete
 	DomainID string
 }
 
-func newOptions() (options *Options, getFlags cmd.Flags) {
+func newOptions() (options *Options, deleteFlags, getFlags cmd.Flags) {
 	options = &Options{
 		CommonOptions: resource.CommonOptions{
-			Name:                "sagemaker-domain",
-			ClusterFlagDisabled: true,
+			ClusterFlagDisabled:    true,
+			DeleteArgumentOptional: true,
+		},
+	}
+
+	deleteFlags = cmd.Flags{
+		&cmd.StringFlag{
+			CommandFlag: cmd.CommandFlag{
+				Name:        "id",
+				Description: "delete by id instead of name",
+				Validate: func(_ *cobra.Command, args []string) error {
+					if len(args) == 0 && options.DomainID == "" {
+						return &cmd.MustIncludeEitherArgumentOrFlag{Arg: "DOMAIN_NAME", Flag: "--id"}
+					}
+					if options.DomainID != "" && len(args) > 0 {
+						return &cmd.ArgumentAndFlagCantBeUsedTogetherError{Arg: "DOMAIN_NAME", Flag: "--id"}
+					}
+					return nil
+				},
+			},
+			Option: &options.DomainID,
 		},
 	}
 

pkg/resource/sagemaker/domain/sagemaker_domain.go

@@ -6,19 +6,23 @@ import (
 )
 
 func New() *resource.Resource {
-	options, getFlags := newOptions()
+	options, deleteFlags, getFlags := newOptions()
 
 	return &resource.Resource{
 		Command: cmd.Command{
 			Name:        "domain",
 			Description: "SageMaker Domain",
+			Aliases:     []string{"do"},
 			Args:        []string{"DOMAIN_NAME"},
 		},
 
-		GetFlags: getFlags,
+		DeleteFlags: deleteFlags,
+		GetFlags:    getFlags,
 
 		Getter: &Getter{},
 
+		Manager: &Manager{},
+
 		Options: options,
 	}
 }