// mautrix-groupme - A Matrix-GroupMe puppeting bridge. // Copyright (C) 2022 Sumner Evans, Karmanyaah Malhotra // // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as published by // the Free Software Foundation, either version 3 of the License, or // (at your option) any later version. // // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU Affero General Public License for more details. // // You should have received a copy of the GNU Affero General Public License // along with this program. If not, see . package main import ( "context" "errors" "fmt" "net/http" "strings" "sync" "time" log "maunium.net/go/maulogger/v2" "github.com/Rhymen/go-whatsapp" "maunium.net/go/mautrix" "maunium.net/go/mautrix/appservice" "maunium.net/go/mautrix/bridge" "maunium.net/go/mautrix/bridge/bridgeconfig" "maunium.net/go/mautrix/event" "maunium.net/go/mautrix/format" "maunium.net/go/mautrix/id" "github.com/beeper/groupme-lib" "github.com/beeper/groupme/database" "github.com/beeper/groupme/groupmeext" ) type User struct { *database.User Conn *groupme.PushSubscription bridge *GMBridge log log.Logger Admin bool Whitelisted bool PermissionLevel bridgeconfig.PermissionLevel BridgeState *bridge.BridgeStateQueue Client *groupmeext.Client ConnectionErrors int CommunityID string ChatList map[groupme.ID]groupme.Chat GroupList map[groupme.ID]groupme.Group RelationList map[groupme.ID]groupme.User cleanDisconnection bool batteryWarningsSent int lastReconnection int64 chatListReceived chan struct{} syncPortalsDone chan struct{} messageInput chan PortalMessage messageOutput chan PortalMessage mgmtCreateLock sync.Mutex spaceCreateLock sync.Mutex spaceMembershipChecked bool } func (br *GMBridge) getUserByMXID(userID id.UserID, onlyIfExists bool) *User { _, isPuppet := br.ParsePuppetMXID(userID) if isPuppet || userID == br.Bot.UserID { return nil } br.usersLock.Lock() defer br.usersLock.Unlock() user, ok := br.usersByMXID[userID] if !ok { userIDPtr := &userID if onlyIfExists { userIDPtr = nil } return br.loadDBUser(br.DB.User.GetByMXID(userID), userIDPtr) } return user } func (br *GMBridge) GetUserByMXID(userID id.UserID) *User { return br.getUserByMXID(userID, false) } func (br *GMBridge) GetIUser(userID id.UserID, create bool) bridge.User { u := br.getUserByMXID(userID, !create) if u == nil { return nil } return u } func (user *User) GetPermissionLevel() bridgeconfig.PermissionLevel { return user.PermissionLevel } func (user *User) GetManagementRoomID() id.RoomID { return user.ManagementRoom } func (user *User) GetMXID() id.UserID { return user.MXID } func (user *User) GetCommandState() map[string]interface{} { return nil } func (br *GMBridge) GetUserByMXIDIfExists(userID id.UserID) *User { return br.getUserByMXID(userID, true) } func (bridge *GMBridge) GetUserByGMID(gmid groupme.ID) *User { bridge.usersLock.Lock() defer bridge.usersLock.Unlock() user, ok := bridge.usersByGMID[gmid] if !ok { return bridge.loadDBUser(bridge.DB.User.GetByGMID(gmid), nil) } return user } func (user *User) addToGMIDMap() { user.bridge.usersLock.Lock() user.bridge.usersByGMID[user.GMID] = user user.bridge.usersLock.Unlock() } func (user *User) removeFromGMIDMap() { user.bridge.usersLock.Lock() jidUser, ok := user.bridge.usersByGMID[user.GMID] if ok && user == jidUser { delete(user.bridge.usersByGMID, user.GMID) } user.bridge.usersLock.Unlock() user.bridge.Metrics.TrackLoginState(user.GMID, false) } func (br *GMBridge) GetAllUsers() []*User { br.usersLock.Lock() defer br.usersLock.Unlock() dbUsers := br.DB.User.GetAll() output := make([]*User, len(dbUsers)) for index, dbUser := range dbUsers { user, ok := br.usersByMXID[dbUser.MXID] if !ok { user = br.loadDBUser(dbUser, nil) } output[index] = user } return output } func (br *GMBridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User { if dbUser == nil { if mxid == nil { return nil } dbUser = br.DB.User.New() dbUser.MXID = *mxid dbUser.Insert() } user := br.NewUser(dbUser) br.usersByMXID[user.MXID] = user if len(user.GMID) > 0 { br.usersByGMID[user.GMID] = user } if len(user.ManagementRoom) > 0 { br.managementRooms[user.ManagementRoom] = user } return user } func (br *GMBridge) NewUser(dbUser *database.User) *User { user := &User{ User: dbUser, bridge: br, log: br.Log.Sub("User").Sub(string(dbUser.MXID)), chatListReceived: make(chan struct{}, 1), syncPortalsDone: make(chan struct{}, 1), messageInput: make(chan PortalMessage), messageOutput: make(chan PortalMessage, br.Config.Bridge.PortalMessageBuffer), } user.PermissionLevel = user.bridge.Config.Bridge.Permissions.Get(user.MXID) user.Whitelisted = user.PermissionLevel >= bridgeconfig.PermissionLevelUser user.Admin = user.PermissionLevel >= bridgeconfig.PermissionLevelAdmin user.BridgeState = br.NewBridgeStateQueue(user, user.log) go user.handleMessageLoop() go user.runMessageRingBuffer() return user } func (user *User) ensureInvited(intent *appservice.IntentAPI, roomID id.RoomID, isDirect bool) (ok bool) { extraContent := make(map[string]interface{}) if isDirect { extraContent["is_direct"] = true } customPuppet := user.bridge.GetPuppetByCustomMXID(user.MXID) if customPuppet != nil && customPuppet.CustomIntent() != nil { extraContent["fi.mau.will_auto_accept"] = true } _, err := intent.InviteUser(roomID, &mautrix.ReqInviteUser{UserID: user.MXID}, extraContent) var httpErr mautrix.HTTPError if err != nil && errors.As(err, &httpErr) && httpErr.RespError != nil && strings.Contains(httpErr.RespError.Err, "is already in the room") { user.bridge.StateStore.SetMembership(roomID, user.MXID, event.MembershipJoin) ok = true return } else if err != nil { user.log.Warnfln("Failed to invite user to %s: %v", roomID, err) } else { ok = true } if customPuppet != nil && customPuppet.CustomIntent() != nil { err = customPuppet.CustomIntent().EnsureJoined(roomID, appservice.EnsureJoinedParams{IgnoreCache: true}) if err != nil { user.log.Warnfln("Failed to auto-join %s: %v", roomID, err) ok = false } else { ok = true } } return } func (user *User) GetSpaceRoom() id.RoomID { if !user.bridge.Config.Bridge.PersonalFilteringSpaces { return "" } if len(user.SpaceRoom) == 0 { user.spaceCreateLock.Lock() defer user.spaceCreateLock.Unlock() if len(user.SpaceRoom) > 0 { return user.SpaceRoom } resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{ Visibility: "private", Name: "GroupMe", Topic: "Your GroupMe bridged chats", InitialState: []*event.Event{{ Type: event.StateRoomAvatar, Content: event.Content{ Parsed: &event.RoomAvatarEventContent{ URL: user.bridge.Config.AppService.Bot.ParsedAvatar, }, }, }}, CreationContent: map[string]interface{}{ "type": event.RoomTypeSpace, }, PowerLevelOverride: &event.PowerLevelsEventContent{ Users: map[id.UserID]int{ user.bridge.Bot.UserID: 9001, user.MXID: 50, }, }, }) if err != nil { user.log.Errorln("Failed to auto-create space room:", err) } else { user.SpaceRoom = resp.RoomID user.Update() user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false) } } else if !user.spaceMembershipChecked && !user.bridge.StateStore.IsInRoom(user.SpaceRoom, user.MXID) { user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false) } user.spaceMembershipChecked = true return user.SpaceRoom } func (user *User) GetManagementRoom() id.RoomID { if len(user.ManagementRoom) == 0 { user.mgmtCreateLock.Lock() defer user.mgmtCreateLock.Unlock() if len(user.ManagementRoom) > 0 { return user.ManagementRoom } creationContent := make(map[string]interface{}) if !user.bridge.Config.Bridge.FederateRooms { creationContent["m.federate"] = false } resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{ Topic: "GroupMe bridge notices", IsDirect: true, CreationContent: creationContent, }) if err != nil { user.log.Errorln("Failed to auto-create management room:", err) } else { user.SetManagementRoom(resp.RoomID) } } return user.ManagementRoom } func (user *User) SetManagementRoom(roomID id.RoomID) { existingUser, ok := user.bridge.managementRooms[roomID] if ok { existingUser.ManagementRoom = "" existingUser.Update() } user.ManagementRoom = roomID user.bridge.managementRooms[user.ManagementRoom] = user user.Update() } func (user *User) SetSession(session *whatsapp.Session) { // user.Session = session // if session == nil { // user.LastConnection = 0 // } // user.Update() } func (user *User) Connect() bool { if user.Conn != nil { return true } else if len(user.Token) == 0 { return false } user.log.Debugln("Connecting to GroupMe") timeout := time.Duration(user.bridge.Config.GroupMe.ConnectionTimeout) if timeout == 0 { timeout = 20 } conn := groupme.NewPushSubscription(context.Background()) user.Conn = &conn user.Conn.StartListening(context.TODO(), groupmeext.NewFayeClient(user.log)) user.Conn.AddFullHandler(user) //TODO: typing notification? return user.RestoreSession() } func (user *User) RestoreSession() bool { if len(user.Token) > 0 { err := user.Conn.SubscribeToUser(context.TODO(), groupme.ID(user.GMID), user.Token) if err != nil { fmt.Println(err) } //TODO: typing notifics user.ConnectionErrors = 0 //user.SetSession(&sess) user.log.Debugln("Session restored successfully") user.PostLogin() return true } else { user.log.Debugln("tried login but no token") return false } } func (user *User) HasSession() bool { return len(user.Token) > 0 } func (user *User) IsConnected() bool { // TODO: better connection check return user.Conn != nil } func (user *User) IsLoggedIn() bool { return true } func (user *User) IsLoginInProgress() bool { // return user.Conn != nil && user.Conn.IsLoginInProgress() return false } func (user *User) GetGMID() groupme.ID { if len(user.GMID) == 0 { u, err := user.Client.MyUser(context.TODO()) if err != nil { user.log.Errorln("Failed to get own GroupMe ID:", err) return "" } user.GMID = u.ID } return user.GMID } func (user *User) Login(token string) error { user.Token = token user.addToGMIDMap() user.PostLogin() if user.Connect() { return nil } return errors.New("failed to connect") } type Chat struct { Portal *Portal LastMessageTime uint64 Group *groupme.Group DM *groupme.Chat } type ChatList []Chat func (cl ChatList) Len() int { return len(cl) } func (cl ChatList) Less(i, j int) bool { return cl[i].LastMessageTime > cl[j].LastMessageTime } func (cl ChatList) Swap(i, j int) { cl[i], cl[j] = cl[j], cl[i] } func (user *User) PostLogin() { user.bridge.Metrics.TrackConnectionState(user.GMID, true) user.bridge.Metrics.TrackLoginState(user.GMID, true) user.bridge.Metrics.TrackBufferLength(user.MXID, 0) // go user.intPostLogin() } func (user *User) tryAutomaticDoublePuppeting() { if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) { return } user.log.Debugln("Checking if double puppeting needs to be enabled") puppet := user.bridge.GetPuppetByGMID(user.GMID) if len(puppet.CustomMXID) > 0 { user.log.Debugln("User already has double-puppeting enabled") // Custom puppet already enabled return } accessToken, err := puppet.loginWithSharedSecret(user.MXID) if err != nil { user.log.Warnln("Failed to login with shared secret:", err) return } err = puppet.SwitchCustomMXID(accessToken, user.MXID) if err != nil { puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err) return } user.log.Infoln("Successfully automatically enabled custom puppet") } func (user *User) sendBridgeNotice(formatString string, args ...interface{}) { notice := fmt.Sprintf(formatString, args...) _, err := user.bridge.Bot.SendNotice(user.GetManagementRoom(), notice) if err != nil { user.log.Warnf("Failed to send bridge notice \"%s\": %v", notice, err) } } func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) { notice := fmt.Sprintf(formatString, args...) content := format.RenderMarkdown(notice, true, false) _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content) if err != nil { user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err) } } func (user *User) postConnPing() bool { // user.log.Debugln("Making post-connection ping") // err := user.Conn.AdminTest() // if err != nil { // user.log.Errorfln("Post-connection ping failed: %v. Disconnecting and then reconnecting after a second", err) // sess, disconnectErr := user.Conn.Disconnect() // if disconnectErr != nil { // user.log.Warnln("Error while disconnecting after failed post-connection ping:", disconnectErr) // } else { // user.Session = &sess // } // user.bridge.Metrics.TrackDisconnection(user.MXID) // go func() { // time.Sleep(1 * time.Second) // user.tryReconnect(fmt.Sprintf("Post-connection ping failed: %v", err)) // }() // return false // } else { // user.log.Debugln("Post-connection ping OK") // return true // } return true } // func (user *User) intPostLogin() { // defer user.syncWait.Done() // user.lastReconnection = time.Now().Unix() // user.Client = groupmeext.NewClient(user.Token) // if len(user.JID) == 0 { // myuser, err := user.Client.MyUser(context.TODO()) // if err != nil { // log.Fatal(err) //TODO // } // user.JID = myuser.ID.String() // } // user.Update() // user.tryAutomaticDoublePuppeting() // user.log.Debugln("Waiting for chat list receive confirmation") // user.HandleChatList() // select { // case <-user.chatListReceived: // user.log.Debugln("Chat list receive confirmation received in PostLogin") // case <-time.After(time.Duration(user.bridge.Config.Bridge.ChatListWait) * time.Second): // user.log.Warnln("Timed out waiting for chat list to arrive!") // user.postConnPing() // return // } // if !user.postConnPing() { // user.log.Debugln("Post-connection ping failed, unlocking processing of incoming messages.") // return // } // user.log.Debugln("Waiting for portal sync complete confirmation") // select { // case <-user.syncPortalsDone: // user.log.Debugln("Post-connection portal sync complete, unlocking processing of incoming messages.") // // TODO this is too short, maybe a per-portal duration? // case <-time.After(time.Duration(user.bridge.Config.Bridge.PortalSyncWait) * time.Second): // user.log.Warnln("Timed out waiting for portal sync to complete! Unlocking processing of incoming messages.") // } // } func (user *User) HandleChatList() { chatMap := map[groupme.ID]groupme.Group{} chats, err := user.Client.IndexAllGroups() if err != nil { user.log.Errorln("chat sync error", err) //TODO: handle return } for _, chat := range chats { chatMap[chat.ID] = *chat } user.GroupList = chatMap dmMap := map[groupme.ID]groupme.Chat{} dms, err := user.Client.IndexAllChats() if err != nil { user.log.Errorln("chat sync error", err) //TODO: handle return } for _, dm := range dms { dmMap[dm.OtherUser.ID] = *dm } user.ChatList = dmMap userMap := map[groupme.ID]groupme.User{} users, err := user.Client.IndexAllRelations() if err != nil { user.log.Errorln("Error syncing user list, continuing sync", err) } for _, u := range users { puppet := user.bridge.GetPuppetByGMID(u.ID) // "" for overall user not related to one group puppet.Sync(nil, &groupme.Member{ UserID: u.ID, Nickname: u.Name, ImageURL: u.AvatarURL, }, false, false) userMap[u.ID] = *u } user.RelationList = userMap user.log.Infoln("Chat list received") user.chatListReceived <- struct{}{} go user.syncPortals(false) } func (user *User) syncPortals(createAll bool) { // user.log.Infoln("Reading chat list") // chats := make(ChatList, 0, len(user.GroupList)+len(user.ChatList)) // portalKeys := make([]database.PortalKeyWithMeta, 0, cap(chats)) // for _, group := range user.GroupList { // portal := user.bridge.GetPortalByJID(database.GroupPortalKey(group.ID.String())) // chats = append(chats, Chat{ // Portal: portal, // LastMessageTime: uint64(group.UpdatedAt.ToTime().Unix()), // Group: &group, // }) // } // for _, dm := range user.ChatList { // portal := user.bridge.GetPortalByJID(database.NewPortalKey(dm.OtherUser.ID.String(), user.JID)) // chats = append(chats, Chat{ // Portal: portal, // LastMessageTime: uint64(dm.UpdatedAt.ToTime().Unix()), // DM: &dm, // }) // } // for _, chat := range chats { // var inCommunity, ok bool // if inCommunity, ok = existingKeys[chat.Portal.Key]; !ok || !inCommunity { // inCommunity = user.addPortalToCommunity(chat.Portal) // if chat.Portal.IsPrivateChat() { // puppet := user.bridge.GetPuppetByJID(chat.Portal.Key.GMID) // user.addPuppetToCommunity(puppet) // } // } // portalKeys = append(portalKeys, database.PortalKeyWithMeta{PortalKey: chat.Portal.Key, InCommunity: inCommunity}) // } // user.log.Infoln("Read chat list, updating user-portal mapping") // err := user.SetPortalKeys(portalKeys) // if err != nil { // user.log.Warnln("Failed to update user-portal mapping:", err) // } // sort.Sort(chats) // limit := user.bridge.Config.Bridge.InitialChatSync // if limit < 0 { // limit = len(chats) // } // now := uint64(time.Now().Unix()) // user.log.Infoln("Syncing portals") // wg := sync.WaitGroup{} // for i, chat := range chats { // if chat.LastMessageTime+user.bridge.Config.Bridge.SyncChatMaxAge < now { // break // } // wg.Add(1) // go func(chat Chat, i int) { // create := (chat.LastMessageTime >= user.LastConnection && user.LastConnection > 0) || i < limit // if len(chat.Portal.MXID) > 0 || create || createAll { // chat.Portal.Sync(user, chat.Group) // err := chat.Portal.BackfillHistory(user, chat.LastMessageTime) // if err != nil { // chat.Portal.log.Errorln("Error backfilling history:", err) // } // } // wg.Done() // }(chat, i) // } // wg.Wait() // //TODO: handle leave from groupme side // user.UpdateDirectChats(nil) // user.log.Infoln("Finished syncing portals") // select { // case user.syncPortalsDone <- struct{}{}: // default: // } } func (user *User) getDirectChats() map[id.UserID][]id.RoomID { res := make(map[id.UserID][]id.RoomID) privateChats := user.bridge.DB.Portal.FindPrivateChats(user.GMID) for _, portal := range privateChats { if len(portal.MXID) > 0 { res[user.bridge.FormatPuppetMXID(portal.Key.GMID)] = []id.RoomID{portal.MXID} } } return res } func (user *User) UpdateDirectChats(chats map[id.UserID][]id.RoomID) { if !user.bridge.Config.Bridge.SyncDirectChatList { return } puppet := user.bridge.GetPuppetByCustomMXID(user.MXID) if puppet == nil || puppet.CustomIntent() == nil { return } intent := puppet.CustomIntent() method := http.MethodPatch if chats == nil { chats = user.getDirectChats() method = http.MethodPut } user.log.Debugln("Updating m.direct list on homeserver") var err error if user.bridge.Config.Homeserver.Software == bridgeconfig.SoftwareAsmux { urlPath := intent.BuildClientURL("unstable", "com.beeper.asmux", "dms") _, err = intent.MakeFullRequest(mautrix.FullRequest{ Method: method, URL: urlPath, Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}}, RequestJSON: chats, }) } else { existingChats := make(map[id.UserID][]id.RoomID) err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats) if err != nil { user.log.Warnln("Failed to get m.direct list to update it:", err) return } for userID, rooms := range existingChats { if _, ok := user.bridge.ParsePuppetMXID(userID); !ok { // This is not a ghost user, include it in the new list chats[userID] = rooms } else if _, ok := chats[userID]; !ok && method == http.MethodPatch { // This is a ghost user, but we're not replacing the whole list, so include it too chats[userID] = rooms } } err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats) } if err != nil { user.log.Warnln("Failed to update m.direct list:", err) } } func (user *User) HandleContactList(contacts []whatsapp.Contact) { contactMap := make(map[string]whatsapp.Contact) for _, contact := range contacts { contactMap[contact.Jid] = contact } go user.syncPuppets(contactMap) } func (user *User) syncPuppets(contacts map[string]whatsapp.Contact) { // if contacts == nil { // contacts = user.Conn.Store.Contacts // } // user.log.Infoln("Syncing puppet info from contacts") // for jid, contact := range contacts { // if strings.HasSuffix(jid, whatsappExt.NewUserSuffix) { // puppet := user.bridge.GetPuppetByJID(contact.Jid) // puppet.Sync(user, contact) // } // } // user.log.Infoln("Finished syncing puppet info from contacts") } func (user *User) updateLastConnectionIfNecessary() { // if user.LastConnection+60 < uint64(time.Now().Unix()) { // user.UpdateLastConnection() // } } func (user *User) HandleError(err error) { if !errors.Is(err, whatsapp.ErrInvalidWsData) { user.log.Errorfln("WhatsApp error: %v", err) } if closed, ok := err.(*whatsapp.ErrConnectionClosed); ok { user.bridge.Metrics.TrackDisconnection(user.MXID) if closed.Code == 1000 && user.cleanDisconnection { user.bridge.Metrics.TrackConnectionState(user.GMID, false) user.cleanDisconnection = false user.log.Infoln("Clean disconnection by server") return } go user.tryReconnect(fmt.Sprintf("Your WhatsApp connection was closed with websocket status code %d", closed.Code)) } else if failed, ok := err.(*whatsapp.ErrConnectionFailed); ok { user.bridge.Metrics.TrackDisconnection(user.MXID) user.ConnectionErrors++ go user.tryReconnect(fmt.Sprintf("Your WhatsApp connection failed: %v", failed.Err)) } // Otherwise unknown error, probably mostly harmless } func (user *User) tryReconnect(msg string) { // user.bridge.Metrics.TrackConnectionState(user.JID, false) // if user.ConnectionErrors > user.bridge.Config.Bridge.MaxConnectionAttempts { // user.sendMarkdownBridgeAlert("%s. Use the `reconnect` command to reconnect.", msg) // return // } // if user.bridge.Config.Bridge.ReportConnectionRetry { // user.sendBridgeNotice("%s. Reconnecting...", msg) // // Don't want the same error to be repeated // msg = "" // } // var tries uint // var exponentialBackoff bool // baseDelay := time.Duration(user.bridge.Config.Bridge.ConnectionRetryDelay) // if baseDelay < 0 { // exponentialBackoff = true // baseDelay = -baseDelay + 1 // } // delay := baseDelay // for user.ConnectionErrors <= user.bridge.Config.Bridge.MaxConnectionAttempts { // err := user.Conn.Restore() // if err == nil { // user.ConnectionErrors = 0 // if user.bridge.Config.Bridge.ReportConnectionRetry { // user.sendBridgeNotice("Reconnected successfully") // } // user.PostLogin() // return // } else if err.Error() == "init responded with 400" { // user.log.Infoln("Got init 400 error when trying to reconnect, resetting connection...") // sess, err := user.Conn.Disconnect() // if err != nil { // user.log.Debugln("Error while disconnecting for connection reset:", err) // } // if len(sess.Wid) > 0 { // user.SetSession(&sess) // } // } // user.log.Errorln("Error while trying to reconnect after disconnection:", err) // tries++ // user.ConnectionErrors++ // if user.ConnectionErrors <= user.bridge.Config.Bridge.MaxConnectionAttempts { // if exponentialBackoff { // delay = (1 << tries) + baseDelay // } // if user.bridge.Config.Bridge.ReportConnectionRetry { // user.sendBridgeNotice("Reconnection attempt failed: %v. Retrying in %d seconds...", err, delay) // } // time.Sleep(delay * time.Second) // } // } // if user.bridge.Config.Bridge.ReportConnectionRetry { // user.sendMarkdownBridgeAlert("%d reconnection attempts failed. Use the `reconnect` command to try to reconnect manually.", tries) // } else { // user.sendMarkdownBridgeAlert("\u26a0 %s. Additionally, %d reconnection attempts failed. Use the `reconnect` command to try to reconnect.", msg, tries) // } } func (user *User) ShouldCallSynchronously() bool { return true } func (user *User) HandleJSONParseError(err error) { user.log.Errorln("WhatsApp JSON parse error:", err) } func (user *User) PortalKey(gmid groupme.ID) database.PortalKey { return database.NewPortalKey(gmid, user.GMID) } func (user *User) GetPortalByGMID(gmid groupme.ID) *Portal { return user.bridge.GetPortalByGMID(user.PortalKey(gmid)) } func (user *User) runMessageRingBuffer() { for msg := range user.messageInput { select { case user.messageOutput <- msg: user.bridge.Metrics.TrackBufferLength(user.MXID, len(user.messageOutput)) default: dropped := <-user.messageOutput user.log.Warnln("Buffer is full, dropping message in", dropped.chat) user.messageOutput <- msg } } } func (user *User) handleMessageLoop() { for { select { case msg := <-user.messageOutput: user.bridge.Metrics.TrackBufferLength(user.MXID, len(user.messageOutput)) puppet := user.bridge.GetPuppetByGMID(msg.data.UserID) portal := user.bridge.GetPortalByGMID(msg.chat) if puppet != nil { puppet.Sync(user, &groupme.Member{ UserID: msg.data.UserID, Nickname: msg.data.Name, ImageURL: msg.data.AvatarURL, }, false, false) } portal.messages <- msg } } } //func (user *User) HandleNewContact(contact whatsapp.Contact) { // user.log.Debugfln("Contact message: %+v", contact) // go func() { // if strings.HasSuffix(contact.Jid, whatsappExt.OldUserSuffix) { // contact.Jid = strings.Replace(contact.Jid, whatsappExt.OldUserSuffix, whatsappExt.NewUserSuffix, -1) // } // puppet := user.bridge.GetPuppetByJID(contact.Jid) // puppet.UpdateName(user, contact) // }() //} //func (user *User) HandleBatteryMessage(battery whatsapp.BatteryMessage) { // user.log.Debugfln("Battery message: %+v", battery) // var notice string // if !battery.Plugged && battery.Percentage < 15 && user.batteryWarningsSent < 1 { // notice = fmt.Sprintf("Phone battery low (%d %% remaining)", battery.Percentage) // user.batteryWarningsSent = 1 // } else if !battery.Plugged && battery.Percentage < 5 && user.batteryWarningsSent < 2 { // notice = fmt.Sprintf("Phone battery very low (%d %% remaining)", battery.Percentage) // user.batteryWarningsSent = 2 // } else if battery.Percentage > 15 || battery.Plugged { // user.batteryWarningsSent = 0 // } // if notice != "" { // go user.sendBridgeNotice("%s", notice) // } //} func (user *User) HandleTextMessage(message groupme.Message) { id := database.ParsePortalKey(message.GroupID.String()) if id == nil { id = database.ParsePortalKey(message.ConversationID.String()) } if id == nil { user.log.Errorln("Error parsing conversationid/portalkey", message.ConversationID.String(), "ignoring message") return } user.messageInput <- PortalMessage{*id, user, &message, uint64(message.CreatedAt.ToTime().Unix())} } func (user *User) HandleLike(msg groupme.Message) { user.HandleTextMessage(msg) } func (user *User) HandleJoin(id groupme.ID) { user.HandleChatList() //TODO: efficient } func (user *User) HandleGroupName(group groupme.ID, newName string) { //p := user.GetPortalByJID(group.String()) //if p != nil { // p.UpdateName(newName, "", false) // get more info abt actual user TODO //} //bugs atm with above? user.HandleChatList() } func (user *User) HandleGroupTopic(_ groupme.ID, _ string) { user.HandleChatList() } func (user *User) HandleGroupMembership(_ groupme.ID, _ string) { user.HandleChatList() //TODO } func (user *User) HandleGroupAvatar(_ groupme.ID, _ string) { user.HandleChatList() } func (user *User) HandleLikeIcon(_ groupme.ID, _, _ int, _ string) { //TODO } func (user *User) HandleNewNickname(groupID, userID groupme.ID, name string) { puppet := user.bridge.GetPuppetByGMID(userID) if puppet != nil { puppet.UpdateName(groupme.Member{ Nickname: name, UserID: userID, }, false) } } func (user *User) HandleNewAvatarInGroup(groupID, userID groupme.ID, url string) { puppet := user.bridge.GetPuppetByGMID(userID) if puppet != nil { puppet.UpdateAvatar(user, false) } } func (user *User) HandleMembers(_ groupme.ID, _ []groupme.Member, _ bool) { user.HandleChatList() } //func (user *User) HandleImageMessage(message whatsapp.ImageMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleStickerMessage(message whatsapp.StickerMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleVideoMessage(message whatsapp.VideoMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleAudioMessage(message whatsapp.AudioMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleDocumentMessage(message whatsapp.DocumentMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleContactMessage(message whatsapp.ContactMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleLocationMessage(message whatsapp.LocationMessage) { // user.messageInput <- PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp} //} // //func (user *User) HandleMessageRevoke(message whatsappExt.MessageRevocation) { // user.messageInput <- PortalMessage{message.RemoteJid, user, message, 0} //} type FakeMessage struct { Text string ID string Alert bool } //func (user *User) HandleCallInfo(info whatsappExt.CallInfo) { // if info.Data != nil { // return // } // data := FakeMessage{ // ID: info.ID, // } // switch info.Type { // case whatsappExt.CallOffer: // if !user.bridge.Config.Bridge.CallNotices.Start { // return // } // data.Text = "Incoming call" // data.Alert = true // case whatsappExt.CallOfferVideo: // if !user.bridge.Config.Bridge.CallNotices.Start { // return // } // data.Text = "Incoming video call" // data.Alert = true // case whatsappExt.CallTerminate: // if !user.bridge.Config.Bridge.CallNotices.End { // return // } // data.Text = "Call ended" // data.ID += "E" // default: // return // } // portal := user.GetPortalByJID(info.From) // if portal != nil { // portal.messages <- PortalMessage{info.From, user, data, 0} // } //} //func (user *User) HandlePresence(info whatsappExt.Presence) { // puppet := user.bridge.GetPuppetByJID(info.SenderJID) // switch info.Status { // case whatsapp.PresenceUnavailable: // _ = puppet.DefaultIntent().SetPresence("offline") // case whatsapp.PresenceAvailable: // if len(puppet.typingIn) > 0 && puppet.typingAt+15 > time.Now().Unix() { // portal := user.bridge.GetPortalByMXID(puppet.typingIn) // _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0) // puppet.typingIn = "" // puppet.typingAt = 0 // } else { // _ = puppet.DefaultIntent().SetPresence("online") // } // case whatsapp.PresenceComposing: // portal := user.GetPortalByJID(info.JID) // if len(puppet.typingIn) > 0 && puppet.typingAt+15 > time.Now().Unix() { // if puppet.typingIn == portal.MXID { // return // } // _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0) // } // puppet.typingIn = portal.MXID // puppet.typingAt = time.Now().Unix() // _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, 15*1000) // } //} // //func (user *User) HandleMsgInfo(info whatsappExt.MsgInfo) { // if (info.Command == whatsappExt.MsgInfoCommandAck || info.Command == whatsappExt.MsgInfoCommandAcks) && info.Acknowledgement == whatsappExt.AckMessageRead { // portal := user.GetPortalByJID(info.ToJID) // if len(portal.MXID) == 0 { // return // } // // go func() { // intent := user.bridge.GetPuppetByJID(info.SenderJID).IntentFor(portal) // for _, msgID := range info.IDs { // msg := user.bridge.DB.Message.GetByJID(portal.Key, msgID) // if msg == nil { // continue // } // // err := intent.MarkRead(portal.MXID, msg.MXID) // if err != nil { // user.log.Warnln("Failed to mark message %s as read by %s: %v", msg.MXID, info.SenderJID, err) // } // } // }() // } //} //func (user *User) HandleReceivedMessage(received whatsapp.ReceivedMessage) { // if received.Type == "read" { // user.markSelfRead(received.Jid, received.Index) // } else { // user.log.Debugfln("Unknown received message type: %+v", received) // } //} // //func (user *User) HandleReadMessage(read whatsapp.ReadMessage) { // user.log.Debugfln("Received chat read message: %+v", read) // user.markSelfRead(read.Jid, "") //} // //func (user *User) markSelfRead(jid, messageID string) { // if strings.HasSuffix(jid, whatsappExt.OldUserSuffix) { // jid = strings.Replace(jid, whatsappExt.OldUserSuffix, whatsappExt.NewUserSuffix, -1) // } // puppet := user.bridge.GetPuppetByJID(user.JID) // if puppet == nil { // return // } // intent := puppet.CustomIntent() // if intent == nil { // return // } // portal := user.GetPortalByJID(jid) // if portal == nil { // return // } // var message *database.Message // if messageID == "" { // message = user.bridge.DB.Message.GetLastInChat(portal.Key) // if message == nil { // return // } // user.log.Debugfln("User read chat %s/%s in WhatsApp mobile (last known event: %s/%s)", portal.Key.JID, portal.MXID, message.JID, message.MXID) // } else { // message = user.bridge.DB.Message.GetByJID(portal.Key, messageID) // if message == nil { // return // } // user.log.Debugfln("User read message %s/%s in %s/%s in WhatsApp mobile", message.JID, message.MXID, portal.Key.JID, portal.MXID) // } // err := intent.MarkRead(portal.MXID, message.MXID) // if err != nil { // user.log.Warnfln("Failed to bridge own read receipt in %s: %v", jid, err) // } //} // //func (user *User) HandleCommand(cmd whatsappExt.Command) { // switch cmd.Type { // case whatsappExt.CommandPicture: // if strings.HasSuffix(cmd.JID, whatsappExt.NewUserSuffix) { // puppet := user.bridge.GetPuppetByJID(cmd.JID) // go puppet.UpdateAvatar(user, cmd.ProfilePicInfo) // } else { // portal := user.GetPortalByJID(cmd.JID) // go portal.UpdateAvatar(user, cmd.ProfilePicInfo, true) // } // case whatsappExt.CommandDisconnect: // user.cleanDisconnection = true // if cmd.Kind == "replaced" { // go user.sendMarkdownBridgeAlert("\u26a0 Your WhatsApp connection was closed by the server because you opened another WhatsApp Web client.\n\n" + // "Use the `reconnect` command to disconnect the other client and resume bridging.") // } else { // user.log.Warnln("Unknown kind of disconnect:", string(cmd.Raw)) // go user.sendMarkdownBridgeAlert("\u26a0 Your WhatsApp connection was closed by the server (reason code: %s).\n\n"+ // "Use the `reconnect` command to reconnect.", cmd.Kind) // } // } //} // //func (user *User) HandleChatUpdate(cmd whatsappExt.ChatUpdate) { // if cmd.Command != whatsappExt.ChatUpdateCommandAction { // return // } // // portal := user.GetPortalByJID(cmd.JID) // if len(portal.MXID) == 0 { // if cmd.Data.Action == whatsappExt.ChatActionIntroduce || cmd.Data.Action == whatsappExt.ChatActionCreate { // go func() { // err := portal.CreateMatrixRoom(user) // if err != nil { // user.log.Errorln("Failed to create portal room after receiving join event:", err) // } // }() // } // return // } // // switch cmd.Data.Action { // case whatsappExt.ChatActionNameChange: // go portal.UpdateName(cmd.Data.NameChange.Name, cmd.Data.SenderJID, true) // case whatsappExt.ChatActionAddTopic: // go portal.UpdateTopic(cmd.Data.AddTopic.Topic, cmd.Data.SenderJID, true) // case whatsappExt.ChatActionRemoveTopic: // go portal.UpdateTopic("", cmd.Data.SenderJID, true) // case whatsappExt.ChatActionPromote: // go portal.ChangeAdminStatus(cmd.Data.UserChange.JIDs, true) // case whatsappExt.ChatActionDemote: // go portal.ChangeAdminStatus(cmd.Data.UserChange.JIDs, false) // case whatsappExt.ChatActionAnnounce: // go portal.RestrictMessageSending(cmd.Data.Announce) // case whatsappExt.ChatActionRestrict: // go portal.RestrictMetadataChanges(cmd.Data.Restrict) // case whatsappExt.ChatActionRemove: // go portal.HandleWhatsAppKick(cmd.Data.SenderJID, cmd.Data.UserChange.JIDs) // case whatsappExt.ChatActionAdd: // go portal.HandleWhatsAppInvite(cmd.Data.SenderJID, cmd.Data.UserChange.JIDs) // //case whatsappExt.ChatActionIntroduce: // // if cmd.Data.SenderJID != "unknown" { // // go portal.Sync(user, whatsapp.Contact{Jid: portal.Key.JID}) // // } // } //} //func (user *User) HandleJsonMessage(message string) { // var msg json.RawMessage // err := json.Unmarshal([]byte(message), &msg) // if err != nil { // return // } // user.log.Debugln("JSON message:", message) // user.updateLastConnectionIfNecessary() //} //func (user *User) HandleRawMessage(message *waProto.WebMessageInfo) { // user.updateLastConnectionIfNecessary() //} // //func (user *User) HandleUnknownBinaryNode(node *waBinary.Node) { // user.log.Debugfln("Unknown binary message: %+v", node) //}