Rollbar notifier

api/v1beta1/provider_types.go

@@ -30,7 +30,7 @@ const (
 // ProviderSpec defines the desired state of Provider
 type ProviderSpec struct {
 	// Type of provider
-	// +kubebuilder:validation:Enum=slack;discord;msteams;rocket;generic;generic-hmac;github;gitlab;bitbucket;azuredevops;googlechat;webex;sentry;azureeventhub;telegram;lark;matrix;opsgenie;alertmanager;grafana;githubdispatch;
+	// +kubebuilder:validation:Enum=slack;discord;msteams;rocket;generic;generic-hmac;github;gitlab;bitbucket;azuredevops;googlechat;webex;sentry;rollbar;azureeventhub;telegram;lark;matrix;opsgenie;alertmanager;grafana;githubdispatch;
 	// +required
 	Type string `json:"type"`
 
@@ -92,6 +92,7 @@ const (
 	GoogleChatProvider     string = "googlechat"
 	WebexProvider          string = "webex"
 	SentryProvider         string = "sentry"
+	RollbarProvider        string = "rollbar"
 	AzureEventHubProvider  string = "azureeventhub"
 	TelegramProvider       string = "telegram"
 	LarkProvider           string = "lark"

api/v1beta2/provider_types.go

@@ -43,6 +43,7 @@ const (
 	GooglePubSubProvider    string = "googlepubsub"
 	WebexProvider           string = "webex"
 	SentryProvider          string = "sentry"
+	RollbarProvider         string = "rollbar"
 	AzureEventHubProvider   string = "azureeventhub"
 	TelegramProvider        string = "telegram"
 	LarkProvider            string = "lark"
@@ -56,7 +57,7 @@ const (
 // ProviderSpec defines the desired state of the Provider.
 type ProviderSpec struct {
 	// Type specifies which Provider implementation to use.
-	// +kubebuilder:validation:Enum=slack;discord;msteams;rocket;generic;generic-hmac;github;gitlab;gitea;bitbucketserver;bitbucket;azuredevops;googlechat;googlepubsub;webex;sentry;azureeventhub;telegram;lark;matrix;opsgenie;alertmanager;grafana;githubdispatch;pagerduty;datadog
+	// +kubebuilder:validation:Enum=slack;discord;msteams;rocket;generic;generic-hmac;github;gitlab;gitea;bitbucketserver;bitbucket;azuredevops;googlechat;googlepubsub;webex;sentry;rollbar;azureeventhub;telegram;lark;matrix;opsgenie;alertmanager;grafana;githubdispatch;pagerduty;datadog
 	// +required
 	Type string `json:"type"`
 

api/v1beta3/provider_types.go

@@ -43,6 +43,7 @@ const (
 	GooglePubSubProvider    string = "googlepubsub"
 	WebexProvider           string = "webex"
 	SentryProvider          string = "sentry"
+	RollbarProvider         string = "rollbar"
 	AzureEventHubProvider   string = "azureeventhub"
 	TelegramProvider        string = "telegram"
 	LarkProvider            string = "lark"
@@ -57,7 +58,7 @@ const (
 // ProviderSpec defines the desired state of the Provider.
 type ProviderSpec struct {
 	// Type specifies which Provider implementation to use.
-	// +kubebuilder:validation:Enum=slack;discord;msteams;rocket;generic;generic-hmac;github;gitlab;gitea;bitbucketserver;bitbucket;azuredevops;googlechat;googlepubsub;webex;sentry;azureeventhub;telegram;lark;matrix;opsgenie;alertmanager;grafana;githubdispatch;pagerduty;datadog;nats
+	// +kubebuilder:validation:Enum=slack;discord;msteams;rocket;generic;generic-hmac;github;gitlab;gitea;bitbucketserver;bitbucket;azuredevops;googlechat;googlepubsub;webex;sentry;rollbar;azureeventhub;telegram;lark;matrix;opsgenie;alertmanager;grafana;githubdispatch;pagerduty;datadog;nats
 	// +required
 	Type string `json:"type"`
 

config/crd/bases/notification.toolkit.fluxcd.io_providers.yaml

@@ -109,6 +109,7 @@ spec:
                 - googlechat
                 - webex
                 - sentry
+                - rollbar
                 - azureeventhub
                 - telegram
                 - lark
@@ -322,6 +323,7 @@ spec:
                 - googlepubsub
                 - webex
                 - sentry
+                - rollbar
                 - azureeventhub
                 - telegram
                 - lark
@@ -538,6 +540,7 @@ spec:
                 - googlepubsub
                 - webex
                 - sentry
+                - rollbar
                 - azureeventhub
                 - telegram
                 - lark

internal/notifier/factory.go

@@ -37,6 +37,7 @@ var (
 		apiv1.GooglePubSubProvider:    googlePubSubNotifierFunc,
 		apiv1.WebexProvider:           webexNotifierFunc,
 		apiv1.SentryProvider:          sentryNotifierFunc,
+		apiv1.RollbarProvider:         rollbarNotifierFunc,
 		apiv1.AzureEventHubProvider:   azureEventHubNotifierFunc,
 		apiv1.TelegramProvider:        telegramNotifierFunc,
 		apiv1.LarkProvider:            larkNotifierFunc,
@@ -164,6 +165,10 @@ func sentryNotifierFunc(opts notifierOptions) (Interface, error) {
 	return NewSentry(opts.CertPool, opts.URL, opts.Channel)
 }
 
+func rollbarNotifierFunc(opts notifierOptions) (Interface, error) {
+	return NewRollbar(opts.Token, opts.Channel, opts.URL)
+}
+
 func azureEventHubNotifierFunc(opts notifierOptions) (Interface, error) {
 	return NewAzureEventHub(opts.URL, opts.Token, opts.Channel)
 }

internal/notifier/rollbar.go

@@ -0,0 +1,111 @@
+package notifier
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+
+	eventv1 "github.com/fluxcd/pkg/apis/event/v1beta1"
+	"github.com/hashicorp/go-retryablehttp"
+)
+
+// Rollbar holds the access token and environment information
+type Rollbar struct {
+	Token       string
+	Environment string
+	RollbarUrl  string
+}
+
+// RollbarPayload represents the payload to send to Rollbar
+type RollbarPayload struct {
+	AccessToken string      `json:"access_token"`
+	Data        RollbarData `json:"data"`
+}
+
+type RollbarData struct {
+	Environment string        `json:"environment"`
+	Body        RollbarBody   `json:"body"`
+	Level       string        `json:"level"`
+	Timestamp   int64         `json:"timestamp"`
+	Platform    string        `json:"platform"`
+	Server      RollbarServer `json:"server"`
+}
+
+type RollbarBody struct {
+	Message RollbarMessage `json:"message"`
+}
+
+type RollbarMessage struct {
+	Body string `json:"body"`
+}
+
+type RollbarServer struct {
+	Host string `json:"host"`
+}
+
+const defaultRollbarURL = "https://api.rollbar.com/api/1/item/"
+
+// NewRollbar validates the Rollbar token and returns a Rollbar object
+func NewRollbar(token string, environment string, url string) (*Rollbar, error) {
+	fmt.Printf("Starting Rollbar notifier!!!! :::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::\n")
+	if token == "" {
+		return nil, fmt.Errorf("Rollbar token cannot be empty")
+	} else {
+		// TODO Disable this
+		fmt.Printf("RollbarNotifier ::: Received token %s\n", token)
+	}
+
+	if url == "" {
+		url = defaultRollbarURL
+	}
+
+	fmt.Printf("RollbarNotifier ::: Posting to url %s\n", url)
+
+	return &Rollbar{
+		Token:       token,
+		Environment: environment,
+		RollbarUrl:  url,
+	}, nil
+}
+
+// Post Rollbar message
+func (r *Rollbar) Post(ctx context.Context, event eventv1.Event) error {
+	// Only post error events
+	// if event.Severity != eventv1.EventSeverityError {
+	// 	return nil
+	// }
+
+	fmt.Printf("RollbarNotifier ::: Event:  %v\n", event)
+	fmt.Printf("RollbarNotifier ::: Rollbar Object %v\n", r)
+	fmt.Printf("RollbarNotifier ::: Severity %v\n", event.Severity)
+
+	host := fmt.Sprintf("%s/%s.%s", strings.ToLower(event.InvolvedObject.Kind), event.InvolvedObject.Name, event.InvolvedObject.Namespace)
+	payload := RollbarPayload{
+		AccessToken: r.Token,
+		Data: RollbarData{
+			Environment: r.Environment,
+			Body: RollbarBody{
+				Message: RollbarMessage{
+					Body: event.Message,
+				},
+			},
+			Level:     "error",
+			Timestamp: time.Now().Unix(),
+			Platform:  "linux",
+			Server: RollbarServer{
+				Host: host,
+			},
+		},
+	}
+
+	err := postMessage(ctx, r.RollbarUrl, "", nil, payload, func(request *retryablehttp.Request) {
+		request.Header.Add("Content-Type", "application/json")
+		request.Header.Add("accept", "application/json")
+		request.Header.Add("X-Rollbar-Access-Token", r.Token)
+	})
+	if err != nil {
+		return fmt.Errorf("postMessage failed: %w\n", err)
+	}
+	return nil
+}

internal/notifier/rollbar_test.go

@@ -0,0 +1,106 @@
+package notifier
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+
+	eventv1 "github.com/fluxcd/pkg/apis/event/v1beta1"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	corev1 "k8s.io/api/core/v1"
+)
+
+// Mock Rollbar server
+func newMockRollbarServer(t *testing.T, expectedPayload RollbarPayload) *httptest.Server {
+	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		var actualPayload RollbarPayload
+		err := json.NewDecoder(r.Body).Decode(&actualPayload)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusBadRequest)
+			return
+		}
+		fmt.Printf("Expected %v", expectedPayload)
+		fmt.Printf("Actual %v", actualPayload)
+		assert.Equal(t, expectedPayload, actualPayload)
+		w.WriteHeader(http.StatusOK)
+	}))
+}
+
+func TestNewRollbar(t *testing.T) {
+	tests := []struct {
+		token       string
+		environment string
+		expectError bool
+	}{
+		{"valid_token", "production", false},
+		{"", "production", true},
+	}
+
+	for _, test := range tests {
+		rollbar, err := NewRollbar(test.token, test.environment, "")
+		if test.expectError {
+			require.Error(t, err)
+		} else {
+			require.NoError(t, err)
+			assert.Equal(t, test.token, rollbar.Token)
+			assert.Equal(t, test.environment, rollbar.Environment)
+		}
+	}
+}
+
+func TestRollbarPost(t *testing.T) {
+	event := eventv1.Event{
+		InvolvedObject: corev1.ObjectReference{
+			Kind:      "Pod",
+			Name:      "rollbar-pod",
+			Namespace: "default",
+		},
+		Severity:            eventv1.EventSeverityError,
+		Message:             "An error occurred",
+		ReportingController: "rollbar-controller",
+	}
+
+	expectedPayload := RollbarPayload{
+		AccessToken: "test_token",
+		Data: RollbarData{
+			Environment: "test_env",
+			Body: RollbarBody{
+				Message: RollbarMessage{
+					Body: event.Message,
+				},
+			},
+			Level:     "error",
+			Timestamp: time.Now().Unix(),
+			Platform:  "linux",
+			Server: RollbarServer{
+				Host: "pod/rollbar-pod.default",
+			},
+		},
+	}
+
+	server := newMockRollbarServer(t, expectedPayload)
+	defer server.Close()
+
+	rollbar, err := NewRollbar("test_token", "test_env", server.URL)
+	require.NoError(t, err)
+
+	err = rollbar.Post(context.Background(), event)
+	require.NoError(t, err)
+}
+
+func TestRollbarPostNonErrorEvent(t *testing.T) {
+	event := eventv1.Event{
+		Severity: eventv1.EventSeverityInfo,
+	}
+
+	rollbar, err := NewRollbar("test_token", "test_env", "")
+	require.NoError(t, err)
+
+	err = rollbar.Post(context.Background(), event)
+	require.NoError(t, err)
+}