package config import ( "errors" "fmt" "io/ioutil" "net/url" "os" "os/exec" "path" "regexp" "strings" "unicode" "github.com/gdamore/tcell" "github.com/go-ini/ini" "github.com/kyoh86/xdg" ) type UIConfig struct { IndexFormat string `ini:"index-format"` TimestampFormat string `ini:"timestamp-format"` ShowHeaders []string `delim:","` RenderAccountTabs string `ini:"render-account-tabs"` SidebarWidth int `ini:"sidebar-width"` PreviewHeight int `ini:"preview-height"` EmptyMessage string `ini:"empty-message"` } const ( FILTER_MIMETYPE = iota FILTER_HEADER ) type AccountConfig struct { CopyTo string Default string From string Name string Source string SourceCredCmd string Folders []string Params map[string]string Outgoing string OutgoingCredCmd string } type BindingConfig struct { Global *KeyBindings AccountWizard *KeyBindings Compose *KeyBindings ComposeEditor *KeyBindings ComposeReview *KeyBindings MessageList *KeyBindings MessageView *KeyBindings Terminal *KeyBindings } type ComposeConfig struct { Editor string `ini:"editor"` } type FilterConfig struct { FilterType int Filter string Command string Header string Regex *regexp.Regexp } type ViewerConfig struct { Pager string Alternatives []string } type AercConfig struct { Bindings BindingConfig Compose ComposeConfig Ini *ini.File `ini:"-"` Accounts []AccountConfig `ini:"-"` Filters []FilterConfig `ini:"-"` Viewer ViewerConfig `ini:"-"` Ui UIConfig } // Input: TimestampFormat // Output: timestamp-format func mapName(raw string) string { newstr := make([]rune, 0, len(raw)) for i, chr := range raw { if isUpper := 'A' <= chr && chr <= 'Z'; isUpper { if i > 0 { newstr = append(newstr, '-') } } newstr = append(newstr, unicode.ToLower(chr)) } return string(newstr) } func loadAccountConfig(path string) ([]AccountConfig, error) { file, err := ini.Load(path) if err != nil { // No config triggers account configuration wizard return nil, nil } file.NameMapper = mapName var accounts []AccountConfig for _, _sec := range file.SectionStrings() { if _sec == "DEFAULT" { continue } sec := file.Section(_sec) account := AccountConfig{ Default: "INBOX", Name: _sec, Params: make(map[string]string), } if err = sec.MapTo(&account); err != nil { return nil, err } for key, val := range sec.KeysHash() { if key == "folders" { account.Folders = strings.Split(val, ",") } else if key == "source-cred-cmd" { account.SourceCredCmd = val } else if key == "outgoing" { account.Outgoing = val } else if key == "outgoing-cred-cmd" { account.OutgoingCredCmd = val } else if key == "from" { account.From = val } else if key == "copy-to" { account.CopyTo = val } else if key != "name" { account.Params[key] = val } } if account.Source == "" { return nil, fmt.Errorf("Expected source for account %s", _sec) } source, err := parseCredential(account.Source, account.SourceCredCmd) if err != nil { return nil, fmt.Errorf("Invalid source credentials for %s: %s", _sec, err) } account.Source = source outgoing, err := parseCredential(account.Outgoing, account.OutgoingCredCmd) if err != nil { return nil, fmt.Errorf("Invalid outgoing credentials for %s: %s", _sec, err) } account.Outgoing = outgoing accounts = append(accounts, account) } return accounts, nil } func parseCredential(cred, command string) (string, error) { if cred == "" || command == "" { return cred, nil } u, err := url.Parse(cred) if err != nil { return "", err } // ignore the command if a password is specified if _, exists := u.User.Password(); exists { return cred, nil } // don't attempt to parse the command if the url is a path (ie /usr/bin/sendmail) if !u.IsAbs() { return cred, nil } cmd := exec.Command("sh", "-c", command) output, err := cmd.Output() if err != nil { return "", fmt.Errorf("failed to read password: %s", err) } pw := strings.TrimSpace(string(output)) u.User = url.UserPassword(u.User.Username(), pw) return u.String(), nil } func installTemplate(root, sharedir, name string) error { if _, err := os.Stat(root); os.IsNotExist(err) { err := os.MkdirAll(root, 0755) if err != nil { return err } } data, err := ioutil.ReadFile(path.Join(sharedir, name)) if err != nil { return err } err = ioutil.WriteFile(path.Join(root, name), data, 0644) if err != nil { return err } return nil } func LoadConfig(root *string, sharedir string) (*AercConfig, error) { if root == nil { _root := path.Join(xdg.ConfigHome(), "aerc") root = &_root } filename := path.Join(*root, "accounts.conf") if err := checkConfigPerms(filename); err != nil { return nil, err } filename = path.Join(*root, "aerc.conf") file, err := ini.Load(filename) if err != nil { if err := installTemplate(*root, sharedir, "aerc.conf"); err != nil { return nil, err } if file, err = ini.Load(filename); err != nil { return nil, err } } file.NameMapper = mapName config := &AercConfig{ Bindings: BindingConfig{ Global: NewKeyBindings(), AccountWizard: NewKeyBindings(), Compose: NewKeyBindings(), ComposeEditor: NewKeyBindings(), ComposeReview: NewKeyBindings(), MessageList: NewKeyBindings(), MessageView: NewKeyBindings(), Terminal: NewKeyBindings(), }, Ini: file, Ui: UIConfig{ IndexFormat: "%4C %Z %D %-17.17n %s", TimestampFormat: "%F %l:%M %p", ShowHeaders: []string{ "From", "To", "Cc", "Bcc", "Subject", "Date", }, RenderAccountTabs: "auto", SidebarWidth: 20, PreviewHeight: 12, EmptyMessage: "(no messages)", }, } // These bindings are not configurable config.Bindings.AccountWizard.ExKey = KeyStroke{ Key: tcell.KeyCtrlE, } quit, _ := ParseBinding("", ":quit") config.Bindings.AccountWizard.Add(quit) if filters, err := file.GetSection("filters"); err == nil { // TODO: Parse the filter more finely, e.g. parse the regex for _, match := range filters.KeyStrings() { cmd := filters.KeysHash()[match] filter := FilterConfig{ Command: cmd, Filter: match, } if strings.Contains(match, ",~") { filter.FilterType = FILTER_HEADER header := filter.Filter[:strings.Index(filter.Filter, ",")] regex := filter.Filter[strings.Index(filter.Filter, "~")+1:] filter.Header = strings.ToLower(header) filter.Regex, err = regexp.Compile(regex) if err != nil { panic(err) } } else if strings.ContainsRune(match, ',') { filter.FilterType = FILTER_HEADER header := filter.Filter[:strings.Index(filter.Filter, ",")] value := filter.Filter[strings.Index(filter.Filter, ",")+1:] filter.Header = strings.ToLower(header) filter.Regex, err = regexp.Compile(regexp.QuoteMeta(value)) } else { filter.FilterType = FILTER_MIMETYPE } config.Filters = append(config.Filters, filter) } } if viewer, err := file.GetSection("viewer"); err == nil { if err := viewer.MapTo(&config.Viewer); err != nil { return nil, err } for key, val := range viewer.KeysHash() { switch key { case "alternatives": config.Viewer.Alternatives = strings.Split(val, ",") } } } if compose, err := file.GetSection("compose"); err == nil { if err := compose.MapTo(&config.Compose); err != nil { return nil, err } } if ui, err := file.GetSection("ui"); err == nil { if err := ui.MapTo(&config.Ui); err != nil { return nil, err } } accountsPath := path.Join(*root, "accounts.conf") if accounts, err := loadAccountConfig(accountsPath); err != nil { return nil, err } else { config.Accounts = accounts } filename = path.Join(*root, "binds.conf") binds, err := ini.Load(filename) if err != nil { if err := installTemplate(*root, sharedir, "binds.conf"); err != nil { return nil, err } if binds, err = ini.Lo