Skip to content

Commit

Permalink
tpl/compare: Add cond (ternary) template func
Browse files Browse the repository at this point in the history
  • Loading branch information
bep committed Sep 8, 2017
1 parent 202510f commit 754854c
Show file tree
Hide file tree
Showing 3 changed files with 25 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tpl/compare/compare.go
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,15 @@ func (n *Namespace) Lt(a, b interface{}) bool {
return left < right
}

// Conditional can be used as a ternary operator.
// It returns a if condition, else b.
func (n *Namespace) Conditional(condition bool, a, b interface{}) interface{} {
if condition {
return a
}
return b
}

func (*Namespace) compareGetFloat(a interface{}, b interface{}) (float64, float64) {
var left, right float64
var leftStr, rightStr *string
Expand Down
9 changes: 9 additions & 0 deletions tpl/compare/compare_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -221,3 +221,12 @@ func TestTimeUnix(t *testing.T) {
toTimeUnix(iv)
}(t)
}

func TestConditional(t *testing.T) {
assert := require.New(t)
n := New()
a, b := "a", "b"

assert.Equal(a, n.Conditional(true, a, b))
assert.Equal(b, n.Conditional(false, a, b))
}
7 changes: 7 additions & 0 deletions tpl/compare/init.go
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,13 @@ func init() {
[][2]string{},
)

ns.AddMethodMapping(ctx.Conditional,
[]string{"cond"},
[][2]string{
{`{{ cond (eq (add 2 2) 4) "2+2 is 4" "what?" | safeHTML }}`, `2+2 is 4`},
},
)

return ns

}
Expand Down

0 comments on commit 754854c

Please sign in to comment.