summaryrefslogtreecommitdiff
path: root/cmd/generate/mailstuff/mbox.go
blob: 8700c24d631fdf5c1f90a37721c00bc786ce293a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
package mailstuff

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

func ReadMBox(r io.Reader) ([]*mail.Message, error) {
	rest, err := io.ReadAll(r)
	if err != nil {
		return nil, err
	}

	const terminator = "\nFrom "

	var parts [][]byte
	for {
		pos := bytes.Index(rest, []byte(terminator))
		if pos < 0 {
			parts = append(parts, rest)
			break
		}
		parts = append(parts, rest[:pos+1])
		rest = rest[pos+1:]
	}

	ret := make([]*mail.Message, len(parts))
	for i := range len(parts) {
		msg, err := mail.ReadMessage(bytes.NewReader(parts[i]))
		if err != nil {
			return nil, err
		}
		ret[i] = msg
	}

	return ret, nil
}