stack/ee/orchestration/internal/triggers/module.go

package triggers

import (
	"net/http"
	"strings"

	"github.com/ThreeDotsLabs/watermill/message"
	"github.com/formancehq/orchestration/internal/workflow"
	"github.com/formancehq/stack/libs/go-libs/logging"
	"github.com/uptrace/bun"
	"go.temporal.io/sdk/client"
	"go.uber.org/fx"
)

func NewModule(taskQueue string) fx.Option {
	return fx.Options(
		fx.Provide(NewManager),
		fx.Provide(func(httpClient *http.Client) *expressionEvaluator {
			return NewExpressionEvaluator(httpClient)
		}),
		fx.Provide(
			fx.Annotate(func(db *bun.DB) *triggerWorkflow {
				return NewWorkflow(db, taskQueue)
			}, fx.As(new(any)), fx.ResultTags(`group:"workflows"`)),
		),
		fx.Provide(
			fx.Annotate(func(db *bun.DB, manager *workflow.WorkflowManager, expressionEvaluator *expressionEvaluator) Activities {
				return NewActivities(db, manager, expressionEvaluator)
			}, fx.As(new(any)), fx.ResultTags(`group:"activities"`)),
		),
	)
}

func NewListenerModule(taskIDPrefix, taskQueue string, topics []string) fx.Option {
	return fx.Options(
		fx.Invoke(func(logger logging.Logger, r *message.Router, s message.Subscriber, temporalClient client.Client) {
			logger.Infof("Listening events from topics: %s", strings.Join(topics, ","))
			registerListener(r, s, temporalClient, taskIDPrefix, taskQueue, topics)
		}),
	)
}