Refactor connector creation logic

This commit is contained in:
Alex 2020-02-28 10:18:47 +01:00
parent 11963aaf3d
commit 30a5cdc2a3
7 changed files with 120 additions and 115 deletions

View file

@ -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)
}

View file

@ -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
}

View file

@ -5,28 +5,31 @@ import (
)
func init() {
Register("irc", ConfigSchema{
&ConfigEntry{
Name: "nick",
Description: "Nickname",
Required: true,
},
&ConfigEntry{
Name: "server",
Description: "Server",
Required: true,
},
&ConfigEntry{
Name: "port",
Description: "Port",
IsNumeric: true,
Default: "6667",
},
&ConfigEntry{
Name: "ssl",
Description: "Use SSL",
IsBoolean: true,
Default: "false",
Register("irc", Protocol{
NewConnector: func() Connector { return &IRC{} },
Schema: ConfigSchema{
&ConfigEntry{
Name: "nick",
Description: "Nickname",
Required: true,
},
&ConfigEntry{
Name: "server",
Description: "Server",
Required: true,
},
&ConfigEntry{
Name: "port",
Description: "Port",
IsNumeric: true,
Default: "6667",
},
&ConfigEntry{
Name: "ssl",
Description: "Use SSL",
IsBoolean: true,
Default: "false",
},
},
})
}

View file

@ -5,48 +5,51 @@ import (
)
func init() {
Register("mattermost", ConfigSchema{
&ConfigEntry{
Name: "server",
Description: "Server",
Required: true,
},
&ConfigEntry{
Name: "username",
Description: "Username",
Required: true,
},
&ConfigEntry{
Name: "password",
Description: "Password",
IsPassword: true,
},
&ConfigEntry{
Name: "token",
Description: "Authentification token (replaces password if set)",
},
&ConfigEntry{
Name: "teams",
Description: "Comma-separated list of teams to follow",
Required: true,
},
&ConfigEntry{
Name: "no_tls",
Description: "Disable SSL/TLS",
IsBoolean: true,
Default: "false",
},
&ConfigEntry{
Name: "initial_backlog",
Description: "Maximum number of messages to load when joining a channel",
IsNumeric: true,
Default: "1000",
},
&ConfigEntry{
Name: "initial_members",
Description: "Maximum number of members to load when joining a channel",
IsNumeric: true,
Default: "100",
Register("mattermost", Protocol{
NewConnector: func() Connector { return &Mattermost{} },
Schema: ConfigSchema{
&ConfigEntry{
Name: "server",
Description: "Server",
Required: true,
},
&ConfigEntry{
Name: "username",
Description: "Username",
Required: true,
},
&ConfigEntry{
Name: "password",
Description: "Password",
IsPassword: true,
},
&ConfigEntry{
Name: "token",
Description: "Authentification token (replaces password if set)",
},
&ConfigEntry{
Name: "teams",
Description: "Comma-separated list of teams to follow",
Required: true,
},
&ConfigEntry{
Name: "no_tls",
Description: "Disable SSL/TLS",
IsBoolean: true,
Default: "false",
},
&ConfigEntry{
Name: "initial_backlog",
Description: "Maximum number of messages to load when joining a channel",
IsNumeric: true,
Default: "1000",
},
&ConfigEntry{
Name: "initial_members",
Description: "Maximum number of members to load when joining a channel",
IsNumeric: true,
Default: "100",
},
},
})
}

View file

@ -5,34 +5,37 @@ import (
)
func init() {
Register("xmpp", ConfigSchema{
&ConfigEntry{
Name: "jid",
Description: "JID",
Required: true,
},
&ConfigEntry{
Name: "password",
Description: "Password",
Required: true,
IsPassword: true,
},
&ConfigEntry{
Name: "nickname",
Description: "Nickname in MUCs",
Required: true,
},
&ConfigEntry{
Name: "port",
Description: "Port",
IsNumeric: true,
Default: "5222",
},
&ConfigEntry{
Name: "ssl",
Description: "Use SSL",
IsBoolean: true,
Default: "true",
Register("xmpp", Protocol{
NewConnector: func() Connector { return &XMPP{} },
Schema: ConfigSchema{
&ConfigEntry{
Name: "jid",
Description: "JID",
Required: true,
},
&ConfigEntry{
Name: "password",
Description: "Password",
Required: true,
IsPassword: true,
},
&ConfigEntry{
Name: "nickname",
Description: "Nickname in MUCs",
Required: true,
},
&ConfigEntry{
Name: "port",
Description: "Port",
IsNumeric: true,
Default: "5222",
},
&ConfigEntry{
Name: "ssl",
Description: "Use SSL",
IsBoolean: true,
Default: "true",
},
},
})
}

18
util.go
View file

@ -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
}
}

2
web.go
View file

@ -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