commands: implement a no-quit mode

Add a mode that prevents aerc from quitting normally when an important
task is performed, i.e. when sending a message. The no-quit mode will be
ignored when quit is used with the -f option to force an exit.

Suggested-by: ph14nix[m]
Signed-off-by: Koni Marti <koni.marti@gmail.com>
Acked-by: Robin Jarry <robin@jarry.cc>

1 files changed, 20 insertions, 3 deletions

diff --git a/commands/quit.go b/commands/quit.go
index fbec2b7d..ee5f46c1 100644
--- a/commands/quit.go
+++ b/commands/quit.go
@@ -2,8 +2,11 @@ package commands
 
 import (
 	"errors"
+	"fmt"
 
+	"git.sr.ht/~rjarry/aerc/commands/mode"
 	"git.sr.ht/~rjarry/aerc/widgets"
+	"git.sr.ht/~sircmpwn/getopt"
 )
 
 type Quit struct{}
@@ -27,8 +30,22 @@ func (err ErrorExit) Error() string {
 }
 
 func (Quit) Execute(aerc *widgets.Aerc, args []string) error {
-	if len(args) != 1 {
-		return errors.New("Usage: quit")
+	force := false
+	opts, optind, err := getopt.Getopts(args, "f")
+	if err != nil {
+		return err
 	}
-	return ErrorExit(1)
+	for _, opt := range opts {
+		switch opt.Option {
+		case 'f':
+			force = true
+		}
+	}
+	if len(args) != optind {
+		return errors.New("Usage: quit [-f]")
+	}
+	if force || mode.QuitAllowed() {
+		return ErrorExit(1)
+	}
+	return fmt.Errorf("A task is not done yet. Use -f to force an exit.")
 }