mirror of
https://gitlab.com/ric_harvey/MailHog.git
synced 2024-11-24 06:44:04 +00:00
85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package http
|
|
|
|
import (
|
|
"regexp"
|
|
"net/http"
|
|
"strings"
|
|
"github.com/ian-kent/MailHog/mailhog/config"
|
|
"github.com/ian-kent/MailHog/mailhog/http/api"
|
|
"github.com/ian-kent/MailHog/mailhog/http/handler"
|
|
)
|
|
|
|
var exitChannel chan int
|
|
var cfg *config.Config
|
|
|
|
// TODO clean this mess up
|
|
|
|
func web_exit(w http.ResponseWriter, r *http.Request, route *handler.Route) {
|
|
web_headers(w)
|
|
w.Write([]byte("Exiting MailHog!"))
|
|
exitChannel <- 1
|
|
}
|
|
|
|
func web_index(w http.ResponseWriter, r *http.Request, route *handler.Route) {
|
|
web_headers(w)
|
|
data, _ := cfg.Assets("assets/templates/index.html")
|
|
w.Write([]byte(web_render(string(data))))
|
|
}
|
|
|
|
func web_jscontroller(w http.ResponseWriter, r *http.Request, route *handler.Route) {
|
|
w.Header().Set("Content-Type", "text/javascript")
|
|
data, _ := cfg.Assets("assets/js/controllers.js")
|
|
w.Write(data)
|
|
}
|
|
|
|
func web_imgcontroller(w http.ResponseWriter, r *http.Request, route *handler.Route) {
|
|
w.Header().Set("Content-Type", "image/png")
|
|
data, _ := cfg.Assets("assets/images/hog.png")
|
|
w.Write(data)
|
|
}
|
|
|
|
func web_img_github(w http.ResponseWriter, r *http.Request, route *handler.Route) {
|
|
w.Header().Set("Content-Type", "image/png")
|
|
data, _ := cfg.Assets("assets/images/github.png")
|
|
w.Write(data)
|
|
}
|
|
|
|
func web_img_ajaxloader(w http.ResponseWriter, r *http.Request, route *handler.Route) {
|
|
w.Header().Set("Content-Type", "image/gif")
|
|
data, _ := cfg.Assets("assets/images/ajax-loader.gif")
|
|
w.Write(data)
|
|
}
|
|
|
|
func web_render(content string) string {
|
|
data, _ := cfg.Assets("assets/templates/layout.html")
|
|
layout := string(data)
|
|
html := strings.Replace(layout, "<%= content %>", content, -1)
|
|
// TODO clean this up
|
|
html = strings.Replace(html, "<%= config[Hostname] %>", cfg.Hostname, -1)
|
|
return html
|
|
}
|
|
|
|
func web_headers(w http.ResponseWriter) {
|
|
w.Header().Set("Content-Type", "text/html")
|
|
}
|
|
|
|
func Start(exitCh chan int, conf *config.Config) {
|
|
exitChannel = exitCh
|
|
cfg = conf
|
|
|
|
server := &http.Server{
|
|
Addr: conf.HTTPBindAddr,
|
|
Handler: &handler.RegexpHandler{},
|
|
}
|
|
|
|
server.Handler.(*handler.RegexpHandler).HandleFunc([]string{"GET"},regexp.MustCompile("^/exit/?$"), web_exit)
|
|
server.Handler.(*handler.RegexpHandler).HandleFunc([]string{"GET"},regexp.MustCompile("^/js/controllers.js$"), web_jscontroller)
|
|
server.Handler.(*handler.RegexpHandler).HandleFunc([]string{"GET"},regexp.MustCompile("^/images/hog.png$"), web_imgcontroller)
|
|
server.Handler.(*handler.RegexpHandler).HandleFunc([]string{"GET"},regexp.MustCompile("^/images/github.png$"), web_img_github)
|
|
server.Handler.(*handler.RegexpHandler).HandleFunc([]string{"GET"},regexp.MustCompile("^/images/ajax-loader.gif$"), web_img_ajaxloader)
|
|
server.Handler.(*handler.RegexpHandler).HandleFunc([]string{"GET"},regexp.MustCompile("^/$"), web_index)
|
|
|
|
api.CreateAPIv1(exitCh, conf, server)
|
|
|
|
server.ListenAndServe()
|
|
}
|