opensearchservice-packages.go

package resources

import (
	"fmt"
	"strings"
	"time"

	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/opensearchservice"
	"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type OSPackage struct {
	svc         *opensearchservice.OpenSearchService
	packageID   *string
	packageName *string
	createdTime *time.Time
}

func init() {
	register("OSPackage", ListOSPackages)
}

func ListOSPackages(sess *session.Session) ([]Resource, error) {
	svc := opensearchservice.New(sess)
	resources := []Resource{}
	var nextToken *string

	for {
		params := &opensearchservice.DescribePackagesInput{
			NextToken: nextToken,
		}
		listResp, err := svc.DescribePackages(params)
		if err != nil {
			return nil, err
		}

		for _, pkg := range listResp.PackageDetailsList {
			resources = append(resources, &OSPackage{
				svc:         svc,
				packageID:   pkg.PackageID,
				packageName: pkg.PackageName,
				createdTime: pkg.CreatedAt,
			})
		}

		// Check if there are more results
		if listResp.NextToken == nil {
			break // No more results, exit the loop
		}

		// Set the nextToken for the next iteration
		nextToken = listResp.NextToken
	}

	return resources, nil
}

func (o *OSPackage) Filter() error {
	if strings.HasPrefix(*o.packageID, "G") {
		return fmt.Errorf("cannot delete default opensearch packages")
	}
	return nil
}

func (o *OSPackage) Remove() error {
	_, err := o.svc.DeletePackage(&opensearchservice.DeletePackageInput{
		PackageID: o.packageID,
	})

	return err
}

func (o *OSPackage) Properties() types.Properties {
	properties := types.NewProperties()
	properties.Set("PackageID", o.packageID)
	properties.Set("PackageName", o.packageName)
	properties.Set("CreatedTime", o.createdTime.Format(time.RFC3339))
	return properties
}

func (o *OSPackage) String() string {
	return *o.packageID
}