restic/vendor/github.com/kurin/blazer/internal/bin/cleanup/cleanup.go

package main

import (
	"context"
	"fmt"
	"os"
	"strings"
	"sync"

	"github.com/kurin/blazer/b2"
)

const (
	apiID  = "B2_ACCOUNT_ID"
	apiKey = "B2_SECRET_KEY"
)

func main() {
	id := os.Getenv(apiID)
	key := os.Getenv(apiKey)
	ctx := context.Background()
	client, err := b2.NewClient(ctx, id, key)
	if err != nil {
		fmt.Println(err)
		return
	}
	buckets, err := client.ListBuckets(ctx)
	if err != nil {
		fmt.Println(err)
		return
	}
	var kill []string
	for _, bucket := range buckets {
		if strings.HasPrefix(bucket.Name(), fmt.Sprintf("%s-b2-tests-", id)) {
			kill = append(kill, bucket.Name())
		}
		if bucket.Name() == fmt.Sprintf("%s-consistobucket", id) || bucket.Name() == fmt.Sprintf("%s-base-tests", id) {
			kill = append(kill, bucket.Name())
		}
	}
	var wg sync.WaitGroup
	for _, name := range kill {
		wg.Add(1)
		go func(name string) {
			defer wg.Done()
			fmt.Println("removing", name)
			if err := killBucket(ctx, client, name); err != nil {
				fmt.Println(err)
			}
		}(name)
	}
	wg.Wait()
}

func killBucket(ctx context.Context, client *b2.Client, name string) error {
	bucket, err := client.NewBucket(ctx, name, nil)
	if b2.IsNotExist(err) {
		return nil
	}
	if err != nil {
		return err
	}
	defer bucket.Delete(ctx)
	iter := bucket.List(ctx, b2.ListHidden())
	for iter.Next() {
		if err := iter.Object().Delete(ctx); err != nil {
			fmt.Println(err)
		}
	}
	return iter.Err()
}