package templates
import (
"bufio"
"bytes"
"fmt"
"os"
"os/exec"
"strings"
"text/template"
"time"
"git.sr.ht/~rjarry/aerc/lib/format"
"git.sr.ht/~rjarry/aerc/lib/parse"
"git.sr.ht/~rjarry/aerc/models"
"github.com/emersion/go-message/mail"
)
var version string
// SetVersion initializes the aerc version displayed in template functions
func SetVersion(v string) {
version = v
}
// wrap allows to chain wrapText
func wrap(lineWidth int, text string) string {
return wrapText(text, lineWidth)
}
func wrapLine(text string, lineWidth int) string {
words := strings.Fields(text)
if len(words) == 0 {
return text
}
var wrapped strings.Builder
wrapped.WriteString(words[0])
spaceLeft := lineWidth - wrapped.Len()
for _, word := range words[1:] {
if len(word)+1 > spaceLeft {
wrapped.WriteRune('\n')
wrapped.WriteString(word)
spaceLeft = lineWidth - len(word)
} else {
wrapped.WriteRune(' ')
wrapped.WriteString(word)
spaceLeft -= 1 + len(word)
}
}
return wrapped.String()
}
func wrapText(text string, lineWidth int) string {
text = strings.ReplaceAll(text, "\r\n", "\n")
text = strings.TrimRight(text, "\n")
lines := strings.Split(text, "\n")
var wrapped strings.Builder
for _, line := range lines {
switch {
case line == "":
// deliberately left blank
case line[0] == '>':
// leave quoted text alone
wrapped.WriteString(line)
default:
wrapped.WriteString(wrapLine(line, lineWidth))
}
wrapped.WriteRune('\n')
}
return wrapped.String()
}
// quote prepends "> " in front of every line in text
func quote(text string) string {
text = strings.ReplaceAll(text, "\r\n", "\n")
text = strings.TrimRight(text, "\n")
lines := strings.Split(text, "\n")
var quoted strings.Builder
for _, line := range lines {
if line == "" {
quoted.WriteString(">\n")
continue
}
quoted.WriteString("> ")
quoted.WriteString(line)
quoted.WriteRune('\n')
}
return quoted.String()
}
// cmd allow to parse reply by shell command
// text have to be passed by cmd param
// if there is error, original string is returned
func cmd(cmd, text string) string {
var out bytes.Buffer
c := exec.Command("sh", "-c", cmd)
c.Stdin = strings.NewReader(text)
c.Stdout = &out
err := c.Run()
if err != nil {
return text
}
return out.String()
}
func toLocal(t time.Time) time.Time {
return time.Time.In(t, time.Local)
}
func names(addresses []*mail.Address) []string {
n := make([]string, len(addresses))
for i, addr := range addresses {
name := addr.Name
if name == "" {
parts := strings.SplitN(addr.Address, "@", 2)
name = parts[0]
}
n[i] = name
}
return n
}
func firstnames(addresses []*mail.Address) []string {
n := make([]string, len(addresses))
for i, addr := range addresses {
var name string
if addr.Name == "" {
parts := strings.SplitN(addr.Address, "@", 2)
parts = strings.SplitN(parts[0], ".", 2)
name = parts[0]
} else {
parts := strings.SplitN(addr.Name, " ", 2)
name = parts[0]
}
n[i] = name
}
return n
}
func initials(addresses []*mail.Address) []string {
n := names(addresses)
ret := make([]string, len(addresses))
for i, name := range n {
split := strings.Split(name, " ")
initial := ""
for _, s := range split {
initial += string([]rune(s)[0:1])
}
ret[i] = initial
}
return ret
}
func emails(addresses []*mail.Address) []string {
e := make([]string, len(addresses))
for i, addr := range addresses {
e[i] = addr.Address
}
return e
}
func mboxes(addresses []*mail.Address) []string {
e := make([]string, len(addresses))
for i, addr := range addresses {
parts := strings.SplitN(addr.Address, "@", 2)
e[i] = parts[0]
}
return e
}
func shortmboxes(addresses []*mail.Address) []string {
e := make([]string, len(addresses))
for i, addr := range addresses {
parts := strings.SplitN(addr.Address, "@", 2)
parts = strings.SplitN(parts[0], ".", 2)
e[i] = parts[0]
}
return e
}
func persons(addresses []*mail.Address) []string {
e := make([]string, len(addresses))
for i, addr := range addresses {
e[i] = format.AddressForHumans(addr)
}
return e
}
var units = []string{"K", "M", "G", "T"}
func humanReadable(value int) string {
sign := ""
if value < 0 {
sign = "-"
value = -value
}
if value < 1000 {
return fmt.Sprintf("%s%d", sign, value)
}
val := float64(value)
unit := ""
for i := 0; val >= 1000 && i < len(units); i++ {
unit = units[i]
val /= 1000.0
}
if val < 100.0 {
return fmt.Sprintf("%s%.1f%s", sign, val, unit)
}
return fmt.Sprintf("%s%.0f%s", sign, val, unit)
}
func cwd() string {
path, err := os.Getwd()
if err != nil {
return err.Error()
}
home, err := os.UserHomeDir()
if err != nil {
return err.Error()
}
if strings.HasPrefix(path, home) {
path = strings.Replace(path, home, "~", 1)
}
return path
}
func join(sep string, elems []string) string {
return strings.Join(elems, sep)
}
// removes a signature from the piped in message
func trimSignature(message string) string {
var res strings.Builder
input := bufio.NewScanner(strings.NewReader(message))
for input.Scan() {
line := input.Text()
if line == "-- " {
break
}
res.WriteString(line)
res.WriteRune('\n')
}
return res.String()
}
func compactDir(path string) string {
return format.CompactPath(path, os.PathSeparator)
}
type (
Case struct{ expr, value string }
Default struct{ value string }
)
func (c *Case) Matches(s string) bool { return parse.MatchCache(s, c.expr) }
func (c *Case) Value() string { return c.value }
func (d *Default) Matches(s string) bool { return true }
func (d *Default) Value() string { return d.value }
func switch_(value string, cases ...models.Case) string {
for _, c := range cases {
if c.Matches(value) {
return c.Value()
}
}
return ""
}
func case_(expr, value string) models.Case {
return &Case{expr: expr, value: value}
}
func default_(value string) models.Case {
return &Default{value: value}
}
var templateFuncs = template.FuncMap{
"quote": quote,
"wrapText": wrapText,
"wrap": wrap,
"now": time.Now,
"dateFormat": time.Time.Format,
"toLocal": toLocal,
"exec": cmd,
"version": func() string { return version },
"names": names,
"firstnames": firstnames,
"initials": initials,
"emails": emails,
"mboxes": mboxes,
"shortmboxes": shortmboxes,
"persons": persons,
"humanReadable": humanReadable,
"cwd": cwd,
"join": join,
"trimSignature": trimSignature,
"compactDir": compactDir,
"match": parse.MatchCache,
"switch": switch_,
"case": case_,
"default": default_,
}
