aerc/aerc.go

162 lines
3.2 KiB
Go
Raw Normal View History

2018-01-10 00:30:46 +01:00
package main
import (
"fmt"
2018-02-01 03:18:21 +01:00
"io"
"io/ioutil"
"log"
"os"
2018-01-10 14:35:26 +01:00
"time"
2018-01-10 01:18:19 +01:00
"git.sr.ht/~sircmpwn/getopt"
2018-02-01 03:18:21 +01:00
"github.com/mattn/go-isatty"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/commands"
"git.sr.ht/~sircmpwn/aerc/commands/account"
"git.sr.ht/~sircmpwn/aerc/commands/compose"
"git.sr.ht/~sircmpwn/aerc/commands/msg"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/commands/msgview"
"git.sr.ht/~sircmpwn/aerc/commands/terminal"
"git.sr.ht/~sircmpwn/aerc/config"
libui "git.sr.ht/~sircmpwn/aerc/lib/ui"
"git.sr.ht/~sircmpwn/aerc/widgets"
2018-01-10 00:30:46 +01:00
)
2019-03-21 21:30:23 +01:00
func getCommands(selected libui.Drawable) []*commands.Commands {
switch selected.(type) {
case *widgets.AccountView:
return []*commands.Commands{
account.AccountCommands,
msg.MessageCommands,
2019-03-21 21:32:22 +01:00
commands.GlobalCommands,
}
2019-05-12 06:06:09 +02:00
case *widgets.Composer:
return []*commands.Commands{
compose.ComposeCommands,
2019-05-12 06:06:09 +02:00
commands.GlobalCommands,
}
2019-03-31 03:45:41 +02:00
case *widgets.MessageViewer:
return []*commands.Commands{
msgview.MessageViewCommands,
msg.MessageCommands,
2019-03-31 03:45:41 +02:00
commands.GlobalCommands,
}
2019-03-30 19:12:04 +01:00
case *widgets.Terminal:
2019-03-21 21:32:22 +01:00
return []*commands.Commands{
terminal.TerminalCommands,
commands.GlobalCommands,
2019-03-21 21:30:23 +01:00
}
default:
return []*commands.Commands{commands.GlobalCommands}
}
}
func execCommand(aerc *widgets.Aerc, ui *libui.UI, cmd string) error {
cmds := getCommands((*aerc).SelectedTab())
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 {
cmds := getCommands((*aerc).SelectedTab())
completions := make([]string, 0)
for _, set := range cmds {
opts := set.GetCompletions(aerc, cmd)
if len(opts) > 0 {
for _, opt := range opts {
completions = append(completions, opt)
}
}
}
return completions
}
var (
Prefix string
ShareDir string
Version string
)
2019-07-16 12:09:25 +02:00
func usage() {
log.Fatal("Usage: aerc [-v]")
}
2018-01-10 00:30:46 +01:00
func main() {
// TODO: Support starting with mailto links, ad-hoc accounts, etc
opts, optind, err := getopt.Getopts(os.Args, "v")
if err != nil {
2019-07-16 12:09:25 +02:00
log.Print(err)
usage()
}
for _, opt := range opts {
switch opt.Option {
case 'v':
fmt.Println("aerc " + Version)
return
}
}
if optind != len(os.Args) {
2019-07-16 12:09:25 +02:00
usage()
}
var (
logOut io.Writer
logger *log.Logger
)
2018-02-01 03:18:21 +01:00
if !isatty.IsTerminal(os.Stdout.Fd()) {
logOut = os.Stdout
} else {
logOut = ioutil.Discard
}
2018-02-02 00:59:13 +01:00
logger = log.New(logOut, "", log.LstdFlags)
2018-02-01 03:18:21 +01:00
logger.Println("Starting up aerc")
conf, err := config.LoadConfigFromFile(nil, ShareDir)
2018-01-10 17:19:45 +01:00
if err != nil {
fmt.Printf("Failed to load config: %v\n", err)
os.Exit(1)
2018-01-10 01:18:19 +01:00
}
2019-03-16 01:32:09 +01:00
var (
aerc *widgets.Aerc
ui *libui.UI
)
2019-03-11 02:15:24 +01:00
aerc = widgets.NewAerc(conf, logger, func(cmd string) error {
return execCommand(aerc, ui, cmd)
}, func(cmd string) []string {
return getCompletions(aerc, cmd)
})
2019-03-11 02:15:24 +01:00
2019-03-16 01:32:09 +01:00
ui, err = libui.Initialize(conf, aerc)
2018-01-11 04:41:15 +01:00
if err != nil {
panic(err)
}
2018-02-27 04:54:39 +01:00
defer ui.Close()
for !ui.ShouldExit() {
for aerc.Tick() {
// Continue updating our internal state
}
2018-02-27 04:54:39 +01:00
if !ui.Tick() {
// ~60 FPS
time.Sleep(16 * time.Millisecond)
2018-01-10 14:35:26 +01:00
}
}
2018-01-10 00:30:46 +01:00
}