57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package response
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/Teajey/sqlstore"
|
|
)
|
|
|
|
type ServeMux struct {
|
|
inner *http.ServeMux
|
|
store *sqlstore.Store
|
|
}
|
|
|
|
func NewServeMux(store *sqlstore.Store) *ServeMux {
|
|
return &ServeMux{
|
|
inner: http.NewServeMux(),
|
|
store: store,
|
|
}
|
|
}
|
|
|
|
func (m *ServeMux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
m.inner.ServeHTTP(w, r)
|
|
}
|
|
|
|
type Handler interface {
|
|
ServeHTTP(h http.Header, r *Request) Response
|
|
}
|
|
|
|
type HandlerFunc func(h http.Header, r *Request) Response
|
|
|
|
func (m *ServeMux) HandleFunc(pattern string, handler HandlerFunc) {
|
|
m.inner.HandleFunc(pattern, func(w http.ResponseWriter, stdReq *http.Request) {
|
|
r := wrapStdRequest(m.store, stdReq)
|
|
|
|
response := handler(w.Header(), &r)
|
|
|
|
err := response.Write(w, stdReq)
|
|
if err != nil {
|
|
response.Data = struct{ Message error }{err}
|
|
response.HtmlTemplateName = "error_page.gotmpl"
|
|
response.Status = http.StatusInternalServerError
|
|
} else {
|
|
return
|
|
}
|
|
err = response.Write(w, stdReq)
|
|
if err != nil {
|
|
log.Printf("Failed to write response.Response to bytes: %s\n", err)
|
|
http.Error(w, "Failed to write response", http.StatusInternalServerError)
|
|
}
|
|
})
|
|
}
|
|
|
|
func (m *ServeMux) Handle(pattern string, handler Handler) {
|
|
m.HandleFunc(pattern, handler.ServeHTTP)
|
|
}
|