pkg/ansible: Adding ansible operator controller and events package

pkg/ansible/controller/controller.go

@@ -0,0 +1,91 @@
+// Copyright 2018 The Operator-SDK Authors
+//
+// 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 controller
+
+import (
+	"fmt"
+	"log"
+	"strings"
+	"time"
+
+	"github.com/operator-framework/operator-sdk/pkg/ansible/events"
+	"github.com/operator-framework/operator-sdk/pkg/ansible/runner"
+
+	"github.com/sirupsen/logrus"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	"sigs.k8s.io/controller-runtime/pkg/controller"
+	crthandler "sigs.k8s.io/controller-runtime/pkg/handler"
+	"sigs.k8s.io/controller-runtime/pkg/manager"
+	"sigs.k8s.io/controller-runtime/pkg/source"
+)
+
+// Options - options for your controller
+type Options struct {
+	EventHandlers []events.EventHandler
+	LoggingLevel  events.LogLevel
+	Runner        runner.Runner
+	Namespace     string
+	GVK           schema.GroupVersionKind
+	// StopChannel is used to deal with the bug:
+	// https://github.com/kubernetes-sigs/controller-runtime/issues/103
+	StopChannel <-chan struct{}
+}
+
+// Add - Creates a new ansible operator controller and adds it to the manager
+func Add(mgr manager.Manager, options Options) {
+	logrus.Infof("Watching %s/%v, %s, %s", options.GVK.Group, options.GVK.Version, options.GVK.Kind, options.Namespace)
+	if options.EventHandlers == nil {
+		options.EventHandlers = []events.EventHandler{}
+	}
+	eventHandlers := append(options.EventHandlers, events.NewLoggingEventHandler(options.LoggingLevel))
+
+	aor := &AnsibleOperatorReconciler{
+		Client:        mgr.GetClient(),
+		GVK:           options.GVK,
+		Runner:        options.Runner,
+		EventHandlers: eventHandlers,
+	}
+
+	// Register the GVK with the schema
+	mgr.GetScheme().AddKnownTypeWithName(options.GVK, &unstructured.Unstructured{})
+	metav1.AddToGroupVersion(mgr.GetScheme(), schema.GroupVersion{
+		Group:   options.GVK.Group,
+		Version: options.GVK.Version,
+	})
+
+	//Create new controller runtime controller and set the controller to watch GVK.
+	c, err := controller.New(fmt.Sprintf("%v-controller", strings.ToLower(options.GVK.Kind)), mgr, controller.Options{
+		Reconciler: aor,
+	})
+	if err != nil {
+		log.Fatal(err)
+	}
+	u := &unstructured.Unstructured{}
+	u.SetGroupVersionKind(options.GVK)
+	if err := c.Watch(&source.Kind{Type: u}, &crthandler.EnqueueRequestForObject{}); err != nil {
+		log.Fatal(err)
+	}
+
+	r := NewReconcileLoop(time.Minute*1, options.GVK, mgr.GetClient())
+	r.Stop = options.StopChannel
+	cs := &source.Channel{Source: r.Source}
+	cs.InjectStopChannel(options.StopChannel)
+	if err := c.Watch(cs, &crthandler.EnqueueRequestForObject{}); err != nil {
+		log.Fatal(err)
+	}
+	r.Start()
+}