add SetBoundary(boundary string) method

message.go

@@ -18,6 +18,7 @@ type Message struct {
 	encoding    Encoding
 	hEncoder    mimeEncoder
 	buf         bytes.Buffer
+	boundary    string
 }
 
 type header map[string][]string
@@ -69,6 +70,13 @@ func (m *Message) applySettings(settings []MessageSetting) {
 // email.
 type MessageSetting func(m *Message)
 
+// SetBoundary is a message setting to set mime separator string
+func SetBoundary(boundary string) MessageSetting {
+	return func(m *Message) {
+		m.boundary = boundary
+	}
+}
+
 // SetCharset is a message setting to set the charset of the email.
 func SetCharset(charset string) MessageSetting {
 	return func(m *Message) {
@@ -97,6 +105,11 @@ const (
 	Unencoded Encoding = "8bit"
 )
 
+// SetBoundary allows to set custom boundary for multypart message generated
+func (m *Message) SetBoundary(boundary string) {
+	m.boundary = boundary
+}
+
 // SetHeader sets a value to the given header field.
 func (m *Message) SetHeader(field string, value ...string) {
 	m.encodeHeader(value)

message_test.go

@@ -200,6 +200,38 @@ func TestPartSetting(t *testing.T) {
 	testMessage(t, m, 1, want)
 }
 
+func TestPartSettingWithCustomBoundary(t *testing.T) {
+	m := NewMessage()
+	m.SetBoundary("lalalaDaiMne3Ryblya")
+	m.SetHeader("From", "[email protected]")
+	m.SetHeader("To", "[email protected]")
+	m.SetBody("text/plain; format=flowed", "¡Hola, señor!", SetPartEncoding(Unencoded))
+	m.AddAlternative("text/html", "¡<b>Hola</b>, <i>señor</i>!</h1>")
+
+	want := &message{
+		from: "[email protected]",
+		to:   []string{"[email protected]"},
+		content: "From: [email protected]\r\n" +
+			"To: [email protected]\r\n" +
+			"Content-Type: multipart/alternative;\r\n" +
+			" boundary=lalalaDaiMne3Ryblya\r\n" +
+			"\r\n" +
+			"--lalalaDaiMne3Ryblya\r\n" +
+			"Content-Type: text/plain; format=flowed; charset=UTF-8\r\n" +
+			"Content-Transfer-Encoding: 8bit\r\n" +
+			"\r\n" +
+			"¡Hola, señor!\r\n" +
+			"--lalalaDaiMne3Ryblya\r\n" +
+			"Content-Type: text/html; charset=UTF-8\r\n" +
+			"Content-Transfer-Encoding: quoted-printable\r\n" +
+			"\r\n" +
+			"=C2=A1<b>Hola</b>, <i>se=C3=B1or</i>!</h1>\r\n" +
+			"--lalalaDaiMne3Ryblya--\r\n",
+	}
+
+	testMessage(t, m, 1, want)
+}
+
 func TestBodyWriter(t *testing.T) {
 	m := NewMessage()
 	m.SetHeader("From", "[email protected]")

writeto.go

@@ -28,15 +28,15 @@ func (w *messageWriter) writeMessage(m *Message) {
 	w.writeHeaders(m.header)
 
 	if m.hasMixedPart() {
-		w.openMultipart("mixed")
+		w.openMultipart("mixed", m.boundary)
 	}
 
 	if m.hasRelatedPart() {
-		w.openMultipart("related")
+		w.openMultipart("related", m.boundary)
 	}
 
 	if m.hasAlternativePart() {
-		w.openMultipart("alternative")
+		w.openMultipart("alternative", m.boundary)
 	}
 	for _, part := range m.parts {
 		w.writePart(part, m.charset)
@@ -77,8 +77,13 @@ type messageWriter struct {
 	err        error
 }
 
-func (w *messageWriter) openMultipart(mimeType string) {
+func (w *messageWriter) openMultipart(mimeType, boundary string) {
 	mw := multipart.NewWriter(w)
+	if boundary == "" {
+		boundary = mw.Boundary()
+	} else {
+		mw.SetBoundary(boundary)
+	}
 	contentType := "multipart/" + mimeType + ";\r\n boundary=" + mw.Boundary()
 	w.writers[w.depth] = mw