init

client/authorization.go

@@ -0,0 +1,179 @@
+package client
+
+import (
+    "errors"
+    "fmt"
+    "time"
+)
+
+var ErrNotSupportedAuthorizationState = errors.New("not supported state")
+
+type AuthorizationStateHandler interface {
+    Handle(client *Client, state AuthorizationState) error
+}
+
+func Authorize(client *Client, authorizationStateHandler AuthorizationStateHandler) error {
+    for {
+        state, err := client.GetAuthorizationState()
+        if err != nil {
+            return err
+        }
+
+        err = authorizationStateHandler.Handle(client, state)
+        if err != nil {
+            return err
+        }
+
+        if state.AuthorizationStateType() == TypeAuthorizationStateReady {
+            // dirty hack for db flush after authorization
+            time.Sleep(1 * time.Second)
+            return nil
+        }
+    }
+}
+
+type clientAuthorizer struct {
+    TdlibParameters chan *TdlibParameters
+    PhoneNumber     chan string
+    Code            chan string
+    State           chan AuthorizationState
+}
+
+func ClientAuthorizer() *clientAuthorizer {
+    return &clientAuthorizer{
+        TdlibParameters: make(chan *TdlibParameters, 1),
+        PhoneNumber:     make(chan string, 1),
+        Code:            make(chan string, 1),
+        State:           make(chan AuthorizationState, 10),
+    }
+}
+
+func (stateHandler *clientAuthorizer) Handle(client *Client, state AuthorizationState) error {
+    stateHandler.State <- state
+
+    switch state.AuthorizationStateType() {
+    case TypeAuthorizationStateWaitTdlibParameters:
+        _, err := client.SetTdlibParameters(<-stateHandler.TdlibParameters)
+        return err
+
+    case TypeAuthorizationStateWaitEncryptionKey:
+        _, err := client.CheckDatabaseEncryptionKey(nil)
+        return err
+
+    case TypeAuthorizationStateWaitPhoneNumber:
+        _, err := client.SetAuthenticationPhoneNumber(<-stateHandler.PhoneNumber, false, false)
+        return err
+
+    case TypeAuthorizationStateWaitCode:
+        _, err := client.CheckAuthenticationCode(<-stateHandler.Code, "", "")
+        return err
+
+    case TypeAuthorizationStateWaitPassword:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateReady:
+        close(stateHandler.TdlibParameters)
+        close(stateHandler.PhoneNumber)
+        close(stateHandler.Code)
+        close(stateHandler.State)
+
+        return nil
+
+    case TypeAuthorizationStateLoggingOut:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateClosing:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateClosed:
+        return ErrNotSupportedAuthorizationState
+    }
+
+    return ErrNotSupportedAuthorizationState
+}
+
+func CliInteractor(clientAuthorizer *clientAuthorizer) {
+    for {
+        select {
+        case state := <-clientAuthorizer.State:
+            switch state.AuthorizationStateType() {
+            case TypeAuthorizationStateWaitPhoneNumber:
+                fmt.Println("Enter phone number: ")
+                var phoneNumber string
+                fmt.Scanln(&phoneNumber)
+
+                clientAuthorizer.PhoneNumber <- phoneNumber
+
+            case TypeAuthorizationStateWaitCode:
+                fmt.Println("Enter code: ")
+                var code string
+                fmt.Scanln(&code)
+
+                clientAuthorizer.Code <- code
+
+            case TypeAuthorizationStateReady:
+                return
+            }
+        }
+    }
+}
+
+type botAuthorizer struct {
+    TdlibParameters chan *TdlibParameters
+    Token           chan string
+    State           chan AuthorizationState
+}
+
+func BotAuthorizer(token string) *botAuthorizer {
+    botAuthorizer := &botAuthorizer{
+        TdlibParameters: make(chan *TdlibParameters, 1),
+        Token:           make(chan string, 1),
+        State:           make(chan AuthorizationState, 10),
+    }
+
+    botAuthorizer.Token <- token
+
+    return botAuthorizer
+}
+
+func (stateHandler *botAuthorizer) Handle(client *Client, state AuthorizationState) error {
+    stateHandler.State <- state
+
+    switch state.AuthorizationStateType() {
+    case TypeAuthorizationStateWaitTdlibParameters:
+        _, err := client.SetTdlibParameters(<-stateHandler.TdlibParameters)
+        return err
+
+    case TypeAuthorizationStateWaitEncryptionKey:
+        _, err := client.CheckDatabaseEncryptionKey(nil)
+        return err
+
+    case TypeAuthorizationStateWaitPhoneNumber:
+        _, err := client.CheckAuthenticationBotToken(<-stateHandler.Token)
+        return err
+
+    case TypeAuthorizationStateWaitCode:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateWaitPassword:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateReady:
+        close(stateHandler.TdlibParameters)
+        close(stateHandler.Token)
+        close(stateHandler.State)
+
+        return nil
+
+    case TypeAuthorizationStateLoggingOut:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateClosing:
+        return ErrNotSupportedAuthorizationState
+
+    case TypeAuthorizationStateClosed:
+        return ErrNotSupportedAuthorizationState
+    }
+
+    return ErrNotSupportedAuthorizationState
+}