110 lines
3 KiB
Go
110 lines
3 KiB
Go
package home
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/FAU-CDI/wisski-distillery/pkg/timex"
|
|
"github.com/tkw1536/goprogram/stream"
|
|
)
|
|
|
|
func (home *Home) updateRedirect(ctx context.Context, io stream.IOStream) {
|
|
timex.SetInterval(ctx, home.RefreshInterval, func(t time.Time) {
|
|
io.Printf("[%s]: reloading overrides", t.String())
|
|
redirect, _ := home.loadRedirect()
|
|
home.redirect.Set(&redirect)
|
|
})
|
|
}
|
|
|
|
func (home *Home) loadRedirect() (redirect Redirect, err error) {
|
|
if redirect.Overrides == nil {
|
|
redirect.Overrides = make(map[string]string)
|
|
}
|
|
redirect.Overrides[""] = home.Config.SelfRedirect.String()
|
|
|
|
redirect.Absolute = false
|
|
redirect.Permanent = false
|
|
|
|
// load the overrides file
|
|
overrides, err := home.Environment.Open(home.Config.SelfOverridesFile)
|
|
if err != nil {
|
|
return redirect, err
|
|
}
|
|
defer overrides.Close()
|
|
|
|
// decode the overrides file
|
|
if err := json.NewDecoder(overrides).Decode(&redirect.Overrides); err != nil {
|
|
return redirect, err
|
|
}
|
|
|
|
// and return!
|
|
return redirect, nil
|
|
}
|
|
|
|
// Redirect implements a redirect server that redirects all requests.
|
|
// It implements http.Handler.
|
|
type Redirect struct {
|
|
// Target is the target URL to redirect to.
|
|
Target string
|
|
|
|
// Fallback is used when target is the empty string.
|
|
Fallback http.Handler
|
|
|
|
// Absolute determines if the request path should be appended to the target URL when redirecting.
|
|
// By default this path is always appended, set Absolute to true to prevent this.
|
|
Absolute bool
|
|
|
|
// Overrides is a map from paths to URLs that should override the default target.
|
|
Overrides map[string]string
|
|
|
|
// Permanent determines if the redirect responses issued should return
|
|
// Permanent Redirect (Status Code 308) or Temporary Redirect (Status Code 307).
|
|
Permanent bool
|
|
}
|
|
|
|
// Redirect determines the redirect URL for a specific incoming request
|
|
// If it returns the empty string, the fallback is used.
|
|
func (redirect Redirect) Redirect(r *http.Request) string {
|
|
// if we have an override for this URL, use it immediatly
|
|
url := strings.TrimSuffix(r.URL.Path, "/")
|
|
if override, ok := redirect.Overrides[url]; ok {
|
|
return override
|
|
}
|
|
|
|
if redirect.Target == "" {
|
|
return ""
|
|
}
|
|
|
|
// if we are in absolute redirect mode, always return the absolute URL
|
|
if redirect.Absolute {
|
|
return redirect.Target
|
|
}
|
|
|
|
// return the target + the redirected URL
|
|
dest := strings.TrimSuffix(redirect.Target, "/") + r.URL.Path
|
|
if len(r.URL.RawQuery) > 0 {
|
|
dest += "?" + r.URL.RawQuery
|
|
}
|
|
return dest
|
|
}
|
|
|
|
// ServeHTTP implements the http.Handler interface and redirects a single request to redirect.Target.
|
|
func (redirect Redirect) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
dest := redirect.Redirect(r)
|
|
if dest == "" {
|
|
redirect.Fallback.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
// determine if we are temporary or permanent redirect
|
|
status := http.StatusTemporaryRedirect
|
|
if redirect.Permanent {
|
|
status = http.StatusPermanentRedirect
|
|
}
|
|
|
|
// and do the redirect
|
|
http.Redirect(w, r, dest, status)
|
|
}
|