baby-nas/internal/web/router.go

package web

import (
	"net/http"
	"strings"

	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
)

func NewRouter(s *Server) chi.Router {
	r := chi.NewRouter()
	r.Use(middleware.RequestID)
	r.Use(middleware.RealIP)
	r.Use(middleware.Logger)
	r.Use(middleware.Recoverer)

	r.Route("/api", func(api chi.Router) {
		// Public auth endpoints.
		api.Post("/auth/login", s.handleLogin)
		api.Post("/auth/logout", s.handleLogout)
		api.Get("/auth/status", s.handleAuthStatus)

		// Everything below requires authentication.
		api.Group(func(protected chi.Router) {
			protected.Use(s.RequireAuth)

			protected.Get("/dirty", s.handleListDirty)
			protected.Post("/apply", s.handleApply)
			protected.Get("/apply/log", s.handleApplyLog)
			protected.Get("/system/status", s.handleSystemStatus)
			protected.Get("/version", s.handleVersion)
			protected.Put("/auth/password", s.handleChangePassword)

			protected.Get("/system/watched-mounts", s.handleListWatchedMounts)
			protected.Post("/system/watched-mounts", s.handleCreateWatchedMount)
			protected.Delete("/system/watched-mounts/{id}", s.handleDeleteWatchedMount)

			protected.Get("/import/status", s.handleImportStatus)
			protected.Post("/import/samba", s.handleImportSamba)
			protected.Post("/import/nfs", s.handleImportNFS)
			protected.Post("/import/users", s.handleImportUsers)

			protected.Route("/samba/shares", func(shares chi.Router) {
				shares.Get("/", s.handleListSambaShares)
				shares.Post("/", s.handleCreateSambaShare)
				shares.Route("/{id}", func(item chi.Router) {
					item.Get("/", s.handleGetSambaShare)
					item.Put("/", s.handleUpdateSambaShare)
					item.Delete("/", s.handleDeleteSambaShare)
				})
			})

			protected.Route("/nfs/exports", func(exports chi.Router) {
				exports.Get("/", s.handleListNFSExports)
				exports.Post("/", s.handleCreateNFSExport)
				exports.Route("/{id}", func(item chi.Router) {
					item.Get("/", s.handleGetNFSExport)
					item.Put("/", s.handleUpdateNFSExport)
					item.Delete("/", s.handleDeleteNFSExport)
				})
			})

			protected.Route("/users", func(users chi.Router) {
				users.Get("/", s.handleListUsers)
				users.Post("/", s.handleCreateUser)
				users.Route("/{id}", func(item chi.Router) {
					item.Get("/", s.handleGetUser)
					item.Put("/", s.handleUpdateUser)
					item.Delete("/", s.handleDeleteUser)
				})
			})

			protected.Route("/files", func(files chi.Router) {
				files.Get("/capabilities", s.handleCapabilities)
				files.Get("/roots", s.handleListRoots)
				files.Get("/", s.handleList)
				files.Get("/info", s.handleInfo)
				files.Post("/mkdir", s.handleMkdir)
				files.Post("/rename", s.handleRename)
				files.Post("/chmod", s.handleChmod)
				files.Post("/chown", s.handleChown)
				files.Delete("/", s.handleDelete)
				files.Post("/upload", s.handleUpload)
				files.Get("/download", s.handleDownload)
				files.Get("/preview", s.handlePreview)
				files.Get("/search", s.handleSearch)
			})

			protected.Route("/storage", func(storageRouter chi.Router) {
			storageRouter.Get("/capabilities", s.handleStorageCapabilities)
			storageRouter.Get("/disk-usage", s.handleStorageDiskUsage)
			storageRouter.Get("/config", s.handleStorageGetConfig)
				storageRouter.Put("/config", s.handleStorageUpdateConfig)
				storageRouter.Get("/conf-file", s.handleStorageGetConfFile)
				storageRouter.Route("/jobs", func(j chi.Router) {
					j.Get("/", s.handleStorageListJobs)
					j.Post("/", s.handleStorageStartJob)
					j.Route("/{id}", func(item chi.Router) {
						item.Get("/", s.handleStorageGetJob)
						item.Post("/cancel", s.handleStorageCancelJob)
						item.Get("/stream", s.handleStorageJobStream)
					})
				})
			})
		})
	})

	r.NotFound(s.handleStatic)
	r.MethodNotAllowed(func(w http.ResponseWriter, _ *http.Request) {
		writeError(w, http.StatusMethodNotAllowed, "method not allowed")
	})

	return r
}

func (s *Server) handleStatic(w http.ResponseWriter, r *http.Request) {
	if r.Method != http.MethodGet && r.Method != http.MethodHead {
		writeError(w, http.StatusMethodNotAllowed, "method not allowed")
		return
	}

	path := strings.TrimPrefix(r.URL.Path, "/")
	if path == "" {
		path = "index.html"
	}

	data, err := Dist.ReadFile("dist/" + path)
	if err != nil {
		if path != "index.html" {
			data, err = Dist.ReadFile("dist/index.html")
		}
		if err != nil {
			http.NotFound(w, r)
			return
		}
	}

	if strings.HasSuffix(path, ".html") {
		w.Header().Set("Content-Type", "text/html; charset=utf-8")
	} else if strings.HasSuffix(path, ".js") {
		w.Header().Set("Content-Type", "application/javascript")
	} else if strings.HasSuffix(path, ".css") {
		w.Header().Set("Content-Type", "text/css")
	}

	w.WriteHeader(http.StatusOK)
	_, _ = w.Write(data)
}