pronounsfu/backend/server/server.go

146 lines
3.6 KiB
Go
Raw Normal View History

2022-05-02 08:19:37 -07:00
package server
import (
"net/http"
2022-05-02 08:19:37 -07:00
"os"
2022-05-25 15:41:06 -07:00
"strconv"
"time"
2022-05-02 08:19:37 -07:00
"codeberg.org/u1f320/pronouns.cc/backend/db"
"codeberg.org/u1f320/pronouns.cc/backend/server/auth"
"codeberg.org/u1f320/pronouns.cc/backend/server/rate"
2022-05-02 08:19:37 -07:00
"github.com/go-chi/chi/v5"
"github.com/go-chi/chi/v5/middleware"
2023-04-13 14:33:48 -07:00
"github.com/go-chi/cors"
2022-05-25 15:41:06 -07:00
"github.com/go-chi/httprate"
"github.com/go-chi/render"
2022-05-02 08:19:37 -07:00
)
// Revision is the git commit, filled at build time
var (
Revision = "[unknown]"
Tag = "[unknown]"
)
2022-05-02 08:19:37 -07:00
// Repository is the URL of the git repository
const Repository = "https://codeberg.org/u1f320/pronouns.cc"
2022-05-02 08:19:37 -07:00
type Server struct {
Router *chi.Mux
DB *db.DB
Auth *auth.Verifier
}
func New() (*Server, error) {
db, err := db.New()
2022-05-02 08:19:37 -07:00
if err != nil {
return nil, err
}
s := &Server{
Router: chi.NewMux(),
DB: db,
Auth: auth.New(),
}
if os.Getenv("DEBUG") == "true" {
s.Router.Use(middleware.Logger)
}
s.Router.Use(middleware.Recoverer)
2023-04-13 14:33:48 -07:00
// add CORS
s.Router.Use(cors.Handler(cors.Options{
AllowedOrigins: []string{"https://*", "http://*"},
AllowedMethods: []string{"HEAD", "GET"},
AllowedHeaders: []string{"Accept", "Authorization", "Content-Type"},
AllowCredentials: false,
MaxAge: 300,
}))
2022-05-02 08:19:37 -07:00
// enable authentication for all routes (but don't require it)
s.Router.Use(s.maybeAuth)
2022-05-25 15:41:06 -07:00
// rate limit handling
// - base is 120 req/minute (2/s)
2022-05-25 15:41:06 -07:00
// - keyed by Authorization header if valid token is provided, otherwise by IP
// - returns rate limit reset info in error
rateLimiter := rate.NewLimiter(120, time.Minute,
2022-05-25 15:41:06 -07:00
httprate.WithKeyFuncs(func(r *http.Request) (string, error) {
_, ok := ClaimsFromContext(r.Context())
if token := r.Header.Get("Authorization"); ok && token != "" {
return token, nil
}
ip, err := httprate.KeyByIP(r)
return ip, err
}),
httprate.WithLimitHandler(func(w http.ResponseWriter, r *http.Request) {
reset, _ := strconv.Atoi(w.Header().Get("X-RateLimit-Reset"))
render.Status(r, http.StatusTooManyRequests)
render.JSON(w, r, APIError{
Code: ErrTooManyRequests,
Message: errCodeMessages[ErrTooManyRequests],
RatelimitReset: &reset,
})
}),
)
// set scopes
// users
rateLimiter.Scope("GET", "/users/*", 60)
2023-03-14 08:16:07 -07:00
rateLimiter.Scope("PATCH", "/users/@me", 10)
// members
rateLimiter.Scope("GET", "/users/*/members", 60)
rateLimiter.Scope("GET", "/users/*/members/*", 60)
2023-03-14 08:16:07 -07:00
rateLimiter.Scope("POST", "/members", 10)
rateLimiter.Scope("GET", "/members/*", 60)
2023-03-14 08:16:07 -07:00
rateLimiter.Scope("PATCH", "/members/*", 20)
rateLimiter.Scope("DELETE", "/members/*", 5)
// auth
rateLimiter.Scope("*", "/auth/*", 20)
rateLimiter.Scope("*", "/auth/tokens", 10)
rateLimiter.Scope("*", "/auth/invites", 10)
rateLimiter.Scope("POST", "/auth/discord/*", 10)
s.Router.Use(rateLimiter.Handler())
2022-05-25 15:41:06 -07:00
2023-04-17 14:44:21 -07:00
// increment the total requests counter whenever a request is made
s.Router.Use(func(next http.Handler) http.Handler {
fn := func(w http.ResponseWriter, r *http.Request) {
s.DB.TotalRequests.Inc()
next.ServeHTTP(w, r)
}
return http.HandlerFunc(fn)
})
// return an API error for not found + method not allowed
s.Router.NotFound(func(w http.ResponseWriter, r *http.Request) {
render.Status(r, errCodeStatuses[ErrNotFound])
render.JSON(w, r, APIError{
Code: ErrNotFound,
Message: errCodeMessages[ErrNotFound],
})
})
s.Router.MethodNotAllowed(func(w http.ResponseWriter, r *http.Request) {
render.Status(r, errCodeStatuses[ErrMethodNotAllowed])
render.JSON(w, r, APIError{
Code: ErrMethodNotAllowed,
Message: errCodeMessages[ErrMethodNotAllowed],
})
})
2022-05-02 08:19:37 -07:00
return s, nil
}
type ctxKey int
const (
ctxKeyClaims ctxKey = 1
)