TrackManiaControl/application/core/Maps/MapManager.php

<?php

namespace ManiaControl\Maps;

require_once __DIR__ . '/Map.php';
require_once __DIR__ . '/MapCommands.php';

use ManiaControl\FileUtil;
use ManiaControl\ManiaControl;
use ManiaControl\Callbacks\CallbackListener;
use ManiaControl\Callbacks\CallbackManager;


/**
 * Manager for maps
 *
 * @author kremsy & steeffeen
 */
class MapManager implements CallbackListener {
	/**
	 * Constants
	 */
	const TABLE_MAPS = 'mc_maps';
	
	/**
	 * Private properties
	 */
	private $maniaControl = null;
	private $mapCommands = null;
	private $mapList = array();
	private $mapListUids = array();
	private $currentMap = null;

	/**
	 * Construct map manager
	 *
	 * @param \ManiaControl\ManiaControl $maniaControl        	
	 */
	public function __construct(ManiaControl $maniaControl) {
		$this->maniaControl = $maniaControl;
		
		// Init database tables
		$this->initTables();
		
		// Create map commands instance
		$this->mapCommands = new MapCommands($maniaControl);
		
		// Register for callbacks
		$this->maniaControl->callbackManager->registerCallbackListener(CallbackManager::CB_MC_ONINIT, $this, 'handleOnInit');
		$this->maniaControl->callbackManager->registerCallbackListener(CallbackManager::CB_MC_BEGINMAP, $this, 'handleBeginMap');
	}

	/**
	 * Initialize necessary database tables
	 *
	 * @return bool
	 */
	private function initTables() {
		$mysqli = $this->maniaControl->database->mysqli;
		$query = "CREATE TABLE IF NOT EXISTS `" . self::TABLE_MAPS . "` (
				`index` int(11) NOT NULL AUTO_INCREMENT,
				`uid` varchar(50) COLLATE utf8_unicode_ci NOT NULL,
				`name` varchar(150) COLLATE utf8_unicode_ci NOT NULL,
				`authorLogin` varchar(100) COLLATE utf8_unicode_ci NOT NULL,
				`fileName` varchar(100) COLLATE utf8_unicode_ci NOT NULL,
				`environment` varchar(50) COLLATE utf8_unicode_ci NOT NULL,
				`mapType` varchar(50) COLLATE utf8_unicode_ci NOT NULL,
				`changed` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
				PRIMARY KEY (`index`),
				UNIQUE KEY `uid` (`uid`)
				) ENGINE=MyISAM DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci COMMENT='Map data' AUTO_INCREMENT=1;";
		$result = $mysqli->query($query);
		if ($mysqli->error) {
			trigger_error($mysqli->error, E_USER_ERROR);
			return false;
		}
		return $result;
	}

	/**
	 * Save map to the database
	 *
	 * @param \ManiaControl\Maps\Map $map        	
	 * @return boolean
	 */
	private function saveMap(Map &$map) {
		$mysqli = $this->maniaControl->database->mysqli;
		$mapQuery = "INSERT INTO `" . self::TABLE_MAPS . "` (
				`uid`,
				`name`,
				`authorLogin`,
				`fileName`,
				`environment`,
				`mapType`
				) VALUES (
				?, ?, ?, ?, ?, ?
				) ON DUPLICATE KEY UPDATE
				`index` = LAST_INSERT_ID(`index`);";
		$mapStatement = $mysqli->prepare($mapQuery);
		if ($mysqli->error) {
			trigger_error($mysqli->error);
			return false;
		}
		$mapStatement->bind_param('ssssss', $map->uid, $map->name, $map->authorLogin, $map->fileName, $map->environment, $map->mapType);
		$mapStatement->execute();
		if ($mapStatement->error) {
			trigger_error($mapStatement->error);
			$mapStatement->close();
			return false;
		}
		$map->index = $mapStatement->insert_id;
		$mapStatement->close();
		return true;
	}

	/**
	 * Add a map to the MapList
	 *
	 * @param \ManiaControl\Maps\Map $map        	
	 * @return bool
	 */
	private function addMap(Map $map) { //TODO needed?
		$this->saveMap($map);
		$this->mapListUids[$map->uid] = $map;
		$this->mapList[] = $map;
		return true;
	}

	/**
	 * Updates the full Map list, needed on Init, addMap and on ShuffleMaps
	 * @return null
	 */
	private function updateFullMapList(){
		if(!$this->maniaControl->client->query('GetMapList', 100,0)){ //fetch 100 Maps
			trigger_error("Couldn't fetch mapList. " . $this->maniaControl->getClientErrorText());
			return null;
		}

		$tempList = array();

		$mapList = $this->maniaControl->client->getResponse();
		foreach($mapList as $rpcMap){
			if(array_key_exists($rpcMap["UId"], $this->mapListUids)){ //Map already exists, only update index
				$tempList[] = $this->mapListUids[$rpcMap["UId"]];
			}else{ //Insert Map Object
				$map = new Map($this->maniaControl, $rpcMap);
				$this->saveMap($map);
				$tempList[] = $map;
				$this->mapListUids[$map->uid] = $map;
			}
		}

		//restore Sorted Maplist
		$this->mapList = $tempList;
	}

	/**
	 * Fetch current map
	 *
	 * @return \ManiaControl\Maps\Map
	 */
	private function fetchCurrentMapInfo() {
		if (!$this->maniaControl->client->query('GetCurrentMapInfo')) {
			trigger_error("Couldn't fetch map info. " . $this->maniaControl->getClientErrorText());
			return null;
		}
		$rpcMap = $this->maniaControl->client->getResponse();
		if(!array_key_exists($rpcMap["UId"], $this->mapListUids)){
			$map = new Map($this->maniaControl, $rpcMap);
			$this->addMap($map);
			return $map;
		}
		return $this->mapListUids[$rpcMap["UId"]];
	}

	/**
	 * Handle OnInit callback
	 *
	 * @param array $callback        	
	 */
	public function handleOnInit(array $callback) {
		$this->updateFullMapList();
		$this->currentMap = $this->fetchCurrentMapInfo();
	}

	/**
	 * @return null
	 */
	public function getCurrentMap(){
		return $this->currentMap;
	}



	/**
	 * Handle BeginMap callback
	 *
	 * @param array $callback        	
	 */
	public function handleBeginMap(array $callback) {
		if(array_key_exists($callback[1][0]["UId"], $this->mapListUids)){ //Map already exists, only update index
			$this->currentMap = $this->mapListUids[$callback[1][0]["UId"]];
		}else{ //can this ever happen?
			$this->currentMap = $this->fetchCurrentMapInfo();
		}
	}



	/**
	 * @return array
	 */
	public function getMapList(){
		return  $this->mapList;
	}

	/**
	 * Adds a Map from Mania Exchange
	 * @param        $mapId
	 * @param 		$login
	 */
	public function addMapFromMx($mapId, $login){
		// Check if ManiaControl can even write to the maps dir
		if (!$this->maniaControl->client->query('GetMapsDirectory')) {
			trigger_error("Couldn't get map directory. " . $this->maniaControl->getClientErrorText());
			$this->maniaControl->chat->sendError("ManiaControl couldn't retrieve the maps directory.", $login);
			return;
		}

		$mapDir = $this->maniaControl->client->getResponse();
		if (!is_dir($mapDir)) {
			trigger_error("ManiaControl doesn't have have access to the maps directory in '{$mapDir}'.");
			$this->maniaControl->chat->sendError("ManiaControl doesn't have access to the maps directory.", $login);
			return;
		}
		$downloadDirectory = $this->maniaControl->settingManager->getSetting($this, 'MapDownloadDirectory', 'MX');
		// Create download directory if necessary
		if (!is_dir($mapDir . $downloadDirectory) && !mkdir($mapDir . $downloadDirectory)) {
			trigger_error("ManiaControl doesn't have to rights to save maps in '{$mapDir}{$downloadDirectory}'.");
			$this->maniaControl->chat->sendError("ManiaControl doesn't have the rights to save maps.", $login);
			return;
		}
		$mapDir .= $downloadDirectory . '/';

		// Download the map
		if (is_numeric($mapId)) {
			// Load from MX
			$serverInfo = $this->maniaControl->server->getSystemInfo();
			$title = strtolower(substr($serverInfo['TitleId'], 0, 2));
			// Check if map exists
			$url = "http://{$title}.mania-exchange.com/api/tracks/get_track_info/id/{$mapId}?format=json";
			$mapInfo = FileUtil::loadFile($url);
			if (!$mapInfo || strlen($mapInfo) <= 0) {
				// Invalid id
				$this->maniaControl->chat->sendError('Invalid MX-Id!', $login);
				return;
			}
			$mapInfo = json_decode($mapInfo, true);
			$url = "http://{$title}.mania-exchange.com/tracks/download/{$mapId}";
			$file = FileUtil::loadFile($url);
			if (!$file) {
				// Download error
				$this->maniaControl->chat->sendError('Download failed!', $login);
				return;
			}
			// Save map
			$fileName = $mapInfo['TrackID'] . '_' . $mapInfo['Name'] . '.Map.Gbx';
			$fileName = FileUtil::getClearedFileName($fileName);
			if (!file_put_contents($mapDir . $fileName, $file)) {
				// Save error
				$this->maniaControl->chat->sendError('Saving map failed!', $login);
				return;
			}
			// Check for valid map
			$mapFileName = $downloadDirectory . '/' . $fileName;
			if (!$this->maniaControl->client->query('CheckMapForCurrentServerParams', $mapFileName)) {
				trigger_error("Couldn't check if map is valid ('{$mapFileName}'). " . $this->maniaControl->getClientErrorText());
				$this->maniaControl->chat->sendError('Error checking map!', $login);
				return;
			}
			$response = $this->maniaControl->client->getResponse();
			if (!$response) {
				// Invalid map type
				$this->maniaControl->chat->sendError("Invalid map type.", $login);
				return;
			}
			// Add map to map list
			if (!$this->maniaControl->client->query('InsertMap', $mapFileName)) {
				$this->maniaControl->chat->sendError("Couldn't add map to match settings!", $login);
				return;
			}
			$this->maniaControl->chat->sendSuccess('Map $<' . $mapInfo['Name'] . '$> added!');

			$this->updateFullMapList();
		}
		// TODO: add local map by filename
	}
}