package env

import (
	"encoding/json"
	"log"
	"os"
	"strconv"
)

func String(key string) string {
	v := os.Getenv(key)
	if v == "" {
		log.Fatalln(key + " is missing")
	}
	return v
}

func StringOptional(key string) string {
	return os.Getenv(key)
}

func Uint64(key string) uint64 {
	v := String(key)
	n, err := strconv.ParseUint(v, 10, 64)
	if err != nil {
		log.Fatalln(key+" has a wrong value. ", err)
	}
	return n
}

func Uint16(key string) uint16 {
	v := String(key)
	n, err := strconv.ParseUint(v, 10, 16)
	if err != nil {
		log.Fatalln(key+" has a wrong value. ", err)
	}
	return uint16(n)
}

const MAX_INT = uint64(^uint(0) >> 1)

func Int(key string) int {
	val := Uint64(key)
	if val > MAX_INT {
		log.Fatalln(key + " is too big. ")
	}
	return int(val)
}

func Bool(key string) bool {
	v := String(key)
	if v != "true" && v != "false" {
		log.Fatalln(key + " has wrong value. Accepted only true or false")
	}
	if v == "true" {
		return true
	}
	return false
}

func StringMapOptional(key string) map[string]string {
	v := StringOptional(key)
	if v == "" {
		return nil
	}
	var stringMap map[string]string

	if err := json.Unmarshal([]byte(v), &stringMap); err != nil {
		log.Fatalln(key + ": wrong json format. Expected string map")
	}
	return stringMap
}