58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
type Server struct {
|
|
fs http.Handler
|
|
ws *WS
|
|
limiter *rate.Limiter
|
|
uuid string
|
|
}
|
|
|
|
func New() *Server {
|
|
fs := http.FileServer(http.Dir(config().GetString("d")))
|
|
return &Server{
|
|
fs: fs,
|
|
ws: NewWS(),
|
|
limiter: rate.NewLimiter(rate.Every(time.Second), 2),
|
|
uuid: uuid.New().String(),
|
|
}
|
|
}
|
|
|
|
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if !s.Authorize(w, r) {
|
|
return
|
|
}
|
|
|
|
log.Println("ext", path.Ext(r.URL.Path))
|
|
if path.Ext(r.URL.Path) != "" {
|
|
s.fs.ServeHTTP(w, r)
|
|
} else if _, err := os.Stat(path.Join(config().GetString("d"), r.URL.Path[1:])); os.IsNotExist(err) {
|
|
s.ws.ServeHTTP(w, r)
|
|
} else {
|
|
s.fs.ServeHTTP(w, r)
|
|
}
|
|
}
|
|
|
|
func (s *Server) Authorize(w http.ResponseWriter, r *http.Request) bool {
|
|
if u, p, ok := r.BasicAuth(); !ok || u != "Q" || p != "Q" {
|
|
s.limiter.Wait(r.Context())
|
|
w.Header().Set("WWW-Authenticate", fmt.Sprintf("Basic realm=%q", r.Host))
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
w.Write([]byte("Unauthorized"))
|
|
return false
|
|
}
|
|
return true
|
|
}
|