envfunc

var EnvBoolFunc = function.New(&function.Spec{
	Params: []function.Parameter{
		{
			Name:             "name",
			Type:             cty.String,
			AllowDynamicType: true,
		},
	},
	Type: function.StaticReturnType(cty.Bool),
	Impl: func(args []cty.Value, retType cty.Type) (cty.Value, error) {
		e := args[0].AsString()

		s, ok := boolEnvs[e]
		if !ok {
			return cty.Value{}, fmt.Errorf("env: %s does not register", e)
		}

		v := os.Getenv(e)
		if v == "" {
			s.v = s.d
			return cty.BoolVal(s.d), nil
		}

		var err error
		s.v, err = strconv.ParseBool(v)
		if err != nil {
			s.v = s.d
		}

		return cty.BoolVal(s.v), nil
	},
})

var EnvIntFunc = function.New(&function.Spec{
	Params: []function.Parameter{
		{
			Name:             "name",
			Type:             cty.String,
			AllowDynamicType: true,
		},
	},
	Type: function.StaticReturnType(cty.Number),
	Impl: func(args []cty.Value, retType cty.Type) (cty.Value, error) {
		e := args[0].AsString()

		s, ok := intEnvs[e]
		if !ok {
			return cty.Value{}, fmt.Errorf("env: %s does not register", e)
		}

		v := os.Getenv(e)
		if v == "" {
			s.v = s.d
			return cty.NumberIntVal(int64(s.d)), nil
		}

		i, err := strconv.ParseInt(v, 10, 64)
		s.v = int(i)
		return cty.NumberIntVal(i), err
	},
})

var EnvStringFunc = function.New(&function.Spec{
	Params: []function.Parameter{
		{
			Name:             "name",
			Type:             cty.String,
			AllowDynamicType: true,
		},
	},
	Type: function.StaticReturnType(cty.String),
	Impl: func(args []cty.Value, retType cty.Type) (cty.Value, error) {
		e := args[0].AsString()

		s, ok := stringEnvs[e]
		if !ok {
			return cty.Value{}, fmt.Errorf("env: %s does not register", e)
		}

		v := os.Getenv(e)
		if v == "" {
			v = s.d
		}
		s.v = v

		return cty.StringVal(v), nil
	},
})