feat: add support for sweepers (scaleway#2374)

Author: remyleone

api/account/v3/sweepers/sweepers.go

@@ -0,0 +1,39 @@
+package sweepers
+
+import (
+	"fmt"
+
+	"github.com/scaleway/scaleway-sdk-go/api/account/v3"
+	"github.com/scaleway/scaleway-sdk-go/internal/testhelpers"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+func SweepProjects(scwClient *scw.Client) error {
+	accountAPI := account.NewProjectAPI(scwClient)
+
+	req := &account.ProjectAPIListProjectsRequest{}
+	listProjects, err := accountAPI.ListProjects(req, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("failed to list projects: %w", err)
+	}
+	for _, project := range listProjects.Projects {
+		// Do not delete default project
+		if project.ID == req.OrganizationID || !testhelpers.IsTestResource(project.Name) {
+			continue
+		}
+		err = accountAPI.DeleteProject(&account.ProjectAPIDeleteProjectRequest{
+			ProjectID: project.ID,
+		})
+		if err != nil {
+			return fmt.Errorf("failed to delete project: %w", err)
+		}
+	}
+	return nil
+}
+
+func SweepAll(scwClient *scw.Client) error {
+	if err := SweepProjects(scwClient); err != nil {
+		return err
+	}
+	return nil
+}

api/applesilicon/v1alpha1/sweepers/sweepers.go

@@ -0,0 +1,40 @@
+package sweepers
+
+import (
+	"fmt"
+
+	applesilicon "github.com/scaleway/scaleway-sdk-go/api/applesilicon/v1alpha1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+func SweepServer(scwClient *scw.Client, zone scw.Zone) error {
+	asAPI := applesilicon.NewAPI(scwClient)
+
+	listServers, err := asAPI.ListServers(&applesilicon.ListServersRequest{Zone: zone}, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("error listing apple silicon servers in (%s) in sweeper: %s", zone, err)
+	}
+
+	for _, server := range listServers.Servers {
+		errDelete := asAPI.DeleteServer(&applesilicon.DeleteServerRequest{
+			ServerID: server.ID,
+			Zone:     zone,
+		})
+		if errDelete != nil {
+			return fmt.Errorf("error deleting apple silicon server in sweeper: %s", err)
+		}
+	}
+
+	return nil
+}
+
+func SweepAllLocalities(scwClient *scw.Client) error {
+	for _, zone := range (&applesilicon.API{}).Zones() {
+		err := SweepServer(scwClient, zone)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

api/baremetal/v1/sweepers/sweepers.go

@@ -0,0 +1,40 @@
+package sweepers
+
+import (
+	"fmt"
+
+	"github.com/scaleway/scaleway-sdk-go/api/baremetal/v1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+func SweepServers(scwClient *scw.Client, zone scw.Zone) error {
+	baremetalAPI := baremetal.NewAPI(scwClient)
+
+	listServers, err := baremetalAPI.ListServers(&baremetal.ListServersRequest{Zone: zone}, scw.WithAllPages())
+	if err != nil {
+		return err
+	}
+
+	for _, server := range listServers.Servers {
+		_, err := baremetalAPI.DeleteServer(&baremetal.DeleteServerRequest{
+			Zone:     zone,
+			ServerID: server.ID,
+		})
+		if err != nil {
+			return fmt.Errorf("error deleting server in sweeper: %s", err)
+		}
+	}
+
+	return nil
+}
+
+func SweepAllLocalities(scwClient *scw.Client) error {
+	for _, zone := range (&baremetal.API{}).Zones() {
+		err := SweepServers(scwClient, zone)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

api/block/v1alpha1/sweepers/sweepers.go

@@ -0,0 +1,71 @@
+package sweepers
+
+import (
+	"fmt"
+
+	block "github.com/scaleway/scaleway-sdk-go/api/block/v1alpha1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+func SweepVolumes(scwClient *scw.Client, zone scw.Zone) error {
+	blockAPI := block.NewAPI(scwClient)
+
+	listVolumes, err := blockAPI.ListVolumes(
+		&block.ListVolumesRequest{
+			Zone: zone,
+		}, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("error listing volume in (%s) in sweeper: %s", zone, err)
+	}
+
+	for _, volume := range listVolumes.Volumes {
+		err := blockAPI.DeleteVolume(&block.DeleteVolumeRequest{
+			VolumeID: volume.ID,
+			Zone:     zone,
+		})
+		if err != nil {
+			return fmt.Errorf("error deleting volume in sweeper: %s", err)
+		}
+	}
+
+	return nil
+}
+
+func SweepSnapshots(scwClient *scw.Client, zone scw.Zone) error {
+	blockAPI := block.NewAPI(scwClient)
+
+	listSnapshots, err := blockAPI.ListSnapshots(
+		&block.ListSnapshotsRequest{
+			Zone: zone,
+		}, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("error listing snapshot in (%s) in sweeper: %s", zone, err)
+	}
+
+	for _, snapshot := range listSnapshots.Snapshots {
+		err := blockAPI.DeleteSnapshot(&block.DeleteSnapshotRequest{
+			SnapshotID: snapshot.ID,
+			Zone:       zone,
+		})
+		if err != nil {
+			return fmt.Errorf("error deleting snapshot in sweeper: %s", err)
+		}
+	}
+
+	return nil
+}
+
+func SweepAllLocalities(scwClient *scw.Client) error {
+	for _, zone := range (&block.API{}).Zones() {
+		err := SweepVolumes(scwClient, zone)
+		if err != nil {
+			return err
+		}
+		err = SweepSnapshots(scwClient, zone)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

api/cockpit/v1/sweepers/sweepers.go

@@ -0,0 +1,131 @@
+package sweepers
+
+import (
+	"fmt"
+	"strings"
+
+	accountSDK "github.com/scaleway/scaleway-sdk-go/api/account/v3"
+	"github.com/scaleway/scaleway-sdk-go/api/cockpit/v1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
+)
+
+func SweepToken(scwClient *scw.Client) error {
+	accountAPI := accountSDK.NewProjectAPI(scwClient)
+	cockpitAPI := cockpit.NewRegionalAPI(scwClient)
+
+	listProjects, err := accountAPI.ListProjects(&accountSDK.ProjectAPIListProjectsRequest{}, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("failed to list projects: %w", err)
+	}
+
+	for _, project := range listProjects.Projects {
+		if !strings.HasPrefix(project.Name, "tf_tests") {
+			continue
+		}
+
+		listTokens, err := cockpitAPI.ListTokens(&cockpit.RegionalAPIListTokensRequest{
+			ProjectID: project.ID,
+		}, scw.WithAllPages())
+		if err != nil {
+			return fmt.Errorf("failed to list tokens: %w", err)
+		}
+
+		for _, token := range listTokens.Tokens {
+			err = cockpitAPI.DeleteToken(&cockpit.RegionalAPIDeleteTokenRequest{
+				TokenID: token.ID,
+			})
+			if err != nil {
+				return fmt.Errorf("failed to delete token: %w", err)
+			}
+		}
+	}
+
+	return nil
+}
+
+func SweepGrafanaUser(scwClient *scw.Client) error {
+	accountAPI := accountSDK.NewProjectAPI(scwClient)
+	cockpitAPI := cockpit.NewGlobalAPI(scwClient)
+
+	listProjects, err := accountAPI.ListProjects(&accountSDK.ProjectAPIListProjectsRequest{}, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("failed to list projects: %w", err)
+	}
+
+	for _, project := range listProjects.Projects {
+		if !strings.HasPrefix(project.Name, "tf_tests") {
+			continue
+		}
+
+		listGrafanaUsers, err := cockpitAPI.ListGrafanaUsers(&cockpit.GlobalAPIListGrafanaUsersRequest{
+			ProjectID: project.ID,
+		}, scw.WithAllPages())
+		if err != nil {
+			return fmt.Errorf("failed to list grafana users: %w", err)
+		}
+
+		for _, grafanaUser := range listGrafanaUsers.GrafanaUsers {
+			err = cockpitAPI.DeleteGrafanaUser(&cockpit.GlobalAPIDeleteGrafanaUserRequest{
+				ProjectID:     project.ID,
+				GrafanaUserID: grafanaUser.ID,
+			})
+			if err != nil {
+				return fmt.Errorf("failed to delete grafana user: %w", err)
+			}
+		}
+	}
+
+	return nil
+}
+
+func SweepSource(scwClient *scw.Client, region scw.Region) error {
+	accountAPI := accountSDK.NewProjectAPI(scwClient)
+	cockpitAPI := cockpit.NewRegionalAPI(scwClient)
+
+	listProjects, err := accountAPI.ListProjects(&accountSDK.ProjectAPIListProjectsRequest{}, scw.WithAllPages())
+	if err != nil {
+		return fmt.Errorf("failed to list projects: %w", err)
+	}
+
+	for _, project := range listProjects.Projects {
+		if !strings.HasPrefix(project.Name, "tf_tests") {
+			continue
+		}
+
+		listDatasources, err := cockpitAPI.ListDataSources(&cockpit.RegionalAPIListDataSourcesRequest{
+			ProjectID: project.ID,
+			Region:    region,
+		}, scw.WithAllPages())
+		if err != nil {
+			return fmt.Errorf("failed to list sources: %w", err)
+		}
+
+		for _, datsource := range listDatasources.DataSources {
+			err = cockpitAPI.DeleteDataSource(&cockpit.RegionalAPIDeleteDataSourceRequest{
+				DataSourceID: datsource.ID,
+				Region:       region,
+			})
+			if err != nil {
+				return fmt.Errorf("failed to delete cockpit source: %w", err)
+			}
+		}
+	}
+
+	return nil
+}
+
+func SweepAllLocalities(scwClient *scw.Client) error {
+	if err := SweepToken(scwClient); err != nil {
+		return err
+	}
+	if err := SweepGrafanaUser(scwClient); err != nil {
+		return err
+	}
+	for _, region := range (&cockpit.RegionalAPI{}).Regions() {
+		if err := SweepSource(scwClient, region); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}