syzkaller/pkg/email/reply.go

43 lines
887 B
Go

// Copyright 2017 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
package email
import (
"bufio"
"bytes"
"strings"
)
func FormReply(email, reply string) string {
s := bufio.NewScanner(strings.NewReader(email))
out := new(bytes.Buffer)
replied := false
for s.Scan() {
ln := s.Bytes()
out.WriteByte('>')
if len(ln) != 0 && ln[0] != '>' {
out.WriteByte(' ')
}
out.Write(ln)
out.WriteByte('\n')
if !replied && bytes.HasPrefix(ln, []byte(commandPrefix)) {
replied = true
writeReply(out, reply)
}
}
if !replied {
writeReply(out, reply)
}
return out.String()
}
func writeReply(out *bytes.Buffer, reply string) {
out.WriteByte('\n')
out.WriteString(reply)
if reply != "" && reply[len(reply)-1] != '\n' {
out.WriteByte('\n')
}
out.WriteByte('\n')
}