1HEADmaster

1 files changed, 392 insertions, 0 deletions

diff --git a/serverbrowser/DialogAddServer.cpp b/serverbrowser/DialogAddServer.cpp
new file mode 100644
index 0000000..cbba58e
--- /dev/null
+++ b/serverbrowser/DialogAddServer.cpp
@@ -0,0 +1,392 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//=============================================================================
+
+#include "pch_serverbrowser.h"
+
+using namespace vgui;
+
+//-----------------------------------------------------------------------------
+// Purpose: Constructor
+// Input  : *gameList - game list to add specified server to
+//-----------------------------------------------------------------------------
+CDialogAddServer::CDialogAddServer(vgui::Panel *parent, IGameList *gameList) : Frame(parent, "DialogAddServer")
+{
+	SetDeleteSelfOnClose(true);
+
+	m_pGameList = gameList;
+
+	SetTitle("#ServerBrowser_AddServersTitle", true);
+	SetSizeable( false );
+
+	m_pTabPanel = new PropertySheet(this, "GameTabs");
+	m_pTabPanel->SetTabWidth(72);
+
+	m_pDiscoveredGames = new ListPanel( this, "Servers" );
+
+	// Add the column headers
+	m_pDiscoveredGames->AddColumnHeader(0, "Password", "#ServerBrowser_Password", 16, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE);
+	m_pDiscoveredGames->AddColumnHeader(1, "Bots", "#ServerBrowser_Bots", 16, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE | ListPanel::COLUMN_HIDDEN);
+	m_pDiscoveredGames->AddColumnHeader(2, "Secure", "#ServerBrowser_Secure", 16, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE);
+
+	bool bGameSupportsReplay = GameSupportsReplay();
+
+	int nReplayWidth = 16;
+	if ( !bGameSupportsReplay )
+	{
+		nReplayWidth = 0;
+	}
+
+	m_pDiscoveredGames->AddColumnHeader(3, "Replay", "#ServerBrowser_Replay", nReplayWidth, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE);
+	m_pDiscoveredGames->AddColumnHeader(4, "Name", "#ServerBrowser_Servers", 20, ListPanel::COLUMN_RESIZEWITHWINDOW | ListPanel::COLUMN_UNHIDABLE);
+	m_pDiscoveredGames->AddColumnHeader(5, "IPAddr", "#ServerBrowser_IPAddress", 60, ListPanel::COLUMN_HIDDEN);
+	m_pDiscoveredGames->AddColumnHeader(6, "GameDesc", "#ServerBrowser_Game", 150);
+	m_pDiscoveredGames->AddColumnHeader(7, "Players", "#ServerBrowser_Players", 60);
+	m_pDiscoveredGames->AddColumnHeader(8, "Map", "#ServerBrowser_Map", 80);
+	m_pDiscoveredGames->AddColumnHeader(9, "Ping", "#ServerBrowser_Latency", 60);
+
+	m_pDiscoveredGames->SetColumnHeaderTooltip(0, "#ServerBrowser_PasswordColumn_Tooltip");
+	m_pDiscoveredGames->SetColumnHeaderTooltip(1, "#ServerBrowser_BotColumn_Tooltip");
+	m_pDiscoveredGames->SetColumnHeaderTooltip(2, "#ServerBrowser_SecureColumn_Tooltip");
+
+	if ( bGameSupportsReplay )
+	{
+		m_pDiscoveredGames->SetColumnHeaderTooltip(3, "#ServerBrowser_ReplayColumn_Tooltip");
+	}
+
+	// setup fast sort functions
+	m_pDiscoveredGames->SetSortFunc(0, PasswordCompare);
+	m_pDiscoveredGames->SetSortFunc(1, BotsCompare);
+	m_pDiscoveredGames->SetSortFunc(2, SecureCompare);
+
+	if ( bGameSupportsReplay )
+	{
+		m_pDiscoveredGames->SetSortFunc(3, ReplayCompare);
+	}
+
+	m_pDiscoveredGames->SetSortFunc(4, ServerNameCompare);
+	m_pDiscoveredGames->SetSortFunc(5, IPAddressCompare);
+	m_pDiscoveredGames->SetSortFunc(6, GameCompare);
+	m_pDiscoveredGames->SetSortFunc(7, PlayersCompare);
+	m_pDiscoveredGames->SetSortFunc(8, MapCompare);
+	m_pDiscoveredGames->SetSortFunc(9, PingCompare);
+
+	m_pDiscoveredGames->SetSortColumn(9); // sort on ping
+
+	m_pTextEntry = new vgui::TextEntry( this, "ServerNameText" );
+	m_pTextEntry->AddActionSignalTarget( this );
+	
+	m_pTestServersButton = new vgui::Button( this, "TestServersButton", "" );
+	m_pAddServerButton = new vgui::Button( this, "OKButton", "" );
+	m_pAddSelectedServerButton = new vgui::Button( this, "SelectedOKButton", "", this, "addselected" );
+
+	m_pTabPanel->AddPage( m_pDiscoveredGames, "#ServerBrowser_Servers" );
+		  
+	LoadControlSettings("Servers/DialogAddServer.res");
+
+	// Setup the buttons. We leave them disabled until there is text in the textbox.
+	m_pAddServerButton->SetEnabled( false );
+	m_pTestServersButton->SetEnabled( false );
+	m_pAddSelectedServerButton->SetEnabled( false );
+	m_pAddSelectedServerButton->SetVisible( false );
+	m_pTabPanel->SetVisible( false );
+
+	m_pTextEntry->RequestFocus();
+
+	// Initially, we aren't high enough to show the tab panel.
+	int x, y;
+	m_pTabPanel->GetPos( x, y );
+	m_OriginalHeight = m_pTabPanel->GetTall() + y + 50;
+	SetTall( y );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: Destructor
+//-----------------------------------------------------------------------------
+CDialogAddServer::~CDialogAddServer()
+{
+	FOR_EACH_VEC( m_Queries, i )
+	{
+		if ( steamapicontext->SteamMatchmakingServers() )
+			steamapicontext->SteamMatchmakingServers()->CancelServerQuery( m_Queries[ i ] );
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Lets us know when the text entry has changed.
+//-----------------------------------------------------------------------------
+void CDialogAddServer::OnTextChanged()
+{
+	bool bAnyText = (m_pTextEntry->GetTextLength() > 0);
+	m_pAddServerButton->SetEnabled( bAnyText );
+	m_pTestServersButton->SetEnabled( bAnyText );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: button command handler
+//-----------------------------------------------------------------------------
+void CDialogAddServer::OnCommand(const char *command)
+{
+	if ( Q_stricmp(command, "OK") == 0 )
+	{
+		OnOK();
+	}
+	else if ( Q_stricmp( command, "TestServers" ) == 0 )
+	{
+		SetTall( m_OriginalHeight );
+		m_pTabPanel->SetVisible( true );
+		m_pAddSelectedServerButton->SetVisible( true );
+	
+		TestServers();
+	}
+	else if ( !Q_stricmp( command, "addselected" ) )
+	{
+		if ( m_pDiscoveredGames->GetSelectedItemsCount() )
+		{
+			// get the server
+			int serverID = m_pDiscoveredGames->GetItemUserData( m_pDiscoveredGames->GetSelectedItem(0) );
+			FinishAddServer( m_Servers[ serverID ] );
+			m_pDiscoveredGames->RemoveItem( m_pDiscoveredGames->GetSelectedItem(0) ); // as we add to favs remove from the list
+			m_pDiscoveredGames->SetEmptyListText( "" );
+		}
+	}
+	else
+	{
+		BaseClass::OnCommand(command);
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Handles the OK button being pressed; adds the server to the game list
+//-----------------------------------------------------------------------------
+void CDialogAddServer::OnOK()
+{
+	// try and parse out IP address
+	const char *address = GetControlString("ServerNameText", "");
+	netadr_t netaddr;
+	netaddr.SetFromString( address, true );
+	if ( !netaddr.GetPort() && !AllowInvalidIPs() )
+	{
+		// use the default port since it was not entered
+		netaddr.SetPort( 27015 );
+	}
+
+	if ( AllowInvalidIPs() || netaddr.IsValid() )
+	{
+		gameserveritem_t server;
+		memset(&server, 0, sizeof(server));
+		server.SetName( address );
+
+		// We assume here that the query and connection ports are the same. This is why it's much
+		// better if they click "Servers" and choose a server in there.
+		server.m_NetAdr.Init( netaddr.GetIPHostByteOrder(), netaddr.GetPort(), netaddr.GetPort() );
+
+		server.m_nAppID = 0;
+		FinishAddServer( server );
+	}
+	else
+	{
+		// could not parse the ip address, popup an error
+		MessageBox *dlg = new MessageBox("#ServerBrowser_AddServerErrorTitle", "#ServerBrowser_AddServerError");
+		dlg->DoModal();
+	}
+
+	// mark ourselves to be closed
+	PostMessage(this, new KeyValues("Close"));
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: Ping a particular IP for server presence
+//-----------------------------------------------------------------------------
+void CDialogAddServer::TestServers()
+{
+	if ( !steamapicontext->SteamMatchmakingServers() )
+		return;
+
+	m_pDiscoveredGames->SetEmptyListText( "" );
+	m_pDiscoveredGames->RemoveAll();
+
+	// If they specified a port, then send a query to that port.
+	const char *address = GetControlString("ServerNameText", "");
+	netadr_t netaddr;
+	netaddr.SetFromString( address, true );
+	
+	m_Servers.RemoveAll();
+	CUtlVector<netadr_t> vecAdress;
+
+	if ( netaddr.GetPort() == 0 )
+	{
+		// No port specified. Go to town on the ports.
+		CUtlVector<uint16> portsToTry;
+		GetMostCommonQueryPorts( portsToTry );
+		
+		for ( int i=0; i < portsToTry.Count(); i++ )
+		{
+			netadr_t newAddr = netaddr;
+			newAddr.SetPort( portsToTry[i] );
+			vecAdress.AddToTail( newAddr );
+		} 
+	}
+	else
+	{
+		vecAdress.AddToTail( netaddr );
+	}
+
+	// Change the text on the tab panel..
+	m_pTabPanel->RemoveAllPages();
+		
+	wchar_t wstr[512];
+	if ( address[0] == 0 )
+	{
+		Q_wcsncpy( wstr, g_pVGuiLocalize->Find( "#ServerBrowser_ServersRespondingLocal"), sizeof( wstr ) );
+	}
+	else
+	{
+		wchar_t waddress[512];
+		Q_UTF8ToUnicode( address, waddress, sizeof( waddress ) );
+		g_pVGuiLocalize->ConstructString( wstr, sizeof( wstr ), g_pVGuiLocalize->Find( "#ServerBrowser_ServersResponding"), 1, waddress );
+	}
+		
+	char str[512];
+	Q_UnicodeToUTF8( wstr, str, sizeof( str ) );
+	m_pTabPanel->AddPage( m_pDiscoveredGames, str );
+	m_pTabPanel->InvalidateLayout();
+	
+	FOR_EACH_VEC( vecAdress, iAddress )
+	{
+		m_Queries.AddToTail( steamapicontext->SteamMatchmakingServers()->PingServer( vecAdress[ iAddress ].GetIPHostByteOrder(), vecAdress[ iAddress ].GetPort(), this ) );
+	}
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: A server answered our ping
+//-----------------------------------------------------------------------------
+void CDialogAddServer::ServerResponded( gameserveritem_t &server )
+{
+	KeyValues *kv = new KeyValues( "Server" );
+
+	kv->SetString( "name", server.GetName() );
+	kv->SetString( "map", server.m_szMap );
+	kv->SetString( "GameDir", server.m_szGameDir );
+	kv->SetString( "GameDesc", server.m_szGameDescription );
+	kv->SetString( "GameTags", server.m_szGameTags );
+	kv->SetInt( "password", server.m_bPassword ? 1 : 0);
+	kv->SetInt( "bots", server.m_nBotPlayers ? 2 : 0);
+	kv->SetInt( "Replay", IsReplayServer( server ) ? 5 : 0 );
+
+	if ( server.m_bSecure )
+	{
+		// show the denied icon if banned from secure servers, the secure icon otherwise
+		kv->SetInt("secure", ServerBrowser().IsVACBannedFromGame( server.m_nAppID ) ?  4 : 3);
+	}
+	else
+	{
+		kv->SetInt("secure", 0);
+	}
+
+	netadr_t reportedIPAddr;
+	reportedIPAddr.SetIP( server.m_NetAdr.GetIP() );
+	reportedIPAddr.SetPort( server.m_NetAdr.GetConnectionPort() );
+	kv->SetString("IPAddr", reportedIPAddr.ToString() );
+
+	char buf[32];
+	Q_snprintf(buf, sizeof(buf), "%d / %d", server.m_nPlayers, server.m_nMaxPlayers);
+	kv->SetString("Players", buf);
+
+	kv->SetInt("Ping", server.m_nPing);
+
+	// new server, add to list
+	int iServer = m_Servers.AddToTail( server );
+	int iListID = m_pDiscoveredGames->AddItem(kv, iServer, false, false);
+	if ( m_pDiscoveredGames->GetItemCount() == 1 )
+	{
+		m_pDiscoveredGames->AddSelectedItem( iListID );
+	}
+	kv->deleteThis();
+
+	m_pDiscoveredGames->InvalidateLayout();
+}
+
+void CDialogAddServer::ServerFailedToRespond()
+{
+	m_pDiscoveredGames->SetEmptyListText( "#ServerBrowser_ServerNotResponding" );
+}
+
+void CDialogAddServer::ApplySchemeSettings( IScheme *pScheme )
+{
+	BaseClass::ApplySchemeSettings( pScheme );
+
+	ImageList *imageList = new ImageList(false);
+	imageList->AddImage(scheme()->GetImage("servers/icon_password", false));
+	imageList->AddImage(scheme()->GetImage("servers/icon_bots", false));
+	imageList->AddImage(scheme()->GetImage("servers/icon_robotron", false));
+	imageList->AddImage(scheme()->GetImage("servers/icon_secure_deny", false));
+	imageList->AddImage(scheme()->GetImage("servers/icon_replay", false));
+
+	int passwordColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_password_column", false));
+	int botColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_bots_column", false));
+	int secureColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_robotron_column", false));
+	int replayColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_replay_column", false));
+
+	m_pDiscoveredGames->SetImageList(imageList, true);
+	vgui::HFont hFont = pScheme->GetFont( "ListSmall", IsProportional() );
+	if ( !hFont )
+		hFont = pScheme->GetFont( "DefaultSmall", IsProportional() );
+
+	m_pDiscoveredGames->SetFont( hFont );
+	m_pDiscoveredGames->SetColumnHeaderImage(0, passwordColumnImage);
+	m_pDiscoveredGames->SetColumnHeaderImage(1, botColumnImage);
+	m_pDiscoveredGames->SetColumnHeaderImage(2, secureColumnImage);
+	m_pDiscoveredGames->SetColumnHeaderImage(3, replayColumnImage);
+}
+
+
+//-----------------------------------------------------------------------------
+// Purpose: A server on the listed IP responded
+//-----------------------------------------------------------------------------
+void CDialogAddServer::OnItemSelected()
+{
+	int nSelectedItem = m_pDiscoveredGames->GetSelectedItem(0);
+	if( nSelectedItem != -1 ) 
+	{
+		m_pAddSelectedServerButton->SetEnabled( true );
+	}
+	else
+	{
+		m_pAddSelectedServerButton->SetEnabled( false );
+	}
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CDialogAddServer::FinishAddServer( gameserveritem_t &pServer )
+{
+	ServerBrowserDialog().AddServerToFavorites( pServer );
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CDialogAddBlacklistedServer::FinishAddServer( gameserveritem_t &pServer )
+{
+	ServerBrowserDialog().AddServerToBlacklist( pServer );
+	ServerBrowserDialog().BlacklistsChanged();
+}
+
+//-----------------------------------------------------------------------------
+// Purpose: 
+//-----------------------------------------------------------------------------
+void CDialogAddBlacklistedServer::ApplySchemeSettings( vgui::IScheme *pScheme )
+{
+	BaseClass::ApplySchemeSettings( pScheme );
+
+	m_pAddServerButton->SetText( "#ServerBrowser_AddAddressToBlacklist" );
+	m_pAddSelectedServerButton->SetText( "#ServerBrowser_AddSelectedToBlacklist" );
+}