package main

import (
	"fmt"
	"github.com/taoshihan1991/imaptool/tools"
	"html/template"
	"log"
	"net/http"
	"strings"
)

type IndexData struct {
	Folders map[string]int
	Mails   interface{}
}

func main() {
	log.Println("listen on 8080...")
	http.HandleFunc("/", index)
	http.HandleFunc("/list", list)
	//登陆界面
	http.HandleFunc("/login", login)
	//监听端口
	http.ListenAndServe(":8080", nil)
}

//首页跳转
func index(w http.ResponseWriter, r *http.Request) {
	if r.URL.RequestURI()=="/favicon.ico"{
		return
	}

	auth := getCookie(r, "auth")
	if !strings.Contains(auth, "|") {
		http.Redirect(w, r, "/login", 302)
	} else {
		authStrings := strings.Split(auth, "|")
		res := tools.CheckEmailPassword(authStrings[0], authStrings[1], authStrings[2])
		if res {
			http.Redirect(w, r, "/list", 302)
		} else {
			http.Redirect(w, r, "/login", 302)
		}
	}
}

//输出列表
func list(w http.ResponseWriter, r *http.Request) {

	auth := getCookie(r, "auth")
	authStrings := strings.Split(auth, "|")

	render := new(IndexData)
	folders := tools.GetFolders(authStrings[0], authStrings[1], authStrings[2])
	render.Folders = folders
	mails := tools.GetFolderMail(authStrings[0], authStrings[1], authStrings[2], "INBOX", 10)
	render.Mails = mails
	t, _ := template.ParseFiles("./tmpl/index.html")
	t.Execute(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 = "连接或验证失败"
			t, _ := template.ParseFiles("./tmpl/login.html")
			t.Execute(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 {
		t, _ := template.ParseFiles("./tmpl/login.html")
		t.Execute(w, nil)
	}
}

//加密cookie
//func authCookie(){
//
//}
//获取cookie
func getCookie(r *http.Request, name string) string {
	cookies := r.Cookies()
	for _, cookie := range cookies {
		if cookie.Name == name {
			return cookie.Value
		}
	}
	return ""
}