automail/commands.go

package main

import (
	"encoding/json"
	"io"

	"github.com/emersion/go-imap"
	"github.com/emersion/go-imap/client"
	"github.com/jhillyerd/enmime"
	"github.com/pkg/errors"
	log "github.com/sirupsen/logrus"
)

type command interface {
	Execute(*client.Client, *imap.Message, *enmime.Envelope, io.Writer) error
}

type commandTypeWrap struct {
	Type string `json:"type"`
}

func (c commandTypeWrap) rewrap(data []byte) (command, error) {
	var out command

	switch c.Type {

	case "move":
		out = new(commandMove)

	case "add_flags":
		out = new(commandAddFlags)

	case "del_flags":
		out = new(commandDelFlags)

	case "get_attachment":
		out = new(commandGetAttachment)

	case "set_flags":
		out = new(commandSetFlags)

	default:
		return nil, errors.New("Command not found")

	}

	return out, errors.Wrap(json.Unmarshal(data, out), "Unable to unmarshal into command")
}

type commandMove struct {
	ToMailbox string `json:"to_mailbox"`
}

func (c commandMove) Execute(client *client.Client, msg *imap.Message, mail *enmime.Envelope, stdin io.Writer) error {
	s := &imap.SeqSet{}
	s.AddNum(msg.Uid)

	if err := client.UidCopy(s, c.ToMailbox); err != nil {
		return errors.Wrap(err, "Unable to copy to target mailbox")
	}

	return errors.Wrap(
		client.UidStore(s, imap.FormatFlagsOp(imap.AddFlags, true), []interface{}{imap.DeletedFlag}, nil),
		"Unable to set deleted flag in original mailbox",
	)
}

type commandAddFlags struct {
	Flags []string `json:"flags"`
}

func (c commandAddFlags) Execute(client *client.Client, msg *imap.Message, mail *enmime.Envelope, stdin io.Writer) error {
	var (
		flags []interface{}
		s     = &imap.SeqSet{}
	)
	s.AddNum(msg.Uid)

	for _, f := range c.Flags {
		flags = append(flags, f)
	}

	return errors.Wrap(
		client.UidStore(s, imap.FormatFlagsOp(imap.AddFlags, true), flags, nil),
		"Unable to add flags",
	)
}

type commandDelFlags struct {
	Flags []string `json:"flags"`
}

func (c commandDelFlags) Execute(client *client.Client, msg *imap.Message, mail *enmime.Envelope, stdin io.Writer) error {
	var (
		flags []interface{}
		s     = &imap.SeqSet{}
	)
	s.AddNum(msg.Uid)

	for _, f := range c.Flags {
		flags = append(flags, f)
	}

	return errors.Wrap(
		client.UidStore(s, imap.FormatFlagsOp(imap.RemoveFlags, true), flags, nil),
		"Unable to remove flags",
	)
}

type commandGetAttachment struct {
	Filename string `json:"filename"`
}

func (c commandGetAttachment) Execute(client *client.Client, msg *imap.Message, mail *enmime.Envelope, stdin io.Writer) error {
	a := attachmentFromMail(mail, c.Filename)
	if a == nil {
		log.WithFields(log.Fields{
			"uid":      msg.Uid,
			"filename": c.Filename,
		}).Error("Requested attachment not found")
		return errors.New("Attachment not found")
	}

	return errors.Wrap(json.NewEncoder(stdin).Encode(a), "Unable to encode attachment to JSON")
}

type commandSetFlags struct {
	Flags []string `json:"flags"`
}

func (c commandSetFlags) Execute(client *client.Client, msg *imap.Message, mail *enmime.Envelope, stdin io.Writer) error {
	var (
		flags []interface{}
		s     = &imap.SeqSet{}
	)
	s.AddNum(msg.Uid)

	for _, f := range c.Flags {
		flags = append(flags, f)
	}

	return errors.Wrap(
		client.UidStore(s, imap.FormatFlagsOp(imap.SetFlags, true), flags, nil),
		"Unable to set flags",
	)
}