package main import ( "encoding/json" "fmt" "github.com/taoshihan1991/imaptool/tmpl" "github.com/taoshihan1991/imaptool/tools" "html/template" "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("/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 } mailServer:=tools.GetMailServerFromCookie(r) render := new(tools.IndexData) render.CurrentPage = currentPage var prePage int if (currentPage - 1) <= 0 { prePage = 1 } else { prePage = currentPage - 1 } render.PrePage = fmt.Sprintf("/list?fid=%s&page=%d", fid, prePage) render.NextPage = fmt.Sprintf("/list?fid=%s&page=%d", fid, currentPage+1) var wg sync.WaitGroup wg.Add(2) go func() { defer wg.Done() folders := tools.GetFolders(mailServer.Server, mailServer.Email,mailServer.Password, fid) render.Folders = folders render.Fid = fid //PageCount:= render.Folders[fid]/PAGE_SIZE numPages := "" start := currentPage - 5 if start <= 0 { start = 1 } end := start + 11 //if end>=PageCount{ // end=PageCount //} for i := start; i < end; i++ { active := "" if currentPage == i { active = "active" } numPages += fmt.Sprintf("
  • %d
  • ", active, fid, i, i) } render.NumPages = template.HTML(numPages) }() go func() { defer wg.Done() mails := tools.GetFolderMail(mailServer.Server, mailServer.Email, mailServer.Password, fid, currentPage, PageSize) render.MailPagelist = mails }() wg.Wait() 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) 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 }() go func() { defer wg.Done() mails := tools.GetFolderMail(mailServer.Server, mailServer.Email, mailServer.Password, fid, currentPage, PageSize) result["mails"]=mails }() wg.Wait() msg, _ := json.Marshal(tools.JsonFolders{ JsonResult: tools.JsonResult{Code: 200, Msg: "获取成功"}, Result: result, }) w.Write(msg) } //加密cookie //func authCookie(){ // //}