stack/components/operator/internal/modules/reconciliation/handler.go

package reconciliation

import (
	stackv1beta3 "github.com/formancehq/operator/apis/stack/v1beta3"
	"github.com/formancehq/operator/internal/modules"
	"github.com/formancehq/operator/internal/modules/ledger"
	"github.com/formancehq/operator/internal/modules/payments"
)

type module struct{}

// DependsOn implements modules.DependsOnAwareModule.
func (*module) DependsOn() []modules.Module {
	return []modules.Module{
		ledger.Module,
		payments.Module,
	}
}

func (o module) Name() string {
	return "reconciliation"
}

func (o module) IsEE() bool {
	return true
}

func (p module) Postgres(ctx modules.ReconciliationConfig) stackv1beta3.PostgresConfig {
	return ctx.Configuration.Spec.Services.Reconciliation.Postgres
}

func (o module) Versions() map[string]modules.Version {
	return map[string]modules.Version{
		"v0.0.0": {
			DatabaseMigration: &modules.DatabaseMigration{
				Shutdown: false,
				Command:  []string{"migrate", "up"},
				AdditionalEnv: func(config modules.ReconciliationConfig) []modules.EnvVar {
					return []modules.EnvVar{}
				},
			},
			Services: func(ctx modules.ReconciliationConfig) modules.Services {
				return modules.Services{
					{
						InjectPostgresVariables: true,
						AuthConfiguration: func(config modules.ReconciliationConfig) stackv1beta3.ClientConfiguration {
							return stackv1beta3.NewClientConfiguration().WithAdditionalScopes(modules.ModulesToScopes(Module.DependsOn()...)...)
						},
						HasVersionEndpoint: true,
						ListenEnvVar:       "LISTEN",
						ExposeHTTP:         modules.DefaultExposeHTTP,
						Annotations:        ctx.Configuration.Spec.Services.Reconciliation.Annotations.Service,
						Container: func(resolveContext modules.ContainerResolutionConfiguration) modules.Container {
							return modules.Container{
								Env:   o.reconciliationEnvVars(resolveContext),
								Image: modules.GetImage("reconciliation", resolveContext.Versions.Spec.Reconciliation),
								Resources: modules.GetResourcesWithDefault(
									resolveContext.Configuration.Spec.Services.Reconciliation.ResourceProperties,
									modules.ResourceSizeSmall(),
								),
								Debug: resolveContext.Configuration.Spec.Services.Reconciliation.Debug,
							}
						},
					},
				}
			},
		},
	}
}

func (o module) reconciliationEnvVars(resolveContext modules.ContainerResolutionConfiguration) modules.ContainerEnv {
	env := modules.NewEnv().Append(
		modules.Env("POSTGRES_DATABASE_NAME", "$(POSTGRES_DATABASE)"),
		modules.Env("STACK_CLIENT_ID", resolveContext.Stack.Status.StaticAuthClients["reconciliation"].ID),
		modules.Env("STACK_CLIENT_SECRET", resolveContext.Stack.Status.StaticAuthClients["reconciliation"].Secrets[0]),
	).Append(
		modules.AuthEnvVars(resolveContext.Stack.URL(), o.Name(), resolveContext.Configuration.Spec.Auth)...,
	)

	return env
}

var Module = &module{}

var _ modules.Module = Module
var _ modules.PostgresAwareModule = Module
var _ modules.DependsOnAwareModule = Module

func init() {
	modules.Register(Module)
}