kubevela/pkg/application/modify.go

package application

import (
	"errors"
	"strings"
)

func (app *Application) SetWorkload(workloadName, workloadType string, workloadData map[string]interface{}) error {
	if app == nil {
		return errors.New("app is nil pointer")
	}
	if workloadData == nil {
		workloadData = make(map[string]interface{})
	}
	workloadData["name"] = strings.ToLower(workloadName)
	if app.Components == nil {
		app.Components = make(map[string]map[string]interface{})
	}
	app.Components[workloadName] = map[string]interface{}{
		workloadType: workloadData,
	}
	return app.Validate()
}

func (app *Application) SetTrait(workloadName, traitType string, traitData map[string]interface{}) error {
	if app == nil {
		return errors.New("app is nil pointer")
	}
	if traitData == nil {
		traitData = make(map[string]interface{})
	}
	traitData["name"] = strings.ToLower(traitType)
	if app.Components == nil {
		app.Components = make(map[string]map[string]interface{})
	}
	comp := app.Components[workloadName]
	if comp == nil {
		comp = make(map[string]interface{})
	}
	traits, err := app.GetTraits(workloadName)
	if err != nil {
		return err
	}
	traits[traitType] = traitData
	comp[Traits] = traits
	app.Components[workloadName] = comp
	return app.Validate()
}

func (app *Application) RemoveTrait(workloadName, traitType string) error {
	if app == nil {
		return errors.New("app is nil pointer")
	}
	if app.Components == nil {
		app.Components = make(map[string]map[string]interface{})
	}
	comp := app.Components[workloadName]
	if comp == nil {
		comp = make(map[string]interface{})
	}
	traits, err := app.GetTraits(workloadName)
	if err != nil {
		return err
	}
	delete(traits, traitType)
	comp[Traits] = traits
	app.Components[workloadName] = comp
	return app.Validate()
}