53 lines
942 B
Go
53 lines
942 B
Go
|
package captiveportal
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (s *Service) Middleware() func(next http.Handler) http.Handler {
|
||
|
|
registry := s.registry
|
||
|
|
matcher := s.options.Matcher
|
||
|
|
liar := s.options.Liar
|
||
|
|
|
||
|
|
return func(next http.Handler) http.Handler {
|
||
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
matches, os, err := matcher.Match(r)
|
||
|
|
if err != nil {
|
||
|
|
panic(errors.WithStack(err))
|
||
|
|
}
|
||
|
|
|
||
|
|
if !matches {
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
id, err := s.options.Identifier.Identify(r)
|
||
|
|
if err != nil {
|
||
|
|
panic(errors.Wrap(err, ErrClientIdentificationFailed.Error()))
|
||
|
|
}
|
||
|
|
|
||
|
|
registry.Touch(id, os)
|
||
|
|
|
||
|
|
if registry.IsLying(id) {
|
||
|
|
liar.Handle(os, w, r)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if registry.IsCaptive(id) {
|
||
|
|
// Redirect to configured URL
|
||
|
|
http.Redirect(w, r, s.captivePortalURL, http.StatusTemporaryRedirect)
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
}
|
||
|
|
|
||
|
|
return http.HandlerFunc(fn)
|
||
|
|
}
|
||
|
|
}
|