Refactor connector creation logic

account.go

@@ -9,6 +9,11 @@ import (
 	log "github.com/sirupsen/logrus"
 
 	. "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector"
+
+	// Necessary for them to register their protocols
+	_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/irc"
+	_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/mattermost"
+	_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/xmpp"
 )
 
 type Account struct {
@@ -45,7 +50,11 @@ func SetAccount(mxid string, name string, protocol string, config map[string]str
 			go prev_acct.connect()
 		}
 	} else {
-		conn := createConnector(protocol)
+		proto, ok := Protocols[protocol]
+		if !ok {
+			return fmt.Errorf("Invalid protocol: %s", protocol)
+		}
+		conn := proto.NewConnector()
 		if conn == nil {
 			return fmt.Errorf("Could not create connector for protocol %s", protocol)
 		}

connector/config.go

@@ -46,6 +46,11 @@ func (c Configuration) GetBool(k string, deflt ...bool) (bool, error) {
 
 // ----
 
+type Protocol struct {
+	NewConnector func() Connector
+	Schema       ConfigSchema
+}
+
 type ConfigSchema []*ConfigEntry
 
 type ConfigEntry struct {
@@ -59,8 +64,8 @@ type ConfigEntry struct {
 	IsBoolean   bool
 }
 
-var Protocols = map[string]ConfigSchema{}
+var Protocols = map[string]Protocol{}
 
-func Register(name string, schema ConfigSchema) {
-	Protocols[name] = schema
+func Register(name string, protocol Protocol) {
+	Protocols[name] = protocol
 }

connector/irc/config.go

@@ -5,7 +5,9 @@ import (
 )
 
 func init() {
-	Register("irc", ConfigSchema{
+	Register("irc", Protocol{
+		NewConnector: func() Connector { return &IRC{} },
+		Schema: ConfigSchema{
 			&ConfigEntry{
 				Name:        "nick",
 				Description: "Nickname",
@@ -28,5 +30,6 @@ func init() {
 				IsBoolean:   true,
 				Default:     "false",
 			},
+		},
 	})
 }

connector/mattermost/config.go

@@ -5,7 +5,9 @@ import (
 )
 
 func init() {
-	Register("mattermost", ConfigSchema{
+	Register("mattermost", Protocol{
+		NewConnector: func() Connector { return &Mattermost{} },
+		Schema: ConfigSchema{
 			&ConfigEntry{
 				Name:        "server",
 				Description: "Server",
@@ -48,5 +50,6 @@ func init() {
 				IsNumeric:   true,
 				Default:     "100",
 			},
+		},
 	})
 }

connector/xmpp/config.go

@@ -5,7 +5,9 @@ import (
 )
 
 func init() {
-	Register("xmpp", ConfigSchema{
+	Register("xmpp", Protocol{
+		NewConnector: func() Connector { return &XMPP{} },
+		Schema: ConfigSchema{
 			&ConfigEntry{
 				Name:        "jid",
 				Description: "JID",
@@ -34,5 +36,6 @@ func init() {
 				IsBoolean:   true,
 				Default:     "true",
 			},
+		},
 	})
 }

util.go

@@ -11,9 +11,6 @@ import (
 	"golang.org/x/crypto/nacl/secretbox"
 
 	. "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector"
-	"git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/irc"
-	"git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/mattermost"
-	"git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/xmpp"
 )
 
 const EASYBRIDGE_SYSTEM_PROTOCOL string = "✯◡✯"
@@ -101,18 +98,3 @@ func decryptAccountConfig(data string, key *[32]byte) (map[string]string, error)
 	err = json.Unmarshal(decoded, &config)
 	return config, err
 }
-
-// ----
-
-func createConnector(protocol string) Connector {
-	switch protocol {
-	case "irc":
-		return &irc.IRC{}
-	case "xmpp":
-		return &xmpp.XMPP{}
-	case "mattermost":
-		return &mattermost.Mattermost{}
-	default:
-		return nil
-	}
-}

web.go

@@ -237,7 +237,7 @@ func configForm(w http.ResponseWriter, r *http.Request,
 		Protocol:     protocol,
 		Config:       map[string]string{},
 		Errors:       map[string]string{},
-		Schema:       connector.Protocols[protocol],
+		Schema:       connector.Protocols[protocol].Schema,
 	}
 	for k, v := range prevConfig {
 		data.Config[k] = v