registry.go

package launcher

import (
	"sort"
	"strings"

	"github.com/spf13/cobra"
	"go.uber.org/zap"
)

var AppRegistry = map[string]*AppDef{}

func RegisterApp(logger *zap.Logger, appDef *AppDef) {
	logger.Debug("registering app", zap.Stringer("app", appDef))
	AppRegistry[appDef.ID] = appDef
}

var RegisterCommonFlags func(logger *zap.Logger, cmd *cobra.Command) error

func RegisterFlags(logger *zap.Logger, cmd *cobra.Command) error {
	for _, appDef := range AppRegistry {
		logger.Debug("trying to register flags", zap.String("app_id", appDef.ID))
		if appDef.RegisterFlags != nil {
			logger.Debug("found non nil flags, registering", zap.String("app_id", appDef.ID))
			err := appDef.RegisterFlags(cmd)
			if err != nil {
				return err
			}
		}
	}

	if RegisterCommonFlags != nil {
		if err := RegisterCommonFlags(logger, cmd); err != nil {
			return err
		}
	}

	return nil
}

func ParseAppsFromArgs(args []string, runByDefault func(string) bool) (apps []string) {
	if len(args) == 0 {
		return ParseAppsFromArgs([]string{"all"}, runByDefault)
	}

	for _, arg := range args {
		chunks := strings.Split(arg, ",")
		for _, app := range chunks {
			app = strings.TrimSpace(app)
			if app == "all" {
				for app := range AppRegistry {
					if !runByDefault(app) {
						continue
					}
					apps = append(apps, app)
				}
			} else {
				if strings.HasPrefix(app, "-") {
					removeApp := app[1:]
					apps = removeElement(apps, removeApp)
				} else {
					apps = append(apps, app)
				}
			}

		}
	}

	sort.Strings(apps)

	return
}

func removeElement(lst []string, el string) (out []string) {
	for _, l := range lst {
		if l != el {
			out = append(out, l)
		}
	}
	return
}