feat: initial release
This commit is contained in:
parent
a3cf21f5bd
commit
761174d035
41 changed files with 2008 additions and 217 deletions
80
internal/web/server/server.go
Normal file
80
internal/web/server/server.go
Normal file
|
|
@ -0,0 +1,80 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
"net/http"
|
||||
|
||||
"git.wzray.com/homelab/mastermind/internal/types"
|
||||
"git.wzray.com/homelab/mastermind/internal/web/middleware"
|
||||
"github.com/rs/zerolog/log"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
mux *http.ServeMux
|
||||
httpServer http.Server
|
||||
}
|
||||
|
||||
func NewServer(addr string, middleware middleware.Middleware) *Server {
|
||||
mux := http.NewServeMux()
|
||||
s := &Server{
|
||||
mux: mux,
|
||||
httpServer: http.Server{
|
||||
Addr: addr,
|
||||
Handler: middleware.Handler(mux),
|
||||
},
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
func (s *Server) Listen() error {
|
||||
return s.httpServer.ListenAndServe()
|
||||
}
|
||||
|
||||
func (s *Server) Shutdown(ctx context.Context) error {
|
||||
return s.httpServer.Shutdown(ctx)
|
||||
}
|
||||
|
||||
func (s *Server) handleFunc(route types.Route) func(w http.ResponseWriter, r *http.Request) {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
log.Debug(). // TODO: make this a middleware
|
||||
Str("method", r.Method).
|
||||
Str("path", r.URL.Path).
|
||||
Str("query", r.URL.RawQuery).
|
||||
Str("remoteAddr", r.RemoteAddr).
|
||||
Send()
|
||||
|
||||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||||
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
w.Write(fail("read request body: %v", err))
|
||||
log.Err(err).Msg("unable to read request body")
|
||||
return
|
||||
}
|
||||
|
||||
raw, err := route.Handle(body)
|
||||
if err != nil {
|
||||
w.Write(fail("handle request: %v", err))
|
||||
log.Err(err).Msg("unable to handle request")
|
||||
return
|
||||
}
|
||||
|
||||
data, err := ok(raw)
|
||||
if err != nil {
|
||||
w.Write(fail("marshal response: %v", err))
|
||||
log.Err(err).Msg("unable to marshal response")
|
||||
return
|
||||
}
|
||||
|
||||
w.Write(data)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) Register(endpoint types.Route) {
|
||||
s.mux.HandleFunc(endpoint.Path(), s.handleFunc(endpoint))
|
||||
}
|
||||
|
||||
func (s *Server) RegisterRaw(method string, pattern string, handler func(http.ResponseWriter, *http.Request)) {
|
||||
s.mux.HandleFunc(method+" "+pattern, handler)
|
||||
}
|
||||
23
internal/web/server/util.go
Normal file
23
internal/web/server/util.go
Normal file
|
|
@ -0,0 +1,23 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
||||
"git.wzray.com/homelab/mastermind/internal/types"
|
||||
)
|
||||
|
||||
func fail(format string, a ...any) []byte {
|
||||
r, _ := json.Marshal(types.Response[string]{
|
||||
Ok: false,
|
||||
Err: fmt.Sprintf(format, a...),
|
||||
})
|
||||
return r
|
||||
}
|
||||
|
||||
func ok[T any](data T) ([]byte, error) {
|
||||
return json.Marshal(types.Response[T]{
|
||||
Ok: true,
|
||||
Data: data,
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue