boltrest/boltrest-server/web_server.go

78 lines
1.9 KiB
Go
Raw Normal View History

2016-02-23 03:23:53 +00:00
package main
import (
"encoding/json"
"log"
"net/http"
"github.com/devict/magopie"
"github.com/devict/magopie/vendor/github.com/gorilla/mux"
"github.com/devict/magopie/vendor/github.com/justinas/alice"
)
// WebServer ...
type WebServer struct {
addr string
tlsCert string
tlsKey string
adminMode bool
}
// Listen makes it listen for connetions
func (s *WebServer) Listen() {
if s.tlsCert != "" && s.tlsKey != "" {
log.Printf("Listening for HTTPS on %s with key %s and cert %s", s.addr, s.tlsKey, s.tlsCert)
log.Fatal(http.ListenAndServeTLS(s.addr, s.tlsCert, s.tlsKey, s.router(a)))
} else {
log.Printf("Listening for HTTP on %s", *addr)
log.Fatal(http.ListenAndServe(*addr, s.router(a)))
}
}
// Router defines all of the routes
func (s *WebServer) Router() http.Handler {
r := mux.NewRouter()
r.HandleFunc("/", s.handleAllSites).Methods("GET")
r.HandleFunc("/genAPIKey/{db}", s.genAPIKey).Methods("POST")
chain := alice.New(mwLogger, mwAuthenticationCheck(a.key)).Then(r)
return chain
}
func (s *WebServer) handleRequest(w http.ResponseWriter, r *http.Request) {
err := json.NewEncoder(w).Encode()
if err != nil {
log.Println(err)
}
}
func (s *WebServer) genAPIKey(w http.ResponseWriter, r *http.Request) {
err := json.NewEncoder(w).Encode()
if err != nil {
log.Println(err)
}
}
func (s *WebServer) mwLogger(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
log.Println("Serving", r.Method, r.URL.String(), "to", r.RemoteAddr)
next.ServeHTTP(w, r)
})
}
func (s *WebServer) mwAuthenticationCheck(key string) func(http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if !magopie.CheckMAC(r.Header.Get("X-Request-ID"), r.Header.Get("X-HMAC"), key) {
log.Println("Request failed HMAC")
w.WriteHeader(http.StatusUnauthorized)
return
}
next.ServeHTTP(w, r)
})
}
}