mirror of
https://github.com/tulir/mautrix-whatsapp
synced 2024-11-15 14:32:43 +01:00
100 lines
3.5 KiB
Go
100 lines
3.5 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 main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"maunium.net/go/mautrix"
|
|
appservice "maunium.net/go/mautrix-appservice"
|
|
)
|
|
|
|
func (user *User) inviteToCommunity() {
|
|
url := user.bridge.Bot.BuildURL("groups", user.CommunityID, "admin", "users", "invite", user.MXID)
|
|
reqBody := map[string]interface{}{}
|
|
_, err := user.bridge.Bot.MakeRequest(http.MethodPut, url, &reqBody, nil)
|
|
if err != nil {
|
|
user.log.Warnfln("Failed to invite user to personal filtering community %s: %v", user.CommunityID, err)
|
|
}
|
|
}
|
|
|
|
func (user *User) updateCommunityProfile() {
|
|
url := user.bridge.Bot.BuildURL("groups", user.CommunityID, "profile")
|
|
profileReq := struct {
|
|
Name string `json:"name"`
|
|
AvatarURL string `json:"avatar_url"`
|
|
ShortDescription string `json:"short_description"`
|
|
}{"WhatsApp", user.bridge.Config.AppService.Bot.Avatar, "Your WhatsApp bridged chats"}
|
|
_, err := user.bridge.Bot.MakeRequest(http.MethodPost, url, &profileReq, nil)
|
|
if err != nil {
|
|
user.log.Warnfln("Failed to update metadata of %s: %v", user.CommunityID, err)
|
|
}
|
|
}
|
|
|
|
func (user *User) createCommunity() {
|
|
if !user.bridge.Config.Bridge.EnableCommunities() {
|
|
return
|
|
}
|
|
|
|
localpart, server := appservice.ParseUserID(user.MXID)
|
|
community := user.bridge.Config.Bridge.FormatCommunity(localpart, server)
|
|
user.log.Debugln("Creating personal filtering community", community)
|
|
bot := user.bridge.Bot
|
|
req := struct {
|
|
Localpart string `json:"localpart"`
|
|
}{community}
|
|
resp := struct {
|
|
GroupID string `json:"group_id"`
|
|
}{}
|
|
_, err := bot.MakeRequest(http.MethodPost, bot.BuildURL("create_group"), &req, &resp)
|
|
if err != nil {
|
|
if httpErr, ok := err.(mautrix.HTTPError); ok {
|
|
if httpErr.RespError.Err != "Group already exists" {
|
|
user.log.Warnln("Server responded with error creating personal filtering community:", err)
|
|
return
|
|
} else {
|
|
user.log.Debugln("Personal filtering community", resp.GroupID, "already existed")
|
|
user.CommunityID = fmt.Sprintf("+%s:%s", req.Localpart, user.bridge.Config.Homeserver.Domain)
|
|
}
|
|
} else {
|
|
user.log.Warnln("Unknown error creating personal filtering community:", err)
|
|
return
|
|
}
|
|
} else {
|
|
user.log.Infoln("Created personal filtering community %s", resp.GroupID)
|
|
user.CommunityID = resp.GroupID
|
|
user.inviteToCommunity()
|
|
user.updateCommunityProfile()
|
|
}
|
|
}
|
|
|
|
func (user *User) addPortalToCommunity(portal *Portal) bool {
|
|
if len(user.CommunityID) == 0 || len(portal.MXID) == 0 {
|
|
return false
|
|
}
|
|
bot := user.bridge.Bot
|
|
url := bot.BuildURL("groups", user.CommunityID, "admin", "rooms", portal.MXID)
|
|
reqBody := map[string]interface{}{}
|
|
_, err := bot.MakeRequest(http.MethodPut, url, &reqBody, nil)
|
|
if err != nil {
|
|
user.log.Warnfln("Failed to add %s to %s: %v", portal.MXID, user.CommunityID, err)
|
|
return false
|
|
}
|
|
user.log.Debugln("Added", portal.MXID, "to", user.CommunityID)
|
|
return true
|
|
}
|