feat: add Inspect option to registry (#1523)

* chore: add alnr to codeowners

* feat: add Inspect option

.github/CODEOWNERS

@@ -1 +1 @@
-*            @zepatrik @hperl
+*            @zepatrik @hperl @alnr

internal/driver/registry_factory.go

@@ -96,6 +96,12 @@ func NewDefaultRegistry(ctx context.Context, flags *pflag.FlagSet, withoutNetwor
 		return nil, errors.Wrap(err, "unable to initialize service registry")
 	}
 
+	if inspect := options.Inspect(); inspect != nil {
+		if err := inspect(r.Persister().Connection(ctx)); err != nil {
+			return nil, errors.Wrap(err, "inspect")
+		}
+	}
+
 	return r, nil
 }
 

ketoctx/options.go

@@ -7,6 +7,7 @@ import (
 	"io/fs"
 	"net/http"
 
+	"github.com/gobuffalo/pop/v6"
 	"github.com/ory/x/healthx"
 	"github.com/ory/x/logrusx"
 	"github.com/ory/x/otelx"
@@ -25,9 +26,11 @@ type (
 		migrationOpts          []popx.MigrationBoxOption
 		readyCheckers          healthx.ReadyCheckers
 		extraMigrations        []fs.FS
+		inspect                InspectFunc
 	}
 	Option        func(o *opts)
 	TracerWrapper func(*otelx.Tracer) *otelx.Tracer
+	InspectFunc   func(*pop.Connection) error
 )
 
 // WithLogger sets the logger.
@@ -96,6 +99,12 @@ func WithReadinessCheck(name string, rc healthx.ReadyChecker) Option {
 	}
 }
 
+func Inspect(f InspectFunc) Option {
+	return func(o *opts) {
+		o.inspect = f
+	}
+}
+
 func (o *opts) Logger() *logrusx.Logger {
 	return o.logger
 }
@@ -128,6 +137,10 @@ func (o *opts) ReadyCheckers() healthx.ReadyCheckers {
 	return o.readyCheckers
 }
 
+func (o *opts) Inspect() InspectFunc {
+	return o.inspect
+}
+
 func Options(options ...Option) *opts {
 	o := &opts{
 		contextualizer: &DefaultContextualizer{},