forked from MirrorHub/mautrix-whatsapp
112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
// mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
|
|
// Copyright (C) 2018 Tulir Asokan
|
|
//
|
|
// 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 <https://www.gnu.org/licenses/>.
|
|
|
|
package main
|
|
|
|
import (
|
|
log "maunium.net/go/maulogger"
|
|
"maunium.net/go/mautrix-appservice"
|
|
"maunium.net/go/gomatrix"
|
|
)
|
|
|
|
type MatrixListener struct {
|
|
bridge *Bridge
|
|
as *appservice.AppService
|
|
log log.Logger
|
|
stop chan struct{}
|
|
}
|
|
|
|
func NewMatrixListener(bridge *Bridge) *MatrixListener {
|
|
return &MatrixListener{
|
|
bridge: bridge,
|
|
as: bridge.AppService,
|
|
stop: make(chan struct{}, 1),
|
|
log: bridge.Log.Sub("Matrix Listener"),
|
|
}
|
|
}
|
|
|
|
func (ml *MatrixListener) Start() {
|
|
for {
|
|
select {
|
|
case evt := <-ml.bridge.AppService.Events:
|
|
ml.log.Debugln("Received Matrix event:", evt)
|
|
switch evt.Type {
|
|
case gomatrix.StateMember:
|
|
ml.HandleMembership(evt)
|
|
case gomatrix.EventMessage:
|
|
ml.HandleMessage(evt)
|
|
}
|
|
case <-ml.stop:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (ml *MatrixListener) HandleBotInvite(evt *gomatrix.Event) {
|
|
intent := ml.as.BotIntent()
|
|
|
|
resp, err := intent.JoinRoom(evt.RoomID, "", nil)
|
|
if err != nil {
|
|
ml.log.Debugln("Failed to join room", evt.RoomID, "with invite from", evt.Sender)
|
|
return
|
|
}
|
|
|
|
members, err := intent.JoinedMembers(resp.RoomID)
|
|
if err != nil {
|
|
ml.log.Debugln("Failed to get members in room", resp.RoomID, "after accepting invite from", evt.Sender)
|
|
intent.LeaveRoom(resp.RoomID)
|
|
return
|
|
}
|
|
|
|
if len(members.Joined) < 2 {
|
|
ml.log.Debugln("Leaving empty room", resp.RoomID, "after accepting invite from", evt.Sender)
|
|
intent.LeaveRoom(resp.RoomID)
|
|
return
|
|
}
|
|
for mxid, _ := range members.Joined {
|
|
if mxid == intent.UserID || mxid == evt.Sender {
|
|
continue
|
|
} else if true { // TODO check if mxid is WhatsApp puppet
|
|
|
|
continue
|
|
}
|
|
ml.log.Debugln("Leaving multi-user room", resp.RoomID, "after accepting invite from", evt.Sender)
|
|
intent.SendNotice(resp.RoomID, "This bridge is user-specific, please don't invite me into rooms with other users.")
|
|
intent.LeaveRoom(resp.RoomID)
|
|
return
|
|
}
|
|
|
|
user := ml.bridge.GetUser(evt.Sender)
|
|
user.ManagementRoom = resp.RoomID
|
|
user.Update()
|
|
intent.SendNotice(user.ManagementRoom, "This room has been registered as your bridge management/status room.")
|
|
ml.log.Debugln(resp.RoomID, "registered as a management room with", evt.Sender)
|
|
}
|
|
|
|
func (ml *MatrixListener) HandleMembership(evt *gomatrix.Event) {
|
|
ml.log.Debugln(evt.Content, evt.Content.Membership, evt.GetStateKey())
|
|
if evt.Content.Membership == "invite" && evt.GetStateKey() == ml.as.BotMXID() {
|
|
ml.HandleBotInvite(evt)
|
|
}
|
|
}
|
|
|
|
func (ml *MatrixListener) HandleMessage(evt *gomatrix.Event) {
|
|
|
|
}
|
|
|
|
func (ml *MatrixListener) Stop() {
|
|
ml.stop <- struct{}{}
|
|
}
|