tools/mailer/mailer.go - master - githubmirror/pocketbase

tools/mailer/mailer.go
72 строки2 KB
Gani Georgiev
use gabriel-vasile/mimetype for the mail attachments
18 мар 2025, 01:52
18 мар 2025, 01:529f19460
package mailer

import (
	"bytes"
	"io"
	"net/mail"

	"github.com/gabriel-vasile/mimetype"
	"github.com/pocketbase/pocketbase/tools/hook"
)

// Message defines a generic email message struct.
type Message struct {
	From              mail.Address         `json:"from"`
	To                []mail.Address       `json:"to"`
	Bcc               []mail.Address       `json:"bcc"`
	Cc                []mail.Address       `json:"cc"`
	Subject           string               `json:"subject"`
	HTML              string               `json:"html"`
	Text              string               `json:"text"`
	Headers           map[string]string    `json:"headers"`
	Attachments       map[string]io.Reader `json:"attachments"`
	InlineAttachments map[string]io.Reader `json:"inlineAttachments"`
}

// Mailer defines a base mail client interface.
type Mailer interface {
	// Send sends an email with the provided Message.
	Send(message *Message) error
}

// SendInterceptor is optional interface for registering mail send hooks.
type SendInterceptor interface {
	OnSend() *hook.Hook[*SendEvent]
}

type SendEvent struct {
	hook.Event
	Message *Message
}

// addressesToStrings converts the provided address to a list of serialized RFC 5322 strings.
//
// To export only the email part of mail.Address, you can set withName to false.
func addressesToStrings(addresses []mail.Address, withName bool) []string {
	result := make([]string, len(addresses))

	for i, addr := range addresses {
		if withName && addr.Name != "" {
			result[i] = addr.String()
		} else {
			// keep only the email part to avoid wrapping in angle-brackets
			result[i] = addr.Address
		}
	}

	return result
}

// detectReaderMimeType reads the first couple bytes of the reader to detect its MIME type.
//
// Returns a new combined reader from the partial read + the remaining of the original reader.
func detectReaderMimeType(r io.Reader) (io.Reader, string, error) {
	readCopy := new(bytes.Buffer)

	mime, err := mimetype.DetectReader(io.TeeReader(r, readCopy))
	if err != nil {
		return nil, "", err
	}

	return io.MultiReader(readCopy, r), mime.String(), nil
}