5
0
mirror of https://github.com/cwinfo/matterbridge.git synced 2024-12-26 20:15:39 +00:00
matterbridge/vendor/go.mau.fi/whatsmeow/privacysettings.go
2022-03-20 14:57:48 +01:00

94 lines
2.9 KiB
Go

// Copyright (c) 2021 Tulir Asokan
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package whatsmeow
import (
waBinary "go.mau.fi/whatsmeow/binary"
"go.mau.fi/whatsmeow/types"
"go.mau.fi/whatsmeow/types/events"
)
// TryFetchPrivacySettings will fetch the user's privacy settings, either from the in-memory cache or from the server.
func (cli *Client) TryFetchPrivacySettings(ignoreCache bool) (*types.PrivacySettings, error) {
if val := cli.privacySettingsCache.Load(); val != nil && !ignoreCache {
return val.(*types.PrivacySettings), nil
}
resp, err := cli.sendIQ(infoQuery{
Namespace: "privacy",
Type: iqGet,
To: types.ServerJID,
Content: []waBinary.Node{{Tag: "privacy"}},
})
if err != nil {
return nil, err
}
privacyNode, ok := resp.GetOptionalChildByTag("privacy")
if !ok {
return nil, &ElementMissingError{Tag: "privacy", In: "response to privacy settings query"}
}
var settings types.PrivacySettings
cli.parsePrivacySettings(&privacyNode, &settings)
cli.privacySettingsCache.Store(&settings)
return &settings, nil
}
// GetPrivacySettings will get the user's privacy settings. If an error occurs while fetching them, the error will be
// logged, but the method will just return an empty struct.
func (cli *Client) GetPrivacySettings() (settings types.PrivacySettings) {
settingsPtr, err := cli.TryFetchPrivacySettings(false)
if err != nil {
cli.Log.Errorf("Failed to fetch privacy settings: %v", err)
} else {
settings = *settingsPtr
}
return
}
func (cli *Client) parsePrivacySettings(privacyNode *waBinary.Node, settings *types.PrivacySettings) *events.PrivacySettings {
var evt events.PrivacySettings
for _, child := range privacyNode.GetChildren() {
if child.Tag != "category" {
continue
}
ag := child.AttrGetter()
name := ag.String("name")
value := types.PrivacySetting(ag.String("value"))
switch name {
case "groupadd":
settings.GroupAdd = value
evt.GroupAddChanged = true
case "last":
settings.LastSeen = value
evt.LastSeenChanged = true
case "status":
settings.Status = value
evt.StatusChanged = true
case "profile":
settings.Profile = value
evt.ProfileChanged = true
case "readreceipts":
settings.ReadReceipts = value
evt.ReadReceiptsChanged = true
}
}
return &evt
}
func (cli *Client) handlePrivacySettingsNotification(privacyNode *waBinary.Node) {
cli.Log.Debugf("Parsing privacy settings change notification")
settings, err := cli.TryFetchPrivacySettings(false)
if err != nil {
cli.Log.Errorf("Failed to fetch privacy settings when handling change: %v", err)
}
evt := cli.parsePrivacySettings(privacyNode, settings)
// The data isn't be reliable if the fetch failed, so only cache if it didn't fail
if err == nil {
cli.privacySettingsCache.Store(settings)
}
cli.dispatchEvent(evt)
}