Sumner Evans
103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
package connector
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
"slices"
|
|
|
|
up "go.mau.fi/util/configupgrade"
|
|
"maunium.net/go/mautrix/bridgev2"
|
|
"maunium.net/go/mautrix/bridgev2/database"
|
|
"maunium.net/go/mautrix/id"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/connector/media"
|
|
)
|
|
|
|
var _ bridgev2.ConfigValidatingNetwork = (*TelegramConnector)(nil)
|
|
|
|
type MemberListConfig struct {
|
|
MaxInitialSync int `yaml:"max_initial_sync"`
|
|
SyncBroadcastChannels bool `yaml:"sync_broadcast_channels"`
|
|
SkipDeleted bool `yaml:"skip_deleted"`
|
|
}
|
|
|
|
func (c MemberListConfig) NormalizedMaxInitialSync() int {
|
|
if c.MaxInitialSync < 0 {
|
|
return 10000
|
|
}
|
|
return c.MaxInitialSync
|
|
}
|
|
|
|
type TelegramConfig struct {
|
|
AppID int `yaml:"app_id"`
|
|
AppHash string `yaml:"app_hash"`
|
|
|
|
AnimatedSticker media.AnimatedStickerConfig `yaml:"animated_sticker"`
|
|
|
|
MemberList MemberListConfig `yaml:"member_list"`
|
|
|
|
MaxMemberCount int `yaml:"max_member_count"`
|
|
}
|
|
|
|
func (c TelegramConfig) ShouldBridge(participantCount int) bool {
|
|
return c.MaxMemberCount < 0 || participantCount <= c.MaxMemberCount
|
|
}
|
|
|
|
//go:embed example-config.yaml
|
|
var ExampleConfig string
|
|
|
|
func upgradeConfig(helper up.Helper) {
|
|
helper.Copy(up.Int, "app_id")
|
|
helper.Copy(up.Str, "app_hash")
|
|
helper.Copy(up.Str, "animated_sticker", "target")
|
|
helper.Copy(up.Bool, "animated_sticker", "convert_from_webm")
|
|
helper.Copy(up.Int, "animated_sticker", "args", "width")
|
|
helper.Copy(up.Int, "animated_sticker", "args", "height")
|
|
helper.Copy(up.Int, "animated_sticker", "args", "fps")
|
|
helper.Copy(up.Int, "member_list", "max_initial_sync")
|
|
helper.Copy(up.Bool, "member_list", "sync_broadcast_channels")
|
|
helper.Copy(up.Bool, "member_list", "skip_deleted")
|
|
helper.Copy(up.Int, "max_member_count")
|
|
}
|
|
|
|
func (tg *TelegramConnector) GetConfig() (example string, data any, upgrader up.Upgrader) {
|
|
return ExampleConfig, tg.Config, up.SimpleUpgrader(upgradeConfig)
|
|
}
|
|
|
|
func (tg *TelegramConnector) ValidateConfig() error {
|
|
if tg.Config.AppID == 0 {
|
|
return fmt.Errorf("app_id is required")
|
|
}
|
|
if tg.Config.AppHash == "" {
|
|
return fmt.Errorf("app_hash is required")
|
|
}
|
|
if !slices.Contains([]string{"disable", "gif", "png", "webp", "webm"}, tg.Config.AnimatedSticker.Target) {
|
|
return fmt.Errorf("unsupported animated sticker target: %s", tg.Config.AnimatedSticker.Target)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (tg *TelegramConnector) GetDBMetaTypes() database.MetaTypes {
|
|
return database.MetaTypes{
|
|
Ghost: func() any { return &GhostMetadata{} },
|
|
Portal: nil,
|
|
Message: func() any { return &MessageMetadata{} },
|
|
Reaction: nil,
|
|
UserLogin: func() any { return &UserLoginMetadata{} },
|
|
}
|
|
}
|
|
|
|
type GhostMetadata struct {
|
|
IsPremium bool `json:"is_premium"`
|
|
AccessHash int64 `json:"access_hash"`
|
|
}
|
|
|
|
type MessageMetadata struct {
|
|
ContentHash []byte `json:"content_hash,omitempty"`
|
|
ContentURI id.ContentURIString `json:"content_uri,omitempty"`
|
|
}
|
|
|
|
type UserLoginMetadata struct {
|
|
Phone string `json:"phone"`
|
|
}
|