GoBlog/regexRedirects.go

45 lines
869 B
Go
Raw Permalink Normal View History

2020-10-15 20:54:43 +02:00
package main
import (
"net/http"
"regexp"
)
type regexRedirect struct {
From *regexp.Regexp
To string
2020-11-10 07:53:08 +01:00
Type int
2020-10-15 20:54:43 +02:00
}
func (a *goBlog) initRegexRedirects() error {
for _, cr := range a.cfg.PathRedirects {
2020-10-15 20:54:43 +02:00
re, err := regexp.Compile(cr.From)
if err != nil {
return err
}
r := &regexRedirect{
2020-10-15 20:54:43 +02:00
From: re,
To: cr.To,
2020-11-10 07:53:08 +01:00
Type: cr.Type,
}
if r.Type == 0 {
r.Type = http.StatusFound
}
a.regexRedirects = append(a.regexRedirects, r)
2020-10-15 20:54:43 +02:00
}
return nil
}
func (a *goBlog) checkRegexRedirects(next http.Handler) http.Handler {
2020-10-15 20:54:43 +02:00
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
for _, re := range a.regexRedirects {
if newPath := re.From.ReplaceAllString(r.URL.Path, re.To); r.URL.Path != newPath {
2020-10-15 20:54:43 +02:00
r.URL.Path = newPath
http.Redirect(w, r, r.URL.String(), re.Type)
2020-10-15 20:54:43 +02:00
return
}
}
next.ServeHTTP(w, r)
})
}