mirrors-mautrix-telegram/pkg/connector/metadata.go

// mautrix-telegram - A Matrix-Telegram puppeting bridge.
// Copyright (C) 2025 Sumner Evans
//
// 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 connector

import (
	"context"

	"go.mau.fi/util/jsontime"
	"maunium.net/go/mautrix/bridgev2"
	"maunium.net/go/mautrix/bridgev2/database"
	"maunium.net/go/mautrix/bridgev2/networkid"
	"maunium.net/go/mautrix/id"

	"go.mau.fi/mautrix-telegram/pkg/gotd/crypto"
	"go.mau.fi/mautrix-telegram/pkg/gotd/session"
)

func (tg *TelegramConnector) GetDBMetaTypes() database.MetaTypes {
	return database.MetaTypes{
		Ghost:     func() any { return &GhostMetadata{} },
		Portal:    func() any { return &PortalMetadata{} },
		Message:   func() any { return &MessageMetadata{} },
		Reaction:  nil,
		UserLogin: func() any { return &UserLoginMetadata{} },
	}
}

type GhostMetadata struct {
	IsPremium bool `json:"is_premium,omitempty"`
	IsChannel bool `json:"is_channel,omitempty"`
	Deleted   bool `json:"deleted,omitempty"`
	NotMin    bool `json:"not_min,omitempty"`

	ContactSource   int64 `json:"contact_source,omitempty"`
	SourceIsContact bool  `json:"source_is_contact,omitempty"`
}

func (gm *GhostMetadata) IsMin() bool {
	return !gm.NotMin
}

type PortalMetadata struct {
	IsSuperGroup      bool          `json:"is_supergroup,omitempty"`
	IsForumGeneral    bool          `json:"is_forum_general,omitempty"`
	ReadUpTo          int           `json:"read_up_to,omitempty"` // FIXME this shouldn't be here
	AllowedReactions  []string      `json:"allowed_reactions"`
	LastSync          jsontime.Unix `json:"last_sync,omitempty"`
	FullSynced        bool          `json:"full_synced,omitempty"`
	ParticipantsCount int           `json:"member_count,omitempty"`
}

func (pm *PortalMetadata) SetIsSuperGroup(isSupergroup bool) (changed bool) {
	changed = pm.IsSuperGroup != isSupergroup
	pm.IsSuperGroup = isSupergroup
	return changed
}

func (pm *PortalMetadata) SetIsForumGeneral(isForumGeneral bool) (changed bool) {
	changed = pm.IsForumGeneral != isForumGeneral
	pm.IsForumGeneral = isForumGeneral
	return changed
}

type MessageMetadata struct {
	ContentHash []byte              `json:"content_hash,omitempty"`
	ContentURI  id.ContentURIString `json:"content_uri,omitempty"`
}

type UserLoginMetadata struct {
	LoginPhone  string           `json:"phone,omitempty"`
	LoginMethod string           `json:"login_method,omitempty"`
	IsBot       bool             `json:"is_bot,omitempty"`
	Session     UserLoginSession `json:"session"`
	TakeoutID   int64            `json:"takeout_id,omitempty"`

	DialogSyncComplete bool               `json:"takeout_portal_crawl_done,omitempty"`
	DialogSyncCursor   networkid.PortalID `json:"takeout_portal_crawl_cursor,omitempty"`
	DialogSyncCount    int                `json:"dialog_sync_count,omitempty"`

	PinnedDialogs []networkid.PortalID `json:"pinned_dialogs,omitempty"`

	PushEncryptionKey []byte `json:"push_encryption_key,omitempty"`
}

func (u *UserLoginMetadata) ResetOnLogout() {
	u.Session.AuthKey = nil
	u.TakeoutID = 0
	u.DialogSyncComplete = false
	u.DialogSyncCursor = networkid.PortalID("")
	u.DialogSyncCount = 0
	u.PushEncryptionKey = nil
}

type UserLoginSession struct {
	AuthKey       []byte `json:"auth_key,omitempty"`
	Datacenter    int    `json:"dc_id,omitempty"`
	ServerAddress string `json:"server_address,omitempty"`
	ServerPort    int    `json:"port,omitempty"`
	Salt          int64  `json:"salt,omitempty"`
}

func (u UserLoginSession) HasAuthKey() bool {
	return len(u.AuthKey) == 256
}

func (s *UserLoginSession) Load(_ context.Context) (*session.Data, error) {
	if !s.HasAuthKey() {
		return nil, session.ErrNotFound
	}
	keyID := crypto.Key(s.AuthKey).ID()
	return &session.Data{
		DC:        s.Datacenter,
		Addr:      s.ServerAddress,
		AuthKey:   s.AuthKey,
		AuthKeyID: keyID[:],
		Salt:      s.Salt,
	}, nil
}

func (s *UserLoginSession) Save(ctx context.Context, data *session.Data) error {
	s.Datacenter = data.DC
	s.ServerAddress = data.Addr
	s.AuthKey = data.AuthKey
	s.Salt = data.Salt
	// TODO save UserLogin to database?
	return nil
}

func updatePortalLastSyncAt(_ context.Context, portal *bridgev2.Portal) bool {
	meta := portal.Metadata.(*PortalMetadata)
	meta.LastSync = jsontime.UnixNow()
	return true
}