payments/internal/storage/module.go

package storage

import (
	"context"

	"github.com/hanzo-fi/go-libs/v5/pkg/observe/log"
	"github.com/hanzo-fi/go-libs/v5/pkg/service"
	"github.com/hanzo-fi/go-libs/v5/pkg/storage/bun/connect"
	"github.com/hanzo-fi/payments/internal/storage/bunconnect"
	"github.com/spf13/cobra"
	"github.com/uptrace/bun"
	"go.uber.org/fx"
)

func Module(cmd *cobra.Command, connectionOptions connect.ConnectionOptions, configEncryptionKey string) fx.Option {
	storageDriver, sqliteDSN, err := bunconnect.FromFlags(cmd.Flags())
	if err != nil {
		return fx.Error(err)
	}
	return fx.Options(
		bunconnect.Module(storageDriver, connectionOptions, sqliteDSN, service.IsDebug(cmd)),
		fx.Provide(func(logger logging.Logger, db *bun.DB) Storage {
			return newStorage(logger, db, configEncryptionKey)
		}),
		fx.Invoke(func(s Storage, lc fx.Lifecycle) {
			lc.Append(fx.Hook{
				OnStop: func(ctx context.Context) error {
					return s.Close()
				},
			})
		}),
	)
}