init: pristine aerc 0.20.0 source

config/columns.go

@@ -0,0 +1,118 @@
+package config
+
+import (
+	"bytes"
+	"fmt"
+	"regexp"
+	"strconv"
+	"strings"
+	"text/template"
+
+	"git.sr.ht/~rjarry/aerc/lib/templates"
+	"github.com/go-ini/ini"
+)
+
+type ColumnFlags uint32
+
+func (f ColumnFlags) Has(o ColumnFlags) bool { return f&o == o }
+
+const (
+	ALIGN_LEFT ColumnFlags = 1 << iota
+	ALIGN_CENTER
+	ALIGN_RIGHT
+	WIDTH_AUTO     // whatever is left
+	WIDTH_FRACTION // ratio of total width
+	WIDTH_EXACT    // exact number of characters
+	WIDTH_FIT      // fit to column content width
+)
+
+type ColumnDef struct {
+	Name     string
+	Flags    ColumnFlags
+	Width    float64
+	Template *template.Template
+}
+
+var columnRe = regexp.MustCompile(`^([\w-]+)(?:([<:>])(=|\*|\d+%?)?)?$`)
+
+func parseColumnDef(col string, section *ini.Section) (*ColumnDef, error) {
+	col = strings.TrimSpace(col)
+	match := columnRe.FindStringSubmatch(col)
+	if match == nil {
+		return nil, fmt.Errorf("invalid column def: %v", col)
+	}
+	name := match[1]
+	keyName := fmt.Sprintf("column-%s", name)
+
+	var flags ColumnFlags
+	switch match[2] {
+	case "<", "":
+		flags |= ALIGN_LEFT
+	case ":":
+		flags |= ALIGN_CENTER
+	case ">":
+		flags |= ALIGN_RIGHT
+	}
+
+	var width float64 = 0
+	switch match[3] {
+	case "=":
+		flags |= WIDTH_FIT
+	case "*", "":
+		flags |= WIDTH_AUTO
+	default:
+		s := match[3]
+		var divider float64 = 1
+		if strings.HasSuffix(s, "%") {
+			divider = 100
+			s = strings.TrimSuffix(s, "%")
+			flags |= WIDTH_FRACTION
+		} else {
+			flags |= WIDTH_EXACT
+		}
+		w, err := strconv.ParseFloat(s, 64)
+		if err != nil {
+			return nil, fmt.Errorf("%s: %w", keyName, err)
+		}
+		if divider == 100 && w > 100 {
+			return nil, fmt.Errorf("%s: invalid width %.0f%%", keyName, w)
+		}
+		width = w / divider
+	}
+	key, err := section.GetKey(keyName)
+	if err != nil {
+		return nil, err
+	}
+
+	t, err := templates.ParseTemplate(keyName, key.String())
+	if err != nil {
+		return nil, err
+	}
+
+	err = templates.Render(t, &bytes.Buffer{}, &dummyData{})
+	if err != nil {
+		return nil, err
+	}
+
+	return &ColumnDef{
+		Name:     name,
+		Flags:    flags,
+		Width:    width,
+		Template: t,
+	}, nil
+}
+
+func ParseColumnDefs(key *ini.Key, section *ini.Section) ([]*ColumnDef, error) {
+	var columns []*ColumnDef
+	for _, col := range key.Strings(",") {
+		c, err := parseColumnDef(col, section)
+		if err != nil {
+			return nil, err
+		}
+		columns = append(columns, c)
+	}
+	if len(columns) == 0 {
+		return nil, nil
+	}
+	return columns, nil
+}