コード例 #1
0
ファイル: gomail.go プロジェクト: baiyuxiong/gomail
func main() {
	cfg := flag.String("c", "cfg.json", "configuration file")
	version := flag.Bool("v", false, "show version")
	flag.Parse()

	if *version {
		fmt.Println(config.VERSION)
		os.Exit(0)
	}

	//parse config file
	config.Parse(*cfg)

	go http.Start()

	go mail.Start()

	go status.CheckInterval()

	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
	<-sigs

	log.Println("gomail quit.")
	os.Exit(0)
}
コード例 #2
0
ファイル: routes.go プロジェクト: baiyuxiong/gomail
func configRoutes() {
	// GET
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {

		if nil != checkLoggedIn(w, r) {
			return
		}

		queries, err := url.ParseQuery(r.URL.RawQuery)

		var perPage int = 20
		var start int = 0
		if err == nil && len(queries["start"]) > 0 {
			start, _ = strconv.Atoi(queries["start"][0])
		}

		var stop int = start + perPage - 1
		if err == nil && len(queries["stop"]) > 0 {
			stop, _ = strconv.Atoi(queries["stop"][0])
		}

		var currentPage int = start/perPage + 1

		var prePageStart int = 0
		var prePageStop int = 19
		var nextPageStart int = 20
		var nextPageStop int = 39

		var showPrePage = false
		if currentPage > 1 {
			showPrePage = true
			prePageStart = (currentPage - 2) * perPage
			prePageStop = prePageStart + perPage - 1
		}
		nextPageStart = currentPage * perPage
		nextPageStop = nextPageStart + perPage - 1

		logs, err := mail.GetMailLog(start, stop)
		if err != nil {
			w.Write([]byte(err.Error()))
			return
		}
		var mailLogs []model.EmailLog

		for _, l := range logs.([]interface{}) {
			var m model.EmailLog
			err := json.Unmarshal(l.([]byte), &m)
			if err != nil {
				log.Println("HandleFunc: / Unmarshal error - ", err.Error())
			} else {
				mailLogs = append(mailLogs, m)
			}
		}

		data := map[string]interface{}{
			"mailLogs":      mailLogs,
			"start":         start,
			"stop":          stop,
			"currentPage":   currentPage,
			"showPrePage":   showPrePage,
			"prePageStart":  prePageStart,
			"prePageStop":   prePageStop,
			"nextPageStart": nextPageStart,
			"nextPageStop":  nextPageStop,
		}

		Render(w, "home/index.html", data)
	})

	// GET
	http.HandleFunc("/status", func(w http.ResponseWriter, r *http.Request) {

		data := status.Check()

		Render(w, "status/index.html", data)
	})

	// GET
	http.HandleFunc("/login", func(w http.ResponseWriter, r *http.Request) {
		Render(w, "home/login.html", nil)
	})

	http.HandleFunc("/doLogin", func(w http.ResponseWriter, r *http.Request) {
		if !lastErrorTime.IsZero() && (time.Now().Sub(lastErrorTime) < time.Minute) {
			w.Write([]byte("Pls try after " + time.Now().Add(time.Minute).String()))
			return
		}

		r.ParseForm()
		if len(r.Form["password"]) < 1 {
			w.Write([]byte("Need password."))
			return
		}
		if len(r.Form["username"]) < 1 {
			w.Write([]byte("Need username."))
			return
		}

		password := r.Form["password"][0]
		username := r.Form["username"][0]

		if password != config.Config().Password || username != config.Config().Username {
			lastErrorTime = time.Now()
			w.Write([]byte("I love you, Pls don't hurt me."))
			return
		} else {
			storeUsername(w, r, username)
			http.Redirect(w, r, "/", http.StatusFound)
			return
		}
	})

	http.HandleFunc("/logout", func(w http.ResponseWriter, r *http.Request) {
		storeUsername(w, r, "")
		http.Redirect(w, r, "/login", http.StatusFound)
		return
	})

	http.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("ok\n"))
	})

	http.HandleFunc("/version", func(w http.ResponseWriter, r *http.Request) {
		if nil != checkLoggedIn(w, r) {
			return
		}
		w.Write([]byte(fmt.Sprintf("%s\n", config.VERSION)))
	})

	http.HandleFunc("/workdir", func(w http.ResponseWriter, r *http.Request) {
		if nil != checkLoggedIn(w, r) {
			return
		}
		w.Write([]byte(fmt.Sprintf("%s\n", file.SelfDir())))
	})

	http.HandleFunc("/config", func(w http.ResponseWriter, r *http.Request) {
		if nil != checkLoggedIn(w, r) {
			return
		}
		RenderOKJson(w, config.Config())
	})

	http.HandleFunc("/config/reload", func(w http.ResponseWriter, r *http.Request) {
		if nil != checkLoggedIn(w, r) {
			return
		}
		config.Parse(config.ConfigFile)
		RenderOKJson(w, config.Config())
	})
	http.HandleFunc("/public/", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, r.URL.Path[1:])
	})
}