/*
 * Irresistible Gaming (c) 2018
 * Developed by Steven Howard
 * Module: cnr/features/minijobs/traindriver.pwn
 * Purpose: traindriver minijob
 */

/* ** Includes ** */
#include 							< YSI\y_hooks >

/* ** Marcos ** */
#define IsVehicleTrain(%0) 			(GetVehicleModel(%0)==538)

/* ** Definitions ** */
#define INVALID_TRAIN_ROUTE 		( 0xFF )

/* ** Variables ** */
enum E_STATION_DATA
{
	E_NAME[ 32 ],
	Float: E_X, 				Float: E_Y, 				Float: E_Z
};

new
	g_StationData[  ][ E_STATION_DATA ] =
	{
		{ "Prickle Pine LV",				1474.3666, 2634.2493, 10.8203 },
		{ "Cranberry Station SF", 			-1944.0211, 180.4831, 25.7109 },
		{ "Market LS", 						816.3867, -1370.4550, -1.6781 },
		{ "El Corona LS", 					1689.9775, -1955.7910, 13.5469 },
		{ "Linden Station LV", 				2866.7095, 1329.3573, 10.8203 }
	},

	bool: p_hasTrainJob 					[ MAX_PLAYERS char ],

	p_TrainMapIcon							[ MAX_PLAYERS ] = { 0xFFFF, ... },
	p_TrainCheckPoint						[ MAX_PLAYERS ] = { 0xFFFF, ... },
	p_TrainPositionTimer					[ MAX_PLAYERS ] = { 0xFFFF, ... },
	p_TrainLoadTimer						[ MAX_PLAYERS ] = { 0xFFFF, ... },
	p_TrainCancelTimer						[ MAX_PLAYERS ] = { 0xFFFF, ... },

	p_TrainTimeElapsed 						[ MAX_PLAYERS ],
	Float: p_TrainDistance					[ MAX_PLAYERS ],
	p_TrainRoute			 				[ MAX_PLAYERS ] [ 2 char ]
;

/* ** Hooks ** */
hook OnScriptInit(  )
{
	AddStaticVehicleEx( 538, 1443.5094, 2636.3716, 10.8910, 270.9954, 1, 98, 800);
	AddStaticVehicleEx( 538, 1465.4567, 2632.7007, 10.7463, 269.1136, 1, 98, 800);
	AddStaticVehicleEx( 538, 785.0410, -1339.9872, -2.0049, 46.9541, 6, 0, 800);
	AddStaticVehicleEx( 538, 842.2191, -1397.2175, -2.0471, 227.3936, 6, 0, 800);
	AddStaticVehicleEx( 538, -1946.3875, 181.3526, 25.7186, 354.3933, 2, 2, 800);
}

hook OnPlayerDisconnect( playerid, reason )
{
	StopPlayerTrainWork( playerid );
	return 1;
}

#if defined AC_INCLUDED
hook OnPlayerDeathEx( playerid, killerid, reason, Float: damage, bodypart )
#else
hook OnPlayerDeath( playerid, killerid, reason )
#endif
{
	StopPlayerTrainWork( playerid );
	return 1;
}

hook OnPlayerUpdateEx( playerid )
{
	new player_vehicle = GetPlayerVehicleID( playerid );

	if ( GetPlayerState( playerid ) == PLAYER_STATE_DRIVER && player_vehicle && p_hasTrainJob{ playerid } && !IsVehicleTrain( player_vehicle ) && p_TrainCancelTimer[ playerid ] == 0xFFFF ) {
 		cancelPlayerTrainWork( playerid, player_vehicle, .ticks = 60 );
	}

	return 1;
}

hook OnPlayerStateChange( playerid, newstate, oldstate )
{
	if ( newstate != PLAYER_STATE_DRIVER && oldstate == PLAYER_STATE_DRIVER && p_hasTrainJob{ playerid } ) {
		cancelPlayerTrainWork( playerid, GetPlayerVehicleID( playerid ), .ticks = 0 );
    }

    else if ( newstate == PLAYER_STATE_DRIVER && oldstate != PLAYER_STATE_DRIVER && p_hasTrainJob{ playerid } ) {
    	KillTimer( p_TrainCancelTimer[ playerid ] );
    	p_TrainCancelTimer[ playerid ] = 0xFFFF;
    }

    else if ( newstate == PLAYER_STATE_DRIVER && IsPlayerInAnyVehicle(playerid) && !p_hasTrainJob{ playerid })
    {
    	if ( IsVehicleTrain( GetPlayerVehicleID( playerid ) ) )
    	{
    		ShowPlayerHelpDialog( playerid, 3000, "You can begin a train job by typing ~g~~h~/train" );
    	}
    }

    return 1;
}

hook OnPlayerEnterDynRaceCP( playerid, checkpointid )
{
	if ( checkpointid == p_TrainCheckPoint[ playerid ] )
	{
		DestroyDynamicRaceCP	( p_TrainCheckPoint[ playerid ] );
		DestroyDynamicMapIcon 	( p_TrainMapIcon[ playerid ] );

		KillTimer 				( p_TrainPositionTimer[ playerid ] );

		if ( p_TrainRoute[ playerid ] { 0 } != INVALID_TRAIN_ROUTE )
		{
			static aPlayer[ 1 ]; aPlayer[ 0 ] = playerid;

			p_TrainCancelTimer[ playerid ] = SetTimerEx( "OnTrainPositionUpdate", 750, false, "dd", playerid, p_TrainRoute[ playerid ] { 1 } );
	  		PlayerTextDrawShow( playerid, p_TruckingTD[ playerid ] );

	  		KillTimer( p_TrainLoadTimer[ playerid ] );
	  		p_TrainLoadTimer		[ playerid ] = 0xFFFF;
	  		p_TrainLoadTimer		[ playerid ] = SetTimerEx( "OnPilotLoadCargo", 5000, false, "d", playerid );

	  		TogglePlayerControllable( playerid, false );

	  		ShowPlayerHelpDialog( playerid, 5000, "Please wait while your passengers are getting inside your train!" );

	  		p_TrainMapIcon		[ playerid ] = CreateDynamicMapIconEx( g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_X ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_Y ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_Z ], 51, 0, MAPICON_GLOBAL, 6000.0, { -1 }, { -1 }, aPlayer );
			p_TrainCheckPoint	[ playerid ] = CreateDynamicRaceCP( 0, g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_X ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_Y ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_Z ], 15.0, 0.0, 0.0, 10.0, -1, -1, playerid );

			return ( p_TrainRoute[ playerid ] { 0 } = INVALID_TRAIN_ROUTE ), 1;
		}

		else if ( p_TrainRoute[ playerid ] { 1 } != INVALID_TRAIN_ROUTE )
		{
			new
				iCashEarned = floatround( p_TrainDistance[ playerid ] * 1.0 );

			GivePlayerScore( playerid, 1 + floatround( p_TrainDistance[ playerid ] / 1000.0 ) );
			GivePlayerCash( playerid, iCashEarned );

			ShowPlayerHelpDialog( playerid, 5000, "You have earned ~y~%s ~w~for transporting passengers!", cash_format( iCashEarned ) );
			StopPlayerTrainWork( playerid );

			return ( p_TrainRoute[ playerid ] { 1 } = INVALID_TRAIN_ROUTE ), 1;
		}

		return 1;
	}

	return 1;
}

/* ** Functions ** */
stock StopPlayerTrainWork( playerid )
{
	DestroyDynamicRaceCP	( p_TrainCheckPoint[ playerid ] );
	DestroyDynamicMapIcon 	( p_TrainMapIcon[ playerid ] );

	KillTimer 				( p_TrainCancelTimer[ playerid ] );
	KillTimer 				( p_TrainPositionTimer[ playerid ] );

	p_TrainDistance			[ playerid ] = 0.0;
	p_hasTrainJob			{ playerid } = false;
	p_TrainCheckPoint		[ playerid ] = 0xFFFF;
	p_TrainMapIcon 			[ playerid ] = 0xFFFF;
	p_TrainCancelTimer		[ playerid ] = 0xFFFF;
	p_TrainPositionTimer 	[ playerid ] = 0xFFFF;
	p_TrainRoute 			[ playerid ] { 0 } = INVALID_PILOT_ROUTE;
	p_TrainRoute 			[ playerid ] { 1 } = INVALID_PILOT_ROUTE;

	PlayerTextDrawHide( playerid, p_TruckingTD[ playerid ] );
}

function OnTrainPositionUpdate( playerid, routeid )
{
	if ( !IsPlayerInAnyVehicle( playerid ) && !p_hasTrainJob{ playerid } && ( p_TrainRoute[ playerid ] { 0 } == 0 && p_TrainRoute[ playerid ] { 1 } == 0 ) ) {
	  	PlayerTextDrawHide( playerid, p_TruckingTD[ playerid ] );
		return ( p_TrainPositionTimer[ playerid ] = 0xFFFF );
	}

	new 
		Float: fDistance = GetPlayerDistanceFromPoint( playerid, g_StationData[ routeid ][ E_X ] , g_StationData[ routeid ][ E_Y ] , g_StationData[ routeid ][ E_Z ] );

	PlayerTextDrawSetString( playerid, p_TruckingTD[ playerid ], sprintf( "~b~Location:~w~ %s~n~~b~Distance:~w~ %0.2fm", g_StationData[ routeid ] [ E_NAME ], fDistance ) );
	return ( p_TrainPositionTimer[ playerid ] = SetTimerEx( "OnTrainPositionUpdate", 750, false, "dd", playerid, routeid ) );
}

function cancelPlayerTrainWork( playerid, vehicleid, ticks )
{
	if ( IsVehicleTrain( vehicleid ) && ticks )
		return KillTimer( p_TrainCancelTimer[ playerid ] ), p_TrainCancelTimer[ playerid ] = 0xFFFF;

	if ( ticks < 1 || !IsPlayerConnected( playerid ) || !IsPlayerSpawned( playerid ) || ! p_hasTrainJob{ playerid } )
	{
		StopPlayerTrainWork( playerid );
		return SendServerMessage( playerid, "Your train mission has been stopped." );
	}
	else
	{
		ShowPlayerHelpDialog( playerid, 1000, "You have %d seconds to get back inside the train you were using.", ticks - 1 );
		p_TrainCancelTimer[ playerid ] = SetTimerEx( "cancelPlayerTrainWork", 980, false, "ddd", playerid, vehicleid, ticks - 1 );
	}

	return 1;
}

/* ** Commands ** */
CMD:train( playerid, params[ ] )
{
	new 
		iVehicle = GetPlayerVehicleID( playerid );

	if ( p_Class[ playerid ] != CLASS_CIVILIAN ) return SendError( playerid, "You must be an ordinary civilian to use this command." );
	else if ( GetPlayerState( playerid ) != PLAYER_STATE_DRIVER ) return SendError( playerid, "You must be a driver of a vehicle to work." );
	else if ( !iVehicle ) return SendError( playerid, "You are not in any vehicle." );
	else if ( strmatch( params, "stop" ))
	{
		StopPlayerTrainWork( playerid );
		return SendServerMessage( playerid, "Your train mission has been stopped." );
	}
	else if ( !strmatch( params, "stop" ) )
	{
		if ( IsVehicleTrain( iVehicle ) )
		{
			if ( ! p_hasTrainJob{ playerid } )
			{
				if ( p_WorkCooldown[ playerid ] > g_iTime )
					return SendError( playerid, "You must wait %d seconds before working again.", p_WorkCooldown[ playerid ] - g_iTime );

				static aPlayer[ 1 ]; aPlayer[ 0 ] = playerid;
				DestroyDynamicMapIcon( p_TrainMapIcon[ playerid ] );

				p_hasTrainJob 			{ playerid } = true;
				p_WorkCooldown			[ playerid ] = g_iTime + 60;

				p_TrainRoute			[ playerid ] { 0 } = random ( sizeof( g_StationData ) );
				p_TrainRoute			[ playerid ] { 1 } = random ( sizeof( g_StationData ) );

				p_TrainTimeElapsed		[ playerid ] = g_iTime;
				p_TrainDistance			[ playerid ] = GetDistanceBetweenPoints( g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_X ], g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_Y ], g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_Z ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_X ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_Y ], g_StationData[ p_TrainRoute[ playerid ] { 1 } ] [ E_Z ] );
				p_TrainMapIcon			[ playerid ] = CreateDynamicMapIconEx( g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_X ], g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_Y ], g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_Z ], 51, 0, MAPICON_GLOBAL, 6000.0, { -1 }, { -1 }, aPlayer );
				p_TrainCheckPoint		[ playerid ] = CreateDynamicRaceCP( 0, g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_X ], g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_Y ], g_StationData[ p_TrainRoute[ playerid ] { 0 } ] [ E_Z ], 0.0, 0.0, 0.0, 10.0, -1, -1, playerid );

				p_PilotPositionTimer[ playerid ] = SetTimerEx( "OnTrainPositionUpdate", 750, false, "dd", playerid, p_TrainRoute[ playerid ] { 0 } );
	  			PlayerTextDrawShow( playerid, p_TruckingTD[ playerid ] );
			}
			else SendError( playerid, "You already have a train job started! Cancel it with "COL_GREY"/train stop"COL_WHITE"." );
		}
		else SendError( playerid, "There are currently no jobs for this particular vehicle." );

		return 1;
	}
	return 1;
}