- HTML templates: layout, login, notebook list, notebook view, page viewer - Web server with chi router, embedded templates via //go:embed - Login/logout flow with session cookies - Notebook list, page grid with SVG thumbnails, page viewer - Share link views (same templates, no auth chrome) - Server command wired to start gRPC + REST + web servers concurrently - Graceful shutdown on SIGINT/SIGTERM Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
79 lines
1.6 KiB
Go
79 lines
1.6 KiB
Go
package webserver
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"html/template"
|
|
"io/fs"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.wntrmute.dev/kyle/eng-pad-server/web"
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
type Config struct {
|
|
Addr string
|
|
DB *sql.DB
|
|
BaseURL string
|
|
}
|
|
|
|
type WebServer struct {
|
|
db *sql.DB
|
|
baseURL string
|
|
tmpl *template.Template
|
|
}
|
|
|
|
func Start(cfg Config) error {
|
|
templateFS, err := fs.Sub(web.Content, "templates")
|
|
if err != nil {
|
|
return fmt.Errorf("template fs: %w", err)
|
|
}
|
|
|
|
tmpl, err := template.ParseFS(templateFS, "*.html")
|
|
if err != nil {
|
|
return fmt.Errorf("parse templates: %w", err)
|
|
}
|
|
|
|
ws := &WebServer{
|
|
db: cfg.DB,
|
|
baseURL: cfg.BaseURL,
|
|
tmpl: tmpl,
|
|
}
|
|
|
|
r := chi.NewRouter()
|
|
|
|
// Static files
|
|
staticFS, _ := fs.Sub(web.Content, "static")
|
|
r.Handle("/static/*", http.StripPrefix("/static/", http.FileServer(http.FS(staticFS))))
|
|
|
|
// Public routes
|
|
r.Get("/login", ws.handleLoginPage)
|
|
r.Post("/login", ws.handleLoginSubmit)
|
|
|
|
// Share routes (no auth)
|
|
r.Get("/s/{token}", ws.handleShareNotebook)
|
|
r.Get("/s/{token}/pages/{num}", ws.handleSharePage)
|
|
|
|
// Authenticated routes
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(ws.authMiddleware)
|
|
r.Get("/", http.RedirectHandler("/notebooks", http.StatusFound).ServeHTTP)
|
|
r.Get("/notebooks", ws.handleNotebooks)
|
|
r.Get("/notebooks/{id}", ws.handleNotebook)
|
|
r.Get("/notebooks/{id}/pages/{num}", ws.handlePage)
|
|
r.Get("/logout", ws.handleLogout)
|
|
})
|
|
|
|
srv := &http.Server{
|
|
Addr: cfg.Addr,
|
|
Handler: r,
|
|
ReadTimeout: 30 * time.Second,
|
|
WriteTimeout: 30 * time.Second,
|
|
IdleTimeout: 120 * time.Second,
|
|
}
|
|
|
|
fmt.Printf("Web UI listening on %s\n", cfg.Addr)
|
|
return srv.ListenAndServe()
|
|
}
|