TrackManiaControl/application/core/Maps/MapCommands.php

<?php

namespace ManiaControl\Maps;

use ManiaControl\Authentication;
use ManiaControl\ManiaControl;
use ManiaControl\Commands\CommandListener;
use ManiaControl\Players\Player;

/**
 * Class offering commands to manage maps
 *
 * @author steeffeen & kremsy
 */
class MapCommands implements CommandListener {
	/**
	 * Private properties
	 */
	private $maniaControl = null;

	/**
	 * Create MapCommands instance
	 *
	 * @param \ManiaControl\ManiaControl $maniaControl        	
	 */
	public function __construct(ManiaControl $maniaControl) {
		$this->maniaControl = $maniaControl;
		
		$this->maniaControl->commandManager->registerCommandListener('nextmap', $this, 'command_NextMap');
		$this->maniaControl->commandManager->registerCommandListener('restartmap', $this, 'command_RestartMap');
		$this->maniaControl->commandManager->registerCommandListener('addmap', $this, 'command_AddMap');
		$this->maniaControl->commandManager->registerCommandListener('removemap', $this, 'command_RemoveMap');
	}

	/**
	 * Handle removemap command
	 *
	 * @param array $chat        	
	 * @param \ManiaControl\Players\Player $player        	
	 * @return bool
	 */
	public function command_RemoveMap(array $chat, Player $player) {
		if (!$this->maniaControl->authentication->checkRight($player, Authentication::AUTH_LEVEL_OPERATOR)) {
			$this->maniaControl->authentication->sendNotAllowed($player);
			return false;
		}
		// Get map name
		$map = $this->maniaControl->server->getMap();
		if (!$map) {
			$this->maniaControl->chat->sendError("Couldn't remove map.", $player->login);
			return false;
		}
		$mapName = $map['FileName'];
		// Remove map
		if (!$this->maniaControl->client->query('RemoveMap', $mapName)) {
			trigger_error("Couldn't remove current map. " . $this->maniaControl->getClientErrorText());
			$this->maniaControl->chat->sendError("Couldn't remove map.", $player->login);
			return false;
		}
		$this->maniaControl->chat->sendSuccess('Map removed.', $player->login);
		return true;
	}

	/**
	 * Handle addmap command
	 *
	 * @param array $chat        	
	 * @param \ManiaControl\Players\Player $player        	
	 * @return bool
	 */
	public function command_AddMap(array $chat, Player $player) {
		if (!$this->maniaControl->authentication->checkRight($player, Authentication::AUTH_LEVEL_OPERATOR)) {
			$this->maniaControl->authentication->sendNotAllowed($player);
			return false;
		}
		$params = explode(' ', $chat[1][2], 2);
		if (count($params) < 2) {
			// TODO: show usage
			return false;
		}
		// 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.", $player->login);
			return false;
		}
		$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.", $player->login);
			return false;
		}
		$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.", $player->login);
			return false;
		}
		$mapDir .= $downloadDirectory . '/';
		// Download the map
		$mapId = $params[1];
		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!', $player->login);
				return false;
			}
			$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!', $player->login);
				return false;
			}
			// Save map
			$fileName = $mapDir . $mapInfo['TrackID'] . '_' . $mapInfo['Name'] . '.Map.Gbx';
			if (!file_put_contents($fileName, $file)) {
				// Save error
				$this->maniaControl->chat->sendError('Saving map failed!', $player->login);
				return false;
			}
			// Check for valid map
			if (!$this->maniaControl->client->query('CheckMapForCurrentServerParams', $fileName)) {
				trigger_error("Couldn't check if map is valid. " . $this->maniaControl->getClientErrorText());
				$this->maniaControl->chat->sendError('Error checking map!', $player->login);
				return false;
			}
			$response = $this->maniaControl->client->getResponse();
			if (!$response) {
				// Inalid map type
				$this->maniaControl->chat->sendError("Invalid map type.", $player->login);
				return false;
			}
			// Add map to map list
			if (!$this->maniaControl->client->query('InsertMap', $fileName)) {
				$this->maniaControl->chat->sendError("Couldn't add map to match settings!", $player->login);
				return false;
			}
			$this->maniaControl->chat->sendSuccess('Map $<' . $mapInfo['Name'] . '$> added!');
			return true;
		}
		// TODO: add local map by filename
	}

	/**
	 * Handle nextmap command
	 *
	 * @param array $chat        	
	 * @param \ManiaControl\Players\Player $player        	
	 * @return bool
	 */
	public function command_NextMap(array $chat, Player $player) {
		if (!$this->maniaControl->authentication->checkRight($player, Authentication::AUTH_LEVEL_OPERATOR)) {
			$this->maniaControl->authentication->sendNotAllowed($player);
			return false;
		}
		return $this->maniaControl->client->query('NextMap');
	}

	/**
	 * Handle retartmap command
	 *
	 * @param array $chat        	
	 * @param \ManiaControl\Players\Player $player        	
	 * @return bool
	 */
	public function command_RestartMap(array $chat, Player $player) {
		if (!$this->maniaControl->authentication->checkRight($player, Authentication::AUTH_LEVEL_OPERATOR)) {
			$this->maniaControl->authentication->sendNotAllowed($player);
			return false;
		}
		return $this->maniaControl->client->query('RestartMap');
	}
}

?>