mirror of
https://github.com/42wim/matterbridge.git
synced 2024-11-15 00:19:24 +01:00
343 lines
8.4 KiB
Go
343 lines
8.4 KiB
Go
package bslack
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/42wim/matterbridge/bridge/config"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/slack-go/slack"
|
|
)
|
|
|
|
const minimumRefreshInterval = 10 * time.Second
|
|
|
|
type users struct {
|
|
log *logrus.Entry
|
|
sc *slack.Client
|
|
|
|
users map[string]*slack.User
|
|
usersMutex sync.RWMutex
|
|
usersSyncPoints map[string]chan struct{}
|
|
|
|
refreshInProgress bool
|
|
earliestRefresh time.Time
|
|
refreshMutex sync.Mutex
|
|
}
|
|
|
|
func newUserManager(log *logrus.Entry, sc *slack.Client) *users {
|
|
return &users{
|
|
log: log,
|
|
sc: sc,
|
|
users: make(map[string]*slack.User),
|
|
usersSyncPoints: make(map[string]chan struct{}),
|
|
earliestRefresh: time.Now(),
|
|
}
|
|
}
|
|
|
|
func (b *users) getUser(id string) *slack.User {
|
|
b.usersMutex.RLock()
|
|
user, ok := b.users[id]
|
|
b.usersMutex.RUnlock()
|
|
if ok {
|
|
return user
|
|
}
|
|
b.populateUser(id)
|
|
b.usersMutex.RLock()
|
|
defer b.usersMutex.RUnlock()
|
|
|
|
return b.users[id]
|
|
}
|
|
|
|
func (b *users) getUsername(id string) string {
|
|
if user := b.getUser(id); user != nil {
|
|
if user.Profile.DisplayName != "" {
|
|
return user.Profile.DisplayName
|
|
}
|
|
return user.Name
|
|
}
|
|
b.log.Warnf("Could not find user with ID '%s'", id)
|
|
return ""
|
|
}
|
|
|
|
func (b *users) getAvatar(id string) string {
|
|
if user := b.getUser(id); user != nil {
|
|
return user.Profile.Image48
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (b *users) populateUser(userID string) {
|
|
for {
|
|
b.usersMutex.Lock()
|
|
_, exists := b.users[userID]
|
|
if exists {
|
|
// already in cache
|
|
b.usersMutex.Unlock()
|
|
return
|
|
}
|
|
|
|
if syncPoint, ok := b.usersSyncPoints[userID]; ok {
|
|
// Another goroutine is already populating this user for us so wait on it to finish.
|
|
b.usersMutex.Unlock()
|
|
<-syncPoint
|
|
// We do not return and iterate again to check that the entry does indeed exist
|
|
// in case the previous query failed for some reason.
|
|
} else {
|
|
b.usersSyncPoints[userID] = make(chan struct{})
|
|
defer func() {
|
|
// Wake up any waiting goroutines and remove the synchronization point.
|
|
close(b.usersSyncPoints[userID])
|
|
delete(b.usersSyncPoints, userID)
|
|
}()
|
|
break
|
|
}
|
|
}
|
|
|
|
// Do not hold the lock while fetching information from Slack
|
|
// as this might take an unbounded amount of time.
|
|
b.usersMutex.Unlock()
|
|
|
|
user, err := b.sc.GetUserInfo(userID)
|
|
if err != nil {
|
|
b.log.Debugf("GetUserInfo failed for %v: %v", userID, err)
|
|
return
|
|
}
|
|
|
|
b.usersMutex.Lock()
|
|
defer b.usersMutex.Unlock()
|
|
|
|
// Register user information.
|
|
b.users[userID] = user
|
|
}
|
|
|
|
func (b *users) invalidateUser(userID string) {
|
|
b.usersMutex.Lock()
|
|
defer b.usersMutex.Unlock()
|
|
delete(b.users, userID)
|
|
}
|
|
|
|
func (b *users) populateUsers(wait bool) {
|
|
b.refreshMutex.Lock()
|
|
if !wait && (time.Now().Before(b.earliestRefresh) || b.refreshInProgress) {
|
|
b.log.Debugf("Not refreshing user list as it was done less than %v ago.", minimumRefreshInterval)
|
|
b.refreshMutex.Unlock()
|
|
|
|
return
|
|
}
|
|
for b.refreshInProgress {
|
|
b.refreshMutex.Unlock()
|
|
time.Sleep(time.Second)
|
|
b.refreshMutex.Lock()
|
|
}
|
|
b.refreshInProgress = true
|
|
b.refreshMutex.Unlock()
|
|
|
|
newUsers := map[string]*slack.User{}
|
|
pagination := b.sc.GetUsersPaginated(slack.GetUsersOptionLimit(200))
|
|
count := 0
|
|
for {
|
|
var err error
|
|
pagination, err = pagination.Next(context.Background())
|
|
time.Sleep(time.Second)
|
|
if err != nil {
|
|
if pagination.Done(err) {
|
|
break
|
|
}
|
|
|
|
if err = handleRateLimit(b.log, err); err != nil {
|
|
b.log.Errorf("Could not retrieve users: %#v", err)
|
|
return
|
|
}
|
|
continue
|
|
}
|
|
|
|
for i := range pagination.Users {
|
|
newUsers[pagination.Users[i].ID] = &pagination.Users[i]
|
|
}
|
|
b.log.Debugf("getting %d users", len(pagination.Users))
|
|
count++
|
|
// more > 2000 users, slack will complain and ratelimit. break
|
|
if count > 10 {
|
|
b.log.Info("Large slack detected > 2000 users, skipping loading complete userlist.")
|
|
break
|
|
}
|
|
}
|
|
|
|
b.usersMutex.Lock()
|
|
defer b.usersMutex.Unlock()
|
|
b.users = newUsers
|
|
|
|
b.refreshMutex.Lock()
|
|
defer b.refreshMutex.Unlock()
|
|
b.earliestRefresh = time.Now().Add(minimumRefreshInterval)
|
|
b.refreshInProgress = false
|
|
}
|
|
|
|
type channels struct {
|
|
log *logrus.Entry
|
|
sc *slack.Client
|
|
|
|
channelsByID map[string]*slack.Channel
|
|
channelsByName map[string]*slack.Channel
|
|
channelsMutex sync.RWMutex
|
|
|
|
channelMembers map[string][]string
|
|
channelMembersMutex sync.RWMutex
|
|
|
|
refreshInProgress bool
|
|
earliestRefresh time.Time
|
|
refreshMutex sync.Mutex
|
|
}
|
|
|
|
func newChannelManager(log *logrus.Entry, sc *slack.Client) *channels {
|
|
return &channels{
|
|
log: log,
|
|
sc: sc,
|
|
channelsByID: make(map[string]*slack.Channel),
|
|
channelsByName: make(map[string]*slack.Channel),
|
|
earliestRefresh: time.Now(),
|
|
}
|
|
}
|
|
|
|
func (b *channels) getChannel(channel string) (*slack.Channel, error) {
|
|
if strings.HasPrefix(channel, "ID:") {
|
|
return b.getChannelByID(strings.TrimPrefix(channel, "ID:"))
|
|
}
|
|
return b.getChannelByName(channel)
|
|
}
|
|
|
|
func (b *channels) getChannelByName(name string) (*slack.Channel, error) {
|
|
return b.getChannelBy(name, b.channelsByName)
|
|
}
|
|
|
|
func (b *channels) getChannelByID(id string) (*slack.Channel, error) {
|
|
return b.getChannelBy(id, b.channelsByID)
|
|
}
|
|
|
|
func (b *channels) getChannelBy(lookupKey string, lookupMap map[string]*slack.Channel) (*slack.Channel, error) {
|
|
b.channelsMutex.RLock()
|
|
defer b.channelsMutex.RUnlock()
|
|
|
|
if channel, ok := lookupMap[lookupKey]; ok {
|
|
return channel, nil
|
|
}
|
|
return nil, fmt.Errorf("channel %s not found", lookupKey)
|
|
}
|
|
|
|
func (b *channels) getChannelMembers(users *users) config.ChannelMembers {
|
|
b.channelMembersMutex.RLock()
|
|
defer b.channelMembersMutex.RUnlock()
|
|
|
|
membersInfo := config.ChannelMembers{}
|
|
for channelID, members := range b.channelMembers {
|
|
for _, member := range members {
|
|
channelName := ""
|
|
userName := ""
|
|
userNick := ""
|
|
user := users.getUser(member)
|
|
if user != nil {
|
|
userName = user.Name
|
|
userNick = user.Profile.DisplayName
|
|
}
|
|
channel, _ := b.getChannelByID(channelID)
|
|
if channel != nil {
|
|
channelName = channel.Name
|
|
}
|
|
memberInfo := config.ChannelMember{
|
|
Username: userName,
|
|
Nick: userNick,
|
|
UserID: member,
|
|
ChannelID: channelID,
|
|
ChannelName: channelName,
|
|
}
|
|
membersInfo = append(membersInfo, memberInfo)
|
|
}
|
|
}
|
|
return membersInfo
|
|
}
|
|
|
|
func (b *channels) registerChannel(channel slack.Channel) {
|
|
b.channelsMutex.Lock()
|
|
defer b.channelsMutex.Unlock()
|
|
|
|
b.channelsByID[channel.ID] = &channel
|
|
b.channelsByName[channel.Name] = &channel
|
|
}
|
|
|
|
func (b *channels) populateChannels(wait bool) {
|
|
b.refreshMutex.Lock()
|
|
if !wait && (time.Now().Before(b.earliestRefresh) || b.refreshInProgress) {
|
|
b.log.Debugf("Not refreshing channel list as it was done less than %v seconds ago.", minimumRefreshInterval)
|
|
b.refreshMutex.Unlock()
|
|
return
|
|
}
|
|
for b.refreshInProgress {
|
|
b.refreshMutex.Unlock()
|
|
time.Sleep(time.Second)
|
|
b.refreshMutex.Lock()
|
|
}
|
|
b.refreshInProgress = true
|
|
b.refreshMutex.Unlock()
|
|
|
|
newChannelsByID := map[string]*slack.Channel{}
|
|
newChannelsByName := map[string]*slack.Channel{}
|
|
newChannelMembers := make(map[string][]string)
|
|
|
|
// We only retrieve public and private channels, not IMs
|
|
// and MPIMs as those do not have a channel name.
|
|
queryParams := &slack.GetConversationsParameters{
|
|
ExcludeArchived: true,
|
|
Types: []string{"public_channel,private_channel"},
|
|
}
|
|
for {
|
|
channels, nextCursor, err := b.sc.GetConversations(queryParams)
|
|
if err != nil {
|
|
if err = handleRateLimit(b.log, err); err != nil {
|
|
b.log.Errorf("Could not retrieve channels: %#v", err)
|
|
return
|
|
}
|
|
continue
|
|
}
|
|
|
|
for i := range channels {
|
|
newChannelsByID[channels[i].ID] = &channels[i]
|
|
newChannelsByName[channels[i].Name] = &channels[i]
|
|
// also find all the members in every channel
|
|
// comment for now, issues on big slacks
|
|
/*
|
|
members, err := b.getUsersInConversation(channels[i].ID)
|
|
if err != nil {
|
|
if err = b.handleRateLimit(err); err != nil {
|
|
b.Log.Errorf("Could not retrieve channel members: %#v", err)
|
|
return
|
|
}
|
|
continue
|
|
}
|
|
newChannelMembers[channels[i].ID] = members
|
|
*/
|
|
}
|
|
|
|
if nextCursor == "" {
|
|
break
|
|
}
|
|
queryParams.Cursor = nextCursor
|
|
}
|
|
|
|
b.channelsMutex.Lock()
|
|
defer b.channelsMutex.Unlock()
|
|
b.channelsByID = newChannelsByID
|
|
b.channelsByName = newChannelsByName
|
|
|
|
b.channelMembersMutex.Lock()
|
|
defer b.channelMembersMutex.Unlock()
|
|
b.channelMembers = newChannelMembers
|
|
|
|
b.refreshMutex.Lock()
|
|
defer b.refreshMutex.Unlock()
|
|
b.earliestRefresh = time.Now().Add(minimumRefreshInterval)
|
|
b.refreshInProgress = false
|
|
}
|