122 lines
2.6 KiB
Go
122 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"math"
|
|
"net/url"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
|
|
"github.com/dustin/go-humanize"
|
|
)
|
|
|
|
var (
|
|
compactNumUnits = []string{"", "k", "M"}
|
|
|
|
tplFuncMap = template.FuncMap{
|
|
// The name "title" is what the function will be called in the template text.
|
|
"color": func(num int) string {
|
|
return string(runeIrcColor) + strconv.Itoa(num)
|
|
},
|
|
"bcolor": func(fgNum, bgNum int) string {
|
|
return string(runeIrcColor) + strconv.Itoa(fgNum) + "," + strconv.Itoa(bgNum)
|
|
},
|
|
"bold": func() string {
|
|
return string(runeIrcBold)
|
|
},
|
|
"italic": func() string {
|
|
return string(runeIrcItalic)
|
|
},
|
|
"reset": func() string {
|
|
return string(runeIrcReset) + string(runeIrcColor)
|
|
},
|
|
"reverse": func() string {
|
|
return string(runeIrcReverse)
|
|
},
|
|
"underline": func() string {
|
|
return string(runeIrcUnderline)
|
|
},
|
|
"urlencode": func(s string) string {
|
|
return url.QueryEscape(s)
|
|
},
|
|
"yesno": func(yes string, no string, value bool) string {
|
|
if value {
|
|
return yes
|
|
}
|
|
|
|
return no
|
|
},
|
|
"excerpt": func(maxLength uint16, text string) string {
|
|
if len(text) > int(maxLength) {
|
|
return text[0:maxLength-1] + "\u2026"
|
|
}
|
|
return text
|
|
},
|
|
"comma": func(num uint64) string {
|
|
return humanize.Comma(int64(num))
|
|
},
|
|
"compactnum": func(num uint64) string {
|
|
// 1 => 0
|
|
// 1000 => 1
|
|
// 1000000 => 2
|
|
log10 := math.Floor(math.Log10(float64(num)) / 3)
|
|
|
|
// Cut to available units
|
|
cut := int(math.Min(float64(len(compactNumUnits)-1), log10))
|
|
|
|
numf := float64(num)
|
|
numf /= math.Pow10(cut * 3)
|
|
|
|
// Rounding
|
|
numf = math.Floor((numf*10)+.5) / 10
|
|
if numf >= 1000 {
|
|
numf /= 1000
|
|
if cut < len(compactNumUnits)-1 {
|
|
cut++
|
|
}
|
|
}
|
|
|
|
unit := compactNumUnits[cut]
|
|
f := "%.1f%s"
|
|
if numf-math.Floor(numf) < 0.05 {
|
|
f = "%.0f%s"
|
|
}
|
|
|
|
return fmt.Sprintf(f, numf, unit)
|
|
},
|
|
"ago": func(t time.Time) string {
|
|
return humanize.Time(t)
|
|
},
|
|
"size": func(s uint64) string {
|
|
return humanize.Bytes(s)
|
|
},
|
|
"playtime": func(s time.Duration) string {
|
|
return fmt.Sprintf("%d", int(math.Floor(s.Minutes()))) + "'"
|
|
},
|
|
}
|
|
|
|
ircTpl = template.Must(
|
|
template.New("").
|
|
Funcs(tplFuncMap).
|
|
ParseGlob("*.tpl"))
|
|
|
|
rxInsignificantWhitespace = regexp.MustCompile(`\s+`)
|
|
)
|
|
|
|
func tplString(name string, data interface{}) (string, error) {
|
|
w := new(bytes.Buffer)
|
|
if err := ircTpl.ExecuteTemplate(w, name, data); err != nil {
|
|
return "", err
|
|
}
|
|
s := w.String()
|
|
s = rxInsignificantWhitespace.ReplaceAllString(s, " ")
|
|
s = strings.Trim(s, " ")
|
|
log.Printf("tplString(%v): %s", name, s)
|
|
return s, nil
|
|
}
|