nuclei/pkg/js/libs/smtp/msg.go

package smtp

import (
	"bufio"
	"net/textproto"
	"strings"

	"github.com/valyala/bytebufferpool"
)

type (
	// SMTPMessage is a message to be sent over SMTP
	// @example
	// ```javascript
	// const smtp = require('nuclei/smtp');
	// const message = new smtp.SMTPMessage();
	// message.From('xyz@projectdiscovery.io');
	// ```
	SMTPMessage struct {
		from string
		to   []string
		sub  string
		msg  []byte
		user string
		pass string
	}
)

// From adds the from field to the message
// @example
// ```javascript
// const smtp = require('nuclei/smtp');
// const message = new smtp.SMTPMessage();
// message.From('xyz@projectdiscovery.io');
// ```
func (s *SMTPMessage) From(email string) *SMTPMessage {
	s.from = email
	return s
}

// To adds the to field to the message
// @example
// ```javascript
// const smtp = require('nuclei/smtp');
// const message = new smtp.SMTPMessage();
// message.To('xyz@projectdiscovery.io');
// ```
func (s *SMTPMessage) To(email string) *SMTPMessage {
	s.to = append(s.to, email)
	return s
}

// Subject adds the subject field to the message
// @example
// ```javascript
// const smtp = require('nuclei/smtp');
// const message = new smtp.SMTPMessage();
// message.Subject('hello');
// ```
func (s *SMTPMessage) Subject(sub string) *SMTPMessage {
	s.sub = sub
	return s
}

// Body adds the message body to the message
// @example
// ```javascript
// const smtp = require('nuclei/smtp');
// const message = new smtp.SMTPMessage();
// message.Body('hello');
// ```
func (s *SMTPMessage) Body(msg []byte) *SMTPMessage {
	s.msg = msg
	return s
}

// Auth when called authenticates using username and password before sending the message
// @example
// ```javascript
// const smtp = require('nuclei/smtp');
// const message = new smtp.SMTPMessage();
// message.Auth('username', 'password');
// ```
func (s *SMTPMessage) Auth(username, password string) *SMTPMessage {
	s.user = username
	s.pass = password
	return s
}

// String returns the string representation of the message
// @example
// ```javascript
// const smtp = require('nuclei/smtp');
// const message = new smtp.SMTPMessage();
// message.From('xyz@projectdiscovery.io');
// message.To('xyz2@projectdiscoveyr.io');
// message.Subject('hello');
// message.Body('hello');
// log(message.String());
// ```
func (s *SMTPMessage) String() string {
	buff := bytebufferpool.Get()
	defer bytebufferpool.Put(buff)
	tw := textproto.NewWriter(bufio.NewWriter(buff))
	_ = tw.PrintfLine("To: %s", strings.Join(s.to, ","))
	if s.sub != "" {
		_ = tw.PrintfLine("Subject: %s", s.sub)
	}
	_ = tw.PrintfLine("\r\n%s", s.msg)
	return buff.String()
}