feat: initial release
This commit is contained in:
parent
a3cf21f5bd
commit
761174d035
41 changed files with 2008 additions and 217 deletions
108
internal/web/client/client.go
Normal file
108
internal/web/client/client.go
Normal file
|
|
@ -0,0 +1,108 @@
|
|||
package client
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"time"
|
||||
|
||||
"git.wzray.com/homelab/mastermind/internal/types"
|
||||
"git.wzray.com/homelab/mastermind/internal/web/middleware"
|
||||
)
|
||||
|
||||
type client struct {
|
||||
http *http.Client
|
||||
middleware middleware.Middleware
|
||||
}
|
||||
|
||||
var defaultClient *client
|
||||
const timeout = time.Duration(2) * time.Second
|
||||
|
||||
func (c *client) makeRequest(method string, host string, path types.Path, data any, out any) error {
|
||||
var body io.Reader
|
||||
if data != nil {
|
||||
raw, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshal body: %w", err)
|
||||
}
|
||||
body = bytes.NewReader(raw)
|
||||
}
|
||||
|
||||
uri := (&url.URL{
|
||||
Scheme: "http",
|
||||
Host: host,
|
||||
Path: path.String(),
|
||||
}).String()
|
||||
|
||||
r, err := http.NewRequest(method, uri, body)
|
||||
if err != nil {
|
||||
return fmt.Errorf("build http request: %w", err)
|
||||
}
|
||||
|
||||
if body != nil {
|
||||
r.Header.Set("Content-Type", "application/json; charset=utf-8")
|
||||
}
|
||||
|
||||
if err := c.middleware.Client(r); err != nil {
|
||||
return fmt.Errorf("apply middleware: %w", err)
|
||||
}
|
||||
|
||||
resp, err := c.http.Do(r)
|
||||
if err != nil {
|
||||
return fmt.Errorf("send request: %w", err)
|
||||
}
|
||||
|
||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||
b, _ := io.ReadAll(resp.Body)
|
||||
return fmt.Errorf("http %d: %s", resp.StatusCode, string(b))
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
if out != nil {
|
||||
if err := json.NewDecoder(resp.Body).Decode(out); err != nil {
|
||||
return fmt.Errorf("decode body: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
io.Copy(io.Discard, resp.Body)
|
||||
return nil
|
||||
}
|
||||
|
||||
func Init(mw middleware.Middleware) {
|
||||
if defaultClient != nil {
|
||||
panic("web.client: Init called twice")
|
||||
}
|
||||
|
||||
defaultClient = &client{
|
||||
http: &http.Client{
|
||||
Timeout: timeout,
|
||||
},
|
||||
middleware: mw,
|
||||
}
|
||||
}
|
||||
|
||||
func request[Out any, In any](method string, host string, path types.Path, data In) (*Out, error) {
|
||||
out := &types.Response[Out]{}
|
||||
err := defaultClient.makeRequest(method, host, path, data, out)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if !out.Ok {
|
||||
return nil, errors.New(out.Err)
|
||||
}
|
||||
|
||||
return &out.Data, err
|
||||
}
|
||||
|
||||
func Get[Out any](host string, path types.Path) (*Out, error) { // TODO: out should not be a pointer
|
||||
return request[Out, any](http.MethodGet, host, path, nil)
|
||||
}
|
||||
|
||||
func Post[Out any, In any](host string, path types.Path, data In) (*Out, error) {
|
||||
return request[Out](http.MethodPost, host, path, data)
|
||||
}
|
||||
53
internal/web/middleware/middleware.go
Normal file
53
internal/web/middleware/middleware.go
Normal file
|
|
@ -0,0 +1,53 @@
|
|||
package middleware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"slices"
|
||||
)
|
||||
|
||||
type Middleware interface {
|
||||
Client(r *http.Request) error
|
||||
Handler(http.Handler) http.Handler
|
||||
}
|
||||
|
||||
type BaseMiddleware struct{}
|
||||
|
||||
func (BaseMiddleware) Client(*http.Request) error { return nil }
|
||||
|
||||
func (BaseMiddleware) Handler(h http.Handler) http.Handler { return h }
|
||||
|
||||
type MiddlewareBuilder struct {
|
||||
middlewares []Middleware
|
||||
}
|
||||
|
||||
func (b *MiddlewareBuilder) Use(middleware Middleware) *MiddlewareBuilder {
|
||||
b.middlewares = append(b.middlewares, middleware)
|
||||
return b
|
||||
}
|
||||
|
||||
type middlewareProxy struct {
|
||||
middlewares []Middleware
|
||||
}
|
||||
|
||||
func (p *middlewareProxy) Client(r *http.Request) error {
|
||||
for _, m := range p.middlewares {
|
||||
if err := m.Client(r); err != nil {
|
||||
return fmt.Errorf("%T: %w", m, err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *middlewareProxy) Handler(h http.Handler) http.Handler {
|
||||
for _, f := range slices.Backward(p.middlewares) {
|
||||
h = f.Handler(h)
|
||||
}
|
||||
return h
|
||||
}
|
||||
|
||||
func (b *MiddlewareBuilder) Prepare() Middleware {
|
||||
return &middlewareProxy{
|
||||
middlewares: append([]Middleware(nil), b.middlewares...),
|
||||
}
|
||||
}
|
||||
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