diff options
Diffstat (limited to 'pkg')
| -rw-r--r-- | pkg/template/template.go | 17 | ||||
| -rw-r--r-- | pkg/tool/tool.go | 8 |
2 files changed, 17 insertions, 8 deletions
diff --git a/pkg/template/template.go b/pkg/template/template.go index f93aca74..dc633885 100644 --- a/pkg/template/template.go +++ b/pkg/template/template.go @@ -60,14 +60,15 @@ func NewFuncMap() []template.FuncMap { "LoadTimes": func(startTime time.Time) string { return fmt.Sprint(time.Since(startTime).Nanoseconds()/1e6) + "ms" }, - "AvatarLink": tool.AvatarLink, - "Safe": Safe, - "Sanitize": bluemonday.UGCPolicy().Sanitize, - "Str2html": Str2html, - "TimeSince": tool.TimeSince, - "RawTimeSince": tool.RawTimeSince, - "FileSize": tool.FileSize, - "Subtract": tool.Subtract, + "AvatarLink": tool.AvatarLink, + "AppendAvatarSize": tool.AppendAvatarSize, + "Safe": Safe, + "Sanitize": bluemonday.UGCPolicy().Sanitize, + "Str2html": Str2html, + "TimeSince": tool.TimeSince, + "RawTimeSince": tool.RawTimeSince, + "FileSize": tool.FileSize, + "Subtract": tool.Subtract, "Add": func(a, b int) int { return a + b }, diff --git a/pkg/tool/tool.go b/pkg/tool/tool.go index 43b4b1d8..333790fe 100644 --- a/pkg/tool/tool.go +++ b/pkg/tool/tool.go @@ -207,6 +207,14 @@ func AvatarLink(email string) (url string) { return url } +// AppendAvatarSize appends avatar size query parameter to the URL in the correct format. +func AppendAvatarSize(url string, size int) string { + if strings.Contains(url, "?") { + return url + "&s=" + com.ToStr(size) + } + return url + "?s=" + com.ToStr(size) +} + // Seconds-based time units const ( Minute = 60 |