feat(controller): ImpVMSnapshot reconciler skeleton

Author: syscod3

cmd/operator/main.go

@@ -162,6 +162,14 @@ func main() {
 		os.Exit(1)
 	}
 
+	if err = (&controller.ImpVMSnapshotReconciler{
+		Client: mgr.GetClient(),
+		Scheme: mgr.GetScheme(),
+	}).SetupWithManager(mgr); err != nil {
+		setupLog.Error(err, "unable to create controller", "controller", "ImpVMSnapshot")
+		os.Exit(1)
+	}
+
 	if err = builder.WebhookManagedBy(mgr, &impv1alpha1.ImpVM{}).
 		WithDefaulter(&webhookv1alpha1.ImpVMWebhook{}).
 		WithValidator(&webhookv1alpha1.ImpVMWebhook{}).

internal/controller/impvmsnapshot_controller.go

@@ -0,0 +1,51 @@
+package controller
+
+import (
+	"context"
+
+	"k8s.io/apimachinery/pkg/runtime"
+	ctrl "sigs.k8s.io/controller-runtime"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	logf "sigs.k8s.io/controller-runtime/pkg/log"
+
+	impv1alpha1 "github.com/syscode-labs/imp/api/v1alpha1"
+)
+
+// ImpVMSnapshotReconciler reconciles ImpVMSnapshot objects.
+type ImpVMSnapshotReconciler struct {
+	client.Client
+	Scheme *runtime.Scheme
+}
+
+// +kubebuilder:rbac:groups=imp.dev,resources=impvmsnapshots,verbs=get;list;watch;create;update;patch;delete
+// +kubebuilder:rbac:groups=imp.dev,resources=impvmsnapshots/status,verbs=get;update;patch
+// +kubebuilder:rbac:groups=imp.dev,resources=impvmsnapshots/finalizers,verbs=update
+
+func (r *ImpVMSnapshotReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
+	log := logf.FromContext(ctx)
+
+	snap := &impv1alpha1.ImpVMSnapshot{}
+	if err := r.Get(ctx, req.NamespacedName, snap); err != nil {
+		return ctrl.Result{}, client.IgnoreNotFound(err)
+	}
+
+	// Set initial phase
+	if snap.Status.Phase == "" {
+		base := snap.DeepCopy()
+		snap.Status.Phase = "Pending"
+		if err := r.Status().Patch(ctx, snap, client.MergeFrom(base)); err != nil {
+			return ctrl.Result{}, err
+		}
+		log.Info("ImpVMSnapshot created, set to Pending", "name", snap.Name)
+	}
+
+	// TODO: trigger agent snapshot, handle cron scheduling, OCI push.
+
+	return ctrl.Result{}, nil
+}
+
+func (r *ImpVMSnapshotReconciler) SetupWithManager(mgr ctrl.Manager) error {
+	return ctrl.NewControllerManagedBy(mgr).
+		For(&impv1alpha1.ImpVMSnapshot{}).
+		Complete(r)
+}

internal/controller/impvmsnapshot_controller_test.go

@@ -0,0 +1,51 @@
+package controller
+
+import (
+	"context"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+	"sigs.k8s.io/controller-runtime/pkg/reconcile"
+
+	impv1alpha1 "github.com/syscode-labs/imp/api/v1alpha1"
+)
+
+func newSnapshotReconciler() *ImpVMSnapshotReconciler {
+	return &ImpVMSnapshotReconciler{
+		Client: k8sClient,
+		Scheme: k8sClient.Scheme(),
+	}
+}
+
+var _ = Describe("ImpVMSnapshot controller", func() {
+	ctx := context.Background()
+
+	It("sets phase to Pending on creation", func() {
+		snap := &impv1alpha1.ImpVMSnapshot{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      "snap-ctrl-test",
+				Namespace: "default",
+			},
+			Spec: impv1alpha1.ImpVMSnapshotSpec{
+				SourceVMName:      "vm-does-not-exist",
+				SourceVMNamespace: "default",
+				Storage:           impv1alpha1.SnapshotStorageSpec{Type: "node-local"},
+			},
+		}
+		Expect(k8sClient.Create(ctx, snap)).To(Succeed())
+		DeferCleanup(func() { k8sClient.Delete(ctx, snap) }) //nolint:errcheck
+
+		r := newSnapshotReconciler()
+		_, err := r.Reconcile(ctx, reconcile.Request{
+			NamespacedName: types.NamespacedName{Name: "snap-ctrl-test", Namespace: "default"},
+		})
+		Expect(err).NotTo(HaveOccurred())
+
+		updated := &impv1alpha1.ImpVMSnapshot{}
+		Expect(k8sClient.Get(ctx, types.NamespacedName{Name: "snap-ctrl-test", Namespace: "default"}, updated)).To(Succeed())
+		Expect(updated.Status.Phase).NotTo(BeEmpty())
+		Expect(updated.Status.Phase).To(Equal("Pending"))
+	})
+})