Add Authorizer override to Publish request

Author: exanubes

client.go

@@ -27,10 +27,16 @@ func (client *appsync_client) Publish(ctx context.Context, input PublishCommandI
 
 	frame := &events.FrameBuilder{}
 
+	var authz app.RequestAuthorizer
+	if input.Authorizer != nil {
+		authz = authorizer.NewInternalAdapter(input.Authorizer)
+	}
+
 	err := client.usecases.Publish.Publish(ctx, publish.PublishCommandInput{
 		Destination: input.Channel,
 		Payload:     input.Payload,
 		Frame:       frame,
+		Authorizer:  authz,
 	})
 	return err
 }

internal/app/usecases/publish/ports.go

@@ -11,6 +11,7 @@ type PublishCommandInput struct {
 	Destination string
 	Payload     app.Payload
 	Frame       app.FrameBuilder
+	Authorizer  app.RequestAuthorizer
 }
 
 type PublishMessage interface {

internal/app/usecases/publish/publish-message.go

@@ -23,7 +23,13 @@ func NewPublishMessageUsecase(
 }
 
 func (usecase *PublishMessageUsecase) Publish(ctx context.Context, input PublishCommandInput) error {
-	signature, err := usecase.authorizer.Authorize(ctx, app.AuthorizeCommandInput{
+	authorizer := usecase.authorizer
+
+	if input.Authorizer != nil {
+		authorizer = input.Authorizer
+	}
+
+	signature, err := authorizer.Authorize(ctx, app.AuthorizeCommandInput{
 		Channel: input.Destination,
 		Payload: input.Payload,
 	})

internal/app/usecases/publish/publish_message_test.go

@@ -57,14 +57,15 @@ func TestPublish(t *testing.T) {
 	destination := "test-channel"
 
 	tests := []struct {
-		name             string
-		authorizer       *mock_authorizer
-		sender           *mock_sender
-		expect_err       error
-		expect_send      bool
-		expect_payload   app.Payload
-		expect_channel   string
-		expect_signature app.Signature
+		name                string
+		authorizer          *mock_authorizer
+		override_authorizer *mock_authorizer
+		sender              *mock_sender
+		expect_err          error
+		expect_send         bool
+		expect_payload      app.Payload
+		expect_channel      string
+		expect_signature    app.Signature
 	}{
 		{
 			name:             "success",
@@ -83,6 +84,17 @@ func TestPublish(t *testing.T) {
 			expect_err:  auth_err,
 			expect_send: false,
 		},
+		{
+			name:                "override authorizer is used instead of default",
+			authorizer:          &mock_authorizer{err: auth_err},
+			override_authorizer: &mock_authorizer{signature: app.Signature{"Authorization": "override-sig"}},
+			sender:              &mock_sender{},
+			expect_err:          nil,
+			expect_send:         true,
+			expect_payload:      payload,
+			expect_channel:      destination,
+			expect_signature:    app.Signature{"Authorization": "override-sig"},
+		},
 		{
 			name:             "writer error is returned",
 			authorizer:       &mock_authorizer{signature: signature},
@@ -100,10 +112,15 @@ func TestPublish(t *testing.T) {
 			frame := &mockFrameBuilder{built_frame: mock_frame{}}
 			usecase := publish.NewPublishMessageUsecase(tt.authorizer, tt.sender)
 
+			var override_authorizer app.RequestAuthorizer
+			if tt.override_authorizer != nil {
+				override_authorizer = tt.override_authorizer
+			}
 			err := usecase.Publish(context.Background(), publish.PublishCommandInput{
 				Destination: destination,
 				Payload:     payload,
 				Frame:       frame,
+				Authorizer:  override_authorizer,
 			})
 
 			if !errors.Is(err, tt.expect_err) {

ports.go

@@ -27,8 +27,9 @@ type Backpressure struct {
 }
 
 type PublishCommandInput struct {
-	Channel string
-	Payload []byte
+	Channel    string
+	Payload    []byte
+	Authorizer authorizer.Authorizer
 }
 
 type SubscribeCommandInput struct {