Implement an own authorizer

1 files changed, 79 insertions, 1 deletions

diff --git a/telegram/connect.go b/telegram/connect.go
index 7e92bc8..4f509d2 100644
--- a/telegram/connect.go
+++ b/telegram/connect.go
@@ -9,6 +9,78 @@ import (
 	"github.com/zelenin/go-tdlib/client"
 )
 
+type clientAuthorizer struct {
+	TdlibParameters chan *client.TdlibParameters
+	PhoneNumber     chan string
+	Code            chan string
+	State           chan client.AuthorizationState
+	Password        chan string
+}
+
+func (stateHandler *clientAuthorizer) Handle(c *client.Client, state client.AuthorizationState) error {
+	stateHandler.State <- state
+
+	switch state.AuthorizationStateType() {
+	case client.TypeAuthorizationStateWaitTdlibParameters:
+		_, err := c.SetTdlibParameters(&client.SetTdlibParametersRequest{
+			Parameters: <-stateHandler.TdlibParameters,
+		})
+		return err
+
+	case client.TypeAuthorizationStateWaitEncryptionKey:
+		_, err := c.CheckDatabaseEncryptionKey(&client.CheckDatabaseEncryptionKeyRequest{})
+		return err
+
+	case client.TypeAuthorizationStateWaitPhoneNumber:
+		_, err := c.SetAuthenticationPhoneNumber(&client.SetAuthenticationPhoneNumberRequest{
+			PhoneNumber: <-stateHandler.PhoneNumber,
+			Settings: &client.PhoneNumberAuthenticationSettings{
+				AllowFlashCall:       false,
+				IsCurrentPhoneNumber: false,
+				AllowSmsRetrieverApi: false,
+			},
+		})
+		return err
+
+	case client.TypeAuthorizationStateWaitCode:
+		_, err := c.CheckAuthenticationCode(&client.CheckAuthenticationCodeRequest{
+			Code: <-stateHandler.Code,
+		})
+		return err
+
+	case client.TypeAuthorizationStateWaitRegistration:
+		return client.ErrNotSupportedAuthorizationState
+
+	case client.TypeAuthorizationStateWaitPassword:
+		_, err := c.CheckAuthenticationPassword(&client.CheckAuthenticationPasswordRequest{
+			Password: <-stateHandler.Password,
+		})
+		return err
+
+	case client.TypeAuthorizationStateReady:
+		return nil
+
+	case client.TypeAuthorizationStateLoggingOut:
+		return client.ErrNotSupportedAuthorizationState
+
+	case client.TypeAuthorizationStateClosing:
+		return client.ErrNotSupportedAuthorizationState
+
+	case client.TypeAuthorizationStateClosed:
+		return client.ErrNotSupportedAuthorizationState
+	}
+
+	return client.ErrNotSupportedAuthorizationState
+}
+
+func (stateHandler *clientAuthorizer) Close() {
+	close(stateHandler.TdlibParameters)
+	close(stateHandler.PhoneNumber)
+	close(stateHandler.Code)
+	close(stateHandler.State)
+	close(stateHandler.Password)
+}
+
 // Connect starts TDlib connection
 func (c *Client) Connect() error {
 	if c.online {
@@ -17,7 +89,13 @@ func (c *Client) Connect() error {
 
 	log.Warn("Connecting to Telegram network...")
 
-	c.authorizer = client.ClientAuthorizer()
+	c.authorizer = &clientAuthorizer{
+		TdlibParameters: make(chan *client.TdlibParameters, 1),
+		PhoneNumber:     make(chan string, 1),
+		Code:            make(chan string, 1),
+		State:           make(chan client.AuthorizationState, 10),
+		Password:        make(chan string, 1),
+	}
 
 	go c.interactor()