feat: initial commit

internal/admin/authz.go

@@ -0,0 +1,101 @@
+package admin
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/auth"
+	"forge.cadoles.com/cadoles/bouncer/internal/auth/jwt"
+	"github.com/pkg/errors"
+	"gitlab.com/wpetit/goweb/api"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+var ErrCodeForbidden api.ErrorCode = "forbidden"
+
+func assertReadAccess(h http.Handler) http.Handler {
+	fn := func(w http.ResponseWriter, r *http.Request) {
+		reqUser, ok := assertRequestUser(w, r)
+		if !ok {
+			return
+		}
+
+		switch user := reqUser.(type) {
+		case *jwt.User:
+			role := user.Role()
+			if role == jwt.RoleReader || role == jwt.RoleWriter {
+				h.ServeHTTP(w, r)
+
+				return
+			}
+
+		default:
+			logUnexpectedUserType(r.Context(), reqUser)
+		}
+
+		forbidden(w, r)
+	}
+
+	return http.HandlerFunc(fn)
+}
+
+func assertWriteAccess(h http.Handler) http.Handler {
+	fn := func(w http.ResponseWriter, r *http.Request) {
+		reqUser, ok := assertRequestUser(w, r)
+		if !ok {
+			return
+		}
+
+		switch user := reqUser.(type) {
+		case *jwt.User:
+			role := user.Role()
+			if role == jwt.RoleWriter {
+				h.ServeHTTP(w, r)
+
+				return
+			}
+
+		default:
+			logUnexpectedUserType(r.Context(), reqUser)
+		}
+
+		forbidden(w, r)
+	}
+
+	return http.HandlerFunc(fn)
+}
+
+func assertRequestUser(w http.ResponseWriter, r *http.Request) (auth.User, bool) {
+	ctx := r.Context()
+	user, err := auth.CtxUser(ctx)
+	if err != nil {
+		logger.Error(ctx, "could not retrieve user", logger.E(errors.WithStack(err)))
+
+		forbidden(w, r)
+
+		return nil, false
+	}
+
+	if user == nil {
+		forbidden(w, r)
+
+		return nil, false
+	}
+
+	return user, true
+}
+
+func forbidden(w http.ResponseWriter, r *http.Request) {
+	logger.Warn(r.Context(), "forbidden", logger.F("path", r.URL.Path))
+
+	api.ErrorResponse(w, http.StatusForbidden, ErrCodeForbidden, nil)
+}
+
+func logUnexpectedUserType(ctx context.Context, user auth.User) {
+	logger.Error(
+		ctx, "unexpected user type",
+		logger.F("subject", user.Subject()),
+		logger.F("type", fmt.Sprintf("%T", user)),
+	)
+}

internal/admin/init.go

@@ -0,0 +1,42 @@
+package admin
+
+import (
+	"context"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/setup"
+	"github.com/pkg/errors"
+)
+
+func (s *Server) initRepositories(ctx context.Context) error {
+	if err := s.initQueueRepository(ctx); err != nil {
+		return errors.WithStack(err)
+	}
+
+	if err := s.initProxyRepository(ctx); err != nil {
+		return errors.WithStack(err)
+	}
+
+	return nil
+}
+
+func (s *Server) initQueueRepository(ctx context.Context) error {
+	queueRepository, err := setup.NewQueueRepository(ctx, s.redisConfig)
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	s.queueRepository = queueRepository
+
+	return nil
+}
+
+func (s *Server) initProxyRepository(ctx context.Context) error {
+	proxyRepository, err := setup.NewProxyRepository(ctx, s.redisConfig)
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	s.proxyRepository = proxyRepository
+
+	return nil
+}

internal/admin/option.go

@@ -0,0 +1,31 @@
+package admin
+
+import (
+	"forge.cadoles.com/cadoles/bouncer/internal/config"
+)
+
+type Option struct {
+	ServerConfig config.AdminServerConfig
+	RedisConfig  config.RedisConfig
+}
+
+type OptionFunc func(*Option)
+
+func defaultOption() *Option {
+	return &Option{
+		ServerConfig: config.NewDefaultAdminServerConfig(),
+		RedisConfig:  config.NewDefaultRedisConfig(),
+	}
+}
+
+func WithServerConfig(conf config.AdminServerConfig) OptionFunc {
+	return func(opt *Option) {
+		opt.ServerConfig = conf
+	}
+}
+
+func WithRedisConfig(conf config.RedisConfig) OptionFunc {
+	return func(opt *Option) {
+		opt.RedisConfig = conf
+	}
+}

internal/admin/proxy.go

@@ -0,0 +1,234 @@
+package admin
+
+import (
+	"net/http"
+	"net/url"
+	"strconv"
+	"strings"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/store"
+	"github.com/go-chi/chi/v5"
+	"github.com/pkg/errors"
+	"gitlab.com/wpetit/goweb/api"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+type QueryProxyResponse struct {
+	Proxies []*store.ProxyHeader `json:"proxies"`
+}
+
+func (s *Server) queryProxy(w http.ResponseWriter, r *http.Request) {
+	limit, ok := getIntQueryParam(w, r, "limit", 10)
+	if !ok {
+		return
+	}
+
+	offset, ok := getIntQueryParam(w, r, "offset", 0)
+	if !ok {
+		return
+	}
+
+	options := []store.QueryProxyOptionFunc{
+		store.WithProxyQueryLimit(int(limit)),
+		store.WithProxyQueryOffset(int(offset)),
+	}
+
+	ids, ok := getStringableSliceValues[store.ProxyID](w, r, "ids", nil)
+	if !ok {
+		return
+	}
+
+	if ids != nil {
+		options = append(options, store.WithProxyQueryIDs(ids...))
+	}
+
+	ctx := r.Context()
+
+	proxies, err := s.proxyRepository.QueryProxy(
+		ctx,
+		options...,
+	)
+	if err != nil {
+		logger.Error(ctx, "could not list proxies", logger.E(errors.WithStack(err)))
+		api.ErrorResponse(w, http.StatusInternalServerError, api.ErrCodeUnknownError, nil)
+
+		return
+	}
+
+	api.DataResponse(w, http.StatusOK, QueryProxyResponse{
+		Proxies: proxies,
+	})
+}
+
+type GetProxyResponse struct {
+	Proxy *store.Proxy `json:"proxy"`
+}
+
+func (s *Server) getProxy(w http.ResponseWriter, r *http.Request) {
+	proxyID, ok := getProxyID(w, r)
+	if !ok {
+		return
+	}
+
+	ctx := r.Context()
+
+	proxy, err := s.proxyRepository.GetProxy(ctx, proxyID)
+	if err != nil {
+		if errors.Is(err, store.ErrNotFound) {
+			api.ErrorResponse(w, http.StatusNotFound, api.ErrCodeNotFound, nil)
+
+			return
+		}
+
+		logger.Error(ctx, "could not get proxy", logger.E(errors.WithStack(err)))
+		api.ErrorResponse(w, http.StatusInternalServerError, api.ErrCodeUnknownError, nil)
+
+		return
+	}
+
+	api.DataResponse(w, http.StatusOK, GetProxyResponse{
+		Proxy: proxy,
+	})
+}
+
+type DeleteProxyResponse struct {
+	ProxyID store.ProxyID `json:"proxyId"`
+}
+
+func (s *Server) deleteProxy(w http.ResponseWriter, r *http.Request) {
+	proxyID, ok := getProxyID(w, r)
+	if !ok {
+		return
+	}
+
+	ctx := r.Context()
+
+	if err := s.proxyRepository.DeleteProxy(ctx, proxyID); err != nil {
+		if errors.Is(err, store.ErrNotFound) {
+			api.ErrorResponse(w, http.StatusNotFound, api.ErrCodeNotFound, nil)
+
+			return
+		}
+
+		logger.Error(ctx, "could not delete proxy", logger.E(errors.WithStack(err)))
+		api.ErrorResponse(w, http.StatusInternalServerError, api.ErrCodeUnknownError, nil)
+
+		return
+	}
+
+	api.DataResponse(w, http.StatusOK, DeleteProxyResponse{
+		ProxyID: proxyID,
+	})
+}
+
+type CreateProxyRequest struct {
+	To   string   `json:"to" validate:"required"`
+	From []string `json:"from" validate:"required"`
+}
+
+type CreateProxyResponse struct {
+	Proxy *store.Proxy `json:"proxy"`
+}
+
+func (s *Server) createProxy(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+
+	createProxyReq := &CreateProxyRequest{}
+	if ok := api.Bind(w, r, createProxyReq); !ok {
+		return
+	}
+
+	to, err := url.Parse(createProxyReq.To)
+	if err != nil {
+		logger.Error(r.Context(), "could not parse to parameter", logger.E(errors.WithStack(err)))
+		api.ErrorResponse(w, http.StatusBadRequest, api.ErrCodeMalformedRequest, nil)
+
+		return
+	}
+
+	proxy, err := s.proxyRepository.CreateProxy(ctx, to, createProxyReq.From...)
+	if err != nil {
+		logger.Error(ctx, "could not update agent", logger.E(errors.WithStack(err)))
+		api.ErrorResponse(w, http.StatusInternalServerError, api.ErrCodeUnknownError, nil)
+
+		return
+	}
+
+	api.DataResponse(w, http.StatusOK, struct {
+		Proxy *store.Proxy `json:"proxy"`
+	}{
+		Proxy: proxy,
+	})
+}
+
+type UpdateProxyRequest struct{}
+
+func (s *Server) updateProxy(w http.ResponseWriter, r *http.Request) {
+}
+
+func getProxyID(w http.ResponseWriter, r *http.Request) (store.ProxyID, bool) {
+	rawProxyID := chi.URLParam(r, "proxyID")
+
+	proxyID, err := store.ParseProxyID(rawProxyID)
+	if err != nil {
+		logger.Error(r.Context(), "could not parse proxy id", logger.E(errors.WithStack(err)))
+		api.ErrorResponse(w, http.StatusBadRequest, api.ErrCodeMalformedRequest, nil)
+
+		return "", false
+	}
+
+	return proxyID, true
+}
+
+func getIntQueryParam(w http.ResponseWriter, r *http.Request, param string, defaultValue int64) (int64, bool) {
+	rawValue := r.URL.Query().Get(param)
+	if rawValue != "" {
+		value, err := strconv.ParseInt(rawValue, 10, 64)
+		if err != nil {
+			logger.Error(r.Context(), "could not parse int param", logger.F("param", param), logger.E(errors.WithStack(err)))
+			api.ErrorResponse(w, http.StatusBadRequest, api.ErrCodeMalformedRequest, nil)
+
+			return 0, false
+		}
+
+		return value, true
+	}
+
+	return defaultValue, true
+}
+
+func getStringSliceValues(w http.ResponseWriter, r *http.Request, param string, defaultValue []string) ([]string, bool) {
+	rawValue := r.URL.Query().Get(param)
+	if rawValue != "" {
+		values := strings.Split(rawValue, ",")
+
+		return values, true
+	}
+
+	return defaultValue, true
+}
+
+func getStringableSliceValues[T ~string](w http.ResponseWriter, r *http.Request, param string, defaultValue []T) ([]T, bool) {
+	rawValue := r.URL.Query().Get(param)
+
+	if rawValue != "" {
+		rawValues := strings.Split(rawValue, ",")
+		ids := make([]T, 0, len(rawValues))
+
+		for _, rv := range rawValues {
+			id, err := store.ParseID[T](rv)
+			if err != nil {
+				logger.Error(r.Context(), "could not parse ids slice param", logger.F("param", param), logger.E(errors.WithStack(err)))
+				api.ErrorResponse(w, http.StatusBadRequest, api.ErrCodeMalformedRequest, nil)
+
+				return nil, false
+			}
+
+			ids = append(ids, id)
+		}
+
+		return ids, true
+	}
+
+	return defaultValue, true
+}

internal/admin/server.go

@@ -0,0 +1,140 @@
+package admin
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"net"
+	"net/http"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/auth"
+	"forge.cadoles.com/cadoles/bouncer/internal/auth/jwt"
+	"forge.cadoles.com/cadoles/bouncer/internal/config"
+	"forge.cadoles.com/cadoles/bouncer/internal/jwk"
+	"forge.cadoles.com/cadoles/bouncer/internal/queue"
+	"forge.cadoles.com/cadoles/bouncer/internal/store"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+	"github.com/go-chi/cors"
+	"github.com/pkg/errors"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+type Server struct {
+	serverConfig    config.AdminServerConfig
+	redisConfig     config.RedisConfig
+	queueRepository queue.Repository
+	proxyRepository store.ProxyRepository
+}
+
+func (s *Server) Start(ctx context.Context) (<-chan net.Addr, <-chan error) {
+	errs := make(chan error)
+	addrs := make(chan net.Addr)
+
+	go s.run(ctx, addrs, errs)
+
+	return addrs, errs
+}
+
+func (s *Server) run(parentCtx context.Context, addrs chan net.Addr, errs chan error) {
+	defer func() {
+		close(errs)
+		close(addrs)
+	}()
+
+	ctx, cancel := context.WithCancel(parentCtx)
+	defer cancel()
+
+	if err := s.initRepositories(ctx); err != nil {
+		errs <- errors.WithStack(err)
+
+		return
+	}
+
+	listener, err := net.Listen("tcp", fmt.Sprintf("%s:%d", s.serverConfig.HTTP.Host, s.serverConfig.HTTP.Port))
+	if err != nil {
+		errs <- errors.WithStack(err)
+
+		return
+	}
+
+	addrs <- listener.Addr()
+
+	defer func() {
+		if err := listener.Close(); err != nil && !errors.Is(err, net.ErrClosed) {
+			errs <- errors.WithStack(err)
+		}
+	}()
+
+	go func() {
+		<-ctx.Done()
+
+		if err := listener.Close(); err != nil && !errors.Is(err, net.ErrClosed) {
+			log.Printf("%+v", errors.WithStack(err))
+		}
+	}()
+
+	key, err := jwk.LoadOrGenerate(string(s.serverConfig.Auth.PrivateKey), jwk.DefaultKeySize)
+	if err != nil {
+		errs <- errors.WithStack(err)
+
+		return
+	}
+
+	keys, err := jwk.PublicKeySet(key)
+	if err != nil {
+		errs <- errors.WithStack(err)
+
+		return
+	}
+
+	router := chi.NewRouter()
+
+	router.Use(middleware.Logger)
+
+	corsMiddleware := cors.New(cors.Options{
+		AllowedOrigins:   s.serverConfig.CORS.AllowedOrigins,
+		AllowedMethods:   s.serverConfig.CORS.AllowedMethods,
+		AllowCredentials: bool(s.serverConfig.CORS.AllowCredentials),
+		AllowedHeaders:   s.serverConfig.CORS.AllowedHeaders,
+		Debug:            bool(s.serverConfig.CORS.Debug),
+	})
+
+	router.Use(corsMiddleware.Handler)
+
+	router.Route("/api/v1", func(r chi.Router) {
+		r.Group(func(r chi.Router) {
+			r.Use(auth.Middleware(
+				jwt.NewAuthenticator(keys, string(s.serverConfig.Auth.Issuer), jwt.DefaultAcceptableSkew),
+			))
+
+			r.Route("/proxies", func(r chi.Router) {
+				r.With(assertReadAccess).Get("/", s.queryProxy)
+				r.With(assertWriteAccess).Post("/", s.createProxy)
+				r.With(assertReadAccess).Get("/{proxyID}", s.getProxy)
+				r.With(assertWriteAccess).Put("/{proxyID}", s.updateProxy)
+				r.With(assertWriteAccess).Delete("/{proxyID}", s.deleteProxy)
+			})
+		})
+	})
+
+	logger.Info(ctx, "http server listening")
+
+	if err := http.Serve(listener, router); err != nil && !errors.Is(err, net.ErrClosed) {
+		errs <- errors.WithStack(err)
+	}
+
+	logger.Info(ctx, "http server exiting")
+}
+
+func NewServer(funcs ...OptionFunc) *Server {
+	opt := defaultOption()
+	for _, fn := range funcs {
+		fn(opt)
+	}
+
+	return &Server{
+		serverConfig: opt.ServerConfig,
+		redisConfig:  opt.RedisConfig,
+	}
+}