payments/cmd/root.go

package cmd

import (
	_ "github.com/bombsimon/logrusr/v3"
	"github.com/hanzo-fi/go-libs/v5/pkg/service"
	"github.com/hanzo-fi/go-libs/v5/pkg/storage/bun/migrate"
	"github.com/spf13/cobra"
)

var (
	ServiceName = "payments"
	Version     = "develop"
	BuildDate   = "-"
	Commit      = "-"
	Edition     = "community"
)

const (
	ConfigEncryptionKeyFlag                      = "config-encryption-key"
	ListenFlag                                   = "listen"
	StackFlag                                    = "stack"
	ConnectorPollingPeriodDefault                = "connector-polling-period-default"
	ConnectorPollingPeriodMinimum                = "connector-polling-period-minimum"
	ConnectorHealthCheckInterval                 = "connector-health-check-interval"
	ConnectorHealthCheckErrorThreshold           = "connector-health-check-error-threshold"
	stackPublicURLFlag                           = "stack-public-url"
	temporalMaxConcurrentWorkflowTaskPollersFlag = "temporal-max-concurrent-workflow-task-pollers"
	temporalMaxConcurrentActivityTaskPollersFlag = "temporal-max-concurrent-activity-task-pollers"
	temporalMaxSlotsPerPollerFlag                = "temporal-max-slots-per-poller"
	temporalMaxLocalActivitySlotsFlag            = "temporal-max-local-activity-slots"
	temporalRateLimitingRetryDelay               = "temporal-rate-limiting-retry-delay"
	OutboxPollingIntervalFlag                    = "outbox-schedule-polling-interval"
	OutboxCleanupIntervalFlag                    = "outbox-schedule-cleanup-interval"
	SkipOutboxScheduleCreationFlag               = "skip-outbox-schedule-creation"
)

func NewRootCommand() *cobra.Command {
	root := &cobra.Command{
		Use:               "payments",
		Short:             "payments",
		DisableAutoGenTag: true,
		Version:           Version,
	}

	root.PersistentFlags().String(ConfigEncryptionKeyFlag, "", "Config encryption key")

	version := newVersion()
	root.AddCommand(version)

	migrate := newMigrate()
	root.AddCommand(migrate)

	server := newServer()
	addAutoMigrateCommand(server)
	root.AddCommand(server)

	worker := newWorker()
	root.AddCommand(worker)

	purge := newPurge()
	purge.Flags().String(StackFlag, "", "Stack name")
	root.AddCommand(purge)

	workbench := newWorkbench()
	root.AddCommand(workbench)

	recreateSchedules := newRecreateSchedules()
	root.AddCommand(recreateSchedules)

	return root
}

func Execute() {
	service.Execute(NewRootCommand())
}

func addAutoMigrateCommand(cmd *cobra.Command) {
	cmd.Flags().Bool(autoMigrateFlag, false, "Auto migrate database")
	cmd.PreRunE = func(cmd *cobra.Command, args []string) error {
		autoMigrate, _ := cmd.Flags().GetBool(autoMigrateFlag)
		if autoMigrate {
			return migrate.Run(cmd, args, Migrate)
		}
		return nil
	}
}