/*
  This file is part of Ingen.
  Copyright 2007-2012 David Robillard <http://drobilla.net/>

  Ingen is free software: you can redistribute it and/or modify it under the
  terms of the GNU Affero General Public License as published by the Free
  Software Foundation, either version 3 of the License, or any later version.

  Ingen 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 Affero General Public License for details.

  You should have received a copy of the GNU Affero General Public License
  along with Ingen.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <cassert>
#include <cstdio>
#include <cstring>
#include <unistd.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <errno.h>
#include "raul/log.hpp"
#include "HTTPSender.hpp"

using namespace std;
using namespace Raul;

namespace Ingen {
namespace Shared {

HTTPSender::HTTPSender()
	: _listen_port(-1)
	, _listen_sock(-1)
	, _client_sock(-1)
	, _send_state(Immediate)
{
	Thread::set_name("HTTPSender");

	struct sockaddr_in addr;

	// Create listen address
	memset(&addr, 0, sizeof(addr));
	addr.sin_family      = AF_INET;
	addr.sin_addr.s_addr = htonl(INADDR_ANY);

	// Create listen socket
	if ((_listen_sock = socket(AF_INET, SOCK_STREAM, 0)) < 0 ) {
		error << "Error creating listening socket (" << strerror(errno) << ")" << endl;
		exit(EXIT_FAILURE);
	}

	// Bind our socket addresss to the listening socket
	if (bind(_listen_sock, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
		error << "Error calling bind (%s)\n" << strerror(errno) << ")" << endl;
		_listen_sock = -1;
	}

	// Find port number
	socklen_t length = sizeof(addr);
	if (getsockname(_listen_sock, (struct sockaddr*)&addr, &length) == -1) {
		error << "Error calling getsockname (" << strerror(errno) << ")" << endl;
		_listen_sock = -1;
		return;
	}

	if (listen(_listen_sock, 1) < 0 ) {
		error << "Error calling listen (" << strerror(errno) << ")" << endl;
		_listen_sock = -1;
		return;
	}

	_listen_port = ntohs(addr.sin_port);
	info << "Opening event stream on TCP port " << _listen_port << endl;
	start();
}

HTTPSender::~HTTPSender()
{
	stop();
	if (_listen_sock != -1)
		close(_listen_sock);
	if (_client_sock != -1)
		close(_client_sock);
}

void
HTTPSender::_run()
{
	if (_listen_sock == -1) {
		error << "Unable to open socket, exiting sender thread" << endl;
		return;
	}

	// Accept connection
	if ((_client_sock = accept(_listen_sock, NULL, NULL) ) < 0) {
		error << "Error calling accept: " << strerror(errno) << endl;
		return;
	}

	// Hold connection open and write when signalled
	while (true) {
		_mutex.lock();
		_signal.wait(_mutex);

		write(_client_sock, _transfer.c_str(), _transfer.length());
		write(_client_sock, "\n\n\n", 3);

		_signal.broadcast();
		_mutex.unlock();
	}

	close(_listen_sock);
	_listen_sock = -1;
}

void
HTTPSender::bundle_begin()
{
	_mutex.lock();
	_send_state = SendingBundle;
	_transfer = "";
}

void
HTTPSender::bundle_end()
{
	assert(_send_state == SendingBundle);
	_signal.broadcast();
	_signal.wait(_mutex);
	_send_state = Immediate;
	_mutex.unlock();
}

void
HTTPSender::send_chunk(const std::string& buf)
{
	if (_send_state == Immediate) {
		_mutex.lock();
		_transfer = buf;
		_signal.broadcast();
		_signal.wait(_mutex);
		_mutex.unlock();
	} else {
		_transfer.append(buf);
	}
}

} // namespace Shared
} // namespace Ingen