1 files changed, 73 insertions, 0 deletions
diff --git a/commands/compose/multipart.go b/commands/compose/multipart.go
new file mode 100644
index 00000000..5a6dd770
--- /dev/null
+++ b/commands/compose/multipart.go
@@ -0,0 +1,73 @@
+package compose
+
+import (
+	"bytes"
+	"fmt"
+
+	"git.sr.ht/~rjarry/aerc/commands"
+	"git.sr.ht/~rjarry/aerc/widgets"
+	"git.sr.ht/~sircmpwn/getopt"
+)
+
+type Multipart struct{}
+
+func init() {
+	register(Multipart{})
+}
+
+func (Multipart) Aliases() []string {
+	return []string{"multipart"}
+}
+
+func (Multipart) Complete(aerc *widgets.Aerc, args []string) []string {
+	var completions []string
+	completions = append(completions, "-d")
+	for mime := range aerc.Config().Converters {
+		completions = append(completions, mime)
+	}
+	return commands.CompletionFromList(aerc, completions, args)
+}
+
+func (a Multipart) Execute(aerc *widgets.Aerc, args []string) error {
+	composer, ok := aerc.SelectedTabContent().(*widgets.Composer)
+	if !ok {
+		return fmt.Errorf(":multipart is only available on the compose::review screen")
+	}
+
+	opts, optind, err := getopt.Getopts(args, "d")
+	if err != nil {
+		return fmt.Errorf("Usage: :multipart [-d] <mime/type>")
+	}
+	var remove bool = false
+	for _, opt := range opts {
+		if opt.Option == 'd' {
+			remove = true
+		}
+	}
+	args = args[optind:]
+	if len(args) != 1 {
+		return fmt.Errorf("Usage: :multipart [-d] <mime/type>")
+	}
+	mime := args[0]
+
+	if remove {
+		return composer.RemovePart(mime)
+	} else {
+		_, found := aerc.Config().Converters[mime]
+		if !found {
+			return fmt.Errorf("no command defined for MIME type: %s", mime)
+		}
+		err = composer.AppendPart(
+			mime,
+			map[string]string{"Charset": "UTF-8"},
+			// the actual content of the part will be rendered
+			// every time the body of the email is updated
+			bytes.NewReader([]byte{}),
+		)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}