mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 07:12:46 +01:00
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http/httputil"
|
|
"net/url"
|
|
|
|
"github.com/arangodb/go-driver"
|
|
"github.com/gin-contrib/sessions"
|
|
"github.com/gin-contrib/sessions/cookie"
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/SecurityBrewery/catalyst"
|
|
"github.com/SecurityBrewery/catalyst/cmd"
|
|
"github.com/SecurityBrewery/catalyst/database/busdb"
|
|
"github.com/SecurityBrewery/catalyst/generated/models"
|
|
"github.com/SecurityBrewery/catalyst/hooks"
|
|
"github.com/SecurityBrewery/catalyst/role"
|
|
"github.com/SecurityBrewery/catalyst/test"
|
|
)
|
|
|
|
func main() {
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
|
|
config, err := cmd.ParseCatalystConfig()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// create app and clear db after start
|
|
theCatalyst, err := catalyst.New(&hooks.Hooks{
|
|
DatabaseAfterConnectFuncs: []func(ctx context.Context, client driver.Client, name string){test.Clear},
|
|
}, config)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
demoUser := &models.UserResponse{ID: "demo", Roles: []string{role.Admin}}
|
|
ctx := busdb.UserContext(context.Background(), demoUser)
|
|
if err := test.SetupTestData(ctx, theCatalyst.DB); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// proxy static requests
|
|
theCatalyst.Server.NoRoute(
|
|
sessions.Sessions(catalyst.SessionName, cookie.NewStore(config.Secret)),
|
|
catalyst.Authenticate(theCatalyst.DB, config.Auth),
|
|
catalyst.AuthorizeBlockedUser,
|
|
proxy,
|
|
)
|
|
|
|
if err = theCatalyst.Server.RunWithSigHandler(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func proxy(ctx *gin.Context) {
|
|
u, _ := url.Parse("http://localhost:8080")
|
|
proxy := httputil.NewSingleHostReverseProxy(u)
|
|
|
|
ctx.Request.Host = ctx.Request.URL.Host
|
|
|
|
proxy.ServeHTTP(ctx.Writer, ctx.Request)
|
|
}
|