/***
    This file is part of snapcast
    Copyright (C) 2014-2018  Johannes Pohl

    This program 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.

    This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
***/

#ifndef CONTROL_SERVER_H
#define CONTROL_SERVER_H

#include <asio.hpp>
#include <vector>
#include <thread>
#include <memory>
#include <set>
#include <sstream>
#include <mutex>

#include "controlSession.h"
#include "common/queue.h"
#include "common/sampleFormat.h"
#include "message/message.h"
#include "message/codecHeader.h"
#include "message/serverSettings.h"


using asio::ip::tcp;
typedef std::shared_ptr<tcp::socket> socket_ptr;


/// Telnet like remote control
/**
 * Telnet like remote control
 */
class ControlServer : public ControlMessageReceiver
{
public:
	ControlServer(asio::io_service* io_service, size_t port, ControlMessageReceiver* controlMessageReceiver = NULL);
	virtual ~ControlServer();

	void start();
	void stop();

	/// Send a message to all connceted clients
	void send(const std::string& message, const ControlSession* excludeSession = NULL);

	/// Clients call this when they receive a message. Implementation of MessageReceiver::onMessageReceived
	virtual void onMessageReceived(ControlSession* connection, const std::string& message);

private:
	void startAccept();
	void handleAccept(socket_ptr socket);
	void cleanup();
//	void acceptor();
	mutable std::recursive_mutex mutex_;
	std::set<std::shared_ptr<ControlSession>> sessions_;
	std::shared_ptr<tcp::acceptor> acceptor_v4_;
	std::shared_ptr<tcp::acceptor> acceptor_v6_;

	Queue<std::shared_ptr<msg::BaseMessage>> messages_;
	asio::io_service* io_service_;
	size_t port_;
	ControlMessageReceiver* controlMessageReceiver_;
};



#endif