refactor: add concept of a pre-init function that gets called before init

protocols/protocols.go

@@ -12,8 +12,9 @@ import (
 
 func RegisterProtocols() {
 	registry.Register(registry.ProtocolEntry{
-		Key:    "s5",
-		Module: s5.ProtocolModule,
+		Key:         "s5",
+		Module:      s5.ProtocolModule,
+		PreInitFunc: s5.PreInit,
 	})
 }
 
@@ -23,6 +24,9 @@ func BuildProtocols(config *viper.Viper) fx.Option {
 	for _, entry := range registry.GetRegistry() {
 		if lo.Contains(enabledProtocols, entry.Key) {
 			options = append(options, entry.Module)
+			if entry.PreInitFunc != nil {
+				options = append(options, fx.Invoke(entry.PreInitFunc))
+			}
 		}
 	}
 

protocols/registry/registry.go

@@ -10,14 +10,15 @@ const GroupName = "protocols"
 
 type Protocol interface {
 	Name() string
-	Init(...any) error
+	Init() error
 	Start(ctx context.Context) error
 	Stop(ctx context.Context) error
 }
 
 type ProtocolEntry struct {
-	Key    string
-	Module fx.Option
+	Key         string
+	Module      fx.Option
+	PreInitFunc interface{}
 }
 
 var protocolEntry []ProtocolEntry