aerc/aerc.go

228 lines
4.8 KiB
Go
Raw Normal View History

2018-01-09 23:30:46 +00:00
package main
import (
"bytes"
"fmt"
2018-02-01 02:18:21 +00:00
"os"
"sort"
2018-01-10 13:35:26 +00:00
"time"
2018-01-10 00:18:19 +00:00
"git.sr.ht/~sircmpwn/getopt"
2018-02-01 02:18:21 +00:00
"github.com/mattn/go-isatty"
"github.com/xo/terminfo"
2018-02-01 02:18:21 +00:00
"git.sr.ht/~rjarry/aerc/commands"
"git.sr.ht/~rjarry/aerc/commands/account"
"git.sr.ht/~rjarry/aerc/commands/compose"
"git.sr.ht/~rjarry/aerc/commands/msg"
"git.sr.ht/~rjarry/aerc/commands/msgview"
"git.sr.ht/~rjarry/aerc/commands/terminal"
"git.sr.ht/~rjarry/aerc/config"
"git.sr.ht/~rjarry/aerc/lib"
"git.sr.ht/~rjarry/aerc/lib/crypto"
"git.sr.ht/~rjarry/aerc/lib/templates"
libui "git.sr.ht/~rjarry/aerc/lib/ui"
"git.sr.ht/~rjarry/aerc/logging"
"git.sr.ht/~rjarry/aerc/widgets"
2018-01-09 23:30:46 +00:00
)
2019-03-21 20:30:23 +00:00
func getCommands(selected libui.Drawable) []*commands.Commands {
switch selected.(type) {
case *widgets.AccountView:
return []*commands.Commands{
account.AccountCommands,
msg.MessageCommands,
2019-03-21 20:32:22 +00:00
commands.GlobalCommands,
}
2019-05-12 04:06:09 +00:00
case *widgets.Composer:
return []*commands.Commands{
compose.ComposeCommands,
2019-05-12 04:06:09 +00:00
commands.GlobalCommands,
}
2019-03-31 01:45:41 +00:00
case *widgets.MessageViewer:
return []*commands.Commands{
msgview.MessageViewCommands,
msg.MessageCommands,
2019-03-31 01:45:41 +00:00
commands.GlobalCommands,
}
2019-03-30 18:12:04 +00:00
case *widgets.Terminal:
2019-03-21 20:32:22 +00:00
return []*commands.Commands{
terminal.TerminalCommands,
commands.GlobalCommands,
2019-03-21 20:30:23 +00:00
}
default:
return []*commands.Commands{commands.GlobalCommands}
}
}
func execCommand(aerc *widgets.Aerc, ui *libui.UI, cmd []string) error {
cmds := getCommands(aerc.SelectedTabContent())
for i, set := range cmds {
err := set.ExecuteCommand(aerc, cmd)
if _, ok := err.(commands.NoSuchCommand); ok {
if i == len(cmds)-1 {
return err
}
continue
} else if _, ok := err.(commands.ErrorExit); ok {
ui.Exit()
return nil
} else if err != nil {
return err
} else {
break
}
}
return nil
}
func getCompletions(aerc *widgets.Aerc, cmd string) []string {
2019-09-05 02:30:49 +00:00
var completions []string
for _, set := range getCommands(aerc.SelectedTabContent()) {
2019-09-05 02:30:49 +00:00
completions = append(completions, set.GetCompletions(aerc, cmd)...)
}
sort.Strings(completions)
return completions
}
// set at build time
var Version string
func usage(msg string) {
fmt.Fprintln(os.Stderr, msg)
fmt.Fprintln(os.Stderr, "usage: aerc [-v] [mailto:...]")
os.Exit(1)
2019-07-16 10:09:25 +00:00
}
func setWindowTitle() {
logging.Debugf("Parsing terminfo")
ti, err := terminfo.LoadFromEnv()
if err != nil {
logging.Warnf("Cannot get terminfo: %v", err)
return
}
if !ti.Has(terminfo.HasStatusLine) {
logging.Infof("Terminal does not have status line support")
return
}
logging.Infof("Setting terminal title")
buf := new(bytes.Buffer)
ti.Fprintf(buf, terminfo.ToStatusLine)
fmt.Fprint(buf, "aerc")
ti.Fprintf(buf, terminfo.FromStatusLine)
os.Stderr.Write(buf.Bytes())
}
2018-01-09 23:30:46 +00:00
func main() {
defer logging.PanicHandler()
opts, optind, err := getopt.Getopts(os.Args, "v")
if err != nil {
usage("error: " + err.Error())
return
}
for _, opt := range opts {
switch opt.Option {
case 'v':
fmt.Println("aerc " + Version)
return
}
}
retryExec := false
args := os.Args[optind:]
if len(args) > 1 {
usage("error: invalid arguments")
return
} else if len(args) == 1 {
arg := args[0]
err := lib.ConnectAndExec(arg)
if err == nil {
return // other aerc instance takes over
}
fmt.Fprintf(os.Stderr, "Failed to communicate to aerc: %v\n", err)
// continue with setting up a new aerc instance and retry after init
retryExec = true
}
2018-02-01 02:18:21 +00:00
if !isatty.IsTerminal(os.Stdout.Fd()) {
logging.Init()
2018-02-01 02:18:21 +00:00
}
logging.Infof("Starting up")
2018-02-01 02:18:21 +00:00
conf, err := config.LoadConfigFromFile(nil)
2018-01-10 16:19:45 +00:00
if err != nil {
fmt.Fprintf(os.Stderr, "Failed to load config: %v\n", err)
os.Exit(1)
2018-01-10 00:18:19 +00:00
}
2019-03-16 00:32:09 +00:00
var (
aerc *widgets.Aerc
ui *libui.UI
)
deferLoop := make(chan struct{})
c := crypto.New(conf.General.PgpProvider)
c.Init()
defer c.Close()
aerc = widgets.NewAerc(conf, c, func(cmd []string) error {
return execCommand(aerc, ui, cmd)
}, func(cmd string) []string {
return getCompletions(aerc, cmd)
}, &commands.CmdHistory, deferLoop)
2019-03-11 01:15:24 +00:00
ui, err = libui.Initialize(aerc)
2018-01-11 03:41:15 +00:00
if err != nil {
panic(err)
}
2018-02-27 03:54:39 +00:00
defer ui.Close()
logging.UICleanup = func() {
ui.Close()
}
close(deferLoop)
if conf.Ui.MouseEnabled {
ui.EnableMouse()
}
as, err := lib.StartServer()
if err != nil {
logging.Warnf("Failed to start Unix server: %v", err)
} else {
defer as.Close()
as.OnMailto = aerc.Mailto
as.OnMbox = aerc.Mbox
}
2020-05-02 12:06:02 +00:00
// set the aerc version so that we can use it in the template funcs
templates.SetVersion(Version)
if retryExec {
// retry execution
arg := args[0]
err := lib.ConnectAndExec(arg)
if err != nil {
fmt.Fprintf(os.Stderr, "Failed to communicate to aerc: %v\n", err)
aerc.CloseBackends()
return
}
}
if isatty.IsTerminal(os.Stderr.Fd()) {
setWindowTitle()
}
for !ui.ShouldExit() {
for aerc.Tick() {
// Continue updating our internal state
}
2018-02-27 03:54:39 +00:00
if !ui.Tick() {
// ~60 FPS
time.Sleep(16 * time.Millisecond)
2018-01-10 13:35:26 +00:00
}
}
aerc.CloseBackends()
2018-01-09 23:30:46 +00:00
}