/*
 * ET: Legacy
 * Copyright (C) 2012 Jan Simek <mail@etlegacy.com>
 *
 * This file is part of ET: Legacy - http://www.etlegacy.com
 *
 * ET: Legacy is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ET: Legacy is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with ET: Legacy. If not, see <http://www.gnu.org/licenses/>.
 */

#include <iostream>
#include <string>

#include <boost/asio.hpp>
#include <boost/program_options.hpp>
#include <boost/lexical_cast.hpp>

#include "connection.h"

using boost::asio::ip::udp;

int main(int argc, char *argv[])
{
	try
	{
		/**
		 * Program options
		 */
		boost::program_options::options_description desc("[OPTIONS]");
		desc.add_options()
		    ("help,h", "show this message")
		    ("server,s", boost::program_options::value<std::string>(),
		    "server to query")
		    ("port,p",
		    boost::program_options::value<unsigned int>()->default_value(27960),
		    "port on the server")
		    ("message,m",
		    boost::program_options::value<std::string>()->default_value("getstatus"),
		    "message to be sent")
		    ("raw,r", "don't parse the server response")
		;
		boost::program_options::variables_map var_map;
		boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), var_map);
		boost::program_options::notify(var_map);

		if (var_map.count("help"))
		{
			std::cout << desc << std::endl;
			return EXIT_SUCCESS;
		}
		else if (!var_map.count("server"))
		{
			std::cerr << "Usage: " << argv[0] << " -s <server>" << std::endl;
			std::cerr << "Append -h or --help to see all program options." << std::endl;
			return 1;
		}

		/**
		 * Send the request
		 */
		Connection client(var_map["server"].as<std::string>(),
		                  var_map["port"].as<unsigned int>(),
		                  var_map["message"].as<std::string>());

		char                      data[1024];
		boost::system::error_code ec;
		std::size_t               n = client.ReceiveMessage(boost::asio::buffer(data),
		                                                    boost::posix_time::seconds(10),
		                                                    ec);

		if (ec)
		{
			std::cout << "Receive error: " << ec.message() << std::endl;
		}
		else
		{
			if (var_map.count("raw"))
			{
				std::cout.write(data, n);
			}
			else
			{
				client.ParseMessage(std::string(data, n));
			}
		}
	}
	catch (std::exception& e)
	{
		std::cerr << "Exception caught: " << e.what() << std::endl;
//         return 1;
	}
	catch (...)
	{
		std::cerr << "Exception of unknown type!" << std::endl;
//         return 1;
	}

	return EXIT_SUCCESS;
}