From 997dc1006a0f096443248e0c1e1b5ca3389cd79f Mon Sep 17 00:00:00 2001
From: Aleksandr Zelenin <aleksandr@zelenin.me>
Date: Wed, 10 Oct 2018 16:35:28 +0300
Subject: [PATCH] registration fix

---
 README.md               | 2 +-
 client/authorization.go | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 01dd9de..cd574a7 100644
--- a/README.md
+++ b/README.md
@@ -59,7 +59,7 @@ func main() {
     
     // client authorizer
     authorizer := client.ClientAuthorizer()
-    go client.CliInteractor(authorizer, false)
+    go client.CliInteractor(authorizer)
     
     // or bot authorizer
     botToken := "000000000:gsVCGG5YbikxYHC7bP5vRvmBqJ7Xz6vG6td"
diff --git a/client/authorization.go b/client/authorization.go
index b8e02ba..588c1de 100644
--- a/client/authorization.go
+++ b/client/authorization.go
@@ -98,7 +98,7 @@ func (stateHandler *clientAuthorizer) Handle(client *Client, state Authorization
     return ErrNotSupportedAuthorizationState
 }
 
-func CliInteractor(clientAuthorizer *clientAuthorizer, registration bool) {
+func CliInteractor(clientAuthorizer *clientAuthorizer) {
     for {
         select {
         case state := <-clientAuthorizer.State:
@@ -118,7 +118,9 @@ func CliInteractor(clientAuthorizer *clientAuthorizer, registration bool) {
                 fmt.Println("Enter code: ")
                 fmt.Scanln(&code)
 
-                if registration {
+                if !state.(*AuthorizationStateWaitCode).IsRegistered {
+                    fmt.Println("Phone number is not registered.")
+
                     fmt.Println("Enter first name: ")
                     fmt.Scanln(&firstName)