func kickOffWatcher(c *imap.Client) {
cmd, _ := c.Idle()
for cmd.InProgress() {
c.Data = nil
c.Recv(-1)
for _, rsp := range c.Data {
if rsp.Label == "EXISTS" {
newEmailCount := rsp.Fields[0].(uint32)
if newEmailCount > curMsgCount {
c.IdleTerm()
notifyNewEmail(c, newEmailCount)
kickOffWatcher(c)
}
curMsgCount = newEmailCount
}
}
}
}
func (emaill *Email) getEmailContentUseIMAP(w rest.ResponseWriter, r *rest.Request) {
addr := r.FormValue("imapAddr")
port := r.FormValue("imapPort")
user := r.FormValue("user")
pwd := r.FormValue("pwd")
id := r.FormValue("id")
address := addr + ":" + port
log.Printf("get email content use imap %s\n", address)
var (
c *imap.Client
cmd *imap.Command
rsp *imap.Response
)
// Connect to the server
c, err := imap.Dial(addr)
if err != nil {
log.Printf("dial %s error\n", address)
return
}
// Remember to log out and close the connection when finished
defer c.Logout(30 * time.Second)
// Print server greeting (first response in the unilateral server data queue)
log.Printf("Server says hello:%s", c.Data[0].Info)
c.Data = nil
// Enable encryption, if supported by the server
if c.Caps["STARTTLS"] {
c.StartTLS(nil)
}
// Authenticate
if c.State() == imap.Login {
c.Login(user, pwd)
}
// List all top-level mailboxes, wait for the command to finish
cmd, _ = imap.Wait(c.List("", "%"))
// Print mailbox information
log.Printf("\nTop-level mailboxes:")
for _, rsp = range cmd.Data {
log.Printf("|--%s", rsp.MailboxInfo())
}
// Check for new unilateral server data responses
for _, rsp = range c.Data {
log.Printf("Server data:%s", rsp)
}
c.Data = nil
// Open a mailbox (synchronous command - no need for imap.Wait)
c.Select("INBOX", true)
log.Printf("\nMailbox status:%s\n", c.Mailbox)
if c.Mailbox == nil {
resp := EmailJsonResponse{Success: true}
w.WriteJson(&resp)
return
}
// Fetch the headers of the 10 most recent messages
set, _ := imap.NewSeqSet("")
set.Add(id)
// if c.Mailbox.Messages >= 10 {
// set.AddRange(c.Mailbox.Messages-9, c.Mailbox.Messages) //测试只取最新一封邮件
// } else {
// set.Add("1:*")
// }
cmd, _ = c.Fetch(set, "RFC822.HEADER", "RFC822.TEXT") //指定要获取的内容
// Process responses while the command is running
log.Printf("\nget mail [%s] messages:", id)
for cmd.InProgress() {
// Wait for the next response (no timeout)
c.Recv(-1)
// Process command data
for _, rsp = range cmd.Data {
header := imap.AsBytes(rsp.MessageInfo().Attrs["RFC822.HEADER"])
if msg, _ := mail.ReadMessage(bytes.NewReader(header)); msg != nil {
subject := msg.Header.Get("Subject")
log.Printf("|--%s", subject)
realSubject := GetRealSubject(subject)
log.Printf("in rest_email.go: get real subject")
log.Printf(realSubject)
senderAddr := msg.Header.Get("From")
recverAddrList := msg.Header.Get("To")
realSenderAddr := GetRealSubject(senderAddr)
realRecverAddrList := GetRealSubject(recverAddrList)
body := imap.AsBytes(rsp.MessageInfo().Attrs["RFC822.TEXT"])
//log.Printf("email body: %s", body)
//realBody := GetRealBody(string(body))
headerAndBody := make([]byte, len(header)+len(body))
copy(headerAndBody, header)
copy(headerAndBody[len(header):], body)
msg, _ := mail.ReadMessage(bytes.NewReader(headerAndBody))
mime, _ := enmime.ParseMIMEBody(msg)
realBody := mime.Text //如果原始内容为html，会去掉html元素标签
log.Printf("real body: %s", realBody)
//获取MIMEPart所有节点内容
// log.Printf("root ======================")
// root := mime.Root
// if root != nil {
// log.Printf(string(root.Content()))
// log.Printf("child==========")
// if child := root.FirstChild(); child != nil {
// log.Printf(string(child.Content()))
// }
// }
attachments := mime.Attachments
log.Printf("attachments len=%d", len(attachments))
count := len(attachments)
var attachmentList []Attachment = nil
var data EmailContent
if count > 0 {
attachmentList = make([]Attachment, count)
for i := 0; i < len(attachments); i++ {
name := attachments[i].FileName()
content := attachments[i].Content() //todo encode by base64
log.Printf("name===%s", name)
attachmentList[i] = Attachment{Name: name, Body: string(content)}
}
}
data = EmailContent{Subject: realSubject, SenderAddr: realSenderAddr,
RecverAddrList: realRecverAddrList, Content: realBody,
Attachments: attachmentList}
resp := EmailJsonResponse{Success: true, Data: data}
w.WriteJson(resp)
}
}
cmd.Data = nil
// Process unilateral server data
for _, rsp = range c.Data {
log.Printf("Server data:%s", rsp)
}
c.Data = nil
}
// Check command completion status
if rsp, err := cmd.Result(imap.OK); err != nil {
if err == imap.ErrAborted {
log.Printf("Fetch command aborted\n")
} else {
log.Printf("Fetch error:%s\n", rsp.Info)
}
}
}
// updateTray is called whenever a client detects that the number of unseen
// messages *may* have changed. It will search the selected folder for unseen
// messages, count them and store the result. Then it will use the notify
// channel to let our main process update the status icon.
func UpdateTray(c *imap.Client, notify chan bool, name string) {
// Send > Search since Search adds CHARSET UTF-8 which might not be supported
cmd, err := c.Send("SEARCH", "UNSEEN")
if err != nil {
fmt.Printf("%s failed to look for new messages\n", name)
fmt.Println(" ", err)
return
}
if _, ok := Unseen[name]; !ok {
Unseen[name] = []uint32{}
}
unseenMessages := []uint32{}
for cmd.InProgress() {
// Wait for the next response (no timeout)
err = c.Recv(-1)
// Process command data
for _, rsp := range cmd.Data {
result := rsp.SearchResults()
unseenMessages = append(unseenMessages, result...)
}
// Reset for next run
cmd.Data = nil
c.Data = nil
}
// Check command completion status
if rsp, err := cmd.Result(imap.OK); err != nil {
if err == imap.ErrAborted {
fmt.Println("fetch command aborted")
} else {
fmt.Println("fetch error:", rsp.Info)
}
return
}
fmt.Printf("%d unseen\n", len(unseenMessages))
// Find messages that the user hasn't been notified of
// TODO: Make this optional/configurable
newUnseen, _ := imap.NewSeqSet("")
numNewUnseen := 0
for _, uid := range unseenMessages {
seen := false
for _, olduid := range Unseen[name] {
if olduid == uid {
seen = true
break
}
}
if !seen {
newUnseen.AddNum(uid)
numNewUnseen++
}
}
// If we do have new unseen messages, fetch and display them
if numNewUnseen > 0 {
messages := make([]string, numNewUnseen)
i := 0
// Fetch headers...
cmd, _ = c.Fetch(newUnseen, "RFC822.HEADER")
for cmd.InProgress() {
c.Recv(-1)
for _, rsp := range cmd.Data {
header := imap.AsBytes(rsp.MessageInfo().Attrs["RFC822.HEADER"])
if msg, _ := mail.ReadMessage(bytes.NewReader(header)); msg != nil {
subject := msg.Header.Get("Subject")
messages[i], err = new(mime.WordDecoder).DecodeHeader(subject)
if err != nil {
messages[i] = subject
}
i++
}
}
cmd.Data = nil
c.Data = nil
}
// Print them in reverse order to get newest first
notification := ""
for ; i > 0; i-- {
notification += "> " + messages[i-1] + "\n"
}
notification = strings.TrimRight(notification, "\n")
fmt.Println(notification)
// And send them with notify-send!
title := fmt.Sprintf("%s has new mail (%d unseen)", name, len(unseenMessages))
sh := exec.Command("/usr/bin/notify-send",
"-i", "notification-message-email",
"-c", "email",
title, notification)
err := sh.Start()
if err != nil {
fmt.Println("Failed to notify user...")
fmt.Println(err)
}
go func() {
// collect exit code to avoid zombies
sh.Wait()
}()
}
Unseen[name] = unseenMessages
// Let main process know something has changed
notify <- true
}
func main() {
token := flag.String("token", "", "The Telegram bot token")
to := flag.String("to", "", "The mail recipient")
from := flag.String("from", "", "The mail sender")
server := flag.String("server", "", "The mail server")
port := flag.Int("port", 587, "The mail server port")
user := flag.String("user", "", "")
pass := flag.String("pass", "", "")
subject := flag.String("subject", "", "")
flag.Parse()
bot, err := telebot.NewBot(*token)
if err != nil {
return
}
// Fetching Mails and send as Telegram messages
var (
c *imap.Client
cmd *imap.Command
rsp *imap.Response
)
c, _ = imap.Dial(*server)
defer c.Logout(30 * time.Second)
c.Data = nil
if c.Caps["STARTTLS"] {
c.StartTLS(nil)
}
if c.State() == imap.Login {
c.Login(*user, *pass)
}
cmd, _ = imap.Wait(c.List("", "%"))
c.Data = nil
c.Select("INBOX", true)
set, _ := imap.NewSeqSet("")
set.Add("1:*")
cmd, _ = c.Fetch(set, "RFC822.HEADER")
for cmd.InProgress() {
c.Recv(-1)
for _, rsp = range cmd.Data {
header := imap.AsBytes(rsp.MessageInfo().Attrs["RFC822.HEADER"])
if msg, _ := mail.ReadMessage(bytes.NewReader(header)); msg != nil {
fmt.Println("|--", msg.Header.Get("FROM"))
}
}
cmd.Data = nil
c.Data = nil
}
if rsp, err := cmd.Result(imap.OK); err != nil {
if err == imap.ErrAborted {
fmt.Println("Fetch command aborted")
} else {
fmt.Println("Fetch error:", rsp.Info)
}
}
/*
* Listen to Telegram messages and send as mail
*/
messages := make(chan telebot.Message)
bot.Listen(messages, 1*time.Second)
for message := range messages {
m := gomail.NewMessage()
m.SetHeader("From", *from)
m.SetHeader("To", *to)
m.SetHeader("Subject", *subject)
m.SetBody("text/html", message.Text)
d := gomail.NewPlainDialer(*server, *port, *user, *pass)
d.TLSConfig = &tls.Config{InsecureSkipVerify: true}
if err := d.DialAndSend(m); err != nil {
panic(err)
}
}
}
func ExampleClient() {
//
// Note: most of error handling code is omitted for brevity
//
var (
c *imap.Client
cmd *imap.Command
rsp *imap.Response
)
// Connect to the server
c, _ = imap.Dial("imap.example.com")
// Remember to log out and close the connection when finished
defer c.Logout(30 * time.Second)
// Print server greeting (first response in the unilateral server data queue)
fmt.Println("Server says hello:", c.Data[0].Info)
c.Data = nil
// Enable encryption, if supported by the server
if c.Caps["STARTTLS"] {
c.StartTLS(nil)
}
// Authenticate
if c.State() == imap.Login {
c.Login("*****@*****.**", "mysupersecretpassword")
}
// List all top-level mailboxes, wait for the command to finish
cmd, _ = imap.Wait(c.List("", "%"))
// Print mailbox information
fmt.Println("\nTop-level mailboxes:")
for _, rsp = range cmd.Data {
fmt.Println("|--", rsp.MailboxInfo())
}
// Check for new unilateral server data responses
for _, rsp = range c.Data {
fmt.Println("Server data:", rsp)
}
c.Data = nil
// Open a mailbox (synchronous command - no need for imap.Wait)
c.Select("INBOX", true)
fmt.Print("\nMailbox status:\n", c.Mailbox)
// Fetch the headers of the 10 most recent messages
set, _ := imap.NewSeqSet("")
if c.Mailbox.Messages >= 10 {
set.AddRange(c.Mailbox.Messages-9, c.Mailbox.Messages)
} else {
set.Add("1:*")
}
cmd, _ = c.Fetch(set, "RFC822.HEADER")
// Process responses while the command is running
fmt.Println("\nMost recent messages:")
for cmd.InProgress() {
// Wait for the next response (no timeout)
c.Recv(-1)
// Process command data
for _, rsp = range cmd.Data {
header := imap.AsBytes(rsp.MessageInfo().Attrs["RFC822.HEADER"])
if msg, _ := mail.ReadMessage(bytes.NewReader(header)); msg != nil {
fmt.Println("|--", msg.Header.Get("Subject"))
}
}
cmd.Data = nil
// Process unilateral server data
for _, rsp = range c.Data {
fmt.Println("Server data:", rsp)
}
c.Data = nil
}
// Check command completion status
if rsp, err := cmd.Result(imap.OK); err != nil {
if err == imap.ErrAborted {
fmt.Println("Fetch command aborted")
} else {
fmt.Println("Fetch error:", rsp.Info)
}
}
}