alanedwardes / statistics_collector.cpp

Created October 06, 2012


    
    View statistics_collector.cpp 
  
  
      statistics_collector.cpp
    
    
        Raw
      
    
    File suppressed. Click to show.
  
  
            1
            2
            3
            4
            5
            6
            7
            8
            9
            10
            11
            12
            13
            14
            15
            16
            17
            18
            19
            20
            21
            22
            23
            24
            25
            26
            27
            28
            29
            30
            31
            32
            33
            34
            35
            36
            37
            38
            39
            40
            41
            42
            43
            44
            45
            46
            47
            48
            49
            50
            51
            52
            53
            54
            55
            56
            57
            58
            59
            60
            61
            62
            63
            64
            65
            66
            67
            68
            69
            70
            71
            72
            73
            74
            75
            76
            77
            78
            79
            80
            81
            82
            83
            84
            85
            86
            87
            88
            89
            90
            91
            92
            93
            94
            95
            96
            97
            98
            99
            100
            101
            102
            103
            104
            105
            106
            107
            108
            109
            110
            111
            112
            113
            114
            115
            116
            117
            118
            119
            120
            121
            122
            123
            124
            125
            126
            127
            128
            129
            130
          
          
            //===== Copyright © 1996-2005, Valve Corporation, All rights reserved. ========
//
// Purpose: Simple entity to log and send gameplay statistics
//
//=============================================================================
 
#include "cbase.h"
#include "vstdlib/jobthread.h"
#include "curl/curl.h"
#include "steam\steam_api.h"
#include "view_shared.h"
#include <time.h>
#include "GameEventListener.h"
 
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
 
#define ENDPOINT "http://example.com/collector.php?collect"
 
ConVar stats_enabled( "stats_enabled", "-1", FCVAR_ARCHIVE);
 
class CStatisticsCollector : public CLogicalEntity, public CGameEventListener
{
	public:
		DECLARE_CLASS( CStatisticsCollector, CLogicalEntity );
		DECLARE_DATADESC();
 
		CStatisticsCollector()
		{
			ListenForGameEvent("player_hurt");
		}
 
		void FireGameEvent( IGameEvent* e );
 
		void Init();
		void GameEvent();
		void SendStatistic( const tchar *m_event );
		void InputSendStatistic( inputdata_t &inputData );
		void cURLThread(const tchar *m_event);
};
 
LINK_ENTITY_TO_CLASS( statistics_collector, CStatisticsCollector  );
 
BEGIN_DATADESC( CStatisticsCollector  )
	DEFINE_INPUTFUNC( FIELD_STRING, "Event", InputSendStatistic )
END_DATADESC()
 
void CStatisticsCollector::FireGameEvent( IGameEvent* e )
{
	CBasePlayer *player = UTIL_GetLocalPlayer();
	if ( Q_strcmp( "player_hurt", e->GetName() ) && player->GetHealth() == 0 )
	{
		// The player has unfortunately passed away
		this->SendStatistic("player_death");
	}
}
 
void CStatisticsCollector::cURLThread(const char *m_event)
{
	DevMsg("Statistics Collectior: Sending event '%s'\n", m_event);
 
	CBasePlayer *player = UTIL_GetLocalPlayer();
	const Vector player_pos = player->GetAbsOrigin();
	uint32 steam_id;
 
	ISteamUser* steam_user = steamapicontext->SteamUser();
	if(steam_user->BLoggedOn())
	{
		steam_id = steam_user->GetSteamID().GetAccountID();
	}
	else
	{
		// Just in case someone isn't logged in. Yarr, pirate Steam
		steam_id = 0;
	}
 
	time_t aclock;
	time( &aclock );
	struct tm *newtime = localtime( &aclock );
 
	char timeString[128];
	Q_strncpy( timeString, asctime( newtime ), sizeof( timeString ) );
	// Get rid of the \n.
	char *pEnd = strstr( timeString, "\n" );
	if ( pEnd )
	{
		*pEnd = 0;
	}
 
	char full_url[512];
	Q_snprintf(full_url, 512, "%s&e=%s&s=%i&p=%.2f,%.2f,%.2f&h=%i&d=%s&m=%s",
		ENDPOINT, m_event, steam_id, player_pos.x, player_pos.y, player_pos.z, player->GetHealth(), timeString, gpGlobals->mapname.ToCStr()
	);
 
	CBaseCombatWeapon *weapon = player->GetActiveWeapon();
	if(weapon)
	{
		Q_snprintf(full_url, 512, "%s&w=%s", full_url, weapon->GetClassname());
	}
 
	DevMsg("Statistics Collectior: GET %s\n", full_url);
 
	CURL *curl;
	curl = curl_easy_init();
	curl_easy_setopt(curl, CURLOPT_URL, full_url);
	curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 10);
	curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10);
	curl_easy_perform(curl);
	curl_easy_cleanup(curl);
 
	DevMsg("Statistics Collectior: Event sent.\n");
}
 
void CStatisticsCollector::SendStatistic( const char *m_event )
{
	if(stats_enabled.GetInt() > 0)
	{
		DevMsg("Statistics Collectior: cURL thread queued.\n");
		ThreadExecute(this, &CStatisticsCollector::cURLThread, m_event);
	}
	else
	{
		DevMsg("Statistics Collectior: Events disabled, ignoring.\n");
	}
}
 
void CStatisticsCollector::InputSendStatistic( inputdata_t &inputData )
{
	this->SendStatistic(inputData.value.String());
}