Registration support

This commit is contained in:
Bohdan Horbeshko 2022-08-15 06:10:29 -04:00
parent 0c71036148
commit e9e65b6778
2 changed files with 36 additions and 12 deletions

View file

@ -290,25 +290,33 @@ func (c *Client) ProcessTransportCommand(cmdline string, resource string) string
}
// set My Name
case "setname":
if !c.Online() {
return notOnline
}
var firstname string
var lastname string
if len(args) > 0 {
firstname = args[0]
}
if firstname == "" {
return "The name should contain at least one character"
}
if len(args) > 1 {
lastname = args[1]
lastname = rawCmdArguments(cmdline, 1)
}
_, err := c.client.SetName(&client.SetNameRequest{
FirstName: firstname,
LastName: lastname,
})
if err != nil {
return errors.Wrap(err, "Couldn't set name").Error()
if c.authorizer != nil && !c.authorizer.isClosed {
c.authorizer.FirstName <- firstname
c.authorizer.LastName <- lastname
} else {
if !c.Online() {
return notOnline
}
_, err := c.client.SetName(&client.SetNameRequest{
FirstName: firstname,
LastName: lastname,
})
if err != nil {
return errors.Wrap(err, "Couldn't set name").Error()
}
}
// set About
case "setbio":

View file

@ -18,6 +18,9 @@ type clientAuthorizer struct {
Code chan string
State chan client.AuthorizationState
Password chan string
FirstName chan string
LastName chan string
isClosed bool
}
func (stateHandler *clientAuthorizer) Handle(c *client.Client, state client.AuthorizationState) error {
@ -52,7 +55,11 @@ func (stateHandler *clientAuthorizer) Handle(c *client.Client, state client.Auth
return err
case client.TypeAuthorizationStateWaitRegistration:
return client.ErrNotSupportedAuthorizationState
_, err := c.RegisterUser(&client.RegisterUserRequest{
FirstName: <-stateHandler.FirstName,
LastName: <-stateHandler.LastName,
})
return err
case client.TypeAuthorizationStateWaitPassword:
_, err := c.CheckAuthenticationPassword(&client.CheckAuthenticationPasswordRequest{
@ -77,11 +84,14 @@ func (stateHandler *clientAuthorizer) Handle(c *client.Client, state client.Auth
}
func (stateHandler *clientAuthorizer) Close() {
stateHandler.isClosed = true
close(stateHandler.TdlibParameters)
close(stateHandler.PhoneNumber)
close(stateHandler.Code)
close(stateHandler.State)
close(stateHandler.Password)
close(stateHandler.FirstName)
close(stateHandler.LastName)
}
// Connect starts TDlib connection
@ -102,6 +112,8 @@ func (c *Client) Connect(resource string) error {
Code: make(chan string, 1),
State: make(chan client.AuthorizationState, 10),
Password: make(chan string, 1),
FirstName: make(chan string, 1),
LastName: make(chan string, 1),
}
c.locks.authorizationReady.Add(1)
@ -212,6 +224,10 @@ func (c *Client) interactor() {
case client.TypeAuthorizationStateWaitCode:
log.Warn("Waiting for authorization code...")
gateway.SendMessage(c.jid, "", "Please, enter authorization code via /code 12345", c.xmpp)
// stage 1b: wait for registration
case client.TypeAuthorizationStateWaitRegistration:
log.Warn("Waiting for full name...")
gateway.SendMessage(c.jid, "", "This number is not registered yet! Please, enter your name via /setname John Doe", c.xmpp)
// stage 2: wait for 2fa
case client.TypeAuthorizationStateWaitPassword:
log.Warn("Waiting for 2FA password...")