diff options
-rw-r--r-- | docs/content/functions/float.md | 26 | ||||
-rw-r--r-- | tpl/cast/cast.go | 23 | ||||
-rw-r--r-- | tpl/cast/cast_test.go | 37 | ||||
-rw-r--r-- | tpl/cast/init.go | 7 |
4 files changed, 87 insertions, 6 deletions
diff --git a/docs/content/functions/float.md b/docs/content/functions/float.md new file mode 100644 index 000000000..2a5f7579c --- /dev/null +++ b/docs/content/functions/float.md @@ -0,0 +1,26 @@ +--- +title: float +linktitle: float +description: Creates a `float` from the argument passed into the function. +godocref: +date: 2017-09-28 +publishdate: 2017-09-28 +lastmod: 2017-09-28 +categories: [functions] +menu: + docs: + parent: "functions" +keywords: [strings,floats] +signature: ["float INPUT"] +workson: [] +hugoversion: +relatedfuncs: [] +deprecated: false +aliases: [] +--- + +Useful for turning strings into floating point numbers. + +``` +{{ float "1.23" }} → 1.23 +``` diff --git a/tpl/cast/cast.go b/tpl/cast/cast.go index 378467178..999b31214 100644 --- a/tpl/cast/cast.go +++ b/tpl/cast/cast.go @@ -30,6 +30,22 @@ type Namespace struct { // ToInt converts the given value to an int. func (ns *Namespace) ToInt(v interface{}) (int, error) { + v = convertTemplateToString(v) + return _cast.ToIntE(v) +} + +// ToString converts the given value to a string. +func (ns *Namespace) ToString(v interface{}) (string, error) { + return _cast.ToStringE(v) +} + +// ToFloat converts the given value to a float. +func (ns *Namespace) ToFloat(v interface{}) (float64, error) { + v = convertTemplateToString(v) + return _cast.ToFloat64E(v) +} + +func convertTemplateToString(v interface{}) interface{} { switch vv := v.(type) { case template.HTML: v = string(vv) @@ -42,10 +58,5 @@ func (ns *Namespace) ToInt(v interface{}) (int, error) { case template.JSStr: v = string(vv) } - return _cast.ToIntE(v) -} - -// ToString converts the given value to a string. -func (ns *Namespace) ToString(v interface{}) (string, error) { - return _cast.ToStringE(v) + return v } diff --git a/tpl/cast/cast_test.go b/tpl/cast/cast_test.go index a5e0db2af..fc20934f8 100644 --- a/tpl/cast/cast_test.go +++ b/tpl/cast/cast_test.go @@ -81,3 +81,40 @@ func TestToString(t *testing.T) { assert.Equal(t, test.expect, result, errMsg) } } + +func TestToFloat(t *testing.T) { + t.Parallel() + + ns := New() + + for i, test := range []struct { + v interface{} + expect interface{} + }{ + {"1", 1.0}, + {template.HTML("2"), 2.0}, + {template.CSS("3"), 3.0}, + {template.HTMLAttr("4"), 4.0}, + {template.JS("-5.67"), -5.67}, + {template.JSStr("6"), 6.0}, + {"1.23", 1.23}, + {"-1.23", -1.23}, + {"0", 0.0}, + {float64(2.12), 2.12}, + {int64(123), 123.0}, + {2, 2.0}, + {t, false}, + } { + errMsg := fmt.Sprintf("[%d] %v", i, test.v) + + result, err := ns.ToFloat(test.v) + + if b, ok := test.expect.(bool); ok && !b { + require.Error(t, err, errMsg) + continue + } + + require.NoError(t, err, errMsg) + assert.Equal(t, test.expect, result, errMsg) + } +} diff --git a/tpl/cast/init.go b/tpl/cast/init.go index d1547310c..3aee6f036 100644 --- a/tpl/cast/init.go +++ b/tpl/cast/init.go @@ -43,6 +43,13 @@ func init() { }, ) + ns.AddMethodMapping(ctx.ToFloat, + []string{"float"}, + [][2]string{ + {`{{ "1234" | float | printf "%T" }}`, `float64`}, + }, + ) + return ns } |