dataupload_retrieve_action.go

/*
Copyright 2020 the Velero contributors.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package actions

import (
	"context"
	"encoding/json"

	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"
	corev1api "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/types"
	"sigs.k8s.io/controller-runtime/pkg/client"

	velerov1api "github.com/vmware-tanzu/velero/pkg/apis/velero/v1"
	velerov2alpha1 "github.com/vmware-tanzu/velero/pkg/apis/velero/v2alpha1"
	veleroclient "github.com/vmware-tanzu/velero/pkg/client"
	"github.com/vmware-tanzu/velero/pkg/label"
	"github.com/vmware-tanzu/velero/pkg/plugin/velero"
)

type DataUploadRetrieveAction struct {
	logger logrus.FieldLogger
	client client.Client
}

func NewDataUploadRetrieveAction(logger logrus.FieldLogger, client client.Client) *DataUploadRetrieveAction {
	return &DataUploadRetrieveAction{
		logger: logger,
		client: client,
	}
}

func (d *DataUploadRetrieveAction) AppliesTo() (velero.ResourceSelector, error) {
	return velero.ResourceSelector{
		IncludedResources: []string{"datauploads.velero.io"},
	}, nil
}

func (d *DataUploadRetrieveAction) Execute(input *velero.RestoreItemActionExecuteInput) (*velero.RestoreItemActionExecuteOutput, error) {
	d.logger.Info("Executing DataUploadRetrieveAction")

	dataUpload := velerov2alpha1.DataUpload{}
	if err := runtime.DefaultUnstructuredConverter.FromUnstructured(input.ItemFromBackup.UnstructuredContent(), &dataUpload); err != nil {
		d.logger.Errorf("unable to convert unstructured item to DataUpload: %s", err.Error())
		return nil, errors.Wrap(err, "unable to convert unstructured item to DataUpload.")
	}

	backup := &velerov1api.Backup{}
	err := d.client.Get(context.Background(), types.NamespacedName{
		Namespace: input.Restore.Namespace,
		Name:      input.Restore.Spec.BackupName,
	}, backup)
	if err != nil {
		d.logger.WithError(err).Errorf("Fail to get backup for restore %s.", input.Restore.Name)
		return nil, errors.Wrapf(err, "error to get backup for restore %s", input.Restore.Name)
	}

	dataUploadResult := velerov2alpha1.DataUploadResult{
		BackupStorageLocation: backup.Spec.StorageLocation,
		DataMover:             dataUpload.Spec.DataMover,
		SnapshotID:            dataUpload.Status.SnapshotID,
		SourceNamespace:       dataUpload.Spec.SourceNamespace,
		DataMoverResult:       dataUpload.Status.DataMoverResult,
	}

	jsonBytes, err := json.Marshal(dataUploadResult)
	if err != nil {
		d.logger.Errorf("fail to convert DataUploadResult to JSON: %s", err.Error())
		return nil, errors.Wrap(err, "fail to convert DataUploadResult to JSON")
	}

	cm := corev1api.ConfigMap{
		TypeMeta: metav1.TypeMeta{
			Kind:       "ConfigMap",
			APIVersion: corev1api.SchemeGroupVersion.String(),
		},
		ObjectMeta: metav1.ObjectMeta{
			GenerateName: dataUpload.Name + "-",
			Namespace:    dataUpload.Namespace,
			Labels: map[string]string{
				velerov1api.RestoreUIDLabel:       label.GetValidName(string(input.Restore.UID)),
				velerov1api.PVCNamespaceNameLabel: label.GetValidName(dataUpload.Spec.SourceNamespace + "." + dataUpload.Spec.SourcePVC),
				velerov1api.ResourceUsageLabel:    label.GetValidName(string(velerov1api.VeleroResourceUsageDataUploadResult)),
			},
		},
		Data: map[string]string{
			string(input.Restore.UID): string(jsonBytes),
		},
	}

	err = veleroclient.CreateRetryGenerateName(d.client, context.Background(), &cm)
	if err != nil {
		d.logger.Errorf("fail to create DataUploadResult ConfigMap %s/%s: %s", cm.Namespace, cm.Name, err.Error())
		return nil, errors.Wrap(err, "fail to create DataUploadResult ConfigMap")
	}

	return &velero.RestoreItemActionExecuteOutput{
		SkipRestore: true,
	}, nil
}