Telegram authorization

This commit is contained in:
bodqhrohro 2019-11-24 19:10:29 +02:00
parent c0c21a35a4
commit 653b1bde94
8 changed files with 223 additions and 45 deletions

2
go.mod
View file

@ -14,3 +14,5 @@ require (
)
replace gosrc.io/xmpp => github.com/bodqhrohro/go-xmpp v0.1.4-0.20191106203535-f3b463f3b26c
replace github.com/zelenin/go-tdlib => github.com/bodqhrohro/go-tdlib v0.1.2-0.20191121233100-48d2382034fb

9
go.sum
View file

@ -1,6 +1,11 @@
github.com/Arman92/go-tdlib v0.0.0-20191002071913-526f4e1d15f7 h1:GbV1Lv3lVHsSeKAqPTBem72OCsGjXntW4jfJdXciE+w=
github.com/Arman92/go-tdlib v0.0.0-20191002071913-526f4e1d15f7/go.mod h1:ZzkRfuaFj8etIYMj/ECtXtgfz72RE6U+dos27b3XIwk=
github.com/agnivade/wasmbrowsertest v0.3.1/go.mod h1:zQt6ZTdl338xxRaMW395qccVE2eQm0SjC/SDz0mPWQI=
github.com/bodqhrohro/go-tdlib v0.1.1 h1:lmHognymABxP3cmHkfAGhGnWaJaZ3htpJ7RSbZacin4=
github.com/bodqhrohro/go-tdlib v0.1.2-0.20191121200156-e826071d3317 h1:+mv4FwWXl8hTa7PrhekwVzPknH+rHqB60jIPBi2XqI8=
github.com/bodqhrohro/go-tdlib v0.1.2-0.20191121200156-e826071d3317/go.mod h1:Xs8fXbk5n7VaPyrSs9DP7QYoBScWYsjX+lUcWmx1DIU=
github.com/bodqhrohro/go-tdlib v0.1.2-0.20191121233100-48d2382034fb h1:y5PnjdAnNVS0q8xuwjm3TxBfLriJmykQdoGiyYZB3s0=
github.com/bodqhrohro/go-tdlib v0.1.2-0.20191121233100-48d2382034fb/go.mod h1:Xs8fXbk5n7VaPyrSs9DP7QYoBScWYsjX+lUcWmx1DIU=
github.com/bodqhrohro/go-xmpp v0.1.4-0.20191106203535-f3b463f3b26c h1:LzcQyE+Gs+0kAbpnPAUD68FvUCieKZip44URAmH70PI=
github.com/bodqhrohro/go-xmpp v0.1.4-0.20191106203535-f3b463f3b26c/go.mod h1:fWixaMaFvx8cxXcJVJ5kU9csMeD/JN8on7ybassU8rY=
github.com/bodqhrohro/go-xmpp v0.2.1-0.20191105232737-9abd5be0aa1b h1:9BLd/SNO4JJZLRl1Qb1v9mNivIlHuwHDe2c8hQvBxFA=
@ -12,6 +17,7 @@ github.com/chromedp/cdproto v0.0.0-20190926234355-1b4886c6fad6/go.mod h1:0YChpVz
github.com/chromedp/chromedp v0.3.1-0.20190619195644-fd957a4d2901/go.mod h1:mJdvfrVn594N9tfiPecUidF6W5jPRKHymqHfzbobPsM=
github.com/chromedp/chromedp v0.4.0/go.mod h1:DC3QUn4mJ24dwjcaGQLoZrhm4X/uPHZ6spDbS2uFhm4=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/edsrzf/mmap-go v1.0.0/go.mod h1:YO35OhQPt3KJa3ryjFM5Bs14WD66h8eGKpfaBNrHW5M=
github.com/fatih/color v1.6.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
@ -25,6 +31,7 @@ github.com/gobwas/ws v1.0.2/go.mod h1:szmBTxLgaFppYjEmNtny/v3w89xOydFnnZMcgRRu/E
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
github.com/google/go-cmp v0.3.0 h1:crn/baboCvb5fXaQ0IJ1SGTsTVrWpDsCWC8EGETZijY=
github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
@ -55,6 +62,7 @@ github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1Cpa
github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/santhosh-tekuri/jsonschema v1.2.4 h1:hNhW8e7t+H1vgY+1QeEQpveR6D4+OwKPXCfD2aieJis=
github.com/santhosh-tekuri/jsonschema v1.2.4/go.mod h1:TEAUOeZSmIxTTuHatJzrvARHiuO9LYd+cIxzgEHCQI4=
@ -67,6 +75,7 @@ github.com/spf13/pflag v1.0.1/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnIn
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=

View file

@ -10,6 +10,7 @@ import (
"dev.narayana.im/narayana/telegabber/persistence"
"github.com/zelenin/go-tdlib/client"
"gosrc.io/xmpp"
)
var logConstants = map[string]int32{
@ -34,6 +35,8 @@ func stringToLogConstant(c string) int32 {
// Client stores the metadata for lazily invoked TDlib instance
type Client struct {
client *client.Client
authorizer *client.ClientAuthorizerType
xmpp *xmpp.Component
jid string
parameters *client.TdlibParameters
Session *persistence.Session
@ -42,14 +45,14 @@ type Client struct {
}
// NewClient instantiates a Telegram App
func NewClient(conf config.TelegramConfig, jid string, session *persistence.Session) (Client, error) {
func NewClient(conf config.TelegramConfig, jid string, component *xmpp.Component, session *persistence.Session) (*Client, error) {
logVerbosity := client.WithLogVerbosity(&client.SetLogVerbosityLevelRequest{
NewVerbosityLevel: stringToLogConstant(conf.Loglevel),
})
apiID, err := strconv.Atoi(conf.Tdlib.Client.APIID)
if err != nil {
return Client{}, errors.Wrap(err, "Wrong api_id")
return &Client{}, errors.Wrap(err, "Wrong api_id")
}
parameters := client.TdlibParameters{
@ -75,8 +78,9 @@ func NewClient(conf config.TelegramConfig, jid string, session *persistence.Sess
IgnoreFileNames: false,
}
return Client{
return &Client{
parameters: &parameters,
xmpp: component,
jid: jid,
Session: session,
logVerbosity: logVerbosity,
@ -93,3 +97,7 @@ func updateHandler(tdlibClient *client.Client) {
}
}
}
func (c *Client) ProcessOutgoingMessage(chatID int, text string, messageID int) {
// TODO
}

33
telegram/commands.go Normal file
View file

@ -0,0 +1,33 @@
package telegram
const notEnoughArguments string = "Not enough arguments"
const telegramNotInitialized string = "Telegram connection is not initialized yet"
// ProcessTransportCommand executes commands sent directly to the component
func (c *Client) ProcessTransportCommand(cmd string, args []string) string {
switch cmd {
case "login", "code", "password":
if cmd == "login" && c.Session.Login != "" {
return ""
}
if len(args) < 1 {
return notEnoughArguments
}
if c.authorizer == nil {
return telegramNotInitialized
}
switch cmd {
case "login":
c.authorizer.PhoneNumber <- args[0]
c.Session.Login = args[0]
case "code":
c.authorizer.Code <- args[0]
case "password":
c.authorizer.Password <- args[0]
}
}
return ""
}

View file

@ -3,6 +3,8 @@ package telegram
import (
"github.com/pkg/errors"
"dev.narayana.im/narayana/telegabber/xmpp/gateway"
log "github.com/sirupsen/logrus"
"github.com/zelenin/go-tdlib/client"
)
@ -15,26 +17,15 @@ func (c *Client) Connect() error {
log.Warn("Connecting to Telegram network...")
authorizer := client.ClientAuthorizer()
go func() {
for {
state, ok := <-authorizer.State
if !ok {
return
}
c.authorizer = client.ClientAuthorizer()
ok = authorizationStateHandler(state)
if !ok {
return
}
}
}()
go c.interactor()
authorizer.TdlibParameters <- c.parameters
c.authorizer.TdlibParameters <- c.parameters
tdlibClient, err := client.NewClient(authorizer, c.logVerbosity)
tdlibClient, err := client.NewClient(c.authorizer, c.logVerbosity)
if err != nil {
return errors.Wrap(err, "Coudn't initialize a Telegram client instance")
return errors.Wrap(err, "Couldn't initialize a Telegram client instance")
}
c.client = tdlibClient
@ -59,10 +50,34 @@ func (c *Client) Disconnect() {
c.online = false
}
func authorizationStateHandler(state client.AuthorizationState) bool {
switch state.AuthorizationStateType() {
// TODO
}
func (c *Client) interactor() {
for {
state, ok := <-c.authorizer.State
if !ok {
return
}
return true
stateType := state.AuthorizationStateType()
log.Infof("Telegram authorization state: %#v", stateType)
switch stateType {
case client.TypeAuthorizationStateWaitPhoneNumber:
log.Warn("Logging in...")
if c.Session.Login != "" {
c.authorizer.PhoneNumber <- c.Session.Login
} else {
gateway.SendMessage(c.jid, "", "Please, enter your Telegram login via /login 12345", c.xmpp)
}
case client.TypeAuthorizationStateWaitCode:
log.Warn("Waiting for authorization code...")
gateway.SendMessage(c.jid, "", "Please, enter authorization code via /code 12345", c.xmpp)
case client.TypeAuthorizationStateWaitPassword:
log.Warn("Waiting for 2FA password...")
gateway.SendMessage(c.jid, "", "Please, enter 2FA passphrase via /password 12345", c.xmpp)
case client.TypeAuthorizationStateReady:
log.Warn("Authorization successful!")
// TODO
return
}
}
}

View file

@ -8,6 +8,7 @@ import (
"dev.narayana.im/narayana/telegabber/config"
"dev.narayana.im/narayana/telegabber/persistence"
"dev.narayana.im/narayana/telegabber/telegram"
"dev.narayana.im/narayana/telegabber/xmpp/gateway"
log "github.com/sirupsen/logrus"
"github.com/soheilhy/args"
@ -17,9 +18,8 @@ import (
const pollingInterval time.Duration = 1e7
var jid *xmpp.Jid
var tgConf config.TelegramConfig
var sessions map[string]telegram.Client
var sessions map[string]*telegram.Client
var queue map[string]*stanza.Presence
var db persistence.SessionsYamlDB
@ -28,18 +28,13 @@ var db persistence.SessionsYamlDB
func NewComponent(conf config.XMPPConfig, tc config.TelegramConfig) (*xmpp.StreamManager, *xmpp.Component, error) {
var err error
jid, err = xmpp.NewJid(conf.Jid)
gateway.Jid, err = xmpp.NewJid(conf.Jid)
if err != nil {
return nil, nil, err
}
tgConf = tc
err = loadSessions(conf.Db)
if err != nil {
return nil, nil, err
}
options := xmpp.ComponentOptions{
Address: conf.Host + ":" + conf.Port,
Domain: conf.Jid,
@ -57,6 +52,11 @@ func NewComponent(conf config.XMPPConfig, tc config.TelegramConfig) (*xmpp.Strea
return nil, nil, err
}
err = loadSessions(conf.Db, component)
if err != nil {
return nil, nil, err
}
sm := xmpp.NewStreamManager(component, nil)
go heartbeat(component)
@ -99,10 +99,10 @@ func heartbeat(component *xmpp.Component) {
}
}
func loadSessions(dbPath string) error {
func loadSessions(dbPath string, component *xmpp.Component) error {
var err error
sessions = make(map[string]telegram.Client)
sessions = make(map[string]*telegram.Client)
db, err = persistence.LoadSessions(dbPath)
if err != nil {
@ -111,7 +111,7 @@ func loadSessions(dbPath string) error {
db.Transaction(func() bool {
for jid, session := range db.Data.Sessions {
getTelegramInstance(jid, &session)
getTelegramInstance(jid, &session, component)
}
return false
@ -120,10 +120,11 @@ func loadSessions(dbPath string) error {
return nil
}
func getTelegramInstance(jid string, savedSession *persistence.Session) (telegram.Client, bool) {
func getTelegramInstance(jid string, savedSession *persistence.Session, component *xmpp.Component) (*telegram.Client, bool) {
var err error
session, ok := sessions[jid]
if !ok {
session, err := telegram.NewClient(tgConf, jid, savedSession)
session, err = telegram.NewClient(tgConf, jid, component, savedSession)
if err != nil {
log.Error(errors.Wrap(err, "TDlib initialization failure"))
return session, false
@ -195,7 +196,7 @@ func newPresence(bareJid string, to string, args ...args.V) stanza.Presence {
func sendPresence(component *xmpp.Component, to string, args ...args.V) error {
var logFrom string
bareJid := jid.Bare()
bareJid := gateway.Jid.Bare()
if SPFrom.IsSet(args) {
logFrom = SPFrom.Get(args)
} else {
@ -212,7 +213,12 @@ func sendPresence(component *xmpp.Component, to string, args ...args.V) error {
// explicit check, as marshalling is expensive
if log.GetLevel() == log.DebugLevel {
log.Debug(xml.Marshal(presence))
xmlPresence, err := xml.Marshal(presence)
if err == nil {
log.Debug(string(xmlPresence))
} else {
log.Debugf("%#v", presence)
}
}
immed := SPImmed.Get(args)

53
xmpp/gateway/gateway.go Normal file
View file

@ -0,0 +1,53 @@
package gateway
import (
"encoding/xml"
log "github.com/sirupsen/logrus"
"gosrc.io/xmpp"
"gosrc.io/xmpp/stanza"
)
// Jid stores the component's JID object
var Jid *xmpp.Jid
// SendMessage creates and sends a message stanza
func SendMessage(to string, from string, body string, component *xmpp.Component) {
componentJid := Jid.Full()
var logFrom string
var messageFrom string
if from == "" {
logFrom = componentJid
messageFrom = componentJid
} else {
logFrom = from
messageFrom = from + "@" + componentJid
}
log.WithFields(log.Fields{
"from": logFrom,
"to": to,
}).Warn("Got message")
message := stanza.Message{
Attrs: stanza.Attrs{
From: messageFrom,
To: to,
Type: "chat",
},
Body: body,
}
// explicit check, as marshalling is expensive
if log.GetLevel() == log.DebugLevel {
xmlMessage, err := xml.Marshal(message)
if err == nil {
log.Debug(string(xmlMessage))
} else {
log.Debugf("%#v", message)
}
}
_ = component.Send(message)
}

View file

@ -2,8 +2,11 @@ package xmpp
import (
"github.com/pkg/errors"
"strconv"
"strings"
"dev.narayana.im/narayana/telegabber/persistence"
"dev.narayana.im/narayana/telegabber/xmpp/gateway"
log "github.com/sirupsen/logrus"
"gosrc.io/xmpp"
@ -33,9 +36,52 @@ func HandleMessage(s xmpp.Sender, p stanza.Packet) {
return
}
log.Printf("Message: %#v\n", msg)
reply := stanza.Message{Attrs: stanza.Attrs{To: msg.From}, Body: msg.Body}
_ = s.Send(reply)
component, ok := s.(*xmpp.Component)
if !ok {
log.Error("Not a component")
return
}
if msg.Type != "error" && msg.Body != "" {
log.WithFields(log.Fields{
"from": msg.From,
"to": msg.To,
}).Warn("Message")
log.Debugf("%#v", msg)
fromJid, err := xmpp.NewJid(msg.From)
if err != nil {
log.Error("Invalid from JID!")
return
}
session, ok := sessions[fromJid.Bare()]
if !ok {
log.Error("Message from stranger")
return
}
toParts := strings.Split(msg.To, "@")
toID := toParts[0]
if len(toParts) > 1 {
toIDInt, err := strconv.Atoi(toID)
if err != nil {
session.ProcessOutgoingMessage(toIDInt, msg.Body, 0)
return
}
} else if toID == gateway.Jid.Bare() {
bodyFields := strings.Fields(msg.Body)
cmd := bodyFields[0]
if strings.HasPrefix(cmd, "/") {
response := session.ProcessTransportCommand(cmd[1:], bodyFields[1:])
if response != "" {
gateway.SendMessage(msg.From, "", response, component)
}
return
}
}
log.Warn("Unknown purpose of the message, skipping")
}
}
// HandlePresence processes an incoming XMPP presence
@ -49,7 +95,7 @@ func HandlePresence(s xmpp.Sender, p stanza.Packet) {
if prs.Type == "subscribe" {
handleSubscription(s, prs)
}
if prs.To == jid.Bare() {
if prs.To == gateway.Jid.Bare() {
handlePresence(s, prs)
}
}
@ -77,6 +123,12 @@ func handlePresence(s xmpp.Sender, p stanza.Presence) {
presenceType = "online"
}
component, ok := s.(*xmpp.Component)
if !ok {
log.Error("Not a component")
return
}
log.WithFields(log.Fields{
"type": presenceType,
"from": p.From,
@ -90,7 +142,7 @@ func handlePresence(s xmpp.Sender, p stanza.Presence) {
return
}
bareFromJid := fromJid.Bare()
session, ok := getTelegramInstance(bareFromJid, &persistence.Session{})
session, ok := getTelegramInstance(bareFromJid, &persistence.Session{}, component)
if !ok {
return
}
@ -101,7 +153,7 @@ func handlePresence(s xmpp.Sender, p stanza.Presence) {
delete(sessions, bareFromJid)
case "unavailable", "error":
session.Disconnect()
case "":
case "", "online":
// due to the weird implentation of go-tdlib wrapper, it won't
// return the client instance until successful authorization
go func() {