repo_description: Declaratively deploy your Kubernetes manifests, Kustomize configs, and Charts as Helm releases. Generate all-in-one manifests for use with ArgoCD.
repo_topics: helm kubernetes kustomize
code: package argparser
  
  import (
  	"fmt"
  	"strings"
  )
  
  type keyVal struct {
  	key       string
  	val       string
  	spaceFlag bool
  }
  type argMap struct {
  	m     map[string][]*keyVal
  	flags []string
  }
  
  // isNewFlag checks if the given arg is a new flag
  func isNewFlag(flag string) bool {
  	return strings.HasPrefix(flag, "--") || strings.HasPrefix(flag, "-")
  }
  
  // removeEmptyArgs removes empty args from the given args
  func removeEmptyArgs(args []string) []string {
  	var newArgs []string
  	for _, arg := range args {
  		if len(arg) > 0 {
  			newArgs = append(newArgs, arg)
  		}
  	}
  	return newArgs
  }
  
  // SetArg sets a flag and value in the map
  func (a *argMap) SetArg(flag, arg string, isSpace bool) {
  	// if flag is empty, return
  	if len(flag) == 0 {
  		return
  	}
  	if _, exists := a.m[flag]; !exists {
  		keyarg := &keyVal{key: flag, val: arg, spaceFlag: isSpace}
  		a.m[flag] = append(a.m[flag], keyarg)
  		a.flags = append(a.flags, flag)
  	} else {
  		keyarg := &keyVal{key: flag, val: arg, spaceFlag: isSpace}
  		a.m[flag] = append(a.m[flag], keyarg)
  	}
  }
  
  // newArgMap creates a new argMap
  func newArgMap() *argMap {
  	return &argMap{m: map[string][]*keyVal{}}
  }
  
  func analyzeArgs(am *argMap, args string) {
  	argsVals := removeEmptyArgs(strings.Split(args, " "))
  	for index, arg := range argsVals {
  		if len(arg) == 0 {
  			continue
  		}
  		if isNewFlag(arg) {
  			argVal := strings.SplitN(arg, "=", 2)
  			if len(argVal) > 1 {
  				arg := argVal[0]
  				value := argVal[1]
  				am.SetArg(arg, value, false)
  			} else {
  				// check if next value is arg to flag
  				if index+1 < len(argsVals) {
  					nextVal := argsVals[index+1]
  					if isNewFlag(nextVal) {
  						am.SetArg(arg, "", false)
  					} else {
  						am.SetArg(arg, nextVal, true)
  					}
  				} else {
  					am.SetArg(arg, "", false)
  				}
  			}
  		}
  	}
  }
  
  func CollectArgs(args string) []string {
  	argsMap := newArgMap()
  	analyzeArgs(argsMap, args)
  	var argArr []string
  	for _, flag := range argsMap.flags {
  		val := argsMap.m[flag]
  
  		for _, obj := range val {
  			if obj.val != "" {
  				if obj.spaceFlag {
  					argArr = append(argArr, obj.key, obj.val)
  				} else {
  					argArr = append(argArr, fmt.Sprintf("%s=%s", obj.key, obj.val))
  				}
  			} else {
  				argArr = append(argArr, obj.key)
  			}
  		}
  	}
  	return argArr
  }
  