package server import ( "context" "fmt" "log" "net" "net/http" "forge.cadoles.com/Cadoles/emissary/internal/auth" "forge.cadoles.com/Cadoles/emissary/internal/auth/agent" "forge.cadoles.com/Cadoles/emissary/internal/auth/user" "forge.cadoles.com/Cadoles/emissary/internal/config" "forge.cadoles.com/Cadoles/emissary/internal/datastore" "forge.cadoles.com/Cadoles/emissary/internal/jwk" "github.com/go-chi/chi" "github.com/go-chi/chi/middleware" "github.com/go-chi/cors" "github.com/pkg/errors" "gitlab.com/wpetit/goweb/logger" ) type Server struct { conf config.ServerConfig agentRepo datastore.AgentRepository } 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.conf.HTTP.Host, s.conf.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.conf.PrivateKeyPath), 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.conf.CORS.AllowedOrigins, AllowedMethods: s.conf.CORS.AllowedMethods, AllowCredentials: bool(s.conf.CORS.AllowCredentials), AllowedHeaders: s.conf.CORS.AllowedHeaders, Debug: bool(s.conf.CORS.Debug), }) router.Use(corsMiddleware.Handler) router.Route("/api/v1", func(r chi.Router) { r.Post("/register", s.registerAgent) r.Group(func(r chi.Router) { r.Use(auth.Middleware( user.NewAuthenticator(keys, string(s.conf.Issuer)), agent.NewAuthenticator(s.agentRepo), )) r.Route("/agents", func(r chi.Router) { r.Get("/", s.queryAgents) r.Get("/{agentID}", s.getAgent) r.Put("/{agentID}", s.updateAgent) r.Delete("/{agentID}", s.deleteAgent) r.Get("/{agentID}/specs", s.getAgentSpecs) r.Post("/{agentID}/specs", s.updateSpec) r.Delete("/{agentID}/specs", s.deleteSpec) }) }) }) 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 New(funcs ...OptionFunc) *Server { opt := defaultOption() for _, fn := range funcs { fn(opt) } return &Server{ conf: opt.Config, } }