feat(iottwinmaker): add workspace, component type, entity, scene, syn…

…cjob

resources/iottwinmaker-component-type.go

@@ -0,0 +1,145 @@
+package resources
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/iottwinmaker"
+
+	"github.com/ekristen/libnuke/pkg/registry"
+	"github.com/ekristen/libnuke/pkg/resource"
+	"github.com/ekristen/libnuke/pkg/types"
+
+	"github.com/ekristen/aws-nuke/v3/pkg/nuke"
+)
+
+const IoTTwinMakerComponentTypeResource = "IoTTwinMakerComponentType"
+
+func init() {
+	registry.Register(&registry.Registration{
+		Name:   IoTTwinMakerComponentTypeResource,
+		Scope:  nuke.Account,
+		Lister: &IoTTwinMakerComponentTypeLister{},
+	})
+}
+
+type IoTTwinMakerComponentTypeLister struct{}
+
+func (l *IoTTwinMakerComponentTypeLister) List(_ context.Context, o interface{}) ([]resource.Resource, error) {
+	opts := o.(*nuke.ListerOpts)
+
+	svc := iottwinmaker.New(opts.Session)
+	resources := make([]resource.Resource, 0)
+
+	// Require to have workspaces identifiers to query components
+	workspaceListResponse, err := ListWorkspacesComponentType(svc)
+
+	if err != nil {
+		return nil, err
+	}
+
+	for _, workspaceResponse := range workspaceListResponse {
+		params := &iottwinmaker.ListComponentTypesInput{
+			WorkspaceId: workspaceResponse.WorkspaceId,
+			MaxResults:  aws.Int64(25),
+		}
+
+		for {
+			resp, err := svc.ListComponentTypes(params)
+			if err != nil {
+				return nil, err
+			}
+
+			for _, item := range resp.ComponentTypeSummaries {
+				// We must filter out amazon-owned component types when querying tags,
+				// because their ARN format causes ListTagsForResource to vail with validation error
+				tags := make(map[string]*string)
+				if !strings.Contains(*item.Arn, "AmazonOwnedTypesWorkspace") {
+					tagResp, err := svc.ListTagsForResource(
+						&iottwinmaker.ListTagsForResourceInput{
+							ResourceARN: item.Arn,
+						})
+					if err != nil {
+						return nil, err
+					}
+					tags = tagResp.Tags
+				}
+
+				resources = append(resources, &IoTTwinMakerComponentType{
+					svc:         svc,
+					ID:          item.ComponentTypeId,
+					arn:         item.Arn,
+					tags:        tags,
+					workspaceID: workspaceResponse.WorkspaceId,
+				})
+			}
+
+			if resp.NextToken == nil {
+				break
+			}
+
+			params.NextToken = resp.NextToken
+		}
+	}
+
+	return resources, nil
+}
+
+// Utility function to list workspaces
+func ListWorkspacesComponentType(svc *iottwinmaker.IoTTwinMaker) ([]*iottwinmaker.WorkspaceSummary, error) {
+	resources := make([]*iottwinmaker.WorkspaceSummary, 0)
+	params := &iottwinmaker.ListWorkspacesInput{
+		MaxResults: aws.Int64(25),
+	}
+	for {
+		resp, err := svc.ListWorkspaces(params)
+		if err != nil {
+			return nil, err
+		}
+		resources = append(resources, resp.WorkspaceSummaries...)
+		if resp.NextToken == nil {
+			break
+		}
+		params.NextToken = resp.NextToken
+	}
+	return resources, nil
+}
+
+type IoTTwinMakerComponentType struct {
+	svc         *iottwinmaker.IoTTwinMaker
+	ID          *string
+	workspaceID *string
+	arn         *string
+	tags        map[string]*string
+}
+
+func (f *IoTTwinMakerComponentType) Filter() error {
+	if strings.Contains(*f.arn, "AmazonOwnedTypesWorkspace") {
+		return fmt.Errorf("cannot delete pre-defined component type")
+	}
+	return nil
+}
+
+func (f *IoTTwinMakerComponentType) Properties() types.Properties {
+	properties := types.NewProperties()
+	properties.Set("workspaceID", f.workspaceID)
+	for k, v := range f.tags {
+		properties.SetTag(&k, v)
+	}
+	return properties
+}
+
+func (f *IoTTwinMakerComponentType) Remove(_ context.Context) error {
+	_, err := f.svc.DeleteComponentType(&iottwinmaker.DeleteComponentTypeInput{
+		ComponentTypeId: f.ID,
+		WorkspaceId:     f.workspaceID,
+	})
+
+	return err
+}
+
+func (f *IoTTwinMakerComponentType) String() string {
+	return *f.ID
+}

resources/iottwinmaker-entity.go

@@ -0,0 +1,124 @@
+package resources
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/iottwinmaker"
+
+	"github.com/ekristen/libnuke/pkg/registry"
+	"github.com/ekristen/libnuke/pkg/resource"
+	"github.com/ekristen/libnuke/pkg/types"
+
+	"github.com/ekristen/aws-nuke/v3/pkg/nuke"
+)
+
+const IoTTwinMakerEntityResource = "IoTTwinMakerEntity"
+
+func init() {
+	registry.Register(&registry.Registration{
+		Name:   IoTTwinMakerEntityResource,
+		Scope:  nuke.Account,
+		Lister: &IoTTwinMakerEntityLister{},
+	})
+}
+
+type IoTTwinMakerEntityLister struct{}
+
+func (l *IoTTwinMakerEntityLister) List(_ context.Context, o interface{}) ([]resource.Resource, error) {
+	opts := o.(*nuke.ListerOpts)
+
+	svc := iottwinmaker.New(opts.Session)
+	resources := make([]resource.Resource, 0)
+
+	// Require to have workspaces identifiers to query entities
+	workspaceListResponse, err := ListWorkspacesEntities(svc)
+
+	if err != nil {
+		return nil, err
+	}
+
+	for _, workspaceResponse := range workspaceListResponse {
+		params := &iottwinmaker.ListEntitiesInput{
+			WorkspaceId: workspaceResponse.WorkspaceId,
+			MaxResults:  aws.Int64(25),
+		}
+
+		for {
+			resp, err := svc.ListEntities(params)
+			if err != nil {
+				return nil, err
+			}
+
+			for _, item := range resp.EntitySummaries {
+				// We must filter out amazon-owned component types when querying tags,
+				// because their ARN format causes ListTagsForResource to vail with validation error
+				resources = append(resources, &IoTTwinMakerEntity{
+					svc:         svc,
+					ID:          item.EntityId,
+					name:        item.EntityName,
+					status:      item.Status.State,
+					workspaceID: workspaceResponse.WorkspaceId,
+				})
+			}
+
+			if resp.NextToken == nil {
+				break
+			}
+
+			params.NextToken = resp.NextToken
+		}
+	}
+
+	return resources, nil
+}
+
+// Utility function to list workspaces
+func ListWorkspacesEntities(svc *iottwinmaker.IoTTwinMaker) ([]*iottwinmaker.WorkspaceSummary, error) {
+	resources := make([]*iottwinmaker.WorkspaceSummary, 0)
+	params := &iottwinmaker.ListWorkspacesInput{
+		MaxResults: aws.Int64(25),
+	}
+	for {
+		resp, err := svc.ListWorkspaces(params)
+		if err != nil {
+			return nil, err
+		}
+		resources = append(resources, resp.WorkspaceSummaries...)
+		if resp.NextToken == nil {
+			break
+		}
+		params.NextToken = resp.NextToken
+	}
+	return resources, nil
+}
+
+type IoTTwinMakerEntity struct {
+	svc         *iottwinmaker.IoTTwinMaker
+	ID          *string
+	name        *string
+	status      *string
+	workspaceID *string
+}
+
+func (f *IoTTwinMakerEntity) Properties() types.Properties {
+	properties := types.NewProperties()
+	properties.Set("name", f.name)
+	properties.Set("status", f.status)
+	properties.Set("workspaceID", f.workspaceID)
+	return properties
+}
+
+func (f *IoTTwinMakerEntity) Remove(_ context.Context) error {
+	_, err := f.svc.DeleteEntity(&iottwinmaker.DeleteEntityInput{
+		EntityId:    f.ID,
+		WorkspaceId: f.workspaceID,
+		IsRecursive: aws.Bool(true),
+	})
+
+	return err
+}
+
+func (f *IoTTwinMakerEntity) String() string {
+	return *f.ID
+}

resources/iottwinmaker-scene.go

@@ -0,0 +1,115 @@
+package resources
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/iottwinmaker"
+
+	"github.com/ekristen/libnuke/pkg/registry"
+	"github.com/ekristen/libnuke/pkg/resource"
+	"github.com/ekristen/libnuke/pkg/types"
+
+	"github.com/ekristen/aws-nuke/v3/pkg/nuke"
+)
+
+const IoTTwinMakerSceneResource = "IoTTwinMakerScene"
+
+func init() {
+	registry.Register(&registry.Registration{
+		Name:   IoTTwinMakerSceneResource,
+		Scope:  nuke.Account,
+		Lister: &IoTTwinMakerSceneLister{},
+	})
+}
+
+type IoTTwinMakerSceneLister struct{}
+
+func (l *IoTTwinMakerSceneLister) List(_ context.Context, o interface{}) ([]resource.Resource, error) {
+	opts := o.(*nuke.ListerOpts)
+
+	svc := iottwinmaker.New(opts.Session)
+	resources := make([]resource.Resource, 0)
+
+	// Require to have workspaces identifiers to query scenes
+	workspaceListResponse, err := ListWorkspacesScene(svc)
+
+	if err != nil {
+		return nil, err
+	}
+
+	for _, workspaceResponse := range workspaceListResponse {
+		params := &iottwinmaker.ListScenesInput{
+			WorkspaceId: workspaceResponse.WorkspaceId,
+			MaxResults:  aws.Int64(25),
+		}
+
+		for {
+			resp, err := svc.ListScenes(params)
+			if err != nil {
+				return nil, err
+			}
+
+			for _, item := range resp.SceneSummaries {
+				resources = append(resources, &IoTTwinMakerScene{
+					svc:         svc,
+					ID:          item.SceneId,
+					workspaceID: workspaceResponse.WorkspaceId,
+				})
+			}
+
+			if resp.NextToken == nil {
+				break
+			}
+
+			params.NextToken = resp.NextToken
+		}
+	}
+
+	return resources, nil
+}
+
+// Utility function to list workspaces
+func ListWorkspacesScene(svc *iottwinmaker.IoTTwinMaker) ([]*iottwinmaker.WorkspaceSummary, error) {
+	resources := make([]*iottwinmaker.WorkspaceSummary, 0)
+	params := &iottwinmaker.ListWorkspacesInput{
+		MaxResults: aws.Int64(25),
+	}
+	for {
+		resp, err := svc.ListWorkspaces(params)
+		if err != nil {
+			return nil, err
+		}
+		resources = append(resources, resp.WorkspaceSummaries...)
+		if resp.NextToken == nil {
+			break
+		}
+		params.NextToken = resp.NextToken
+	}
+	return resources, nil
+}
+
+type IoTTwinMakerScene struct {
+	svc         *iottwinmaker.IoTTwinMaker
+	ID          *string
+	workspaceID *string
+}
+
+func (f *IoTTwinMakerScene) Properties() types.Properties {
+	properties := types.NewProperties()
+	properties.Set("workspaceID", f.workspaceID)
+	return properties
+}
+
+func (f *IoTTwinMakerScene) Remove(_ context.Context) error {
+	_, err := f.svc.DeleteScene(&iottwinmaker.DeleteSceneInput{
+		SceneId:     f.ID,
+		WorkspaceId: f.workspaceID,
+	})
+
+	return err
+}
+
+func (f *IoTTwinMakerScene) String() string {
+	return *f.ID
+}