36 lines
1.2 KiB
Go
36 lines
1.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/r3labs/sse/v2"
|
|
"github.com/zitadel/oidc/v3/pkg/client/rp"
|
|
)
|
|
|
|
func SetupRoutes(router *http.ServeMux, sse *sse.Server, appHandler *AppHandler, authHandler *AuthHandler) {
|
|
router.HandleFunc("GET /sse", sse.ServeHTTP)
|
|
|
|
fsAssets := http.FileServer(http.Dir("assets"))
|
|
router.Handle("GET /assets/", http.StripPrefix("/assets/", fsAssets))
|
|
|
|
icons := http.FileServer(http.Dir("storage/icons"))
|
|
router.Handle("GET /icons/", http.StripPrefix("/icons/", icons))
|
|
|
|
state := func() string {
|
|
return uuid.New().String()
|
|
}
|
|
router.Handle("GET /login", rp.AuthURLHandler(state, authHandler.provider, authHandler.urlOptions...))
|
|
router.Handle("GET /auch/callback", rp.CodeExchangeHandler(authHandler.marshalToken, authHandler.provider))
|
|
|
|
router.HandleFunc("GET /", authMiddleware(http.HandlerFunc(appHandler.appHandler), authHandler))
|
|
}
|
|
|
|
func authMiddleware(next http.Handler, authHandler *AuthHandler) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := context.Background()
|
|
clains, err := rp.VerifyTokens(ctx, authHandler.provider)
|
|
next.ServeHTTP(w, r)
|
|
}
|
|
}
|