feat: GenericEventSource to support authentication. Closes argoproj#928…

… (argoproj#935) Signed-off-by: Derek Wang <whynowy@gmail.com>
juliev0 · Dec 21, 2020 · 7c09b97 · 7c09b97
1 parent c742f19
commit 7c09b97
Show file tree

Hide file tree

Showing 9 changed files with 444 additions and 325 deletions.
diff --git a/api/event-source.html b/api/event-source.html
diff --git a/api/event-source.md b/api/event-source.md
diff --git a/api/openapi-spec/swagger.json b/api/openapi-spec/swagger.json
diff --git a/eventsources/sources/generic/start.go b/eventsources/sources/generic/start.go
@@ -3,16 +3,20 @@ package generic
 import (
 	"context"
 	"encoding/json"
+	fmt "fmt"
 	"time"
 
+	"go.uber.org/zap"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/connectivity"
+	"google.golang.org/grpc/metadata"
+
+	"github.com/argoproj/argo-events/common"
 	"github.com/argoproj/argo-events/common/logging"
 	"github.com/argoproj/argo-events/eventsources/sources"
 	apicommon "github.com/argoproj/argo-events/pkg/apis/common"
 	"github.com/argoproj/argo-events/pkg/apis/events"
 	"github.com/argoproj/argo-events/pkg/apis/eventsource/v1alpha1"
-	"go.uber.org/zap"
-	"google.golang.org/grpc"
-	"google.golang.org/grpc/connectivity"
 )
 
 // EventListener implements Eventing for generic event source
@@ -110,7 +114,16 @@ func (el *EventListener) connect() (Eventing_StartEventSourceClient, error) {
 	}
 	el.conn = conn
 	client := NewEventingClient(el.conn)
-	return client.StartEventSource(context.Background(), &EventSource{
+	ctx := context.Background()
+	if el.GenericEventSource.AuthSecret != nil {
+		token, err := common.GetSecretFromVolume(el.GenericEventSource.AuthSecret)
+		if err != nil {
+			return nil, err
+		}
+		auth := fmt.Sprintf("Bearer %s", token)
+		ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs("authorization", auth))
+	}
+	return client.StartEventSource(ctx, &EventSource{
 		Name:   el.GetEventName(),
 		Config: []byte(el.GenericEventSource.Config),
 	})

diff --git a/pkg/apis/eventsource/v1alpha1/generated.pb.go b/pkg/apis/eventsource/v1alpha1/generated.pb.go
diff --git a/pkg/apis/eventsource/v1alpha1/generated.proto b/pkg/apis/eventsource/v1alpha1/generated.proto
diff --git a/pkg/apis/eventsource/v1alpha1/openapi_generated.go b/pkg/apis/eventsource/v1alpha1/openapi_generated.go
diff --git a/pkg/apis/eventsource/v1alpha1/types.go b/pkg/apis/eventsource/v1alpha1/types.go
@@ -823,6 +823,9 @@ type GenericEventSource struct {
 	// Metadata holds the user defined metadata which will passed along the event payload.
 	// +optional
 	Metadata map[string]string `json:"metadata,omitempty" protobuf:"bytes,5,rep,name=metadata"`
+	// AuthSecret holds a secret selector that contains a bearer token for authentication
+	// +optional
+	AuthSecret *corev1.SecretKeySelector `json:"authSecret,omitempty" protobuf:"bytes,6,opt,name=authSecret"`
 }
 
 const (

diff --git a/pkg/apis/eventsource/v1alpha1/zz_generated.deepcopy.go b/pkg/apis/eventsource/v1alpha1/zz_generated.deepcopy.go