connectrpcofx

var Module = fx.Module("connectrpc",
	provideConnectConfigPointer,
	provideConnectConfig,
	provideConnectServiceNames,
	provideAllConnectHandlers,
	invokeServer,
)

var WithCorsHttpMiddleware = fx.Provide(
	fx.Annotate(
		connectrpco.NewCorsHandler,
		fx.ParamTags(fxtags.Named(corso.NamedConfig)),
		fx.ResultTags(fxtags.Group(connectrpco.GroupHttpMiddleware)),
	),
)

var WithGrpcHealthHandler = fx.Provide(
	fx.Annotate(
		connectrpco.NewGrpcHealthHandler,
		fx.ParamTags(
			fxtags.Group(connectrpco.GroupConnectServiceNames),
			fxtags.Group(connectrpco.GroupConnectHandlerOptions),
		),
		fx.ResultTags(fxtags.GroupFlatten(connectrpco.GroupUtilityConnectHandler)),
	),
)

var WithGrpcReflectHandler = fx.Provide(
	fx.Annotate(
		connectrpco.NewGrpcReflectHandler,
		fx.ParamTags(
			fxtags.Group(connectrpco.GroupConnectServiceNames),
			fxtags.Group(connectrpco.GroupConnectHandlerOptions),
		),
		fx.ResultTags(fxtags.GroupFlatten(connectrpco.GroupUtilityConnectHandler)),
	),
)

var WithOtelConnectInterceptor = func(trustRemote bool, omitTraceEvents bool) fx.Option {
	var opts []otelconnect.Option

	if trustRemote {
		opts = append(opts, otelconnect.WithTrustRemote())
	}

	if omitTraceEvents {
		opts = append(opts, otelconnect.WithoutTraceEvents())
	}

	return fx.Provide(
		fx.Annotate(
			func() (connect.HandlerOption, error) {
				otelInterceptor, err := otelconnect.NewInterceptor(opts...)
				if err != nil {
					return nil, err
				}
				return connect.WithInterceptors(otelInterceptor), nil
			},
			fx.ResultTags(fxtags.Group(connectrpco.GroupConnectHandlerOptions)),
		),
	)
}