feat: Add parsing message content

This commit is contained in:
ptrcnull 2021-03-23 17:42:59 +01:00
parent ab2ddeed87
commit 6a73913e0c
4 changed files with 113 additions and 4 deletions

2
go.mod
View File

@ -1,3 +1,5 @@
module tgmail
go 1.15
require git.ddd.rip/ptrcnull/telegram v0.0.4

2
go.sum Normal file
View File

@ -0,0 +1,2 @@
git.ddd.rip/ptrcnull/telegram v0.0.4 h1:b4VaLAdIUuEFG5DRM+JzfLEVoe4tdHhrgxtmoF2UJOQ=
git.ddd.rip/ptrcnull/telegram v0.0.4/go.mod h1:SSSKvfhw7mDx/8UPoLdtP9J74z2/pXccHnKzdi16nLA=

View File

@ -1,13 +1,19 @@
package main
import (
"git.ddd.rip/ptrcnull/telegram"
"log"
"net"
"os"
"tgmail/server"
)
func main() {
l, err := net.Listen("tcp", "0.0.0.0:2525")
server.Tg = telegram.Client{
Token: os.Getenv("TELEGRAM_TOKEN"),
}
l, err := net.Listen("tcp", "0.0.0.0:25")
if err != nil {
panic(err)
}
@ -21,6 +27,7 @@ func main() {
log.Println(err)
return
}
log.Println("UwU")
go server.Handle(c)
}
}

View File

@ -2,15 +2,25 @@ package server
import (
"bytes"
"encoding/json"
"git.ddd.rip/ptrcnull/telegram"
"io"
"io/ioutil"
"log"
"mime"
"mime/multipart"
"mime/quotedprintable"
"net"
"net/http"
"net/mail"
"strings"
)
const response = "uwu"
var Tg telegram.Client
const chatId = "456311800"
func Handle(conn net.Conn) {
c := wrapper{Conn: conn}
log.Println("Connection received from", c.RemoteAddr().String())
@ -77,6 +87,59 @@ func readContent(c wrapper) (string, error) {
return data, nil
}
type Header interface {
Get(key string) string
}
func readTextBody(header Header, body io.Reader) (string, error) {
log.Println("reading text body")
var bodyBytes []byte
var err error
if header.Get("Content-Transfer-Encoding") == "quoted-printable" {
bodyBytes, err = ioutil.ReadAll(quotedprintable.NewReader(body))
} else {
bodyBytes, err = ioutil.ReadAll(body)
}
log.Println("ended reading text body")
return string(bodyBytes), err
}
func readMultipartBody(msg *mail.Message, boundary string) (string, error) {
r := multipart.NewReader(msg.Body, boundary)
body := ""
for {
part, err := r.NextPart()
if err == io.EOF {
break
}
if err != nil {
log.Println("error parsing multipart:", err)
break
}
mimetype, _, err := mime.ParseMediaType(part.Header.Get("Content-Type"))
if err != nil {
log.Println("error parsing multipart mimetype:", err)
continue
}
if mimetype == "text/plain" {
body, err = readTextBody(part.Header, part)
if err != nil {
log.Println("error reading text body from multipart:", err)
}
} else if mimetype == "text/html" && body == "" {
body, err = readTextBody(part.Header, part)
if err != nil {
log.Println("error reading html body from multipart:", err)
}
}
}
return body, nil
}
func handleMessage(data string) {
msg, err := mail.ReadMessage(bytes.NewReader([]byte(data)))
if err != nil {
@ -84,12 +147,47 @@ func handleMessage(data string) {
return
}
body, err := ioutil.ReadAll(msg.Body)
res := "New mail!"
res += "\nFrom: " + msg.Header.Get("From")
res += "\nTo: " + msg.Header.Get("To")
mimetype, mimeparams, err := mime.ParseMediaType(msg.Header.Get("Content-Type"))
if err != nil {
log.Println(err)
return
}
log.Println("FROM: ", msg.Header.Get("From"))
log.Println(string(body))
log.Println("handling email with content-type", mimetype)
var body string
if strings.HasPrefix(mimetype, "text/") {
body, err = readTextBody(msg.Header, msg.Body)
} else if strings.HasPrefix(mimetype, "multipart/") {
body, err = readMultipartBody(msg, mimeparams["boundary"])
} else {
body = "check console for more info"
}
if err != nil {
log.Println("reading body failed:", err)
return
}
if len(body) > 1000 {
// curl 'https://bin.ddd.rip/documents' --compressed -H 'Content-Type: application/json; charset=utf-8' -H 'X-Requested-With: XMLHttpRequest' -H 'Origin: https://bin.ddd.rip' -H 'DNT: 1' -H 'Connection: keep-alive' -H 'Referer: https://bin.ddd.rip/' -H 'TE: Trailers' --data-raw dupa
log.Println("mail too long, uploading to hastebin")
res, err := http.Post("https://bin.ddd.rip/documents", "text/plain", bytes.NewReader([]byte(body)))
if err != nil {
body = "error uploading email: " + err.Error()
} else {
var response map[string]string
json.NewDecoder(res.Body).Decode(&response)
body = "https://bin.ddd.rip/" + response["key"]
}
}
res += "\n\n" + body
log.Println("sending telegram message")
_, err = Tg.SendMessage(chatId, res)
if err != nil {
log.Println(err)
}
}