From adfd6ca644ed2a8cb7205502075ac582fcf12c96 Mon Sep 17 00:00:00 2001
From: Learath2 <learath2@gmail.com>
Date: Thu, 2 Mar 2017 18:16:50 +0300
Subject: [PATCH] Enable the username prompt on client side

---
 src/game/client/components/console.cpp | 33 +++++++++++++++++++++++---
 src/game/client/components/console.h   |  2 ++
 2 files changed, 32 insertions(+), 3 deletions(-)

diff --git a/src/game/client/components/console.cpp b/src/game/client/components/console.cpp
index 7b5f130eb..0b38e7c01 100644
--- a/src/game/client/components/console.cpp
+++ b/src/game/client/components/console.cpp
@@ -9,6 +9,7 @@
 
 #include <base/system.h>
 
+#include <engine/serverbrowser.h>
 #include <engine/shared/ringbuffer.h>
 #include <engine/shared/config.h>
 #include <engine/graphics.h>
@@ -47,6 +48,8 @@ CGameConsole::CInstance::CInstance(int Type)
 	m_CompletionRenderOffset = 0.0f;
 	m_ReverseTAB = false;
 
+	m_aUser[0] = '\0';
+
 	m_IsCommand = false;
 }
 
@@ -76,7 +79,18 @@ void CGameConsole::CInstance::ExecuteLine(const char *pLine)
 		if(m_pGameConsole->Client()->RconAuthed())
 			m_pGameConsole->Client()->Rcon(pLine);
 		else
-			m_pGameConsole->Client()->RconAuth("", pLine);
+		{
+			CServerInfo pServerInfo;
+			m_pGameConsole->Client()->GetServerInfo(&pServerInfo);
+
+			if(!m_aUser[0] && IsDDNet(&pServerInfo))
+				str_copy(m_aUser, pLine, sizeof m_aUser);
+			else
+			{
+				m_pGameConsole->Client()->RconAuth(m_aUser, pLine);
+				m_aUser[0] = '\0';
+			}
+		}
 	}
 }
 
@@ -459,6 +473,9 @@ void CGameConsole::OnRender()
 		Info.m_pCurrentCmd = pConsole->m_aCompletionBuffer;
 		TextRender()->SetCursor(&Info.m_Cursor, x+Info.m_Offset, y+RowHeight+2.0f, FontSize, TEXTFLAG_RENDER);
 
+		CServerInfo pServerInfo;
+		Client()->GetServerInfo(&pServerInfo);
+
 		// render prompt
 		CTextCursor Cursor;
 		TextRender()->SetCursor(&Cursor, x, y, FontSize, TEXTFLAG_RENDER);
@@ -470,7 +487,17 @@ void CGameConsole::OnRender()
 				if(Client()->RconAuthed())
 					pPrompt = "rcon> ";
 				else
-					pPrompt = "ENTER PASSWORD> ";
+				{
+					if(IsDDNet(&pServerInfo))
+					{
+						if(!pConsole->m_aUser[0])
+							pPrompt = "Enter Username> ";
+						else
+							pPrompt = "Enter Password> ";
+					}
+					else
+						pPrompt = "ENTER PASSWORD> ";
+				}
 			}
 			else
 				pPrompt = "NOT CONNECTED> ";
@@ -495,7 +522,7 @@ void CGameConsole::OnRender()
 		//hide rcon password
 		char aInputString[512];
 		str_copy(aInputString, pConsole->m_Input.GetString(Editing), sizeof(aInputString));
-		if(m_ConsoleType == CONSOLETYPE_REMOTE && Client()->State() == IClient::STATE_ONLINE && !Client()->RconAuthed())
+		if(m_ConsoleType == CONSOLETYPE_REMOTE && Client()->State() == IClient::STATE_ONLINE && !Client()->RconAuthed() && (pConsole->m_aUser[0] || !IsDDNet(&pServerInfo)))
 		{
 			for(int i = 0; i < pConsole->m_Input.GetLength(Editing); ++i)
 				aInputString[i] = '*';
diff --git a/src/game/client/components/console.h b/src/game/client/components/console.h
index 05ffb70f4..40863752a 100644
--- a/src/game/client/components/console.h
+++ b/src/game/client/components/console.h
@@ -43,6 +43,8 @@ class CGameConsole : public CComponent
 		float m_CompletionRenderOffset;
 		bool m_ReverseTAB;
 
+		char m_aUser[32];
+
 		bool m_IsCommand;
 		char m_aCommandName[IConsole::TEMPCMD_NAME_LENGTH];
 		char m_aCommandHelp[IConsole::TEMPCMD_HELP_LENGTH];