github.com/caos/orbos@v1.5.14-0.20221103111702-e6cd0cea7ad4/internal/operator/boom/bundle/bundle.go

package bundle

import (
	"errors"
	"fmt"
	"sync"

	"github.com/caos/orbos/internal/operator/boom/api/latest"

	"github.com/caos/orbos/internal/operator/boom/metrics"

	"github.com/caos/orbos/internal/operator/boom/application"
	"github.com/caos/orbos/internal/operator/boom/bundle/bundles"
	"github.com/caos/orbos/internal/operator/boom/bundle/config"
	"github.com/caos/orbos/internal/operator/boom/current"
	"github.com/caos/orbos/internal/operator/boom/name"
	"github.com/caos/orbos/internal/operator/boom/templator"
	"github.com/caos/orbos/internal/operator/boom/templator/helm"
	helperTemp "github.com/caos/orbos/internal/operator/boom/templator/helper"
	"github.com/caos/orbos/internal/operator/boom/templator/yaml"
	"github.com/caos/orbos/internal/utils/clientgo"
	"github.com/caos/orbos/mntr"
)

var (
	Testmode bool = false
)

type Bundle struct {
	baseDirectoryPath string
	crdName           string
	predefinedBundle  name.Bundle
	orb               string
	Applications      map[name.Application]application.Application
	HelmTemplator     templator.Templator
	YamlTemplator     templator.Templator
	monitor           mntr.Monitor
}

func New(conf *config.Config) *Bundle {
	apps := make(map[name.Application]application.Application, 0)
	helmTemplator := helperTemp.NewTemplator(conf.Monitor, conf.CrdName, conf.BaseDirectoryPath, helm.GetName())
	yamlTemplator := helperTemp.NewTemplator(conf.Monitor, conf.CrdName, conf.BaseDirectoryPath, yaml.GetName())

	return &Bundle{
		crdName:           conf.CrdName,
		orb:               conf.Orb,
		baseDirectoryPath: conf.BaseDirectoryPath,
		monitor:           conf.Monitor,
		HelmTemplator:     helmTemplator,
		YamlTemplator:     yamlTemplator,
		Applications:      apps,
		predefinedBundle:  "",
	}
}

func (b *Bundle) GetPredefinedBundle() string {
	return b.predefinedBundle.String()
}

func (b *Bundle) CleanUp() error {

	err := b.HelmTemplator.CleanUp()
	if err != nil {
		return err
	}

	return b.YamlTemplator.CleanUp()
}

func (b *Bundle) GetApplications() map[name.Application]application.Application {
	return b.Applications
}

func (b *Bundle) AddApplicationsByBundleName(name name.Bundle) error {

	appNames := bundles.Get(name)
	if appNames == nil {
		return fmt.Errorf("No bundle known with name %s", name)
	}
	b.predefinedBundle = name

	for _, appName := range appNames {
		if err := b.AddApplicationByName(appName); err != nil {
			return err
		}
	}
	return nil
}

func (b *Bundle) AddApplicationByName(appName name.Application) error {

	app := application.New(b.monitor, appName, b.orb)
	return b.AddApplication(app)
}

func (b *Bundle) AddApplication(app application.Application) error {

	if _, found := b.Applications[app.GetName()]; found {
		return errors.New("Application already in bundle")
	}

	b.Applications[app.GetName()] = app
	return nil
}

func (b *Bundle) Reconcile(currentResourceList []*clientgo.Resource, spec *latest.ToolsetSpec) error {

	applicationCount := 0
	// go through list of application until every application is reconciled
	// and this orderNumber by orderNumber (default is 1)
	for orderNumber := 0; applicationCount < len(b.Applications); orderNumber++ {
		var wg sync.WaitGroup
		errList := make(map[name.Application]chan error, len(b.Applications))
		for appName := range b.Applications {
			//if application has the same orderNumber as currently iterating the reconcile the application
			if application.GetOrderNumber(appName) == orderNumber {
				wg.Add(1)
				errChan := make(chan error)
				go b.ReconcileApplication(currentResourceList, appName, spec, &wg, errChan)
				applicationCount++
				errList[appName] = errChan
			}
		}
		for appName, errChan := range errList {
			if err := <-errChan; err != nil {
				return fmt.Errorf("Error while reconciling application %s: %w", appName.String(), err)
			}
		}
		wg.Wait()
	}

	return nil
}

func (b *Bundle) ReconcileApplication(currentResourceList []*clientgo.Resource, appName name.Application, spec *latest.ToolsetSpec, wg *sync.WaitGroup, errChan chan error) {
	defer wg.Done()

	logFields := map[string]interface{}{
		"application": appName,
		"action":      "reconciling",
	}
	monitor := b.monitor.WithFields(logFields)

	app, found := b.Applications[appName]
	if !found {
		err := errors.New("Application not found")
		monitor.Error(err)
		errChan <- err
		return
	}
	monitor.Info("Start")

	deploy := app.Deploy(spec)
	currentApplicationResourceList := current.FilterForApplication(appName, currentResourceList)

	var resultFunc func(string, string) error
	if Testmode {
		resultFunc = func(resultFilePath, namespace string) error {
			return nil
		}
	} else {
		if deploy {
			resultFunc = applyWithCurrentState(monitor, currentApplicationResourceList, app, spec.ForceApply)
		} else {
			resultFunc = deleteWithCurrentState(monitor, currentApplicationResourceList, app)
		}
	}

	_, usedHelm := app.(application.HelmApplication)
	if usedHelm {
		templatorName := helm.GetName()
		err := b.HelmTemplator.Template(app, spec, resultFunc)
		if err != nil {
			metrics.FailureReconcilingApplication(appName.String(), templatorName.String(), deploy)
			errChan <- err
			return
		}
		metrics.SuccessfulReconcilingApplication(appName.String(), templatorName.String(), deploy)
	}
	_, usedYaml := app.(application.YAMLApplication)
	if usedYaml {
		templatorName := yaml.GetName()
		err := b.YamlTemplator.Template(app, spec, resultFunc)
		if err != nil {
			metrics.FailureReconcilingApplication(appName.String(), templatorName.String(), deploy)
			errChan <- err
			return
		}
		metrics.SuccessfulReconcilingApplication(appName.String(), templatorName.String(), deploy)
	}

	monitor.Info("Done")
	errChan <- nil
}