initial commit

pkg/api/routes.go

@@ -0,0 +1,112 @@
+package api
+
+import (
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+
+	"jagacloud/node-agent/pkg/compute/libvirt"
+	"jagacloud/node-agent/pkg/config"
+	"jagacloud/node-agent/pkg/containers/lxc"
+	"jagacloud/node-agent/pkg/containers/podman"
+	"jagacloud/node-agent/pkg/state"
+	"jagacloud/node-agent/pkg/tasks"
+)
+
+// Services bundles runtime services passed to handlers.
+type Services struct {
+	Tasks     *tasks.Registry
+	Libvirt   libvirtClient
+	LXC       lxcManager
+	Podman    podmanClient
+	Store     *state.Store
+	Validator ResourceValidator
+}
+
+// StoreLock returns an unlock function for the given ID; no-op if store is nil.
+func (s Services) StoreLock(id string) func() {
+	if s.Store == nil {
+		return func() {}
+	}
+	return s.Store.Lock(id)
+}
+
+// ResourceValidator validates network/storage resources before actions.
+type ResourceValidator interface {
+	ValidateVM(libvirt.VMSpec) error
+	ValidateCT(lxc.Spec) error
+}
+
+// Interfaces narrowed for injection/testing.
+type libvirtClient interface {
+	ListVMs() ([]libvirt.VM, error)
+	CreateVM(libvirt.VMSpec) (libvirt.VM, error)
+	StartVM(id string) error
+	StopVM(id string) error
+	RebootVM(id string) error
+	DeleteVM(id string) error
+}
+
+type lxcManager interface {
+	List() ([]lxc.Container, error)
+	Create(lxc.Spec) (lxc.Container, error)
+	Start(id string) error
+	Stop(id string) error
+	Delete(id string) error
+}
+
+type podmanClient interface {
+	List(ctID string) ([]podman.OCIContainer, error)
+	Create(ctID string, spec podman.CreateSpec) (podman.OCIContainer, error)
+	Start(ctID, cid string) error
+	Stop(ctID, cid string) error
+	Delete(ctID, cid string) error
+}
+
+// RegisterRoutes wires HTTP routes.
+func RegisterRoutes(r *chi.Mux, cfg config.Config, svc Services) {
+	r.Use(middleware.RequestID)
+	r.Use(middleware.Logger)
+	if cfg.AuthToken != "" {
+		r.Use(authMiddleware(cfg.AuthToken))
+	}
+
+	r.Get("/api/v1/node", handleNodeInfo(cfg, svc))
+
+	r.Route("/api/v1/vms", func(r chi.Router) {
+		r.Get("/", handleListVMs(cfg, svc))
+		r.Post("/", handleCreateVM(cfg, svc))
+		r.Route("/{id}", func(r chi.Router) {
+			r.Get("/", handleGetVM(cfg, svc))
+			r.Post("/start", handleStartVM(cfg, svc))
+			r.Post("/stop", handleStopVM(cfg, svc))
+			r.Post("/reboot", handleRebootVM(cfg, svc))
+			r.Post("/delete", handleDeleteVM(cfg, svc))
+		})
+	})
+
+	r.Route("/api/v1/containers", func(r chi.Router) {
+		r.Get("/", handleListCT(cfg, svc))
+		r.Post("/", handleCreateCT(cfg, svc))
+		r.Route("/{id}", func(r chi.Router) {
+			r.Get("/", handleGetCT(cfg, svc))
+			r.Post("/start", handleStartCT(cfg, svc))
+			r.Post("/stop", handleStopCT(cfg, svc))
+			r.Post("/delete", handleDeleteCT(cfg, svc))
+
+			r.Route("/oci", func(r chi.Router) {
+				r.Get("/", handleListOCI(cfg, svc))
+				r.Post("/", handleCreateOCI(cfg, svc))
+				r.Route("/{cid}", func(r chi.Router) {
+					r.Get("/", handleGetOCI(cfg, svc))
+					r.Post("/start", handleStartOCI(cfg, svc))
+					r.Post("/stop", handleStopOCI(cfg, svc))
+					r.Post("/delete", handleDeleteOCI(cfg, svc))
+				})
+			})
+		})
+	})
+
+	r.Route("/api/v1/tasks", func(r chi.Router) {
+		r.Get("/{id}", handleGetTask(cfg, svc))
+	})
+}