|
|
package main
|
|
|
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
"github.com/taoshihan1991/imaptool/tmpl"
|
|
|
"github.com/taoshihan1991/imaptool/tools"
|
|
|
"log"
|
|
|
"net/http"
|
|
|
"strconv"
|
|
|
"sync"
|
|
|
)
|
|
|
|
|
|
const PageSize = 20
|
|
|
|
|
|
func main() {
|
|
|
log.Println("listen on 8080...\r\ngo：http://127.0.0.1:8080")
|
|
|
//根路径
|
|
|
http.HandleFunc("/", index)
|
|
|
//邮件夹
|
|
|
http.HandleFunc("/list", list)
|
|
|
//登陆界面
|
|
|
http.HandleFunc("/login", login)
|
|
|
//验证接口
|
|
|
http.HandleFunc("/check", check)
|
|
|
//邮件夹接口
|
|
|
http.HandleFunc("/folders", folders)
|
|
|
//邮件接口
|
|
|
http.HandleFunc("/mail", mail)
|
|
|
//详情界面
|
|
|
http.HandleFunc("/view", view)
|
|
|
//监听端口
|
|
|
http.ListenAndServe(":8080", nil)
|
|
|
}
|
|
|
|
|
|
//首页跳转
|
|
|
func index(w http.ResponseWriter, r *http.Request) {
|
|
|
if r.URL.RequestURI() == "/favicon.ico" {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
mailServer := tools.GetMailServerFromCookie(r)
|
|
|
if mailServer == nil {
|
|
|
http.Redirect(w, r, "/login", 302)
|
|
|
} else {
|
|
|
res := tools.CheckEmailPassword(mailServer.Server, mailServer.Email, mailServer.Password)
|
|
|
if res {
|
|
|
http.Redirect(w, r, "/list", 302)
|
|
|
} else {
|
|
|
http.Redirect(w, r, "/login", 302)
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
//输出列表
|
|
|
func list(w http.ResponseWriter, r *http.Request) {
|
|
|
values := r.URL.Query()
|
|
|
fid := ""
|
|
|
currentPage := 0
|
|
|
if len(values["fid"]) != 0 {
|
|
|
fid = values["fid"][0]
|
|
|
}
|
|
|
if len(values["page"]) != 0 {
|
|
|
currentPage, _ = strconv.Atoi(values["page"][0])
|
|
|
}
|
|
|
if fid == "" {
|
|
|
fid = "INBOX"
|
|
|
}
|
|
|
if currentPage == 0 {
|
|
|
currentPage = 1
|
|
|
}
|
|
|
render := new(tools.IndexData)
|
|
|
render.CurrentPage = currentPage
|
|
|
render.Fid = fid
|
|
|
tmpl.RenderList(w, render)
|
|
|
}
|
|
|
|
|
|
//详情界面
|
|
|
func view(w http.ResponseWriter, r *http.Request) {
|
|
|
values := r.URL.Query()
|
|
|
fid := ""
|
|
|
if len(values["fid"]) != 0 {
|
|
|
fid = values["fid"][0]
|
|
|
} else {
|
|
|
fid = "INBOX"
|
|
|
}
|
|
|
var id uint32
|
|
|
if len(values["id"]) != 0 {
|
|
|
i, _ := strconv.Atoi(values["id"][0])
|
|
|
id = uint32(i)
|
|
|
} else {
|
|
|
id = 0
|
|
|
}
|
|
|
//
|
|
|
//mailServer:=tools.GetMailServerFromCookie(r)
|
|
|
//var wg sync.WaitGroup
|
|
|
var render = new(tools.ViewData)
|
|
|
render.Fid = fid
|
|
|
render.Id = id
|
|
|
//wg.Add(1)
|
|
|
//go func() {
|
|
|
// defer wg.Done()
|
|
|
// folders := tools.GetFolders(mailServer.Server, mailServer.Email, mailServer.Password, fid)
|
|
|
// render.Folders = folders
|
|
|
// render.Fid = fid
|
|
|
//}()
|
|
|
//wg.Add(1)
|
|
|
//go func() {
|
|
|
// defer wg.Done()
|
|
|
// mail := tools.GetMessage(mailServer.Server, mailServer.Email, mailServer.Password, fid, id)
|
|
|
// render.From = mail.From
|
|
|
// render.To = mail.To
|
|
|
// render.Subject = mail.Subject
|
|
|
// render.Date = mail.Date
|
|
|
// render.HtmlBody = template.HTML(mail.Body)
|
|
|
//}()
|
|
|
//wg.Wait()
|
|
|
tmpl.RenderView(w, render)
|
|
|
}
|
|
|
|
|
|
//登陆界面
|
|
|
func login(w http.ResponseWriter, r *http.Request) {
|
|
|
email := r.PostFormValue("email")
|
|
|
server := r.PostFormValue("server")
|
|
|
password := r.PostFormValue("password")
|
|
|
var errStr string
|
|
|
if email != "" && server != "" && password != "" {
|
|
|
res := tools.CheckEmailPassword(server, email, password)
|
|
|
if !res {
|
|
|
errStr = "连接或验证失败"
|
|
|
tmpl.RenderLogin(w, errStr)
|
|
|
} else {
|
|
|
auth := fmt.Sprintf("%s|%s|%s", server, email, password)
|
|
|
cookie := http.Cookie{
|
|
|
Name: "auth",
|
|
|
Value: auth,
|
|
|
}
|
|
|
http.SetCookie(w, &cookie)
|
|
|
http.Redirect(w, r, "/", 302)
|
|
|
}
|
|
|
} else {
|
|
|
tmpl.RenderLogin(w, errStr)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
//验证接口
|
|
|
func check(w http.ResponseWriter, r *http.Request) {
|
|
|
email := r.PostFormValue("email")
|
|
|
server := r.PostFormValue("server")
|
|
|
password := r.PostFormValue("password")
|
|
|
msg, _ := json.Marshal(tools.JsonResult{Code: 400, Msg: "验证失败"})
|
|
|
|
|
|
w.Header().Set("content-type", "text/json;charset=utf-8;")
|
|
|
if email != "" && server != "" && password != "" {
|
|
|
res := tools.CheckEmailPassword(server, email, password)
|
|
|
if res {
|
|
|
msg, _ = json.Marshal(tools.JsonResult{Code: 200, Msg: "验证成功,正在跳转..."})
|
|
|
auth := fmt.Sprintf("%s|%s|%s", server, email, password)
|
|
|
tools.SetCookie("auth", auth, &w)
|
|
|
w.Write(msg)
|
|
|
} else {
|
|
|
w.Write(msg)
|
|
|
}
|
|
|
} else {
|
|
|
w.Write(msg)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
//邮件夹接口
|
|
|
func folders(w http.ResponseWriter, r *http.Request) {
|
|
|
values := r.URL.Query()
|
|
|
fid := ""
|
|
|
currentPage := 0
|
|
|
if len(values["fid"]) != 0 {
|
|
|
fid = values["fid"][0]
|
|
|
}
|
|
|
if len(values["page"]) != 0 {
|
|
|
currentPage, _ = strconv.Atoi(values["page"][0])
|
|
|
}
|
|
|
if fid == "" {
|
|
|
fid = "INBOX"
|
|
|
}
|
|
|
if currentPage == 0 {
|
|
|
currentPage = 1
|
|
|
}
|
|
|
|
|
|
mailServer := tools.GetMailServerFromCookie(r)
|
|
|
w.Header().Set("content-type", "text/json;charset=utf-8;")
|
|
|
|
|
|
if mailServer == nil {
|
|
|
msg, _ := json.Marshal(tools.JsonResult{Code: 400, Msg: "验证失败"})
|
|
|
w.Write(msg)
|
|
|
return
|
|
|
}
|
|
|
var wg sync.WaitGroup
|
|
|
wg.Add(2)
|
|
|
result := make(map[string]interface{})
|
|
|
go func() {
|
|
|
defer wg.Done()
|
|
|
folders := tools.GetFolders(mailServer.Server, mailServer.Email, mailServer.Password, fid)
|
|
|
result["folders"] = folders
|
|
|
result["total"] = folders[fid]
|
|
|
}()
|
|
|
go func() {
|
|
|
defer wg.Done()
|
|
|
mails := tools.GetFolderMail(mailServer.Server, mailServer.Email, mailServer.Password, fid, currentPage, PageSize)
|
|
|
result["mails"] = mails
|
|
|
}()
|
|
|
wg.Wait()
|
|
|
result["pagesize"] = PageSize
|
|
|
result["fid"] = fid
|
|
|
|
|
|
msg, _ := json.Marshal(tools.JsonFolders{
|
|
|
JsonResult: tools.JsonResult{Code: 200, Msg: "获取成功"},
|
|
|
Result: result,
|
|
|
})
|
|
|
w.Write(msg)
|
|
|
}
|
|
|
|
|
|
//邮件接口
|
|
|
func mail(w http.ResponseWriter, r *http.Request) {
|
|
|
values := r.URL.Query()
|
|
|
fid := ""
|
|
|
if len(values["fid"]) != 0 {
|
|
|
fid = values["fid"][0]
|
|
|
} else {
|
|
|
fid = "INBOX"
|
|
|
}
|
|
|
var id uint32
|
|
|
if len(values["id"]) != 0 {
|
|
|
i, _ := strconv.Atoi(values["id"][0])
|
|
|
id = uint32(i)
|
|
|
} else {
|
|
|
id = 0
|
|
|
}
|
|
|
mailServer := tools.GetMailServerFromCookie(r)
|
|
|
w.Header().Set("content-type", "text/json;charset=utf-8;")
|
|
|
|
|
|
if mailServer == nil {
|
|
|
msg, _ := json.Marshal(tools.JsonResult{Code: 400, Msg: "验证失败"})
|
|
|
w.Write(msg)
|
|
|
return
|
|
|
}
|
|
|
var wg sync.WaitGroup
|
|
|
result := make(map[string]interface{})
|
|
|
wg.Add(2)
|
|
|
go func() {
|
|
|
defer wg.Done()
|
|
|
folders := tools.GetFolders(mailServer.Server, mailServer.Email, mailServer.Password, fid)
|
|
|
result["folders"] = folders
|
|
|
result["total"] = folders[fid]
|
|
|
}()
|
|
|
go func() {
|
|
|
defer wg.Done()
|
|
|
mail := tools.GetMessage(mailServer.Server, mailServer.Email, mailServer.Password, fid, id)
|
|
|
result["from"] = mail.From
|
|
|
result["to"] = mail.To
|
|
|
result["subject"] = mail.Subject
|
|
|
result["date"] = mail.Date
|
|
|
result["html"] = mail.Body
|
|
|
}()
|
|
|
wg.Wait()
|
|
|
result["fid"] = fid
|
|
|
|
|
|
msg, _ := json.Marshal(tools.JsonFolders{
|
|
|
JsonResult: tools.JsonResult{Code: 200, Msg: "获取成功"},
|
|
|
Result: result,
|
|
|
})
|
|
|
w.Write(msg)
|
|
|
}
|
