109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package httpserver
|
|
|
|
import (
|
|
"embed"
|
|
"html/template"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
|
|
"github.com/tim/md-hub-secure/apps/server/internal/config"
|
|
"github.com/tim/md-hub-secure/apps/server/internal/docs"
|
|
"github.com/tim/md-hub-secure/apps/server/internal/realtime"
|
|
"github.com/tim/md-hub-secure/apps/server/internal/store"
|
|
)
|
|
|
|
//go:embed templates/*.gohtml static/*
|
|
var assets embed.FS
|
|
|
|
type Dependencies struct {
|
|
Config config.Config
|
|
Logger *slog.Logger
|
|
Documents *docs.Service
|
|
Repository *docs.Repository
|
|
ContentStore *store.ContentStore
|
|
Hub *realtime.Hub
|
|
}
|
|
|
|
type Server struct {
|
|
config config.Config
|
|
logger *slog.Logger
|
|
documents *docs.Service
|
|
repository *docs.Repository
|
|
contentStore *store.ContentStore
|
|
hub *realtime.Hub
|
|
templates *template.Template
|
|
webEnabled bool
|
|
}
|
|
|
|
func New(deps Dependencies) (http.Handler, error) {
|
|
templates, err := template.New("").Funcs(template.FuncMap{
|
|
"trimMd": func(path string) string {
|
|
return strings.TrimSuffix(path, ".md")
|
|
},
|
|
}).ParseFS(assets, "templates/*.gohtml")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
server := &Server{
|
|
config: deps.Config,
|
|
logger: deps.Logger,
|
|
documents: deps.Documents,
|
|
repository: deps.Repository,
|
|
contentStore: deps.ContentStore,
|
|
hub: deps.Hub,
|
|
templates: templates,
|
|
}
|
|
|
|
if _, err := os.Stat(deps.Config.Web.DistDir); err == nil {
|
|
server.webEnabled = true
|
|
}
|
|
|
|
router := chi.NewRouter()
|
|
router.Use(middleware.RequestID)
|
|
router.Use(middleware.RealIP)
|
|
router.Use(middleware.Recoverer)
|
|
router.Use(server.timeoutExceptWebSocket(30 * time.Second))
|
|
router.Use(server.requestLogger)
|
|
router.Use(server.securityHeaders)
|
|
|
|
router.Get("/", server.handleIndex)
|
|
router.Get("/health", server.handleHealth)
|
|
router.Get("/ws", server.handleWebSocket)
|
|
router.Post("/api/admin/login", server.handleAdminLogin)
|
|
router.Get("/api/admin/users", server.handleAdminUsers)
|
|
router.Post("/api/admin/users", server.handleAdminCreateUser)
|
|
router.Get("/api/admin/workspace", server.handleAdminWorkspace)
|
|
router.Post("/api/admin/workspace/sync", server.handleAdminWorkspaceSync)
|
|
router.Get("/docs", server.handleDocsIndex)
|
|
router.Get("/docs/*", server.handleDocument)
|
|
router.Post("/api/uploads", server.handleUpload)
|
|
router.Get("/attachments/{hash}", server.handleAttachment)
|
|
router.Handle("/static/*", http.StripPrefix("/static/", http.FileServer(mustSub("static"))))
|
|
|
|
if server.webEnabled {
|
|
router.Get("/app", server.handleAppRedirect)
|
|
router.Handle("/app/*", http.StripPrefix("/app/", server.appFileServer()))
|
|
}
|
|
|
|
return router, nil
|
|
}
|
|
|
|
func (s *Server) handleAppRedirect(w http.ResponseWriter, r *http.Request) {
|
|
http.Redirect(w, r, "/app/", http.StatusMovedPermanently)
|
|
}
|
|
|
|
func mustSub(path string) http.FileSystem {
|
|
sub, err := fsSub(path)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return http.FS(sub)
|
|
}
|