58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
|
package setup
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"forge.cadoles.com/cadoles/bouncer/internal/config"
|
||
|
|
"forge.cadoles.com/cadoles/bouncer/internal/proxy/director"
|
||
|
|
"forge.cadoles.com/cadoles/bouncer/internal/store"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Registry struct {
|
||
|
|
layers map[store.LayerType]LayerSetupFunc
|
||
|
|
}
|
||
|
|
|
||
|
|
type LayerSetupFunc func(context.Context, *config.Config) (director.Layer, error)
|
||
|
|
|
||
|
|
func (r *Registry) RegisterLayer(layerType store.LayerType, layerSetup LayerSetupFunc) {
|
||
|
|
r.layers[layerType] = layerSetup
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Registry) CreateLayers(ctx context.Context, conf *config.Config) ([]director.Layer, error) {
|
||
|
|
layers := make([]director.Layer, 0, len(r.layers))
|
||
|
|
|
||
|
|
for layerType, layerSetup := range r.layers {
|
||
|
|
layer, err := layerSetup(ctx, conf)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrapf(err, "could not create layer '%s'", layerType)
|
||
|
|
}
|
||
|
|
|
||
|
|
layers = append(layers, layer)
|
||
|
|
}
|
||
|
|
|
||
|
|
return layers, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Registry) LayerTypeExists(layerType store.LayerType) bool {
|
||
|
|
_, exists := r.layers[layerType]
|
||
|
|
|
||
|
|
return exists
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Registry) GetLayerTypes() []store.LayerType {
|
||
|
|
layerTypes := make([]store.LayerType, 0, len(r.layers))
|
||
|
|
|
||
|
|
for layerType := range r.layers {
|
||
|
|
layerTypes = append(layerTypes, layerType)
|
||
|
|
}
|
||
|
|
|
||
|
|
return layerTypes
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRegistry() *Registry {
|
||
|
|
return &Registry{
|
||
|
|
layers: make(map[store.LayerType]LayerSetupFunc),
|
||
|
|
}
|
||
|
|
}
|