Compare commits

..

No commits in common. "fcb2552e73245ce4d214c4644a95198bbea6a512" and "8dfd5378c087967dd66bbe3434a26251b213ba7a" have entirely different histories.

2 changed files with 15 additions and 179 deletions

View File

@ -22,7 +22,6 @@ type cValue struct {
durval time.Duration
boolval bool
strval string
mapval map[string]cValue
err error
}

View File

@ -15,14 +15,10 @@ type cEntry struct {
dtype DataType
unset bool
empty bool
defval string // The default value
hasdef bool // Default value is defined
}
type Config struct {
parsed bool
env map[string]cEntry
mapEnv map[string]map[string]cEntry
}
// NewConfig returns an envconf.Config that is used to read configuration from environment variables.
@ -32,21 +28,10 @@ func NewConfig() *Config {
config := new(Config)
config.parsed = false
config.env = make(map[string]cEntry)
config.mapEnv = make(map[string]map[string]cEntry)
for _, v := range os.Environ() {
splitted := strings.SplitN(v, "=", 2)
if len(splitted) == 2 {
key := cleanKey(splitted[0])
keysplit := strings.Split(key, "_")
left := ""
right := ""
if len(keysplit) > 1 {
left = strings.Join(keysplit[:len(keysplit)-1], "_")
right = keysplit[len(keysplit)-1]
}
key := strings.TrimSpace(strings.ToUpper(splitted[0]))
if unicode.IsLetter(getFirstRune(key)) {
var entry cEntry
entry.value = splitted[1]
@ -54,14 +39,13 @@ func NewConfig() *Config {
entry.unset = false
entry.empty = false
config.env[key] = entry
config.mapEnv[left][right] = entry
}
}
}
return config
}
// Define the type of an environment variable.
// Define defines the type of an environment variable.
// Variables without a defined type will be ignored by Parse.
func (c *Config) Define(key string, dtype DataType) {
upper := strings.ToUpper(key)
@ -78,20 +62,7 @@ func (c *Config) Define(key string, dtype DataType) {
}
}
// Define the type of an environment variable.
// Variables without a defined type will be ignored by Parse.
func (c *Config) DefineMap(key string, dtype DataType) {
upper := strings.ToUpper(key)
entries, ok := c.mapEnv[upper]
if ok {
for mapKey, entry := range entries {
entry.dtype = dtype
c.mapEnv[key][mapKey] = entry
}
}
}
// Define the type and default value of an environment variable.
// DefineDefault defines the type and default value of an environment variable.
// Variables without a defined type will be ignored by Parse.
func (c *Config) DefineDefault(key string, val string, dtype DataType) {
upper := strings.ToUpper(key)
@ -102,8 +73,6 @@ func (c *Config) DefineDefault(key string, val string, dtype DataType) {
}
entry.dtype = dtype
entry.empty = false
entry.defval = val
entry.hasdef = true
c.env[upper] = entry
} else {
var entry cEntry
@ -111,8 +80,6 @@ func (c *Config) DefineDefault(key string, val string, dtype DataType) {
entry.unset = true
entry.empty = false
entry.value = val
entry.defval = val
entry.hasdef = true
c.env[upper] = entry
}
}
@ -135,14 +102,6 @@ func (c *Config) Parse() {
c.env[k] = v
}
}
for k, v := range c.mapEnv {
for mk, mv := range v {
mv.parsed = mv.dtype.parse(k+"_"+mk, mv.value)
c.mapEnv[k][mk] = mv
}
}
if failed {
for k, v := range c.env {
if (v.parsed.err == nil) && v.unset {
@ -158,65 +117,28 @@ func (c *Config) Parse() {
}
}
func (c *Config) help() {
max := make([]int, 2, 2)
preq := false
pdef := false
func (c *Config) Help() {
max := 0
for k, v := range c.env {
if v.dtype != TypeNone {
if len(k) > max[0] {
max[0] = len(k)
if len(k) > max {
max = len(k)
}
if len(v.dtype.String()) > max[1] {
max[1] = len(v.dtype.String())
if len(v.dtype.String()) > max {
max = len(v.dtype.String())
}
if v.hasdef {
pdef = true
} else {
preq = true
if len(v.value) > max {
max = len(v.value)
}
}
}
if pdef {
fmt.Println()
for k, v := range c.env {
if v.dtype != TypeNone {
if v.hasdef {
format := fmt.Sprintf("Variable %%-%ds| Type %%-%ds| Default %%s\n", max[0]+5, max[1]+3)
fmt.Printf(format, fmt.Sprintf(`"%s"`, k), v.dtype, fmt.Sprintf(`"%s"`, v.defval))
format := fmt.Sprintf("Variable %%%ds|Type %%%ds|Default %%s\n", max, max)
fmt.Printf(format, k, v.dtype, v.value)
}
}
}
}
if preq {
fmt.Println()
for k, v := range c.env {
if v.dtype != TypeNone {
if !v.hasdef {
format := fmt.Sprintf("Variable %%-%ds| Type %%-%ds| Required\n", max[0]+5, max[1]+3)
fmt.Printf(format, fmt.Sprintf(`"%s"`, k), v.dtype)
}
}
}
}
fmt.Println()
}
func (c *Config) StatusHelp() bool {
flags := make(map[string]bool)
flags["--help"] = true
flags["-help"] = true
flags["-h"] = true
if len(os.Args) > 1 {
key := strings.ToLower(strings.TrimSpace(os.Args[1]))
_, ok := flags[key]
if ok {
c.help()
return false
}
}
return c.Status()
}
// Status prints out failures that occured while parsing the environment to os.Stderr.
// Variables that have been defined without a default value and are
@ -234,17 +156,6 @@ func (c *Config) Status() (ok bool) {
}
}
}
for _, v := range c.mapEnv {
for _, mv := range v {
err := mv.parsed.err
if err != nil {
ok = false
if !mv.empty {
fmt.Fprintln(os.Stderr, err)
}
}
}
}
if !ok {
fmt.Fprintln(os.Stderr, "")
for _, v := range c.env {
@ -258,27 +169,6 @@ func (c *Config) Status() (ok bool) {
return
}
func cleanKey(str string) string {
fn := func(r rune) rune {
if (r >= '0' && r <= '9') || (r >= 'A' && r <= 'Z') || r == '_' {
return r
}
return -1
}
str = strings.Map(fn, strings.ToUpper(str))
oldlen := len(str) + 1
newlen := len(str)
for newlen < oldlen {
oldlen = newlen
str = strings.ReplaceAll(str, "__", "_")
newlen = len(str)
}
return str
}
func (c *Config) getRaw(key string, dtype DataType) (val cValue) {
val.dtype = TypeNone
if c.parsed {
@ -291,27 +181,6 @@ func (c *Config) getRaw(key string, dtype DataType) (val cValue) {
return
}
func (c *Config) getRawMap(key string, dtype DataType) (empty map[string]cValue) {
empty = make(map[string]cValue)
retval := make(map[string]cValue)
if c.parsed {
key = cleanKey(key)
entries, ok := c.mapEnv[key]
if ok {
for k, v := range entries {
if v.dtype == dtype {
retval[k] = v.parsed
} else {
return
}
}
return retval
}
}
return
}
// GetInt returns the value of an environment variable.
// If the variable is not defined as envconf.TypeInt the function will return 0.
func (c *Config) GetInt(key string) int64 {
@ -358,38 +227,6 @@ func (c *Config) GetBool(key string) bool {
return val.boolval
}
func (c *Config) GetMapInt(key string) (retval map[string]int64) {
retval = make(map[string]int64)
for k, v := range c.getRawMap(key, TypeInt) {
retval[k] = v.intval
}
return
}
func (c *Config) GetMapDuration(key string) (retval map[string]time.Duration) {
retval = make(map[string]time.Duration)
for k, v := range c.getRawMap(key, TypeDuration) {
retval[k] = v.durval
}
return
}
func (c *Config) GetMapString(key string) (retval map[string]string) {
retval = make(map[string]string)
for k, v := range c.getRawMap(key, TypeString) {
retval[k] = v.strval
}
return
}
func (c *Config) GetMapBool(key string) (retval map[string]bool) {
retval = make(map[string]bool)
for k, v := range c.getRawMap(key, TypeBool) {
retval[k] = v.boolval
}
return
}
func getFirstRune(str string) rune {
for _, v := range str {
return v