env.go

package generators

import (
	"os"

	stringsutil "github.com/projectdiscovery/utils/strings"
)

var envVars map[string]interface{}

func parseEnvVars() map[string]interface{} {
	sliceEnvVars := os.Environ()
	parsedEnvVars := make(map[string]interface{}, len(sliceEnvVars))
	for _, envVar := range sliceEnvVars {
		key, _ := stringsutil.Before(envVar, "=")
		val, _ := stringsutil.After(envVar, "=")
		parsedEnvVars[key] = val
	}
	return parsedEnvVars
}

// EnvVars returns a map with all environment variables into a map
func EnvVars() map[string]interface{} {
	if envVars == nil {
		envVars = parseEnvVars()
	}

	return envVars
}