Telegram: device registration

1 files changed, 145 insertions, 2 deletions

diff --git a/protocols/Telegram/src/server.cpp b/protocols/Telegram/src/server.cpp
index f8985e5ecb..1905b4ae70 100644
--- a/protocols/Telegram/src/server.cpp
+++ b/protocols/Telegram/src/server.cpp
@@ -22,6 +22,8 @@ void __cdecl CMTProto::ServerThread(void *)
 	m_bRunning = true;
 	m_bTerminated = m_bAuthorized = false;
 
+	SendQuery(new td::td_api::getOption("version"));
+
 	while (!m_bTerminated) {
 		ProcessResponse(m_pClientMmanager->receive(10));
 	}
@@ -29,6 +31,138 @@ void __cdecl CMTProto::ServerThread(void *)
 	m_bRunning = false;
 }
 
+void CMTProto::LogOut()
+{
+	if (m_bTerminated)
+		return;
+
+	debugLogA("CMTProto::OnLoggedOut");
+	m_bTerminated = true;
+	m_bAuthorized = false;
+
+	ProtoBroadcastAck(0, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)m_iStatus, ID_STATUS_OFFLINE);
+	m_iStatus = m_iDesiredStatus = ID_STATUS_OFFLINE;
+
+	setAllContactStatuses(ID_STATUS_OFFLINE, false);
+}
+
+void CMTProto::OnLoggedIn()
+{
+	m_bAuthorized = true;
+
+	debugLogA("CMTProto::OnLoggedIn");
+
+	ProtoBroadcastAck(0, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)m_iStatus, m_iDesiredStatus);
+	m_iStatus = m_iDesiredStatus;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+INT_PTR CALLBACK CMTProto::EnterPhoneCode(void *param)
+{
+	auto *ppro = (CMTProto *)param;
+
+	ENTER_STRING es = {};
+	es.szModuleName = ppro->m_szModuleName;
+	es.caption = TranslateT("Enter secret code sent to your phone");
+	if (EnterString(&es)) {
+		ppro->SendQuery(new td::td_api::checkAuthenticationCode(_T2A(es.ptszResult).get()), &CMTProto::OnUpdateAuth);
+		mir_free(es.ptszResult);
+	}
+	else ppro->LogOut();
+	return 0;
+}
+
+INT_PTR CALLBACK CMTProto::EnterPassword(void *param)
+{
+	auto *ppro = (CMTProto *)param;
+	CMStringW wszTitle(TranslateT("Enter password"));
+	
+	auto *pAuth = (td::td_api::authorizationStateWaitPassword *)ppro->pAuthState.get();
+	if (!pAuth->password_hint_.empty())
+		wszTitle.AppendFormat(TranslateT(" (hint: %s)"), Utf2T(pAuth->password_hint_.c_str()).get());
+
+	ENTER_STRING es = {};
+	es.szModuleName = ppro->m_szModuleName;
+	es.caption = wszTitle;
+	es.type = ESF_PASSWORD;
+	if (EnterString(&es)) {
+		ppro->SendQuery(new td::td_api::checkAuthenticationPassword(_T2A(es.ptszResult).get()), &CMTProto::OnUpdateAuth);
+		mir_free(es.ptszResult);
+	}
+	else ppro->LogOut();
+	return 0;
+}
+
+void CMTProto::ProcessAuth(td::td_api::updateAuthorizationState *pObj)
+{
+	pAuthState = std::move(pObj->authorization_state_);
+	switch (pAuthState->get_id()) {
+	case td::td_api::authorizationStateWaitTdlibParameters::ID:
+		{
+			char text[100];
+			Miranda_GetVersionText(text, sizeof(text));
+
+			CMStringW wszPath(GetProtoFolder());
+
+			auto *request = new td::td_api::setTdlibParameters();
+			request->database_directory_ = T2Utf(wszPath).get();
+			request->use_message_database_ = false;
+			request->use_secret_chats_ = true;
+			request->api_id_ = 94575;
+			request->api_hash_ = "a3406de8d171bb422bb6ddf3bbd800e2";
+			request->system_language_code_ = "en";
+			request->device_model_ = "Miranda NG";
+			request->application_version_ = text;
+			request->enable_storage_optimizer_ = true;
+			SendQuery(request, &CMTProto::OnUpdateAuth);
+		}
+		break;
+
+	case td::td_api::authorizationStateWaitPhoneNumber::ID:
+		SendQuery(new td::td_api::setAuthenticationPhoneNumber(_T2A(m_szOwnPhone).get(), nullptr), &CMTProto::OnUpdateAuth);
+		break;
+
+	case td::td_api::authorizationStateWaitCode::ID:
+		CallFunctionSync(EnterPhoneCode, this);
+		break;
+
+	case td::td_api::authorizationStateWaitPassword::ID:
+		CallFunctionSync(EnterPassword, this);
+		break;
+
+	case td::td_api::authorizationStateReady::ID:
+		OnLoggedIn();
+		break;
+
+	case td::td_api::authorizationStateLoggingOut::ID:
+		debugLogA("Server required us to log out, exiting");
+		LogOut();
+		break;
+
+	case td::td_api::authorizationStateClosing::ID:
+		debugLogA("Connection terminated, exiting");
+		LogOut();
+		break;
+	}
+}
+
+void CMTProto::OnUpdateAuth(td::ClientManager::Response &response)
+{
+	if (response.object->get_id() == td::td_api::error::ID) {
+		auto *pError = (td::td_api::error*)response.object.get();
+		debugLogA("error happened: %s", to_string(*pError).c_str());
+		
+		if (pError->message_ == "PHONE_CODE_EXPIRED")
+			Popup(0, TranslateT("Phone code expired"), TranslateT("Error"));
+		else if(pError->message_ == "INVALID_PHONE_CODE")
+			Popup(0, TranslateT("Invalid phone code"), TranslateT("Error"));
+
+		pAuthState = std::move(nullptr);
+		LogOut();
+	}
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 
 void CMTProto::ProcessResponse(td::ClientManager::Response response)
@@ -36,12 +170,21 @@ void CMTProto::ProcessResponse(td::ClientManager::Response response)
 	if (!response.object)
 		return;
 
-	if (response.client_id) {
-		auto *p = m_arRequests.find((TG_REQUEST *)&response.client_id);
+	debugLogA("ProcessResponse: id=%d (%s)", int(response.request_id), to_string(response.object).c_str());
+
+	if (response.request_id) {
+		auto *p = m_arRequests.find((TG_REQUEST *)&response.request_id);
 		if (p) {
 			(this->*p->pHandler)(response);
 			m_arRequests.remove(p);
 		}
+		return;
+	}
+
+	switch (response.object->get_id()) {
+	case td::td_api::updateAuthorizationState::ID:
+		ProcessAuth((td::td_api::updateAuthorizationState *)response.object.get());
+		break;
 	}
 }