github/hugo
1
0
Fork 0
mirror of https://github.com/gohugoio/hugo.git synced 2025-02-17 05:11:47 +00:00
Code Issues Projects Releases Packages Wiki Activity

tpl: Make the safeHTML and cousins accept more than strings

Browse source
This commit is contained in:
Bjørn Erik Pedersen 2016-03-03 20:30:18 +01:00
parent 70739c972e
commit 3a2a4c3b07
2 changed files with 19 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tpl/template_funcs.go
View file

@ -1459,25 +1459,25 @@ func index(item interface{}, indices ...interface{}) (interface{}, error) {
// //
// safeHTMLAttr is currently disabled, pending further discussion // safeHTMLAttr is currently disabled, pending further discussion
// on its use case. 2015-01-19 // on its use case. 2015-01-19
func safeHTMLAttr(text string) template.HTMLAttr { func safeHTMLAttr(a interface{}) template.HTMLAttr {
return template.HTMLAttr(text) return template.HTMLAttr(cast.ToString(a))
} }
// safeCSS returns a given string as html/template CSS content. // safeCSS returns a given string as html/template CSS content.
func safeCSS(text string) template.CSS { func safeCSS(a interface{}) template.CSS {
return template.CSS(text) return template.CSS(cast.ToString(a))
} }
// safeURL returns a given string as html/template URL content. // safeURL returns a given string as html/template URL content.
func safeURL(text string) template.URL { func safeURL(a interface{}) template.URL {
return template.URL(text) return template.URL(cast.ToString(a))
} }
// safeHTML returns a given string as html/template HTML content. // safeHTML returns a given string as html/template HTML content.
func safeHTML(a string) template.HTML { return template.HTML(a) } func safeHTML(a interface{}) template.HTML { return template.HTML(cast.ToString(a)) }
// safeJS returns the given string as a html/template JS content. // safeJS returns the given string as a html/template JS content.
func safeJS(a string) template.JS { return template.JS(a) } func safeJS(a interface{}) template.JS { return template.JS(cast.ToString(a)) }
func doArithmetic(a, b interface{}, op rune) (interface{}, error) { func doArithmetic(a, b interface{}, op rune) (interface{}, error) {
av := reflect.ValueOf(a) av := reflect.ValueOf(a)

12
tpl/template_funcs_test.go
View file

@ -103,6 +103,11 @@ jsonify: {{ (slice "A" "B" "C") | jsonify }}
md5: {{ md5 "Hello world, gophers!" }} md5: {{ md5 "Hello world, gophers!" }}
sha1: {{ sha1 "Hello world, gophers!" }} sha1: {{ sha1 "Hello world, gophers!" }}
emojify: {{ "I :heart: Hugo" | emojify }} emojify: {{ "I :heart: Hugo" | emojify }}
safeHTML: {{ "Bat&Man" | safeHTML }}
safeHTML: {{ "Bat&Man" | safeHTML | safeHTML }}
safeCSS: {{ "Bat&Man" | safeCSS | safeCSS }}
safeURL: {{ "http://gohugo.io" | safeURL | safeURL }}
safeJS: {{ "(1*2)" | safeJS | safeJS }}
` `
expected := `chomp: <p>Blockhead</p> expected := `chomp: <p>Blockhead</p>
dateFormat: Wednesday, Jan 21, 2015 dateFormat: Wednesday, Jan 21, 2015
@ -141,6 +146,11 @@ jsonify: ["A","B","C"]
md5: b3029f756f98f79e7f1b7f1d1f0dd53b md5: b3029f756f98f79e7f1b7f1d1f0dd53b
sha1: c8b5b0e33d408246e30f53e32b8f7627a7a649d4 sha1: c8b5b0e33d408246e30f53e32b8f7627a7a649d4
emojify: I ❤️ Hugo emojify: I ❤️ Hugo
safeHTML: Bat&Man
safeHTML: Bat&Man
safeCSS: Bat&amp;Man
safeURL: http://gohugo.io
safeJS: (1*2)
` `
var b bytes.Buffer var b bytes.Buffer
@ -166,7 +176,7 @@ emojify: I ❤️ Hugo
} }
if b.String() != expected { if b.String() != expected {
t.Errorf("Got\n%q\nExpected\n>%q<", b.String(), expected) t.Errorf("Got\n%q\nExpected\n%q", b.String(), expected)
} }
} }