easybridge/connector/mattermost/mattermost.go

387 lines
8.5 KiB
Go
Raw Normal View History

2020-02-17 22:17:13 +00:00
package mattermost
import (
"fmt"
_ "os"
"strings"
"time"
2020-02-18 16:17:04 +00:00
"encoding/json"
2020-02-17 22:17:13 +00:00
"github.com/mattermost/mattermost-server/model"
"github.com/42wim/matterbridge/matterclient"
2020-02-18 16:17:04 +00:00
log "github.com/sirupsen/logrus"
2020-02-17 22:17:13 +00:00
. "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector"
)
// User id format: nickname@server
// Room id format: room_name@team@server
type Mattermost struct {
handler Handler
server string
username string
team string
conn *matterclient.MMClient
handlerStopChan chan bool
2020-02-18 16:17:04 +00:00
usermap map[string]string // map username to mm user id
sentjoinedmap map[string]bool // map username/room name to bool
userdisplaynamemap map[UserID]string // map username to last displayname
2020-02-17 22:17:13 +00:00
}
func (mm *Mattermost) SetHandler(h Handler) {
mm.handler = h
}
func (mm *Mattermost) Protocol() string {
return "mattermost"
}
func (mm *Mattermost) Configure(c Configuration) error {
if mm.conn != nil {
mm.Close()
}
var err error
mm.server, err = c.GetString("server")
if err != nil {
return err
}
mm.username, err = c.GetString("username")
if err != nil {
return err
}
mm.team, err = c.GetString("team")
if err != nil {
return err
}
2020-02-18 16:17:04 +00:00
notls, err := c.GetBool("no_tls", false)
if err != nil {
return err
}
2020-02-17 22:17:13 +00:00
password, _ := c.GetString("password", "")
token, _ := c.GetString("token", "")
if token != "" {
password = "token=" + token
}
mm.conn = matterclient.New(mm.username, password, mm.team, mm.server)
2020-02-18 16:17:04 +00:00
mm.conn.Credentials.NoTLS = notls
2020-02-17 22:17:13 +00:00
err = mm.conn.Login()
if err != nil {
return err
}
go mm.conn.WsReceiver()
go mm.conn.StatusLoop()
fmt.Printf("CLIENT4: %#v\n", mm.conn.Client)
for i := 0; i < 42; i++ {
time.Sleep(time.Duration(1) * time.Second)
if mm.conn.WsConnected {
mm.handleConnected()
return nil
}
}
return fmt.Errorf("Failed to connect after 42s attempting")
}
func (mm *Mattermost) User() UserID {
return UserID(mm.username + "@" + mm.server)
}
func (mm *Mattermost) getTeamIdByName(name string) string {
for _, team := range mm.conn.OtherTeams {
if team.Team.Name == name {
return team.Id
}
}
return ""
}
func (mm *Mattermost) checkRoomId(id RoomID) (string, error) {
x := strings.Split(string(id), "@")
if len(x) == 1 {
return "", fmt.Errorf("Please write whole room ID with team and server: %s@%s@%s", id, mm.team, mm.server)
}
if len(x) == 2 {
return x[0], nil
}
if len(x) != 3 || x[2] != mm.server {
return "", fmt.Errorf("Invalid room ID: %s", id)
}
team_id := mm.getTeamIdByName(x[1])
if team_id == "" {
return "", fmt.Errorf("Team not found: %s", id)
}
ch_id := mm.conn.GetChannelId(x[0], team_id)
if ch_id == "" {
return "", fmt.Errorf("Channel not found: %s", id)
}
return ch_id, nil
}
func (mm *Mattermost) reverseRoomId(id string) RoomID {
team := mm.conn.GetChannelTeamId(id)
if team == "" {
return RoomID(fmt.Sprintf("%s@%s", id, mm.server))
} else {
teamName := mm.conn.GetTeamName(team)
name := mm.conn.GetChannelName(id)
fmt.Printf("CHANNEL NAME: %s TEAM: %s\n", name, teamName)
return RoomID(fmt.Sprintf("%s@%s@%s", name, teamName, mm.server))
}
}
func (mm *Mattermost) checkUserId(id UserID) (string, error) {
x := strings.Split(string(id), "@")
if len(x) == 1 {
return "", fmt.Errorf("Please write whole user ID with server: %s@%s", id, mm.server)
}
if len(x) != 2 || x[1] != mm.server {
return "", fmt.Errorf("Invalid user ID: %s", id)
}
2020-02-18 16:17:04 +00:00
if user_id, ok := mm.usermap[x[0]]; ok {
return user_id, nil
}
u, resp := mm.conn.Client.GetUserByUsername(x[0], "")
if u == nil || resp.Error != nil {
return "", fmt.Errorf("Not found: %s (%s)", x[0], resp.Error)
}
mm.usermap[x[0]] = u.Id
return u.Id, nil
2020-02-17 22:17:13 +00:00
}
func (mm *Mattermost) SetUserInfo(info *UserInfo) error {
return fmt.Errorf("Not implemented")
}
func (mm *Mattermost) SetRoomInfo(roomId RoomID, info *RoomInfo) error {
ch, err := mm.checkRoomId(roomId)
if err != nil {
return err
}
if info.Topic != "" {
mm.conn.UpdateChannelHeader(ch, info.Topic)
}
if info.Picture != nil {
err = fmt.Errorf("Not supported: channel picture on mattermost")
}
if info.Name != "" {
err = fmt.Errorf("Not supported: channel name on mattermost")
}
return err
}
func (mm *Mattermost) Join(roomId RoomID) error {
ch, err := mm.checkRoomId(roomId)
if err != nil {
return err
}
return mm.conn.JoinChannel(ch)
}
func (mm *Mattermost) Invite(userId UserID, roomId RoomID) error {
2020-02-18 16:17:04 +00:00
if roomId == "" {
_, err := mm.checkUserId(userId)
return err
}
2020-02-17 22:17:13 +00:00
return fmt.Errorf("Not supported: invite on mattermost")
}
func (mm *Mattermost) Leave(roomId RoomID) {
// Not supported? TODO
}
func (mm *Mattermost) Send(event *Event) error {
// TODO: attachements
2020-02-18 16:17:04 +00:00
// TODO: verify private messages work
post := &model.Post{
Message: event.Text,
}
if event.Type == EVENT_ACTION {
post.Type = "me"
}
2020-02-17 22:17:13 +00:00
if event.Room != "" {
ch, err := mm.checkRoomId(event.Room)
if err != nil {
return err
}
2020-02-18 16:17:04 +00:00
post.ChannelId = ch
2020-02-17 22:17:13 +00:00
} else if event.Recipient != "" {
ui, err := mm.checkUserId(event.Recipient)
if err != nil {
return err
}
2020-02-18 16:17:04 +00:00
_, resp := mm.conn.Client.CreateDirectChannel(mm.conn.User.Id, ui)
if resp.Error != nil {
return resp.Error
}
channelName := model.GetDMNameFromIds(ui, mm.conn.User.Id)
err = mm.conn.UpdateChannels()
if err != nil {
return err
}
post.ChannelId = mm.conn.GetChannelId(channelName, "")
2020-02-17 22:17:13 +00:00
} else {
return fmt.Errorf("Invalid target")
}
2020-02-18 16:17:04 +00:00
_, resp := mm.conn.Client.CreatePost(post)
if resp.Error != nil {
return resp.Error
}
return nil
2020-02-17 22:17:13 +00:00
}
func (mm *Mattermost) Close() {
mm.conn.WsQuit = true
if mm.handlerStopChan != nil {
mm.handlerStopChan <- true
mm.handlerStopChan = nil
}
}
func (mm *Mattermost) handleConnected() {
mm.handlerStopChan = make(chan bool)
2020-02-18 16:17:04 +00:00
mm.usermap = make(map[string]string)
mm.sentjoinedmap = make(map[string]bool)
mm.userdisplaynamemap = make(map[UserID]string)
2020-02-17 22:17:13 +00:00
go mm.handleLoop(mm.conn.MessageChan, mm.handlerStopChan)
fmt.Printf("Connected to mattermost\n")
chans := mm.conn.GetChannels()
for _, ch := range chans {
if len(strings.Split(ch.Name, "__")) == 2 {
continue // This is a DM channel
}
id := mm.reverseRoomId(ch.Id)
chName := ch.DisplayName
if teamName := mm.conn.GetTeamName(ch.TeamId); teamName != "" {
chName = teamName + " / " + chName
}
mm.handler.Joined(id)
mm.handler.RoomInfoUpdated(id, UserID(""), &RoomInfo{
Name: chName,
Topic: ch.Header,
})
}
}
func (mm *Mattermost) handleLoop(msgCh chan *matterclient.Message, quitCh chan bool) {
for {
select {
case <-quitCh:
break
case msg := <-msgCh:
fmt.Printf("Mattermost: %#v\n", msg)
2020-02-18 16:17:04 +00:00
fmt.Printf("Mattermost raw: %#v\n", msg.Raw)
err := mm.handlePosted(msg.Raw)
if err != nil {
log.Warnf("Mattermost error: %s", err)
2020-02-17 22:17:13 +00:00
}
}
}
}
2020-02-18 16:17:04 +00:00
func (mm *Mattermost) handlePosted(msg *model.WebSocketEvent) error {
channel_name := msg.Data["channel_name"].(string)
post_str := msg.Data["post"].(string)
var post model.Post
err := json.Unmarshal([]byte(post_str), &post)
if err != nil {
return err
}
// Skip self messages
if post.UserId == mm.conn.User.Id {
return nil
}
// Find sending user
user := mm.conn.GetUser(post.UserId)
if user == nil {
return fmt.Errorf("Invalid user")
}
userId := UserID(fmt.Sprintf("%s@%s", user.Username, mm.server))
if lastdn, ok := mm.userdisplaynamemap[userId]; !ok || lastdn != user.Nickname {
log.Warnf("Update displayname %s %s", userId, user.Nickname)
mm.handler.UserInfoUpdated(userId, &UserInfo{
DisplayName: user.Nickname,
})
mm.userdisplaynamemap[userId] = user.Nickname
}
2020-02-18 16:17:04 +00:00
// Build message event
msg_ev := &Event{
Author: userId,
Text: post.Message,
Type: EVENT_MESSAGE,
}
if post.Type == "me" {
msg_ev.Type = EVENT_ACTION
}
// Dispatch as PM or as room message
if len(strings.Split(channel_name, "__")) == 2 {
// Private message, no need to find room id
if user.Id == mm.conn.User.Id {
// Skip self sent messages
return nil
}
mm.handler.Event(msg_ev)
} else {
roomId := mm.reverseRoomId(post.ChannelId)
if roomId == "" {
return fmt.Errorf("Invalid channel id")
}
cache_key := fmt.Sprintf("%s / %s", userId, roomId)
if _, ok := mm.sentjoinedmap[cache_key]; !ok {
mm.handler.Event(&Event{
Author: userId,
Room: roomId,
Type: EVENT_JOIN,
})
mm.sentjoinedmap[cache_key] = true
}
2020-02-18 16:17:04 +00:00
if post.Type == "system_header_change" {
new_header := post.Props["new_header"].(string)
mm.handler.RoomInfoUpdated(roomId, userId, &RoomInfo{
Topic: new_header,
})
} else if post.Type == "" || post.Type == "me" {
msg_ev.Room = roomId
mm.handler.Event(msg_ev)
} else {
return fmt.Errorf("Unhandled post type: %s", post.Type)
}
}
return nil
}