package telegram import ( "github.com/pkg/errors" "path/filepath" "strconv" "sync" "time" "dev.narayana.im/narayana/telegabber/config" "dev.narayana.im/narayana/telegabber/persistence" "dev.narayana.im/narayana/telegabber/telegram/cache" "github.com/zelenin/go-tdlib/client" "gosrc.io/xmpp" ) var logConstants = map[string]int32{ ":fatal": 0, ":error": 1, ":warn": 2, ":info": 3, ":debug": 4, ":verbose": 5, ":all": 1023, } func stringToLogConstant(c string) int32 { level, ok := logConstants[c] if !ok { level = 0 } return level } // Client stores the metadata for lazily invoked TDlib instance type Client struct { client *client.Client authorizer *clientAuthorizer parameters *client.TdlibParameters options []client.Option me *client.User xmpp *xmpp.Component jid string Session *persistence.Session resources map[string]bool content *config.TelegramContentConfig cache *cache.Cache online bool locks clientLocks } type clientLocks struct { authorizationReady sync.WaitGroup chatMessageLocks map[int64]*sync.Mutex resourcesLock sync.Mutex } // NewClient instantiates a Telegram App func NewClient(conf config.TelegramConfig, jid string, component *xmpp.Component, session *persistence.Session) (*Client, error) { var options []client.Option options = append(options, client.WithLogVerbosity(&client.SetLogVerbosityLevelRequest{ NewVerbosityLevel: stringToLogConstant(conf.Loglevel), })) if conf.Tdlib.Client.CatchTimeout != 0 { options = append(options, client.WithCatchTimeout( time.Duration(conf.Tdlib.Client.CatchTimeout)*time.Second, )) } apiID, err := strconv.ParseInt(conf.Tdlib.Client.APIID, 10, 32) if err != nil { return &Client{}, errors.Wrap(err, "Wrong api_id") } parameters := client.TdlibParameters{ UseTestDc: false, DatabaseDirectory: filepath.Join("./sessions/", jid), FilesDirectory: filepath.Join("./sessions/", jid, "/files/"), UseFileDatabase: true, UseChatInfoDatabase: conf.Tdlib.Client.UseChatInfoDatabase, UseMessageDatabase: true, UseSecretChats: conf.Tdlib.Client.UseSecretChats, ApiId: int32(apiID), ApiHash: conf.Tdlib.Client.APIHash, SystemLanguageCode: "en", DeviceModel: conf.Tdlib.Client.DeviceModel, SystemVersion: "1.0.0", ApplicationVersion: conf.Tdlib.Client.ApplicationVersion, EnableStorageOptimizer: true, IgnoreFileNames: false, } return &Client{ parameters: ¶meters, xmpp: component, jid: jid, Session: session, resources: make(map[string]bool), content: &conf.Content, cache: cache.NewCache(), options: options, locks: clientLocks{ chatMessageLocks: make(map[int64]*sync.Mutex), }, }, nil }