0.29.1原版

2026-07-02 19:14:14 +08:00
commit d94008f0fb
947 changed files with 174905 additions and 0 deletions
@@ -0,0 +1,173 @@
+package email
+
+import (
+	"crypto/tls"
+	"net"
+	"net/smtp"
+	"time"
+
+	"github.com/pkg/errors"
+)
+
+const smtpOperationTimeout = 15 * time.Second
+
+// Client represents an SMTP email client.
+type Client struct {
+	config *Config
+}
+
+// NewClient creates a new email client with the given configuration.
+func NewClient(config *Config) *Client {
+	return &Client{
+		config: config,
+	}
+}
+
+// validateConfig validates the client configuration.
+func (c *Client) validateConfig() error {
+	if c.config == nil {
+		return errors.New("email configuration is required")
+	}
+	return c.config.Validate()
+}
+
+// createAuth creates an SMTP auth mechanism if credentials are provided.
+func (c *Client) createAuth() smtp.Auth {
+	if c.config.SMTPUsername == "" && c.config.SMTPPassword == "" {
+		return nil
+	}
+	return smtp.PlainAuth("", c.config.SMTPUsername, c.config.SMTPPassword, c.config.SMTPHost)
+}
+
+// createTLSConfig creates a TLS configuration for secure connections.
+func (c *Client) createTLSConfig() *tls.Config {
+	return &tls.Config{
+		ServerName: c.config.SMTPHost,
+		MinVersion: tls.VersionTLS12,
+	}
+}
+
+// Send sends an email message via SMTP.
+func (c *Client) Send(message *Message) error {
+	// Validate configuration
+	if err := c.validateConfig(); err != nil {
+		return errors.Wrap(err, "invalid email configuration")
+	}
+
+	// Validate message
+	if message == nil {
+		return errors.New("message is required")
+	}
+	if err := message.Validate(); err != nil {
+		return errors.Wrap(err, "invalid email message")
+	}
+
+	// Format the message
+	body := message.Format(c.config.FromEmail, c.config.FromName)
+
+	// Get all recipients
+	recipients := message.GetAllRecipients()
+
+	// Create auth
+	auth := c.createAuth()
+
+	// Send based on encryption type
+	if c.config.UseSSL {
+		return c.sendWithSSL(auth, recipients, body)
+	}
+	return c.sendWithTLS(auth, recipients, body)
+}
+
+// sendWithTLS sends email using STARTTLS (port 587).
+func (c *Client) sendWithTLS(auth smtp.Auth, recipients []string, body string) error {
+	serverAddr := c.config.GetServerAddress()
+
+	dialer := &net.Dialer{Timeout: smtpOperationTimeout}
+	conn, err := dialer.Dial("tcp", serverAddr)
+	if err != nil {
+		return errors.Wrapf(err, "failed to connect to SMTP server: %s", serverAddr)
+	}
+	defer conn.Close()
+	if err := conn.SetDeadline(time.Now().Add(smtpOperationTimeout)); err != nil {
+		return errors.Wrap(err, "failed to set SMTP connection deadline")
+	}
+
+	client, err := smtp.NewClient(conn, c.config.SMTPHost)
+	if err != nil {
+		return errors.Wrap(err, "failed to create SMTP client")
+	}
+	defer client.Quit()
+
+	if c.config.UseTLS {
+		if ok, _ := client.Extension("STARTTLS"); !ok {
+			return errors.New("SMTP server does not support STARTTLS")
+		}
+		if err := client.StartTLS(c.createTLSConfig()); err != nil {
+			return errors.Wrap(err, "failed to start SMTP STARTTLS")
+		}
+	}
+
+	return c.sendWithClient(client, auth, recipients, body)
+}
+
+// sendWithSSL sends email using SSL/TLS (port 465).
+func (c *Client) sendWithSSL(auth smtp.Auth, recipients []string, body string) error {
+	serverAddr := c.config.GetServerAddress()
+
+	// Create TLS connection
+	tlsConfig := c.createTLSConfig()
+	dialer := &net.Dialer{Timeout: smtpOperationTimeout}
+	conn, err := tls.DialWithDialer(dialer, "tcp", serverAddr, tlsConfig)
+	if err != nil {
+		return errors.Wrapf(err, "failed to connect to SMTP server with SSL: %s", serverAddr)
+	}
+	defer conn.Close()
+	if err := conn.SetDeadline(time.Now().Add(smtpOperationTimeout)); err != nil {
+		return errors.Wrap(err, "failed to set SMTP connection deadline")
+	}
+
+	// Create SMTP client
+	client, err := smtp.NewClient(conn, c.config.SMTPHost)
+	if err != nil {
+		return errors.Wrap(err, "failed to create SMTP client")
+	}
+	defer client.Quit()
+
+	return c.sendWithClient(client, auth, recipients, body)
+}
+
+func (c *Client) sendWithClient(client *smtp.Client, auth smtp.Auth, recipients []string, body string) error {
+	if auth != nil {
+		if err := client.Auth(auth); err != nil {
+			return errors.Wrap(err, "SMTP authentication failed")
+		}
+	}
+
+	// Set sender
+	if err := client.Mail(c.config.FromEmail); err != nil {
+		return errors.Wrap(err, "failed to set sender")
+	}
+
+	// Set recipients
+	for _, recipient := range recipients {
+		if err := client.Rcpt(recipient); err != nil {
+			return errors.Wrapf(err, "failed to set recipient: %s", recipient)
+		}
+	}
+
+	// Send message body
+	writer, err := client.Data()
+	if err != nil {
+		return errors.Wrap(err, "failed to send DATA command")
+	}
+
+	if _, err := writer.Write([]byte(body)); err != nil {
+		return errors.Wrap(err, "failed to write message body")
+	}
+
+	if err := writer.Close(); err != nil {
+		return errors.Wrap(err, "failed to close message writer")
+	}
+
+	return nil
+}