mirror of
https://github.com/yusing/godoxy.git
synced 2026-01-11 22:30:47 +01:00
590743f1ef
- Added ShortLinkMatcher to handle short link routing. - Integrated short link handling in Entrypoint. - Introduced tests for short link matching and dispatching. - Configured default domain suffix for subdomain aliases.
100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
package errorpage
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"sync"
|
|
|
|
"github.com/puzpuzpuz/xsync/v4"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/yusing/godoxy/internal/common"
|
|
"github.com/yusing/godoxy/internal/watcher"
|
|
"github.com/yusing/godoxy/internal/watcher/events"
|
|
gperr "github.com/yusing/goutils/errs"
|
|
"github.com/yusing/goutils/fs"
|
|
"github.com/yusing/goutils/task"
|
|
)
|
|
|
|
const errPagesBasePath = common.ErrorPagesBasePath
|
|
|
|
var (
|
|
setupOnce sync.Once
|
|
dirWatcher watcher.Watcher
|
|
fileContentMap = xsync.NewMap[string, []byte](xsync.WithGrowOnly())
|
|
)
|
|
|
|
func setup() {
|
|
t := task.RootTask("error_page", false)
|
|
dirWatcher = watcher.NewDirectoryWatcher(t, errPagesBasePath)
|
|
loadContent()
|
|
go watchDir()
|
|
}
|
|
|
|
func GetStaticFile(filename string) ([]byte, bool) {
|
|
if common.IsTest {
|
|
return nil, false
|
|
}
|
|
setupOnce.Do(setup)
|
|
return fileContentMap.Load(filename)
|
|
}
|
|
|
|
// try <statusCode>.html -> 404.html -> not ok.
|
|
func GetErrorPageByStatus(statusCode int) (content []byte, ok bool) {
|
|
content, ok = GetStaticFile(fmt.Sprintf("%d.html", statusCode))
|
|
if !ok && statusCode != 404 {
|
|
return fileContentMap.Load("404.html")
|
|
}
|
|
return content, ok
|
|
}
|
|
|
|
func loadContent() {
|
|
files, err := fs.ListFiles(errPagesBasePath, 0)
|
|
if err != nil {
|
|
log.Err(err).Msg("failed to list error page resources")
|
|
return
|
|
}
|
|
for _, file := range files {
|
|
if _, ok := fileContentMap.Load(file); ok {
|
|
continue
|
|
}
|
|
content, err := os.ReadFile(file)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msgf("failed to read error page resource %s", file)
|
|
continue
|
|
}
|
|
file = path.Base(file)
|
|
log.Info().Msgf("error page resource %s loaded", file)
|
|
fileContentMap.Store(file, content)
|
|
}
|
|
}
|
|
|
|
func watchDir() {
|
|
eventCh, errCh := dirWatcher.Events(task.RootContext())
|
|
for {
|
|
select {
|
|
case <-task.RootContextCanceled():
|
|
return
|
|
case event, ok := <-eventCh:
|
|
if !ok {
|
|
return
|
|
}
|
|
filename := event.ActorName
|
|
switch event.Action {
|
|
case events.ActionFileWritten:
|
|
fileContentMap.Delete(filename)
|
|
loadContent()
|
|
case events.ActionFileDeleted:
|
|
fileContentMap.Delete(filename)
|
|
log.Warn().Msgf("error page resource %s deleted", filename)
|
|
case events.ActionFileRenamed:
|
|
log.Warn().Msgf("error page resource %s deleted", filename)
|
|
fileContentMap.Delete(filename)
|
|
loadContent()
|
|
}
|
|
case err := <-errCh:
|
|
gperr.LogError("error watching error page directory", err)
|
|
}
|
|
}
|
|
}
|