forked from lug/matterbridge
7886f05e88
An extra avatarMap (cache) is created for mattermost and telegram. If MediaServerUpload is configured, the avatar images of users are downloaded the first time a user sends a message. If this download succeeds a message with EVENT_AVATAR_DOWNLOAD is sent to the originating protocol. This message also contains a SHA field (in msg.Extra["file"]), if this is not empty, the sha will be added to the avatarMap. (so we now have a userid-sha cache) Next time this user sends a message, the MediaServerUpload/sha/userid.png URL will be used as the avatar field.
261 lines
8.1 KiB
Go
261 lines
8.1 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/BurntSushi/toml"
|
|
"log"
|
|
"os"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
EVENT_JOIN_LEAVE = "join_leave"
|
|
EVENT_TOPIC_CHANGE = "topic_change"
|
|
EVENT_FAILURE = "failure"
|
|
EVENT_FILE_FAILURE_SIZE = "file_failure_size"
|
|
EVENT_AVATAR_DOWNLOAD = "avatar_download"
|
|
EVENT_REJOIN_CHANNELS = "rejoin_channels"
|
|
EVENT_USER_ACTION = "user_action"
|
|
EVENT_MSG_DELETE = "msg_delete"
|
|
)
|
|
|
|
type Message struct {
|
|
Text string `json:"text"`
|
|
Channel string `json:"channel"`
|
|
Username string `json:"username"`
|
|
UserID string `json:"userid"` // userid on the bridge
|
|
Avatar string `json:"avatar"`
|
|
Account string `json:"account"`
|
|
Event string `json:"event"`
|
|
Protocol string `json:"protocol"`
|
|
Gateway string `json:"gateway"`
|
|
Timestamp time.Time `json:"timestamp"`
|
|
ID string `json:"id"`
|
|
Extra map[string][]interface{}
|
|
}
|
|
|
|
type FileInfo struct {
|
|
Name string
|
|
Data *[]byte
|
|
Comment string
|
|
URL string
|
|
Size int64
|
|
Avatar bool
|
|
SHA string
|
|
}
|
|
|
|
type ChannelInfo struct {
|
|
Name string
|
|
Account string
|
|
Direction string
|
|
ID string
|
|
SameChannel map[string]bool
|
|
Options ChannelOptions
|
|
}
|
|
|
|
type Protocol struct {
|
|
AuthCode string // steam
|
|
BindAddress string // mattermost, slack // DEPRECATED
|
|
Buffer int // api
|
|
Charset string // irc
|
|
Debug bool // general
|
|
EditSuffix string // mattermost, slack, discord, telegram, gitter
|
|
EditDisable bool // mattermost, slack, discord, telegram, gitter
|
|
IconURL string // mattermost, slack
|
|
IgnoreNicks string // all protocols
|
|
IgnoreMessages string // all protocols
|
|
Jid string // xmpp
|
|
Login string // mattermost, matrix
|
|
MediaDownloadSize int // all protocols
|
|
MediaServerDownload string
|
|
MediaServerUpload string
|
|
MessageDelay int // IRC, time in millisecond to wait between messages
|
|
MessageFormat string // telegram
|
|
MessageLength int // IRC, max length of a message allowed
|
|
MessageQueue int // IRC, size of message queue for flood control
|
|
MessageSplit bool // IRC, split long messages with newlines on MessageLength instead of clipping
|
|
Muc string // xmpp
|
|
Name string // all protocols
|
|
Nick string // all protocols
|
|
NickFormatter string // mattermost, slack
|
|
NickServNick string // IRC
|
|
NickServUsername string // IRC
|
|
NickServPassword string // IRC
|
|
NicksPerRow int // mattermost, slack
|
|
NoHomeServerSuffix bool // matrix
|
|
NoTLS bool // mattermost
|
|
Password string // IRC,mattermost,XMPP,matrix
|
|
PrefixMessagesWithNick bool // mattemost, slack
|
|
Protocol string // all protocols
|
|
RejoinDelay int // IRC
|
|
ReplaceMessages [][]string // all protocols
|
|
ReplaceNicks [][]string // all protocols
|
|
RemoteNickFormat string // all protocols
|
|
Server string // IRC,mattermost,XMPP,discord
|
|
ShowJoinPart bool // all protocols
|
|
ShowTopicChange bool // slack
|
|
ShowEmbeds bool // discord
|
|
SkipTLSVerify bool // IRC, mattermost
|
|
StripNick bool // all protocols
|
|
Team string // mattermost
|
|
Token string // gitter, slack, discord, api
|
|
URL string // mattermost, slack // DEPRECATED
|
|
UseAPI bool // mattermost, slack
|
|
UseSASL bool // IRC
|
|
UseTLS bool // IRC
|
|
UseFirstName bool // telegram
|
|
UseUserName bool // discord
|
|
UseInsecureURL bool // telegram
|
|
WebhookBindAddress string // mattermost, slack
|
|
WebhookURL string // mattermost, slack
|
|
WebhookUse string // mattermost, slack, discord
|
|
}
|
|
|
|
type ChannelOptions struct {
|
|
Key string // irc
|
|
WebhookURL string // discord
|
|
}
|
|
|
|
type Bridge struct {
|
|
Account string
|
|
Channel string
|
|
Options ChannelOptions
|
|
SameChannel bool
|
|
}
|
|
|
|
type Gateway struct {
|
|
Name string
|
|
Enable bool
|
|
In []Bridge
|
|
Out []Bridge
|
|
InOut []Bridge
|
|
}
|
|
|
|
type SameChannelGateway struct {
|
|
Name string
|
|
Enable bool
|
|
Channels []string
|
|
Accounts []string
|
|
}
|
|
|
|
type Config struct {
|
|
Api map[string]Protocol
|
|
IRC map[string]Protocol
|
|
Mattermost map[string]Protocol
|
|
Matrix map[string]Protocol
|
|
Slack map[string]Protocol
|
|
Steam map[string]Protocol
|
|
Gitter map[string]Protocol
|
|
Xmpp map[string]Protocol
|
|
Discord map[string]Protocol
|
|
Telegram map[string]Protocol
|
|
Rocketchat map[string]Protocol
|
|
Sshchat map[string]Protocol
|
|
General Protocol
|
|
Gateway []Gateway
|
|
SameChannelGateway []SameChannelGateway
|
|
}
|
|
|
|
type BridgeConfig struct {
|
|
Config Protocol
|
|
General *Protocol
|
|
Account string
|
|
Remote chan Message
|
|
}
|
|
|
|
func NewConfig(cfgfile string) *Config {
|
|
var cfg Config
|
|
if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
fail := false
|
|
for k, v := range cfg.Mattermost {
|
|
res := Deprecated(v, "mattermost."+k)
|
|
if res {
|
|
fail = res
|
|
}
|
|
}
|
|
for k, v := range cfg.Slack {
|
|
res := Deprecated(v, "slack."+k)
|
|
if res {
|
|
fail = res
|
|
}
|
|
}
|
|
for k, v := range cfg.Rocketchat {
|
|
res := Deprecated(v, "rocketchat."+k)
|
|
if res {
|
|
fail = res
|
|
}
|
|
}
|
|
if fail {
|
|
log.Fatalf("Fix your config. Please see changelog for more information")
|
|
}
|
|
if cfg.General.MediaDownloadSize == 0 {
|
|
cfg.General.MediaDownloadSize = 1000000
|
|
}
|
|
return &cfg
|
|
}
|
|
|
|
func OverrideCfgFromEnv(cfg *Config, protocol string, account string) {
|
|
var protoCfg Protocol
|
|
val := reflect.ValueOf(cfg).Elem()
|
|
// loop over the Config struct
|
|
for i := 0; i < val.NumField(); i++ {
|
|
typeField := val.Type().Field(i)
|
|
// look for the protocol map (both lowercase)
|
|
if strings.ToLower(typeField.Name) == protocol {
|
|
// get the Protocol struct from the map
|
|
data := val.Field(i).MapIndex(reflect.ValueOf(account))
|
|
protoCfg = data.Interface().(Protocol)
|
|
protoStruct := reflect.ValueOf(&protoCfg).Elem()
|
|
// loop over the found protocol struct
|
|
for i := 0; i < protoStruct.NumField(); i++ {
|
|
typeField := protoStruct.Type().Field(i)
|
|
// build our environment key (eg MATTERBRIDGE_MATTERMOST_WORK_LOGIN)
|
|
key := "matterbridge_" + protocol + "_" + account + "_" + typeField.Name
|
|
key = strings.ToUpper(key)
|
|
// search the environment
|
|
res := os.Getenv(key)
|
|
// if it exists and the current field is a string
|
|
// then update the current field
|
|
if res != "" {
|
|
fieldVal := protoStruct.Field(i)
|
|
if fieldVal.Kind() == reflect.String {
|
|
log.Printf("config: overriding %s from env with %s\n", key, res)
|
|
fieldVal.Set(reflect.ValueOf(res))
|
|
}
|
|
}
|
|
}
|
|
// update the map with the modified Protocol (cfg.Protocol[account] = Protocol)
|
|
val.Field(i).SetMapIndex(reflect.ValueOf(account), reflect.ValueOf(protoCfg))
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetIconURL(msg *Message, cfg *Protocol) string {
|
|
iconURL := cfg.IconURL
|
|
info := strings.Split(msg.Account, ".")
|
|
protocol := info[0]
|
|
name := info[1]
|
|
iconURL = strings.Replace(iconURL, "{NICK}", msg.Username, -1)
|
|
iconURL = strings.Replace(iconURL, "{BRIDGE}", name, -1)
|
|
iconURL = strings.Replace(iconURL, "{PROTOCOL}", protocol, -1)
|
|
return iconURL
|
|
}
|
|
|
|
func Deprecated(cfg Protocol, account string) bool {
|
|
if cfg.BindAddress != "" {
|
|
log.Printf("ERROR: %s BindAddress is deprecated, you need to change it to WebhookBindAddress.", account)
|
|
} else if cfg.URL != "" {
|
|
log.Printf("ERROR: %s URL is deprecated, you need to change it to WebhookURL.", account)
|
|
} else if cfg.UseAPI {
|
|
log.Printf("ERROR: %s UseAPI is deprecated, it's enabled by default, please remove it from your config file.", account)
|
|
} else {
|
|
return false
|
|
}
|
|
return true
|
|
//log.Fatalf("ERROR: Fix your config: %s", account)
|
|
}
|