add support for Rollout in migrations

Author: Zach Swanson

components/migrations.go

@@ -39,6 +39,7 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/reconcile"
 	"sigs.k8s.io/controller-runtime/pkg/source"
 
+	"github.com/coderanger/migrations-operator/api/stubs/argoproj"
 	migrationsv1beta1 "github.com/coderanger/migrations-operator/api/v1beta1"
 	"github.com/coderanger/migrations-operator/utils"
 	"github.com/coderanger/migrations-operator/webhook"
@@ -332,6 +333,8 @@ func (_ *migrationsComponent) findSpecFor(obj cu.Object) *corev1.PodSpec {
 		return &v.Spec
 	case *appsv1.Deployment:
 		return &v.Spec.Template.Spec
+	case *argoproj.Rollout:
+		return &v.Spec.Template.Spec
 	// TODO other types. lots of them.
 	default:
 		return nil

config/rbac/role.yaml

@@ -32,6 +32,14 @@ rules:
   - get
   - list
   - watch
+- apiGroups:
+  - argoproj.io
+  resources:
+  - rollouts
+  verbs:
+  - get
+  - list
+  - watch
 - apiGroups:
   - batch
   resources:

controllers/migrator.go

@@ -31,6 +31,7 @@ import (
 // +kubebuilder:rbac:groups="",resources=events,verbs=create;patch
 // +kubebuilder:rbac:groups=apps,resources=replicasets;deployments;statfulsets;daemonsets,verbs=get;list;watch
 // +kubebuilder:rbac:groups=batch,resources=cronjobs,verbs=get;list;watch
+// +kubebuilder:rbac:groups=argoproj.io,resources=rollouts,verbs=get;list;watch
 
 func Migrator(mgr ctrl.Manager) error {
 	return cu.NewReconciler(mgr).

main.go

@@ -28,6 +28,7 @@ import (
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/log/zap"
 
+	"github.com/coderanger/migrations-operator/api/stubs/argoproj"
 	migrationsv1beta1 "github.com/coderanger/migrations-operator/api/v1beta1"
 	"github.com/coderanger/migrations-operator/controllers"
 	"github.com/coderanger/migrations-operator/http"
@@ -43,6 +44,7 @@ var (
 func init() {
 	_ = clientgoscheme.AddToScheme(scheme)
 
+	_ = argoproj.AddToScheme(scheme)
 	_ = migrationsv1beta1.AddToScheme(scheme)
 	// +kubebuilder:scaffold:scheme
 }