mirror of
https://gitlab.com/ric_harvey/MailHog.git
synced 2024-11-27 08:14:04 +00:00
Remove data package
This commit is contained in:
Ian Kent
parent
5948ba152d
commit
ac84efffe4
1 changed files with 0 additions and 203 deletions
203
data/message.go
203
data/message.go
|
|
@ -1,203 +0,0 @@
|
|||
package data
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"log"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// MessageID represents the ID of an SMTP message including the hostname part
|
||||
type MessageID string
|
||||
|
||||
// NewMessageID generates a new message ID
|
||||
func NewMessageID(hostname string) (MessageID, error) {
|
||||
size := 32
|
||||
|
||||
rb := make([]byte, size)
|
||||
_, err := rand.Read(rb)
|
||||
|
||||
if err != nil {
|
||||
return MessageID(""), err
|
||||
}
|
||||
|
||||
rs := base64.URLEncoding.EncodeToString(rb)
|
||||
|
||||
return MessageID(rs + "@" + hostname), nil
|
||||
}
|
||||
|
||||
// Messages represents an array of Messages
|
||||
// - TODO is this even required?
|
||||
type Messages []Message
|
||||
|
||||
// Message represents a parsed SMTP message
|
||||
type Message struct {
|
||||
ID MessageID
|
||||
From *Path
|
||||
To []*Path
|
||||
Content *Content
|
||||
Created time.Time
|
||||
MIME *MIMEBody // FIXME refactor to use Content.MIME
|
||||
}
|
||||
|
||||
// Path represents an SMTP forward-path or return-path
|
||||
type Path struct {
|
||||
Relays []string
|
||||
Mailbox string
|
||||
Domain string
|
||||
Params string
|
||||
}
|
||||
|
||||
// Content represents the body content of an SMTP message
|
||||
type Content struct {
|
||||
Headers map[string][]string
|
||||
Body string
|
||||
Size int
|
||||
MIME *MIMEBody
|
||||
}
|
||||
|
||||
// SMTPMessage represents a raw SMTP message
|
||||
type SMTPMessage struct {
|
||||
From string
|
||||
To []string
|
||||
Data string
|
||||
Helo string
|
||||
}
|
||||
|
||||
// MIMEBody represents a collection of MIME parts
|
||||
type MIMEBody struct {
|
||||
Parts []*Content
|
||||
}
|
||||
|
||||
// Parse converts a raw SMTP message to a parsed MIME message
|
||||
func (m *SMTPMessage) Parse(hostname string) *Message {
|
||||
var arr []*Path
|
||||
for _, path := range m.To {
|
||||
arr = append(arr, PathFromString(path))
|
||||
}
|
||||
|
||||
id, _ := NewMessageID(hostname)
|
||||
msg := &Message{
|
||||
ID: id,
|
||||
From: PathFromString(m.From),
|
||||
To: arr,
|
||||
Content: ContentFromString(m.Data),
|
||||
Created: time.Now(),
|
||||
}
|
||||
|
||||
if msg.Content.IsMIME() {
|
||||
log.Printf("Parsing MIME body")
|
||||
msg.MIME = msg.Content.ParseMIMEBody()
|
||||
}
|
||||
|
||||
msg.Content.Headers["Message-ID"] = []string{string(id)}
|
||||
msg.Content.Headers["Received"] = []string{"from " + m.Helo + " by " + hostname + " (Go-MailHog)\r\n id " + string(id) + "; " + time.Now().Format(time.RFC1123Z)}
|
||||
msg.Content.Headers["Return-Path"] = []string{"<" + m.From + ">"}
|
||||
return msg
|
||||
}
|
||||
|
||||
// IsMIME detects a valid MIME header
|
||||
func (content *Content) IsMIME() bool {
|
||||
header, ok := content.Headers["Content-Type"]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
return strings.HasPrefix(header[0], "multipart/")
|
||||
}
|
||||
|
||||
// ParseMIMEBody parses SMTP message content into multiple MIME parts
|
||||
func (content *Content) ParseMIMEBody() *MIMEBody {
|
||||
var parts []*Content
|
||||
|
||||
if hdr, ok := content.Headers["Content-Type"]; ok {
|
||||
if len(hdr) > 0 {
|
||||
re := regexp.MustCompile("boundary=\"([^\"]+)\"")
|
||||
match := re.FindStringSubmatch(hdr[0])
|
||||
if len(match) < 2 {
|
||||
log.Printf("Boundary not found: %s", hdr[0])
|
||||
}
|
||||
log.Printf("Got boundary: %s", match[1])
|
||||
|
||||
p := strings.Split(content.Body, "--"+match[1])
|
||||
|
||||
for _, s := range p {
|
||||
if len(s) > 0 {
|
||||
part := ContentFromString(strings.Trim(s, "\r\n"))
|
||||
if part.IsMIME() {
|
||||
log.Printf("Parsing inner MIME body")
|
||||
part.MIME = part.ParseMIMEBody()
|
||||
}
|
||||
parts = append(parts, part)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return &MIMEBody{
|
||||
Parts: parts,
|
||||
}
|
||||
}
|
||||
|
||||
// PathFromString parses a forward-path or reverse-path into its parts
|
||||
func PathFromString(path string) *Path {
|
||||
var relays []string
|
||||
email := path
|
||||
if strings.Contains(path, ":") {
|
||||
x := strings.SplitN(path, ":", 2)
|
||||
r, e := x[0], x[1]
|
||||
email = e
|
||||
relays = strings.Split(r, ",")
|
||||
}
|
||||
mailbox, domain := "", ""
|
||||
if strings.Contains(email, "@") {
|
||||
x := strings.SplitN(email, "@", 2)
|
||||
mailbox, domain = x[0], x[1]
|
||||
} else {
|
||||
mailbox = email
|
||||
}
|
||||
|
||||
return &Path{
|
||||
Relays: relays,
|
||||
Mailbox: mailbox,
|
||||
Domain: domain,
|
||||
Params: "", // FIXME?
|
||||
}
|
||||
}
|
||||
|
||||
// ContentFromString parses SMTP content into separate headers and body
|
||||
func ContentFromString(data string) *Content {
|
||||
log.Printf("Parsing Content from string: '%s'", data)
|
||||
x := strings.SplitN(data, "\r\n\r\n", 2)
|
||||
h := make(map[string][]string, 0)
|
||||
|
||||
if len(x) == 2 {
|
||||
headers, body := x[0], x[1]
|
||||
hdrs := strings.Split(headers, "\r\n")
|
||||
var lastHdr = ""
|
||||
for _, hdr := range hdrs {
|
||||
if lastHdr != "" && (strings.HasPrefix(hdr, " ") || strings.HasPrefix(hdr, "\t")) {
|
||||
h[lastHdr][len(h[lastHdr])-1] = h[lastHdr][len(h[lastHdr])-1] + hdr
|
||||
} else if strings.Contains(hdr, ": ") {
|
||||
y := strings.SplitN(hdr, ": ", 2)
|
||||
key, value := y[0], y[1]
|
||||
// TODO multiple header fields
|
||||
h[key] = []string{value}
|
||||
lastHdr = key
|
||||
} else {
|
||||
log.Printf("Found invalid header: '%s'", hdr)
|
||||
}
|
||||
}
|
||||
return &Content{
|
||||
Size: len(data),
|
||||
Headers: h,
|
||||
Body: body,
|
||||
}
|
||||
}
|
||||
return &Content{
|
||||
Size: len(data),
|
||||
Headers: h,
|
||||
Body: x[0],
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue