feat: adding key-vault resource

resources/key-vault.go

@@ -0,0 +1,82 @@
+package resources
+
+import (
+	"context"
+	"time"
+
+	"github.com/sirupsen/logrus"
+
+	"github.com/Azure/azure-sdk-for-go/services/keyvault/mgmt/2019-09-01/keyvault"
+
+	"github.com/ekristen/azure-nuke/pkg/resource"
+	"github.com/ekristen/azure-nuke/pkg/types"
+)
+
+type KeyVault struct {
+	client keyvault.VaultsClient
+	name   string
+	rg     string
+}
+
+func init() {
+	resource.RegisterV2(resource.Registration{
+		Name:   "KeyVault",
+		Scope:  resource.ResourceGroup,
+		Lister: ListKeyVault,
+	})
+}
+
+func ListKeyVault(opts resource.ListerOpts) ([]resource.Resource, error) {
+	logrus.Tracef("subscription id: %s", opts.SubscriptionId)
+
+	client := keyvault.NewVaultsClient(opts.SubscriptionId)
+	client.Authorizer = opts.Authorizers.Management
+	client.RetryAttempts = 1
+	client.RetryDuration = time.Second * 2
+
+	resources := make([]resource.Resource, 0)
+
+	logrus.Trace("attempting to list ssh key")
+
+	ctx := context.Background()
+	list, err := client.ListByResourceGroup(ctx, opts.ResourceGroup, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	logrus.Trace("listing ....")
+
+	for list.NotDone() {
+		logrus.Trace("list not done")
+		for _, g := range list.Values() {
+			resources = append(resources, &KeyVault{
+				client: client,
+				name:   *g.Name,
+				rg:     opts.ResourceGroup,
+			})
+		}
+
+		if err := list.NextWithContext(ctx); err != nil {
+			return nil, err
+		}
+	}
+
+	return resources, nil
+}
+
+func (r *KeyVault) Remove() error {
+	_, err := r.client.Delete(context.TODO(), r.rg, r.name)
+	return err
+}
+
+func (r *KeyVault) Properties() types.Properties {
+	properties := types.NewProperties()
+
+	properties.Set("Name", r.name)
+
+	return properties
+}
+
+func (r *KeyVault) String() string {
+	return r.name
+}