init: pristine aerc 0.20.0 source

commands/msg/invite.go

@@ -0,0 +1,180 @@
+package msg
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"time"
+
+	"git.sr.ht/~rjarry/aerc/app"
+	"git.sr.ht/~rjarry/aerc/commands"
+	"git.sr.ht/~rjarry/aerc/config"
+	"git.sr.ht/~rjarry/aerc/lib"
+	"git.sr.ht/~rjarry/aerc/lib/calendar"
+	"git.sr.ht/~rjarry/aerc/lib/format"
+	"git.sr.ht/~rjarry/aerc/lib/log"
+	"git.sr.ht/~rjarry/aerc/models"
+	"github.com/emersion/go-message/mail"
+)
+
+type invite struct {
+	Edit       bool `opt:"-e" desc:"Force [compose].edit-headers = true."`
+	NoEdit     bool `opt:"-E" desc:"Force [compose].edit-headers = false."`
+	SkipEditor bool `opt:"-s" desc:"Skip the editor and go directly to the review screen."`
+}
+
+func init() {
+	commands.Register(invite{})
+}
+
+func (invite) Description() string {
+	return "Accept or decline a meeting invitation."
+}
+
+func (invite) Context() commands.CommandContext {
+	return commands.MESSAGE_LIST | commands.MESSAGE_VIEWER
+}
+
+func (invite) Aliases() []string {
+	return []string{"accept", "accept-tentative", "decline"}
+}
+
+func (i invite) Execute(args []string) error {
+	acct := app.SelectedAccount()
+	if acct == nil {
+		return errors.New("no account selected")
+	}
+	store := acct.Store()
+	if store == nil {
+		return errors.New("cannot perform action: messages still loading")
+	}
+	msg, err := acct.SelectedMessage()
+	if err != nil {
+		return err
+	}
+
+	part := lib.FindCalendartext(msg.BodyStructure, nil)
+	if part == nil {
+		return fmt.Errorf("no invitation found (missing text/calendar)")
+	}
+
+	editHeaders := (config.Compose.EditHeaders || i.Edit) && !i.NoEdit
+
+	subject := trimLocalizedRe(msg.Envelope.Subject, acct.AccountConfig().LocalizedRe)
+	switch args[0] {
+	case "accept":
+		subject = "Accepted: " + subject
+	case "accept-tentative":
+		subject = "Tentatively Accepted: " + subject
+	case "decline":
+		subject = "Declined: " + subject
+	default:
+		return fmt.Errorf("no participation status defined")
+	}
+
+	from := chooseFromAddr(acct.AccountConfig(), msg)
+
+	var to []*mail.Address
+
+	if len(msg.Envelope.ReplyTo) != 0 {
+		to = msg.Envelope.ReplyTo
+	} else {
+		to = msg.Envelope.From
+	}
+
+	if !config.Compose.ReplyToSelf {
+		for i, v := range to {
+			if v.Address == from.Address {
+				to = append(to[:i], to[i+1:]...)
+				break
+			}
+		}
+		if len(to) == 0 {
+			to = msg.Envelope.To
+		}
+	}
+
+	recSet := newAddrSet() // used for de-duping
+	recSet.AddList(to)
+
+	h := &mail.Header{}
+	h.SetAddressList("from", []*mail.Address{from})
+	h.SetSubject(subject)
+	h.SetMsgIDList("in-reply-to", []string{msg.Envelope.MessageId})
+	err = setReferencesHeader(h, msg.RFC822Headers)
+	if err != nil {
+		app.PushError(fmt.Sprintf("could not set references: %v", err))
+	}
+	original := models.OriginalMail{
+		From:          format.FormatAddresses(msg.Envelope.From),
+		Date:          msg.Envelope.Date,
+		RFC822Headers: msg.RFC822Headers,
+	}
+
+	handleInvite := func(reader io.Reader) (*calendar.Reply, error) {
+		cr, err := calendar.CreateReply(reader, from, args[0])
+		if err != nil {
+			return nil, err
+		}
+		for _, org := range cr.Organizers {
+			organizer, err := mail.ParseAddress(org)
+			if err != nil {
+				continue
+			}
+			if !recSet.Contains(organizer) {
+				to = append(to, organizer)
+			}
+		}
+		h.SetAddressList("to", to)
+		return cr, nil
+	}
+
+	addTab := func(cr *calendar.Reply) error {
+		composer, err := app.NewComposer(acct,
+			acct.AccountConfig(), acct.Worker(), editHeaders,
+			"", h, &original, cr.PlainText)
+		if err != nil {
+			app.PushError("Error: " + err.Error())
+			return err
+		}
+		err = composer.AppendPart(cr.MimeType, cr.Params, cr.CalendarText)
+		if err != nil {
+			return fmt.Errorf("failed to write invitation: %w", err)
+		}
+		if i.SkipEditor {
+			composer.Terminal().Close()
+		} else {
+			composer.FocusTerminal()
+		}
+
+		composer.Tab = app.NewTab(composer, subject)
+
+		composer.OnClose(func(c *app.Composer) {
+			switch {
+			case c.Sent() && c.Archive() != "":
+				store.Answered([]models.UID{msg.Uid}, true, nil)
+				err := archive([]*models.MessageInfo{msg}, nil, c.Archive())
+				if err != nil {
+					app.PushStatus("Archive failed", 10*time.Second)
+				}
+			case c.Sent():
+				store.Answered([]models.UID{msg.Uid}, true, nil)
+			}
+		})
+
+		return nil
+	}
+
+	store.FetchBodyPart(msg.Uid, part, func(reader io.Reader) {
+		if cr, err := handleInvite(reader); err != nil {
+			app.PushError(err.Error())
+			return
+		} else {
+			err := addTab(cr)
+			if err != nil {
+				log.Warnf("failed to add tab: %v", err)
+			}
+		}
+	})
+	return nil
+}