aerc/lib/format/format.go

388 lines
9.1 KiB
Go

package format
import (
"errors"
"mime"
gomail "net/mail"
"strings"
"time"
"unicode"
"git.sr.ht/~sircmpwn/aerc/models"
"github.com/emersion/go-message"
)
func ParseAddress(address string) (*models.Address, error) {
addrs, err := gomail.ParseAddress(address)
if err != nil {
return nil, err
}
return (*models.Address)(addrs), nil
}
func ParseAddressList(s string) ([]*models.Address, error) {
if len(s) == 0 {
// workaround for go versions < 1.15
// 1.15 returns an empty list if "" is provided as input, prior versions
// return an error which is not what we want
return nil, nil
}
parser := gomail.AddressParser{
&mime.WordDecoder{message.CharsetReader},
}
list, err := parser.ParseList(s)
if err != nil {
return nil, err
}
addrs := make([]*models.Address, len(list))
for i, a := range list {
addrs[i] = (*models.Address)(a)
}
return addrs, nil
}
func FormatAddresses(l []*models.Address) string {
formatted := make([]string, len(l))
for i, a := range l {
formatted[i] = a.Format()
}
return strings.Join(formatted, ", ")
}
func ParseMessageFormat(
fromAddress string,
format string, timestampformat string,
accountName string, number int, msg *models.MessageInfo,
marked bool) (string,
[]interface{}, error) {
retval := make([]byte, 0, len(format))
var args []interface{}
accountFromAddress, err := ParseAddress(fromAddress)
if err != nil {
return "", nil, err
}
var c rune
for i, ni := 0, 0; i < len(format); {
ni = strings.IndexByte(format[i:], '%')
if ni < 0 {
ni = len(format)
retval = append(retval, []byte(format[i:ni])...)
break
}
ni += i + 1
// Check for fmt flags
if ni == len(format) {
goto handle_end_error
}
c = rune(format[ni])
if c == '+' || c == '-' || c == '#' || c == ' ' || c == '0' {
ni++
}
// Check for precision and width
if ni == len(format) {
goto handle_end_error
}
c = rune(format[ni])
for unicode.IsDigit(c) {
ni++
c = rune(format[ni])
}
if c == '.' {
ni++
c = rune(format[ni])
for unicode.IsDigit(c) {
ni++
c = rune(format[ni])
}
}
retval = append(retval, []byte(format[i:ni])...)
// Get final format verb
if ni == len(format) {
goto handle_end_error
}
c = rune(format[ni])
switch c {
case '%':
retval = append(retval, '%')
case 'a':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender")
}
addr := msg.Envelope.From[0]
retval = append(retval, 's')
args = append(args, addr.Address)
case 'A':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
var addr *models.Address
if len(msg.Envelope.ReplyTo) == 0 {
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender or reply-to")
} else {
addr = msg.Envelope.From[0]
}
} else {
addr = msg.Envelope.ReplyTo[0]
}
retval = append(retval, 's')
args = append(args, addr.Address)
case 'C':
retval = append(retval, 'd')
args = append(args, number)
case 'd':
date := msg.Envelope.Date
if date.IsZero() {
date = msg.InternalDate
}
retval = append(retval, 's')
args = append(args,
dummyIfZeroDate(date.Local(), timestampformat))
case 'D':
date := msg.Envelope.Date
if date.IsZero() {
date = msg.InternalDate
}
retval = append(retval, 's')
args = append(args,
dummyIfZeroDate(date.Local(), timestampformat))
case 'f':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender")
}
addr := msg.Envelope.From[0].Format()
retval = append(retval, 's')
args = append(args, addr)
case 'F':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender")
}
addr := msg.Envelope.From[0]
var val string
if addr.Name == accountFromAddress.Name && len(msg.Envelope.To) != 0 {
addr = msg.Envelope.To[0]
}
if addr.Name != "" {
val = addr.Name
} else {
val = addr.Address
}
retval = append(retval, 's')
args = append(args, val)
case 'g':
retval = append(retval, 's')
args = append(args, strings.Join(msg.Labels, ", "))
case 'i':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
retval = append(retval, 's')
args = append(args, msg.Envelope.MessageId)
case 'n':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender")
}
addr := msg.Envelope.From[0]
var val string
if addr.Name != "" {
val = addr.Name
} else {
val = addr.Address
}
retval = append(retval, 's')
args = append(args, val)
case 'r':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
addrs := FormatAddresses(msg.Envelope.To)
retval = append(retval, 's')
args = append(args, addrs)
case 'R':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
addrs := FormatAddresses(msg.Envelope.Cc)
retval = append(retval, 's')
args = append(args, addrs)
case 's':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
retval = append(retval, 's')
args = append(args, msg.Envelope.Subject)
case 't':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.To) == 0 {
return "", nil,
errors.New("found no address for recipient")
}
addr := msg.Envelope.To[0]
retval = append(retval, 's')
args = append(args, addr.Address)
case 'T':
retval = append(retval, 's')
args = append(args, accountName)
case 'u':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender")
}
addr := msg.Envelope.From[0]
mailbox := addr.Address // fallback if there's no @ sign
if split := strings.SplitN(addr.Address, "@", 2); len(split) == 2 {
mailbox = split[1]
}
retval = append(retval, 's')
args = append(args, mailbox)
case 'v':
if msg.Envelope == nil {
return "", nil,
errors.New("no envelope available for this message")
}
if len(msg.Envelope.From) == 0 {
return "", nil,
errors.New("found no address for sender")
}
addr := msg.Envelope.From[0]
// check if message is from current user
if addr.Name != "" {
retval = append(retval, 's')
args = append(args,
strings.Split(addr.Name, " ")[0])
}
case 'Z':
// calculate all flags
var readReplyFlag = ""
var delFlag = ""
var flaggedFlag = ""
var markedFlag = ""
seen := false
recent := false
answered := false
for _, flag := range msg.Flags {
if flag == models.SeenFlag {
seen = true
} else if flag == models.RecentFlag {
recent = true
} else if flag == models.AnsweredFlag {
answered = true
}
if flag == models.DeletedFlag {
delFlag = "D"
// TODO: check if attachments
}
if flag == models.FlaggedFlag {
flaggedFlag = "!"
}
// TODO: check gpg stuff
}
if seen {
if answered {
readReplyFlag = "r" // message has been replied to
}
} else {
if recent {
readReplyFlag = "N" // message is new
} else {
readReplyFlag = "O" // message is old
}
}
if marked {
markedFlag = "*"
}
retval = append(retval, '4', 's')
args = append(args, readReplyFlag+delFlag+flaggedFlag+markedFlag)
// Move the below cases to proper alphabetical positions once
// implemented
case 'l':
// TODO: number of lines in the message
retval = append(retval, 'd')
args = append(args, msg.Size)
case 'e':
// TODO: current message number in thread
fallthrough
case 'E':
// TODO: number of messages in current thread
fallthrough
case 'H':
// TODO: spam attribute(s) of this message
fallthrough
case 'L':
// TODO:
fallthrough
case 'X':
// TODO: number of attachments
fallthrough
case 'y':
// TODO: X-Label field
fallthrough
case 'Y':
// TODO: X-Label field and some other constraints
fallthrough
default:
// Just ignore it and print as is
// so %k in index format becomes %%k to Printf
retval = append(retval, '%')
retval = append(retval, byte(c))
}
i = ni + 1
}
return string(retval), args, nil
handle_end_error:
return "", nil,
errors.New("reached end of string while parsing message format")
}
func dummyIfZeroDate(date time.Time, format string) string {
if date.IsZero() {
return strings.Repeat("?", len(format))
}
return date.Format(format)
}