package transform

import "github.com/drone/drone/yaml"

// PluginDisable is a transform function that alters the Yaml configuration to
// disables plugins. This is intended for use when executing the pipeline
// locally on your own computer.
func PluginDisable(conf *yaml.Config, local bool) error {
	for _, container := range conf.Pipeline {
		if local && !container.Constraints.Local.Bool() {
			container.Disabled = true
		}

		if isClone(container) {
			container.Disabled = true
			continue
		}
	}
	return nil
}

// PluginParams is a transform function that alters the Yaml configuration to
// include plugin vargs parameters as environment variables.
func PluginParams(conf *yaml.Config) error {
	for _, container := range conf.Pipeline {
		if len(container.Vargs) == 0 {
			continue
		}
		if container.Environment == nil {
			container.Environment = map[string]string{}
		}
		err := argsToEnv(container.Vargs, container.Environment)
		if err != nil {
			return err
		}
	}
	return nil
}