mirror of https://github.com/jlelse/GoBlog
624 lines
18 KiB
Go
624 lines
18 KiB
Go
package main
|
|
|
|
import (
|
|
"compress/flate"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/dchest/captcha"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
servertiming "github.com/mitchellh/go-server-timing"
|
|
"golang.org/x/crypto/acme"
|
|
"golang.org/x/crypto/acme/autocert"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
const (
|
|
contentType = "Content-Type"
|
|
userAgent = "User-Agent"
|
|
appUserAgent = "GoBlog"
|
|
)
|
|
|
|
func (a *goBlog) startServer() (err error) {
|
|
log.Println("Start server(s)...")
|
|
// Start
|
|
a.d = &dynamicHandler{}
|
|
// Set basic middlewares
|
|
var finalHandler http.Handler = a.d
|
|
if a.cfg.Server.PublicHTTPS || a.cfg.Server.SecurityHeaders {
|
|
finalHandler = a.securityHeaders(finalHandler)
|
|
}
|
|
finalHandler = servertiming.Middleware(finalHandler, nil)
|
|
finalHandler = middleware.Heartbeat("/ping")(finalHandler)
|
|
finalHandler = middleware.Compress(flate.DefaultCompression)(finalHandler)
|
|
finalHandler = middleware.Recoverer(finalHandler)
|
|
if a.cfg.Server.Logging {
|
|
finalHandler = a.logMiddleware(finalHandler)
|
|
}
|
|
// Create routers that don't change
|
|
if err = a.buildStaticHandlersRouters(); err != nil {
|
|
return err
|
|
}
|
|
// Load router
|
|
if err = a.reloadRouter(); err != nil {
|
|
return err
|
|
}
|
|
// Start Onion service
|
|
if a.cfg.Server.Tor {
|
|
go func() {
|
|
if err := a.startOnionService(finalHandler); err != nil {
|
|
log.Println("Tor failed:", err.Error())
|
|
}
|
|
}()
|
|
}
|
|
// Start server
|
|
s := &http.Server{
|
|
Handler: finalHandler,
|
|
ReadTimeout: 5 * time.Minute,
|
|
WriteTimeout: 5 * time.Minute,
|
|
}
|
|
a.shutdown.Add(shutdownServer(s, "main server"))
|
|
if a.cfg.Server.PublicHTTPS {
|
|
// Start HTTP server for redirects
|
|
httpServer := &http.Server{
|
|
Addr: ":http",
|
|
Handler: http.HandlerFunc(redirectToHttps),
|
|
ReadTimeout: 5 * time.Minute,
|
|
WriteTimeout: 5 * time.Minute,
|
|
}
|
|
a.shutdown.Add(shutdownServer(httpServer, "http server"))
|
|
go func() {
|
|
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Println("Failed to start HTTP server:", err.Error())
|
|
}
|
|
}()
|
|
// Start HTTPS
|
|
s.Addr = ":https"
|
|
hosts := []string{a.cfg.Server.publicHostname}
|
|
if a.cfg.Server.shortPublicHostname != "" {
|
|
hosts = append(hosts, a.cfg.Server.shortPublicHostname)
|
|
}
|
|
acmeDir := acme.LetsEncryptURL
|
|
// acmeDir := "https://acme-staging-v02.api.letsencrypt.org/directory"
|
|
m := &autocert.Manager{
|
|
Prompt: autocert.AcceptTOS,
|
|
HostPolicy: autocert.HostWhitelist(hosts...),
|
|
Cache: &httpsCache{db: a.db},
|
|
Client: &acme.Client{DirectoryURL: acmeDir},
|
|
}
|
|
if err = s.Serve(m.Listener()); err != nil && err != http.ErrServerClosed {
|
|
return err
|
|
}
|
|
} else {
|
|
s.Addr = ":" + strconv.Itoa(a.cfg.Server.Port)
|
|
if err = s.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func shutdownServer(s *http.Server, name string) func() {
|
|
return func() {
|
|
toc, c := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer c()
|
|
if err := s.Shutdown(toc); err != nil {
|
|
log.Printf("Error on server shutdown (%v): %v", name, err)
|
|
}
|
|
log.Println("Stopped server:", name)
|
|
}
|
|
}
|
|
|
|
func redirectToHttps(w http.ResponseWriter, r *http.Request) {
|
|
requestHost, _, err := net.SplitHostPort(r.Host)
|
|
if err != nil {
|
|
requestHost = r.Host
|
|
}
|
|
w.Header().Set("Connection", "close")
|
|
http.Redirect(w, r, fmt.Sprintf("https://%s%s", requestHost, r.URL.RequestURI()), http.StatusMovedPermanently)
|
|
}
|
|
|
|
func (a *goBlog) reloadRouter() error {
|
|
h, err := a.buildDynamicRouter()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
a.d.swapHandler(h)
|
|
a.cache.purge()
|
|
return nil
|
|
}
|
|
|
|
const (
|
|
paginationPath = "/page/{page:[0-9-]+}"
|
|
feedPath = ".{feed:rss|json|atom}"
|
|
)
|
|
|
|
func (a *goBlog) buildStaticHandlersRouters() error {
|
|
if pm := a.cfg.PrivateMode; pm != nil && pm.Enabled {
|
|
a.privateMode = true
|
|
a.privateModeHandler = append(a.privateModeHandler, a.authMiddleware)
|
|
} else {
|
|
a.privateMode = false
|
|
a.privateModeHandler = []func(http.Handler) http.Handler{}
|
|
}
|
|
|
|
a.captchaHandler = captcha.Server(500, 250)
|
|
|
|
a.micropubRouter = chi.NewRouter()
|
|
a.micropubRouter.Use(a.checkIndieAuth)
|
|
a.micropubRouter.Get("/", a.serveMicropubQuery)
|
|
a.micropubRouter.Post("/", a.serveMicropubPost)
|
|
a.micropubRouter.Post(micropubMediaSubPath, a.serveMicropubMedia)
|
|
|
|
a.indieAuthRouter = chi.NewRouter()
|
|
a.indieAuthRouter.Get("/", a.indieAuthRequest)
|
|
a.indieAuthRouter.With(a.authMiddleware).Post("/accept", a.indieAuthAccept)
|
|
a.indieAuthRouter.Post("/", a.indieAuthVerification)
|
|
a.indieAuthRouter.Get("/token", a.indieAuthToken)
|
|
a.indieAuthRouter.Post("/token", a.indieAuthToken)
|
|
|
|
a.webmentionsRouter = chi.NewRouter()
|
|
if wm := a.cfg.Webmention; wm != nil && !wm.DisableReceiving {
|
|
a.webmentionsRouter.Post("/", a.handleWebmention)
|
|
a.webmentionsRouter.Group(func(r chi.Router) {
|
|
// Authenticated routes
|
|
r.Use(a.authMiddleware)
|
|
r.Get("/", a.webmentionAdmin)
|
|
r.Get(paginationPath, a.webmentionAdmin)
|
|
r.Post("/delete", a.webmentionAdminDelete)
|
|
r.Post("/approve", a.webmentionAdminApprove)
|
|
r.Post("/reverify", a.webmentionAdminReverify)
|
|
})
|
|
}
|
|
|
|
a.notificationsRouter = chi.NewRouter()
|
|
a.notificationsRouter.Use(a.authMiddleware)
|
|
a.notificationsRouter.Get("/", a.notificationsAdmin)
|
|
a.notificationsRouter.Get(paginationPath, a.notificationsAdmin)
|
|
a.notificationsRouter.Post("/delete", a.notificationsAdminDelete)
|
|
|
|
if ap := a.cfg.ActivityPub; ap != nil && ap.Enabled {
|
|
a.activitypubRouter = chi.NewRouter()
|
|
a.activitypubRouter.Post("/inbox/{blog}", a.apHandleInbox)
|
|
a.activitypubRouter.Post("/{blog}/inbox", a.apHandleInbox)
|
|
}
|
|
|
|
a.editorRouter = chi.NewRouter()
|
|
a.editorRouter.Use(a.authMiddleware)
|
|
a.editorRouter.Get("/", a.serveEditor)
|
|
a.editorRouter.Post("/", a.serveEditorPost)
|
|
a.editorRouter.Get("/files", a.serveEditorFiles)
|
|
a.editorRouter.Post("/files/view", a.serveEditorFilesView)
|
|
a.editorRouter.Post("/files/delete", a.serveEditorFilesDelete)
|
|
|
|
a.commentsRouter = chi.NewRouter()
|
|
a.commentsRouter.Use(a.privateModeHandler...)
|
|
a.commentsRouter.With(a.cache.cacheMiddleware, noIndexHeader).Get("/{id:[0-9]+}", a.serveComment)
|
|
a.commentsRouter.With(a.captchaMiddleware).Post("/", a.createComment)
|
|
a.commentsRouter.Group(func(r chi.Router) {
|
|
// Admin
|
|
r.Use(a.authMiddleware)
|
|
r.Get("/", a.commentsAdmin)
|
|
r.Get(paginationPath, a.commentsAdmin)
|
|
r.Post("/delete", a.commentsAdminDelete)
|
|
})
|
|
|
|
a.searchRouter = chi.NewRouter()
|
|
a.searchRouter.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware)
|
|
r.Get("/", a.serveSearch)
|
|
r.Post("/", a.serveSearch)
|
|
searchResultPath := "/" + searchPlaceholder
|
|
r.Get(searchResultPath, a.serveSearchResult)
|
|
r.Get(searchResultPath+feedPath, a.serveSearchResult)
|
|
r.Get(searchResultPath+paginationPath, a.serveSearchResult)
|
|
})
|
|
a.searchRouter.With(a.cache.cacheMiddleware).Get("/opensearch.xml", a.serveOpenSearch)
|
|
|
|
a.setBlogMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.sectionMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.taxonomyMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.taxValueMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.photosMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.searchMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.customPagesMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
a.commentsMiddlewares = map[string]func(http.Handler) http.Handler{}
|
|
|
|
for blog, blogConfig := range a.cfg.Blogs {
|
|
sbm := middleware.WithValue(blogContextKey, blog)
|
|
a.setBlogMiddlewares[blog] = sbm
|
|
|
|
for _, section := range blogConfig.Sections {
|
|
if section.Name != "" {
|
|
secPath := blogConfig.getRelativePath(section.Name)
|
|
a.sectionMiddlewares[secPath] = middleware.WithValue(indexConfigKey, &indexConfig{
|
|
path: secPath,
|
|
section: section,
|
|
})
|
|
}
|
|
}
|
|
|
|
for _, taxonomy := range blogConfig.Taxonomies {
|
|
if taxonomy.Name != "" {
|
|
taxPath := blogConfig.getRelativePath(taxonomy.Name)
|
|
a.taxonomyMiddlewares[taxPath] = middleware.WithValue(taxonomyContextKey, taxonomy)
|
|
}
|
|
}
|
|
|
|
if blogConfig.Photos != nil && blogConfig.Photos.Enabled {
|
|
a.photosMiddlewares[blog] = middleware.WithValue(indexConfigKey, &indexConfig{
|
|
path: blogConfig.getRelativePath(blogConfig.Photos.Path),
|
|
parameter: blogConfig.Photos.Parameter,
|
|
title: blogConfig.Photos.Title,
|
|
description: blogConfig.Photos.Description,
|
|
summaryTemplate: templatePhotosSummary,
|
|
})
|
|
}
|
|
|
|
if blogConfig.Search != nil && blogConfig.Search.Enabled {
|
|
a.searchMiddlewares[blog] = middleware.WithValue(pathContextKey, blogConfig.getRelativePath(blogConfig.Search.Path))
|
|
}
|
|
|
|
for _, cp := range blogConfig.CustomPages {
|
|
a.customPagesMiddlewares[cp.Path] = middleware.WithValue(customPageContextKey, cp)
|
|
}
|
|
|
|
if commentsConfig := blogConfig.Comments; commentsConfig != nil && commentsConfig.Enabled {
|
|
a.commentsMiddlewares[blog] = middleware.WithValue(pathContextKey, blogConfig.getRelativePath("/comment"))
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *goBlog) buildDynamicRouter() (*chi.Mux, error) {
|
|
r := chi.NewRouter()
|
|
|
|
// Basic middleware
|
|
r.Use(a.redirectShortDomain)
|
|
r.Use(middleware.RedirectSlashes)
|
|
r.Use(middleware.CleanPath)
|
|
r.Use(middleware.GetHead)
|
|
if !a.cfg.Cache.Enable {
|
|
r.Use(middleware.NoCache)
|
|
}
|
|
|
|
// No Index Header
|
|
if a.privateMode {
|
|
r.Use(noIndexHeader)
|
|
}
|
|
|
|
// Login middleware etc.
|
|
r.Use(a.checkIsLogin)
|
|
r.Use(a.checkIsCaptcha)
|
|
r.Use(a.checkLoggedIn)
|
|
|
|
// Logout
|
|
r.With(a.authMiddleware).Get("/login", serveLogin)
|
|
r.With(a.authMiddleware).Get("/logout", a.serveLogout)
|
|
|
|
// Micropub
|
|
r.Mount(micropubPath, a.micropubRouter)
|
|
|
|
// IndieAuth
|
|
r.Mount("/indieauth", a.indieAuthRouter)
|
|
|
|
// ActivityPub and stuff
|
|
if ap := a.cfg.ActivityPub; ap != nil && ap.Enabled {
|
|
r.Mount("/activitypub", a.activitypubRouter)
|
|
r.With(a.cache.cacheMiddleware).Get("/.well-known/webfinger", a.apHandleWebfinger)
|
|
r.With(a.cache.cacheMiddleware).Get("/.well-known/host-meta", handleWellKnownHostMeta)
|
|
r.With(a.cache.cacheMiddleware).Get("/.well-known/nodeinfo", a.serveNodeInfoDiscover)
|
|
r.With(a.cache.cacheMiddleware).Get("/nodeinfo", a.serveNodeInfo)
|
|
}
|
|
|
|
// Webmentions
|
|
r.Mount(webmentionPath, a.webmentionsRouter)
|
|
|
|
// Notifications
|
|
r.Mount(notificationsPath, a.notificationsRouter)
|
|
|
|
// Posts
|
|
pp, err := a.db.allPostPaths(statusPublished)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.checkActivityStreamsRequest, a.cache.cacheMiddleware)
|
|
for _, path := range pp {
|
|
r.Get(path, a.servePost)
|
|
}
|
|
})
|
|
|
|
// Drafts
|
|
dp, err := a.db.allPostPaths(statusDraft)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.authMiddleware)
|
|
for _, path := range dp {
|
|
r.Get(path, a.servePost)
|
|
}
|
|
})
|
|
|
|
// Post aliases
|
|
allPostAliases, err := a.db.allPostAliases()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware)
|
|
for _, path := range allPostAliases {
|
|
r.Get(path, a.servePostAlias)
|
|
}
|
|
})
|
|
|
|
// Assets
|
|
for _, path := range a.allAssetPaths() {
|
|
r.Get(path, a.serveAsset)
|
|
}
|
|
|
|
// Static files
|
|
for _, path := range allStaticPaths() {
|
|
r.Get(path, a.serveStaticFile)
|
|
}
|
|
|
|
// Media files
|
|
r.With(a.privateModeHandler...).Get(`/m/{file:[0-9a-fA-F]+(\.[0-9a-zA-Z]+)?}`, a.serveMediaFile)
|
|
|
|
// Captcha
|
|
r.Handle("/captcha/*", a.captchaHandler)
|
|
|
|
// Short paths
|
|
r.With(a.privateModeHandler...).With(a.cache.cacheMiddleware).Get("/s/{id:[0-9a-fA-F]+}", a.redirectToLongPath)
|
|
|
|
for blog, blogConfig := range a.cfg.Blogs {
|
|
sbm := a.setBlogMiddlewares[blog]
|
|
|
|
// Sections
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware, sbm)
|
|
for _, section := range blogConfig.Sections {
|
|
if section.Name != "" {
|
|
secPath := blogConfig.getRelativePath(section.Name)
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.sectionMiddlewares[secPath])
|
|
r.Get(secPath, a.serveIndex)
|
|
r.Get(secPath+feedPath, a.serveIndex)
|
|
r.Get(secPath+paginationPath, a.serveIndex)
|
|
})
|
|
}
|
|
}
|
|
})
|
|
|
|
// Taxonomies
|
|
for _, taxonomy := range blogConfig.Taxonomies {
|
|
if taxonomy.Name != "" {
|
|
taxPath := blogConfig.getRelativePath(taxonomy.Name)
|
|
taxValues, err := a.db.allTaxonomyValues(blog, taxonomy.Name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware, sbm)
|
|
r.With(a.taxonomyMiddlewares[taxPath]).Get(taxPath, a.serveTaxonomy)
|
|
for _, tv := range taxValues {
|
|
r.Group(func(r chi.Router) {
|
|
vPath := taxPath + "/" + urlize(tv)
|
|
if _, ok := a.taxValueMiddlewares[vPath]; !ok {
|
|
a.taxValueMiddlewares[vPath] = middleware.WithValue(indexConfigKey, &indexConfig{
|
|
path: vPath,
|
|
tax: taxonomy,
|
|
taxValue: tv,
|
|
})
|
|
}
|
|
r.Use(a.taxValueMiddlewares[vPath])
|
|
r.Get(vPath, a.serveIndex)
|
|
r.Get(vPath+feedPath, a.serveIndex)
|
|
r.Get(vPath+paginationPath, a.serveIndex)
|
|
})
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// Photos
|
|
if blogConfig.Photos != nil && blogConfig.Photos.Enabled {
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware, sbm, a.photosMiddlewares[blog])
|
|
photoPath := blogConfig.getRelativePath(blogConfig.Photos.Path)
|
|
r.Get(photoPath, a.serveIndex)
|
|
r.Get(photoPath+feedPath, a.serveIndex)
|
|
r.Get(photoPath+paginationPath, a.serveIndex)
|
|
})
|
|
}
|
|
|
|
// Search
|
|
if blogConfig.Search != nil && blogConfig.Search.Enabled {
|
|
searchPath := blogConfig.getRelativePath(blogConfig.Search.Path)
|
|
r.With(sbm, a.searchMiddlewares[blog]).Mount(searchPath, a.searchRouter)
|
|
}
|
|
|
|
// Stats
|
|
if blogConfig.BlogStats != nil && blogConfig.BlogStats.Enabled {
|
|
statsPath := blogConfig.getRelativePath(blogConfig.BlogStats.Path)
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware, sbm)
|
|
r.Get(statsPath, a.serveBlogStats)
|
|
r.Get(statsPath+".table.html", a.serveBlogStatsTable)
|
|
})
|
|
}
|
|
|
|
// Date archives
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware, sbm)
|
|
|
|
yearRegex := `/{year:x|\d\d\d\d}`
|
|
monthRegex := `/{month:x|\d\d}`
|
|
dayRegex := `/{day:\d\d}`
|
|
|
|
yearPath := blogConfig.getRelativePath(yearRegex)
|
|
r.Get(yearPath, a.serveDate)
|
|
r.Get(yearPath+feedPath, a.serveDate)
|
|
r.Get(yearPath+paginationPath, a.serveDate)
|
|
|
|
monthPath := yearPath + monthRegex
|
|
r.Get(monthPath, a.serveDate)
|
|
r.Get(monthPath+feedPath, a.serveDate)
|
|
r.Get(monthPath+paginationPath, a.serveDate)
|
|
|
|
dayPath := monthPath + dayRegex
|
|
r.Get(dayPath, a.serveDate)
|
|
r.Get(dayPath+feedPath, a.serveDate)
|
|
r.Get(dayPath+paginationPath, a.serveDate)
|
|
})
|
|
|
|
// Blog
|
|
if !blogConfig.PostAsHome {
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(sbm)
|
|
r.With(a.checkActivityStreamsRequest, a.cache.cacheMiddleware).Get(blogConfig.getRelativePath(""), a.serveHome)
|
|
r.With(a.cache.cacheMiddleware).Get(blogConfig.getRelativePath("")+feedPath, a.serveHome)
|
|
r.With(a.cache.cacheMiddleware).Get(blogConfig.getRelativePath(paginationPath), a.serveHome)
|
|
})
|
|
}
|
|
|
|
// Custom pages
|
|
for _, cp := range blogConfig.CustomPages {
|
|
scp := a.customPagesMiddlewares[cp.Path]
|
|
if cp.Cache {
|
|
r.With(a.privateModeHandler...).With(a.cache.cacheMiddleware, sbm, scp).Get(cp.Path, a.serveCustomPage)
|
|
} else {
|
|
r.With(a.privateModeHandler...).With(sbm, scp).Get(cp.Path, a.serveCustomPage)
|
|
}
|
|
}
|
|
|
|
// Random post
|
|
if rp := blogConfig.RandomPost; rp != nil && rp.Enabled {
|
|
randomPath := rp.Path
|
|
if randomPath == "" {
|
|
randomPath = "/random"
|
|
}
|
|
r.With(a.privateModeHandler...).With(sbm).Get(blogConfig.getRelativePath(randomPath), a.redirectToRandomPost)
|
|
}
|
|
|
|
// Editor
|
|
r.With(sbm).Mount(blogConfig.getRelativePath("/editor"), a.editorRouter)
|
|
|
|
// Comments
|
|
if commentsConfig := blogConfig.Comments; commentsConfig != nil && commentsConfig.Enabled {
|
|
r.With(sbm, a.commentsMiddlewares[blog]).Mount(blogConfig.getRelativePath("/comment"), a.commentsRouter)
|
|
}
|
|
|
|
// Blogroll
|
|
if brConfig := blogConfig.Blogroll; brConfig != nil && brConfig.Enabled {
|
|
brPath := blogConfig.getRelativePath(brConfig.Path)
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(a.privateModeHandler...)
|
|
r.Use(a.cache.cacheMiddleware, sbm)
|
|
r.Get(brPath, a.serveBlogroll)
|
|
r.Get(brPath+".opml", a.serveBlogrollExport)
|
|
})
|
|
}
|
|
}
|
|
|
|
// Sitemap
|
|
r.With(a.privateModeHandler...).With(a.cache.cacheMiddleware).Get(sitemapPath, a.serveSitemap)
|
|
|
|
// Robots.txt - doesn't need cache, because it's too simple
|
|
if !a.privateMode {
|
|
r.Get("/robots.txt", a.serveRobotsTXT)
|
|
} else {
|
|
r.Get("/robots.txt", servePrivateRobotsTXT)
|
|
}
|
|
|
|
// Check redirects, then serve 404
|
|
r.With(a.cache.cacheMiddleware, a.checkRegexRedirects).NotFound(a.serve404)
|
|
|
|
r.MethodNotAllowed(a.serveNotAllowed)
|
|
|
|
return r, nil
|
|
}
|
|
|
|
const blogContextKey contextKey = "blog"
|
|
const pathContextKey contextKey = "httpPath"
|
|
|
|
func (a *goBlog) refreshCSPDomains() {
|
|
a.cspDomains = ""
|
|
if mp := a.cfg.Micropub.MediaStorage; mp != nil && mp.MediaURL != "" {
|
|
if u, err := url.Parse(mp.MediaURL); err == nil {
|
|
a.cspDomains += " " + u.Hostname()
|
|
}
|
|
}
|
|
if len(a.cfg.Server.CSPDomains) > 0 {
|
|
a.cspDomains += " " + strings.Join(a.cfg.Server.CSPDomains, " ")
|
|
}
|
|
}
|
|
|
|
func (a *goBlog) securityHeaders(next http.Handler) http.Handler {
|
|
a.refreshCSPDomains()
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Strict-Transport-Security", "max-age=31536000;")
|
|
w.Header().Set("Referrer-Policy", "no-referrer")
|
|
w.Header().Set("X-Content-Type-Options", "nosniff")
|
|
w.Header().Set("X-Frame-Options", "SAMEORIGIN")
|
|
w.Header().Set("X-Xss-Protection", "1; mode=block")
|
|
w.Header().Set("Content-Security-Policy", "default-src 'self'"+a.cspDomains)
|
|
if a.cfg.Server.Tor && a.torAddress != "" {
|
|
w.Header().Set("Onion-Location", fmt.Sprintf("http://%v%v", a.torAddress, r.RequestURI))
|
|
}
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func noIndexHeader(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("X-Robots-Tag", "noindex")
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
type dynamicHandler struct {
|
|
router *chi.Mux
|
|
mutex sync.RWMutex
|
|
initialized bool
|
|
}
|
|
|
|
func (d *dynamicHandler) swapHandler(h *chi.Mux) {
|
|
d.mutex.Lock()
|
|
d.router = h
|
|
d.initialized = true
|
|
d.mutex.Unlock()
|
|
}
|
|
|
|
func (d *dynamicHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
// Fix to use Path routing instead of RawPath routing in Chi
|
|
r.URL.RawPath = ""
|
|
// Serve request
|
|
d.mutex.RLock()
|
|
for !d.initialized {
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
router := d.router
|
|
d.mutex.RUnlock()
|
|
router.ServeHTTP(w, r)
|
|
}
|