Bridge own read receipts from WhatsApp mobile with double puppeting

This commit is contained in:
Tulir Asokan 2020-07-30 18:08:26 +03:00
parent 51f0be415b
commit 25a99c4464
5 changed files with 48 additions and 9 deletions

View file

@ -218,7 +218,7 @@ func (puppet *Puppet) handleReceiptEvent(portal *Portal, event *event.Event) {
if message == nil { if message == nil {
continue continue
} }
puppet.customUser.log.Infofln("Marking %s/%s in %s/%s as read", message.JID, message.MXID, portal.Key.JID, portal.MXID) puppet.customUser.log.Debugfln("Marking %s/%s in %s/%s as read", message.JID, message.MXID, portal.Key.JID, portal.MXID)
_, err := puppet.customUser.Conn.Read(portal.Key.JID, message.JID) _, err := puppet.customUser.Conn.Read(portal.Key.JID, message.JID)
if err != nil { if err != nil {
puppet.customUser.log.Warnln("Error marking read:", err) puppet.customUser.log.Warnln("Error marking read:", err)
@ -238,10 +238,10 @@ func (puppet *Puppet) handleTypingEvent(portal *Portal, evt *event.Event) {
puppet.customTypingIn[evt.RoomID] = isTyping puppet.customTypingIn[evt.RoomID] = isTyping
presence := whatsapp.PresenceComposing presence := whatsapp.PresenceComposing
if !isTyping { if !isTyping {
puppet.customUser.log.Infofln("Marking not typing in %s/%s", portal.Key.JID, portal.MXID) puppet.customUser.log.Debugfln("Marking not typing in %s/%s", portal.Key.JID, portal.MXID)
presence = whatsapp.PresencePaused presence = whatsapp.PresencePaused
} else { } else {
puppet.customUser.log.Infofln("Marking typing in %s/%s", portal.Key.JID, portal.MXID) puppet.customUser.log.Debugfln("Marking typing in %s/%s", portal.Key.JID, portal.MXID)
} }
_, err := puppet.customUser.Conn.Presence(portal.Key.JID, presence) _, err := puppet.customUser.Conn.Presence(portal.Key.JID, presence)
if err != nil { if err != nil {

2
go.mod
View file

@ -19,4 +19,4 @@ require (
maunium.net/go/mautrix v0.6.1 maunium.net/go/mautrix v0.6.1
) )
replace github.com/Rhymen/go-whatsapp => github.com/tulir/go-whatsapp v0.3.4 replace github.com/Rhymen/go-whatsapp => github.com/tulir/go-whatsapp v0.3.5

2
go.sum
View file

@ -117,6 +117,8 @@ github.com/tulir/go-whatsapp v0.3.3 h1:R/SRdgjG1rdmegxx1CE2KmVBKzI8xvC9EE+NNApb4
github.com/tulir/go-whatsapp v0.3.3/go.mod h1:7yGOBdWidM6gsmbAFwgkwHEIhzVrm01+6UbImpMWfTM= github.com/tulir/go-whatsapp v0.3.3/go.mod h1:7yGOBdWidM6gsmbAFwgkwHEIhzVrm01+6UbImpMWfTM=
github.com/tulir/go-whatsapp v0.3.4 h1:MnfKMj8QOZpZ0SBRXOzhTmoMRF+KqsSDLI+R/glw3rs= github.com/tulir/go-whatsapp v0.3.4 h1:MnfKMj8QOZpZ0SBRXOzhTmoMRF+KqsSDLI+R/glw3rs=
github.com/tulir/go-whatsapp v0.3.4/go.mod h1:7yGOBdWidM6gsmbAFwgkwHEIhzVrm01+6UbImpMWfTM= github.com/tulir/go-whatsapp v0.3.4/go.mod h1:7yGOBdWidM6gsmbAFwgkwHEIhzVrm01+6UbImpMWfTM=
github.com/tulir/go-whatsapp v0.3.5 h1:cFw8MWhoLTqR0h2kSkSvz866rggRIAx4X2l8I65gARk=
github.com/tulir/go-whatsapp v0.3.5/go.mod h1:7yGOBdWidM6gsmbAFwgkwHEIhzVrm01+6UbImpMWfTM=
golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20200604202706-70a84ac30bf9 h1:vEg9joUBmeBcK9iSJftGNf3coIG4HqZElCPehJsfAYM= golang.org/x/crypto v0.0.0-20200604202706-70a84ac30bf9 h1:vEg9joUBmeBcK9iSJftGNf3coIG4HqZElCPehJsfAYM=

View file

@ -189,7 +189,7 @@ func (mh *MetricsHandler) Start() {
go mh.startUpdatingStats() go mh.startUpdatingStats()
err := mh.server.ListenAndServe() err := mh.server.ListenAndServe()
mh.running = false mh.running = false
if err != nil { if err != nil && err != http.ErrServerClosed {
mh.log.Fatalln("Error in metrics listener:", err) mh.log.Fatalln("Error in metrics listener:", err)
} }
} }

41
user.go
View file

@ -31,6 +31,7 @@ import (
"maunium.net/go/mautrix" "maunium.net/go/mautrix"
"github.com/Rhymen/go-whatsapp" "github.com/Rhymen/go-whatsapp"
waBinary "github.com/Rhymen/go-whatsapp/binary"
waProto "github.com/Rhymen/go-whatsapp/binary/proto" waProto "github.com/Rhymen/go-whatsapp/binary/proto"
"maunium.net/go/mautrix/event" "maunium.net/go/mautrix/event"
@ -251,8 +252,13 @@ func (user *User) RestoreSession() bool {
return true return true
} else if err != nil { } else if err != nil {
user.log.Errorln("Failed to restore session:", err) user.log.Errorln("Failed to restore session:", err)
if errors.Is(err, whatsapp.ErrUnpaired) {
user.sendMarkdownBridgeAlert("\u26a0 Failed to connect to WhatsApp: unpaired from phone. " +
"To re-pair your phone, use `delete-session` and then `login`.")
} else {
user.sendMarkdownBridgeAlert("\u26a0 Failed to connect to WhatsApp. Make sure WhatsApp " + user.sendMarkdownBridgeAlert("\u26a0 Failed to connect to WhatsApp. Make sure WhatsApp " +
"on your phone is reachable and use `reconnect` to try connecting again.") "on your phone is reachable and use `reconnect` to try connecting again.")
}
user.log.Debugln("Disconnecting due to failed session restore...") user.log.Debugln("Disconnecting due to failed session restore...")
_, err := user.Conn.Disconnect() _, err := user.Conn.Disconnect()
if err != nil { if err != nil {
@ -850,8 +856,8 @@ func (user *User) HandleMsgInfo(info whatsappExt.MsgInfo) {
go func() { go func() {
intent := user.bridge.GetPuppetByJID(info.SenderJID).IntentFor(portal) intent := user.bridge.GetPuppetByJID(info.SenderJID).IntentFor(portal)
for _, id := range info.IDs { for _, msgID := range info.IDs {
msg := user.bridge.DB.Message.GetByJID(portal.Key, id) msg := user.bridge.DB.Message.GetByJID(portal.Key, msgID)
if msg == nil { if msg == nil {
continue continue
} }
@ -865,6 +871,33 @@ func (user *User) HandleMsgInfo(info whatsappExt.MsgInfo) {
} }
} }
func (user *User) HandleReadMessage(read whatsapp.ReadMessage) {
if strings.HasSuffix(read.Jid, whatsappExt.OldUserSuffix) {
read.Jid = strings.Replace(read.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(read.Jid)
if portal == nil {
return
}
message := user.bridge.DB.Message.GetLastInChat(portal.Key)
if message == nil {
return
}
user.log.Debugfln("User read %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", read.Jid, err)
}
}
func (user *User) HandleCommand(cmd whatsappExt.Command) { func (user *User) HandleCommand(cmd whatsappExt.Command) {
switch cmd.Type { switch cmd.Type {
case whatsappExt.CommandPicture: case whatsappExt.CommandPicture:
@ -946,6 +979,10 @@ func (user *User) HandleRawMessage(message *waProto.WebMessageInfo) {
user.updateLastConnectionIfNecessary() user.updateLastConnectionIfNecessary()
} }
func (user *User) HandleUnknownBinaryNode(node *waBinary.Node) {
user.log.Debugfln("Unknown binary message: %+v", node)
}
func (user *User) NeedsRelaybot(portal *Portal) bool { func (user *User) NeedsRelaybot(portal *Portal) bool {
return !user.HasSession() || !user.IsInPortal(portal.Key) return !user.HasSession() || !user.IsInPortal(portal.Key)
} }