89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package admin
|
|
|
|
import (
|
|
"context"
|
|
|
|
"forge.cadoles.com/cadoles/bouncer/internal/integration"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/jwk"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/setup"
|
|
"github.com/pkg/errors"
|
|
"gitlab.com/wpetit/goweb/logger"
|
|
)
|
|
|
|
func (s *Server) initRepositories(ctx context.Context) error {
|
|
if err := s.initRedisClient(ctx); err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
if err := s.initLayerRepository(ctx); err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
if err := s.initProxyRepository(ctx); err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) initRedisClient(ctx context.Context) error {
|
|
client := setup.NewRedisClient(ctx, s.redisConfig)
|
|
|
|
s.redisClient = client
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) initLayerRepository(ctx context.Context) error {
|
|
layerRepository, err := setup.NewLayerRepository(ctx, s.redisClient)
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
s.layerRepository = layerRepository
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) initProxyRepository(ctx context.Context) error {
|
|
proxyRepository, err := setup.NewProxyRepository(ctx, s.redisClient)
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
s.proxyRepository = proxyRepository
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) initPrivateKey(ctx context.Context) error {
|
|
localKey, err := jwk.LoadOrGenerate(string(s.serverConfig.Auth.PrivateKey), jwk.DefaultKeySize)
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
ctx = integration.WithPrivateKey(ctx, localKey)
|
|
|
|
key, err := integration.RunOnKeyLoad(ctx, s.integrations)
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
if key != nil {
|
|
s.privateKey = key
|
|
} else {
|
|
s.privateKey = localKey
|
|
}
|
|
|
|
logger.Info(ctx, "using private key", logger.F("keyID", s.privateKey.KeyID()))
|
|
|
|
publicKeys, err := jwk.PublicKeySet(s.privateKey)
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
s.publicKeys = publicKeys
|
|
|
|
return nil
|
|
}
|