165 lines
4.1 KiB
Go
165 lines
4.1 KiB
Go
// mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
|
|
// Copyright (C) 2019 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 whatsappExt
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/Rhymen/go-whatsapp"
|
|
)
|
|
|
|
const (
|
|
OldUserSuffix = "@c.us"
|
|
NewUserSuffix = "@s.whatsapp.net"
|
|
)
|
|
|
|
type ExtendedConn struct {
|
|
*whatsapp.Conn
|
|
|
|
handlers []whatsapp.Handler
|
|
}
|
|
|
|
func ExtendConn(conn *whatsapp.Conn) *ExtendedConn {
|
|
ext := &ExtendedConn{
|
|
Conn: conn,
|
|
}
|
|
ext.Conn.AddHandler(ext)
|
|
return ext
|
|
}
|
|
|
|
func (ext *ExtendedConn) AddHandler(handler whatsapp.Handler) {
|
|
ext.Conn.AddHandler(handler)
|
|
ext.handlers = append(ext.handlers, handler)
|
|
}
|
|
|
|
func (ext *ExtendedConn) RemoveHandler(handler whatsapp.Handler) bool {
|
|
ext.Conn.RemoveHandler(handler)
|
|
for i, v := range ext.handlers {
|
|
if v == handler {
|
|
ext.handlers = append(ext.handlers[:i], ext.handlers[i+1:]...)
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (ext *ExtendedConn) RemoveHandlers() {
|
|
ext.Conn.RemoveHandlers()
|
|
ext.handlers = make([]whatsapp.Handler, 0)
|
|
}
|
|
|
|
func (ext *ExtendedConn) shouldCallSynchronously(handler whatsapp.Handler) bool {
|
|
sh, ok := handler.(whatsapp.SyncHandler)
|
|
return ok && sh.ShouldCallSynchronously()
|
|
}
|
|
|
|
func (ext *ExtendedConn) ShouldCallSynchronously() bool {
|
|
return true
|
|
}
|
|
|
|
type GroupInfo struct {
|
|
JID string `json:"jid"`
|
|
OwnerJID string `json:"owner"`
|
|
|
|
Name string `json:"subject"`
|
|
NameSetTime int64 `json:"subjectTime"`
|
|
NameSetBy string `json:"subjectOwner"`
|
|
|
|
Announce bool `json:"announce"` // Can only admins send messages?
|
|
|
|
Topic string `json:"desc"`
|
|
TopicID string `json:"descId"`
|
|
TopicSetAt int64 `json:"descTime"`
|
|
TopicSetBy string `json:"descOwner"`
|
|
|
|
GroupCreated int64 `json:"creation"`
|
|
|
|
Status int16 `json:"status"`
|
|
|
|
Participants []struct {
|
|
JID string `json:"id"`
|
|
IsAdmin bool `json:"isAdmin"`
|
|
IsSuperAdmin bool `json:"isSuperAdmin"`
|
|
} `json:"participants"`
|
|
}
|
|
|
|
func (ext *ExtendedConn) GetGroupMetaData(jid string) (*GroupInfo, error) {
|
|
data, err := ext.Conn.GetGroupMetaData(jid)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get group metadata: %v", err)
|
|
}
|
|
content := <-data
|
|
|
|
info := &GroupInfo{}
|
|
err = json.Unmarshal([]byte(content), info)
|
|
if err != nil {
|
|
return info, fmt.Errorf("failed to unmarshal group metadata: %v", err)
|
|
}
|
|
|
|
for index, participant := range info.Participants {
|
|
info.Participants[index].JID = strings.Replace(participant.JID, OldUserSuffix, NewUserSuffix, 1)
|
|
}
|
|
info.NameSetBy = strings.Replace(info.NameSetBy, OldUserSuffix, NewUserSuffix, 1)
|
|
info.TopicSetBy = strings.Replace(info.TopicSetBy, OldUserSuffix, NewUserSuffix, 1)
|
|
|
|
return info, nil
|
|
}
|
|
|
|
type ProfilePicInfo struct {
|
|
URL string `json:"eurl"`
|
|
Tag string `json:"tag"`
|
|
|
|
Status int `json:"status"`
|
|
}
|
|
|
|
func (ppi *ProfilePicInfo) Download() (io.ReadCloser, error) {
|
|
resp, err := http.Get(ppi.URL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return resp.Body, nil
|
|
}
|
|
|
|
func (ppi *ProfilePicInfo) DownloadBytes() ([]byte, error) {
|
|
body, err := ppi.Download()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer body.Close()
|
|
data, err := ioutil.ReadAll(body)
|
|
return data, err
|
|
}
|
|
|
|
func (ext *ExtendedConn) GetProfilePicThumb(jid string) (*ProfilePicInfo, error) {
|
|
data, err := ext.Conn.GetProfilePicThumb(jid)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get avatar: %v", err)
|
|
}
|
|
content := <-data
|
|
info := &ProfilePicInfo{}
|
|
err = json.Unmarshal([]byte(content), info)
|
|
if err != nil {
|
|
return info, fmt.Errorf("failed to unmarshal avatar info: %v", err)
|
|
}
|
|
return info, nil
|
|
}
|