CLOUDP-141796: Remake cleaner (#743)

Author: Sugar-pack

.github/actions/cleanup/Dockerfile

@@ -0,0 +1,26 @@
+package atlasclient
+
+import (
+	"fmt"
+	"net/http"
+	"runtime"
+
+	"go.mongodb.org/atlas/mongodbatlas"
+
+	"github.com/mongodb/mongodb-atlas-kubernetes/pkg/util/httputil"
+)
+
+func SetupClient(publicKey, privateKey, managerUrl string) (mongodbatlas.Client, error) {
+	withDigest := httputil.Digest(publicKey, privateKey)
+
+	httpClient, err := httputil.DecorateClient(&http.Client{Transport: http.DefaultTransport}, withDigest)
+	if err != nil {
+		return mongodbatlas.Client{}, err
+	}
+	client, err := mongodbatlas.New(httpClient, mongodbatlas.SetBaseURL(managerUrl))
+	if err != nil {
+		return mongodbatlas.Client{}, err
+	}
+	client.UserAgent = fmt.Sprintf("%s/%s (%s;%s)", "MongoDBAtlasKubernetesOperator", "unknown", runtime.GOOS, runtime.GOARCH)
+	return *client, nil
+}

.github/actions/cleanup/action.yml

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+)
+
+type Config struct {
+	DeleteAll  bool
+	Lifetime   int // lifetime in hours
+	PublicKey  string
+	PrivateKey string
+	ManagerUrl string
+}
+
+const DefaultManagerURL = "https://cloud-qa.mongodb.com/"
+
+func NewConfig() (*Config, error) {
+	deleteAllStr := os.Getenv("CLEAN_ALL")
+	deleteAll := deleteAllStr != "false"
+
+	var lifetime int
+	if !deleteAll {
+		lifetimeStr := os.Getenv("MAX_PROJECT_LIFETIME")
+		if lifetimeStr == "" {
+			return nil, fmt.Errorf("MAX_PROJECT_LIFETIME is not set")
+		}
+		var err error
+		lifetime, err = strconv.Atoi(lifetimeStr)
+		if err != nil {
+			return nil, fmt.Errorf("MAX_PROJECT_LIFETIME must be an integer")
+		}
+	}
+
+	publicKey := os.Getenv("MCLI_PUBLIC_API_KEY")
+	if publicKey == "" {
+		return nil, fmt.Errorf("MCLI_PUBLIC_API_KEY must be set")
+	}
+	privateKey := os.Getenv("MCLI_PRIVATE_API_KEY")
+	if privateKey == "" {
+		return nil, fmt.Errorf("MCLI_PRIVATE_API_KEY must be set")
+	}
+	managerUrl := os.Getenv("MCLI_OPS_MANAGER_URL")
+	if managerUrl == "" {
+		managerUrl = DefaultManagerURL
+	}
+
+	return &Config{
+		DeleteAll:  deleteAll,
+		Lifetime:   lifetime,
+		PublicKey:  publicKey,
+		PrivateKey: privateKey,
+		ManagerUrl: managerUrl,
+	}, nil
+}

.github/actions/cleanup/atlasclient/client.go

@@ -0,0 +1,29 @@
+package deployment
+
+import (
+	"context"
+	"fmt"
+
+	"go.mongodb.org/atlas/mongodbatlas"
+)
+
+func DeleteAllDeployments(ctx context.Context, client mongodbatlas.ClustersService, projectID string) error {
+	deployments, err := GetAllDeployments(ctx, client, projectID)
+	if err != nil {
+		return fmt.Errorf("error getting deployments: %s", err)
+	}
+	for _, deployment := range deployments {
+		if _, err = client.Delete(ctx, projectID, deployment.Name); err != nil {
+			return fmt.Errorf("error deleting deployment: %s", err)
+		}
+	}
+	return nil
+}
+
+func GetAllDeployments(ctx context.Context, client mongodbatlas.ClustersService, projectID string) ([]mongodbatlas.Cluster, error) {
+	deployments, _, err := client.List(ctx, projectID, nil)
+	if err != nil {
+		return nil, fmt.Errorf("error getting deployments: %s", err)
+	}
+	return deployments, nil
+}

.github/actions/cleanup/config.go

@@ -0,0 +1,29 @@
+package deployment
+
+import (
+	"context"
+	"fmt"
+
+	"go.mongodb.org/atlas/mongodbatlas"
+)
+
+func DeleteAllServerless(ctx context.Context, client mongodbatlas.ServerlessInstancesService, projectID string) error {
+	serverless, err := GetAllServerless(ctx, client, projectID)
+	if err != nil {
+		return fmt.Errorf("error getting serverless: %s", err)
+	}
+	for _, s := range serverless {
+		if _, err = client.Delete(ctx, projectID, s.Name); err != nil {
+			return fmt.Errorf("error deleting serverless: %s", err)
+		}
+	}
+	return nil
+}
+
+func GetAllServerless(ctx context.Context, client mongodbatlas.ServerlessInstancesService, projectID string) ([]*mongodbatlas.Cluster, error) {
+	serverless, _, err := client.List(ctx, projectID, nil)
+	if err != nil {
+		return nil, fmt.Errorf("error getting serverless: %s", err)
+	}
+	return serverless.Results, nil
+}