edge/pkg/module/auth/module.go

102 lines
2.6 KiB
Go
Raw Normal View History

2023-02-21 12:14:29 +01:00
package auth
import (
"net/http"
"forge.cadoles.com/arcad/edge/pkg/app"
edgeHTTP "forge.cadoles.com/arcad/edge/pkg/http"
"forge.cadoles.com/arcad/edge/pkg/jwtutil"
2023-02-21 12:14:29 +01:00
"forge.cadoles.com/arcad/edge/pkg/module/util"
"github.com/dop251/goja"
"github.com/pkg/errors"
"gitlab.com/wpetit/goweb/logger"
2023-02-21 12:14:29 +01:00
)
const (
CookieName string = "edge-auth"
)
2023-02-21 12:14:29 +01:00
const (
ClaimSubject = "sub"
ClaimIssuer = "iss"
ClaimPreferredUsername = "preferred_username"
ClaimEdgeRole = "edge_role"
ClaimEdgeTenant = "edge_tenant"
ClaimEdgeEntrypoint = "edge_entrypoint"
2023-02-21 12:14:29 +01:00
)
type Module struct {
server *app.Server
getClaimFn GetClaimFunc
2023-02-21 12:14:29 +01:00
}
func (m *Module) Name() string {
return "auth"
}
func (m *Module) Export(export *goja.Object) {
if err := export.Set("getClaim", m.getClaim); err != nil {
panic(errors.Wrap(err, "could not set 'getClaim' function"))
2023-02-21 12:14:29 +01:00
}
if err := export.Set("CLAIM_SUBJECT", ClaimSubject); err != nil {
panic(errors.Wrap(err, "could not set 'CLAIM_SUBJECT' property"))
2023-02-21 12:14:29 +01:00
}
if err := export.Set("CLAIM_TENANT", ClaimEdgeTenant); err != nil {
panic(errors.Wrap(err, "could not set 'CLAIM_TENANT' property"))
}
if err := export.Set("CLAIM_ENTRYPOINT", ClaimEdgeEntrypoint); err != nil {
panic(errors.Wrap(err, "could not set 'CLAIM_ENTRYPOINT' property"))
}
if err := export.Set("CLAIM_ROLE", ClaimEdgeRole); err != nil {
panic(errors.Wrap(err, "could not set 'CLAIM_ROLE' property"))
}
if err := export.Set("CLAIM_PREFERRED_USERNAME", ClaimPreferredUsername); err != nil {
panic(errors.Wrap(err, "could not set 'CLAIM_PREFERRED_USERNAME' property"))
}
if err := export.Set("CLAIM_ISSUER", ClaimIssuer); err != nil {
panic(errors.Wrap(err, "could not set 'CLAIM_ISSUER' property"))
}
2023-02-21 12:14:29 +01:00
}
func (m *Module) getClaim(call goja.FunctionCall, rt *goja.Runtime) goja.Value {
2023-02-21 12:14:29 +01:00
ctx := util.AssertContext(call.Argument(0), rt)
claimName := util.AssertString(call.Argument(1), rt)
2023-02-21 12:14:29 +01:00
req, ok := ctx.Value(edgeHTTP.ContextKeyOriginRequest).(*http.Request)
2023-02-21 12:14:29 +01:00
if !ok {
panic(rt.ToValue(errors.New("could not find http request in context")))
2023-02-21 12:14:29 +01:00
}
claim, err := m.getClaimFn(ctx, req, claimName)
2023-02-21 12:14:29 +01:00
if err != nil {
if errors.Is(err, jwtutil.ErrUnauthenticated) {
return nil
}
2023-02-21 12:14:29 +01:00
logger.Error(ctx, "could not retrieve claim", logger.CapturedE(errors.WithStack(err)))
return nil
}
return rt.ToValue(claim)
}
2023-02-21 12:14:29 +01:00
func ModuleFactory(funcs ...OptionFunc) app.ServerModuleFactory {
opt := defaultOptions()
for _, fn := range funcs {
fn(opt)
2023-02-21 12:14:29 +01:00
}
return func(server *app.Server) app.ServerModule {
return &Module{
server: server,
getClaimFn: opt.GetClaim,
2023-02-21 12:14:29 +01:00
}
}
}