1
0
Fork 0

feat: initial release

This commit is contained in:
Arthur K. 2026-01-17 18:14:50 +03:00
parent a3cf21f5bd
commit 1e0ee5bffe
Signed by: wzray
GPG key ID: B97F30FDC4636357
40 changed files with 2007 additions and 217 deletions

View 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)
}

View 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...),
}
}

View 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)
}

View 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,
})
}