/* This file is part of Ingen.
 * Copyright (C) 2007 Dave Robillard <http://drobilla.net>
 * 
 * Ingen 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 2 of the License, or (at your option) 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 General Public License for details.
 * 
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "OSCClientSender.hpp"
#include <cassert>
#include <iostream>
#include <unistd.h>
#include <raul/AtomLiblo.hpp>
#include "ObjectStore.hpp"
#include "NodeFactory.hpp"
#include "util.hpp"
#include "Patch.hpp"
#include "Node.hpp"
#include "Plugin.hpp"
#include "Port.hpp"
#include "Connection.hpp"
#include "AudioDriver.hpp"
#include "interface/ClientInterface.hpp"

using namespace std;

namespace Ingen {

void
OSCClientSender::bundle_begin()
{
	// FIXME: Don't split bundles as for 'transfers'
	_transfer = lo_bundle_new(LO_TT_IMMEDIATE);
}

void
OSCClientSender::bundle_end()
{
	// FIXME: Don't split bundles as for 'transfers'
	transfer_end();
}


void
OSCClientSender::transfer_begin()
{
	_transfer = lo_bundle_new(LO_TT_IMMEDIATE);
}


void
OSCClientSender::transfer_end()
{
	assert(_transfer);
	lo_send_bundle(_address, _transfer);
	lo_bundle_free(_transfer);
	_transfer = NULL;
}


/*! \page client_osc_namespace Client OSC Namespace Documentation
 *
 * <p>NOTE: this comment doesn't really apply any longer.. sort of.
 * but maybe it still should.. maybe.  so it remains...</p>
 *
 * <p>These are all the messages sent from the engine to the client.
 * Communication takes place over two distinct bands: control band and
 * notification band.</p>
 * <p>The control band is where clients send commands, and receive a simple
 * response, either OK or an error.</p>
 * <p>All notifications of engine state (ie new nodes) are sent over the
 * notification band <em>which is seperate from the control band</em>.  The
 * reasoning behind this is that many clients may be connected at the same
 * time - a client may receive notifications that are not a direct consequence
 * of some message it sent.</p>
 * <p>The notification band can be thought of as a stream of events representing
 * the changing engine state.  For example, It is possible for a client to send
 * commands and receive aknowledgements, and not listen to the notification band
 * at all; or (in the near future anyway) for a client to use UDP for the control
 * band (for speed), and TCP for the notification band (for reliability and
 * order guarantees).</p>
 * \n\n
 */


/** \page client_osc_namespace
 * \n
 * <h2>Control Band</h2>
 */

/** \page client_osc_namespace
 * <p> \b /ingen/ok - Respond to a successful user command
 * \arg \b response-id (int) - Request ID this is a response to
 * </p> \n \n
 */
void
OSCClientSender::response_ok(int32_t id)
{
	if (!_enabled)
		return;

	if (lo_send(_address, "/ingen/ok", "i", id) < 0) {
		cerr << "Unable to send ok " << id << "! ("
			<< lo_address_errstr(_address) << ")" << endl;
	}
}
	

/** \page client_osc_namespace
 * <p> \b /ingen/response - Respond to a user command
 * \arg \b response-id (int) - Request ID this is a response to
 * \arg \b message (const std::string&) - Error message (natural language text)
 * </p> \n \n
 */
void
OSCClientSender::response_error(int32_t id, const std::string& msg)
{
	if (!_enabled)
		return;

	if (lo_send(_address, "/ingen/error", "is", id, msg.c_str()) < 0) {
		cerr << "Unable to send error " << id << "! ("
			<< lo_address_errstr(_address) << ")" << endl;
	}
}


/** \page client_osc_namespace
 * \n
 * <h2>Notification Band</h2>
 */


/** \page client_osc_namespace
 * <p> \b /ingen/error - Notification that an error has occurred
 * \arg \b message (const std::string&) - Error message (natural language text) \n\n
 * \li This is for notification of errors that aren't a direct response to a
 * user command, ie "unexpected" errors.</p> \n \n
 */
void
OSCClientSender::error(const std::string& msg)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/error", "s", msg.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/num_plugins
 * \arg \b num (int) - Number of plugins engine has loaded\n\n
 * \li This is sent before sending the list of plugins, so the client is aware
 * of how many plugins (/ingen/plugin messages) to expect.</p> \n \n
 */


/** \page client_osc_namespace
 * <p> \b /ingen/num_plugins
 * \arg \b num (int) - Number of plugins engine has loaded\n\n
 * \li This is sent before sending the list of plugins, so the client is aware
 * of how many plugins (/ingen/plugin messages) to expect.</p> \n \n
 */
void
OSCClientSender::num_plugins(uint32_t num)
{
	if (!_enabled)
		return;

	lo_message m = lo_message_new();
	lo_message_add_int32(m, num);
	lo_send_message(_address, "/ingen/num_plugins", m);
}


/** \page client_osc_namespace
 * <p> \b /ingen/plugin - Notification of the existance of a plugin
 * \arg \b type (const std::string&) - Type if plugin ("LADSPA", "DSSI", or "Internal")
 * \arg \b uri (const std::string&) - URI of the plugin (see engine namespace documentation) \n
 * \arg \b lib-name (const std::string&) - Name of shared library plugin resides in (ie "cmt.so")
 * \arg \b plug-label (const std::string&) - Label of the plugin (ie "dahdsr_iaoa")
 * \arg \b name (const std::string&) - Descriptive human-readable name of plugin (ie "ADSR Envelope")
 * </p> \n \n
 */
/*
void
OSCClientSender::plugins()
{
	Engine::instance().node_factory()->lock_plugin_list();
	
	const list<Plugin*>& plugs = Engine::instance().node_factory()->plugins();
	const Plugin* plugin;

	lo_timetag tt;
	lo_timetag_now(&tt);
	lo_bundle b = lo_bundle_new(tt);
	lo_message m = lo_message_new();
	list<lo_message> msgs;

	lo_message_add_int32(m, plugs.size());
	lo_bundle_add_message(b, "/ingen/num_plugins", m);
	msgs.push_back(m);

	for (list<Plugin*>::const_iterator j = plugs.begin(); j != plugs.end(); ++j) {
		plugin = (*j);
		m = lo_message_new();

		lo_message_add_string(m, plugin->type_const std::string&());
		lo_message_add_string(m, plugin->uri().c_str());
		lo_message_add_string(m, plugin->plug_label().c_str());
		lo_message_add_string(m, plugin->name().c_str());
		lo_bundle_add_message(b, "/ingen/plugin", m);
		msgs.push_back(m);
		if (lo_bundle_length(b) > 1024) {
			lo_send_bundle(_address, b);
			lo_bundle_free(b);
			b = lo_bundle_new(tt);
		}
	}
	
	if (lo_bundle_length(b) > 0) {
		lo_send_bundle(_address, b);
		lo_bundle_free(b);
	} else {
		lo_bundle_free(b);
	}
	for (list<lo_bundle>::const_iterator i = msgs.begin(); i != msgs.end(); ++i)
		lo_message_free(*i);

	Engine::instance().node_factory()->unlock_plugin_list();
}
*/

/** \page client_osc_namespace
 * <p> \b /ingen/new_node - Notification of a new node's creation.
 * \arg \b plug-uri (const std::string&) - URI of the plugin new node is an instance of
 * \arg \b path (const std::string&) - Path of the new node
 * \arg \b polyphonic (integer-boolean) - Node is polyphonic (1 = yes, 0 = no)
 * \arg \b num-ports (integer) - Number of ports (number of new_port messages to expect)\n\n
 * \li New nodes are sent as a bundle.  The first message in the bundle will be
 * this one (/ingen/new_node), followed by a series of /ingen/new_port commands,
 * followed by /ingen/new_node_end. </p> \n \n
 */
void OSCClientSender::new_node(const std::string&   plugin_uri,
                               const std::string&   node_path,
                               bool     is_polyphonic,
                               uint32_t num_ports)
{
	if (!_enabled)
		return;

	//cerr << "Sending node " << node_path << endl;

	lo_send(_address, "/ingen/new_node", "ssii", plugin_uri.c_str(),
	        node_path.c_str(), is_polyphonic ? 1 : 0, num_ports);
#if 0
	/*
	lo_timetag tt;
	lo_timetag_now(&tt);
	lo_bundle b = lo_bundle_new(tt);
	lo_message m = lo_message_new();
	list<lo_message> msgs;

	lo_message_add_string(m, plugin_type.c_str());
	lo_message_add_string(m, plugin_uri.c_str());
	lo_message_add_string(m, node_path.c_str());
	lo_message_add_int32(m, is_polyphonic ? 1 : 0);
	lo_message_add_int32(m, num_ports);

	lo_bundle_add_message(b, "/ingen/new_node", m);
	msgs.push_back(m);
*/


	/*
	const Raul::Array<Port*>& ports = node->ports();
	Port*     port;
	PortInfo* info;
	for (size_t j=0; j < ports.size(); ++j) {
		port = ports.at(j);
		info = port->port_info();

		assert(port != NULL);
		assert(info != NULL);
		
		m = lo_message_new();
		lo_message_add_string(m, port->path().c_str());
		lo_message_add_string(m, info->type_const std::string&().c_str());
		lo_message_add_string(m, info->direction_const std::string&().c_str());
		lo_message_add_string(m, info->hint_const std::string&().c_str());
		lo_message_add_float(m, info->default_val());
		lo_message_add_float(m, info->min_val());
		lo_message_add_float(m, info->max_val());
		lo_bundle_add_message(b, "/ingen/new_port", m);
		msgs.push_back(m);
		
		// If the bundle is getting very large, send it and start
		// a new one
		if (lo_bundle_length(b) > 1024) {
		  lo_send_bundle(_address, b);
		  lo_bundle_free(b);
		  b = lo_bundle_new(tt);
		}
	}
*/
	/*m = lo_message_new();
	//lo_bundle_add_message(b, "/ingen/new_node_end", m);
	//msgs.push_back(m);

	lo_send_bundle(_address, b);
	lo_bundle_free(b);

	for (list<lo_bundle>::const_iterator i = msgs.begin(); i != msgs.end(); ++i)
		lo_message_free(*i);

	usleep(100);
*/
	/*
	const map<const std::string&, const std::string&>& data = node->metadata();
	// Send node metadata
	for (map<const std::string&, const std::string&>::const_iterator i = data.begin(); i != data.end(); ++i)
		metadata_update(node->path(), (*i).first, (*i).second);

	
	// Send port metadata
	for (size_t j=0; j < ports.size(); ++j) {
		port = ports.at(j);
		const map<const std::string&, const std::string&>& data = port->metadata();
		for (map<const std::string&, const std::string&>::const_iterator i = data.begin(); i != data.end(); ++i)
			metadata_update(port->path(), (*i).first, (*i).second);
	}

	// Send control values
	for (size_t i=0; i < node->ports().size(); ++i) {
		TypedPort<Sample>* port = (TypedPort<Sample>*)node->ports().at(i);
		if (port->port_info()->is_input() && port->port_info()->is_control())
			control_change(port->path(), port->buffer(0)->value_at(0));
	}
	*/
#endif
}



/** \page client_osc_namespace
 * <p> \b /ingen/new_port - Notification of a new port's creation.
 * \arg \b path (const std::string&) - Path of new port
 * \arg \b data-type (const std::string&) - Type of port (ingen:audio, ingen:control, ingen:midi, or ingen:osc)
 * \arg \b direction ("is-output") (integer) - Direction of data flow (Input = 0, Output = 1)
 *
 * \li Note that in the event of loading a patch, this message could be
 * followed immediately by a control change, meaning the default-value is
 * not actually the current value of the port.
 * \li The minimum and maximum values are suggestions only, they are not
 * enforced in any way, and going outside them is perfectly fine.  Also note
 * that the port ranges in om_gtk are not these ones!  Those ranges are set
 * as metadata.</p> \n \n
 */
void
OSCClientSender::new_port(const std::string& path,
                          const std::string& data_type,
                          bool   is_output)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/new_port", "ssi", path.c_str(), data_type.c_str(), is_output);
}


/** \page client_osc_namespace
 * <p> \b /ingen/destroyed - Notification an object has been destroyed
 * \arg \b path (const std::string&) - Path of object (which no longer exists) </p> \n \n
 */
void
OSCClientSender::object_destroyed(const std::string& path)
{
	if (!_enabled)
		return;

	assert(path != "/");
	
	lo_send(_address, "/ingen/destroyed", "s", path.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/patch_cleared - Notification a patch has been cleared (all children destroyed)
 * \arg \b path (const std::string&) - Path of patch (which is now empty)</p> \n \n
 */
void
OSCClientSender::patch_cleared(const std::string& patch_path)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/patch_cleared", "s", patch_path.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/patch_enabled - Notification a patch's DSP processing has been enabled.
 * \arg \b path (const std::string&) - Path of enabled patch</p> \n \n
 */
void
OSCClientSender::patch_enabled(const std::string& patch_path)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/patch_enabled", "s", patch_path.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/patch_disabled - Notification a patch's DSP processing has been disabled.
 * \arg \b path (const std::string&) - Path of disabled patch</p> \n \n
 */
void
OSCClientSender::patch_disabled(const std::string& patch_path)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/patch_disabled", "s", patch_path.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/new_connection - Notification a new connection has been made.
 * \arg \b src-path (const std::string&) - Path of the source port
 * \arg \b dst-path (const std::string&) - Path of the destination port</p> \n \n
 */
void
OSCClientSender::connection(const std::string& src_port_path, const std::string& dst_port_path)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/new_connection", "ss", src_port_path.c_str(), dst_port_path.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/disconnection - Notification a connection has been unmade.
 * \arg \b src-path (const std::string&) - Path of the source port
 * \arg \b dst-path (const std::string&) - Path of the destination port</p> \n \n
 */
void
OSCClientSender::disconnection(const std::string& src_port_path, const std::string& dst_port_path)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/disconnection", "ss", src_port_path.c_str(), dst_port_path.c_str());
}


/** \page client_osc_namespace
 * <p> \b /ingen/metadata_update - Notification of a piece of metadata.
 * \arg \b path (const std::string&) - Path of the object associated with metadata (can be a node, patch, or port)
 * \arg \b key (const std::string&)
 * \arg \b value (const std::string&)</p> \n \n
 */
void
OSCClientSender::metadata_update(const std::string& path, const std::string& key, const Atom& value)
{
	if (!_enabled)
		return;

	lo_message m = lo_message_new();
	lo_message_add_string(m, path.c_str());
	lo_message_add_string(m, key.c_str());
	Raul::AtomLiblo::lo_message_add_atom(m, value);
	lo_send_message(_address, "/ingen/metadata_update", m);
}


/** \page client_osc_namespace
 * <p> \b /ingen/control_change - Notification the value of a port has changed
 * \arg \b path (const std::string&) - Path of port
 * \arg \b value (float) - New value of port
 *
 * \li This will only send updates for values set by clients of course - not values
 * changing because of connections to other ports!</p> \n \n
 */
void
OSCClientSender::control_change(const std::string& port_path, float value)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/control_change", "sf", port_path.c_str(), value);
}


/** \page client_osc_namespace
 * <p> \b /ingen/plugin - Notification of the existance of a plugin
 * \arg \b type (const std::string&) - Type of plugin ("LADSPA", "DSSI", or "Internal")
 * \arg \b uri (const std::string&) - Type of plugin ("LADSPA", "DSSI", or "Internal")
 * \arg \b name (const std::string&) - Descriptive human-readable name of plugin (ie "ADSR Envelope")
 */
void
OSCClientSender::new_plugin(const std::string& uri, const std::string& type_uri, const std::string& name)
{
	if (!_enabled)
		return;

	// FIXME: size?  liblo doesn't export this.
	// Don't want to exceed max UDP packet size (1500 bytes)
	static const size_t MAX_BUNDLE_SIZE = 1500 - 32*5;
		
	lo_message m = lo_message_new();
	lo_message_add_string(m, uri.c_str());
	lo_message_add_string(m, type_uri.c_str());
	lo_message_add_string(m, name.c_str());
	
	if (_transfer) {

		if (lo_bundle_length(_transfer) + lo_message_length(m, "/ingen/plugin")
				> MAX_BUNDLE_SIZE) {
			lo_send_bundle(_address, _transfer);
			_transfer = lo_bundle_new(LO_TT_IMMEDIATE);
		}
		lo_bundle_add_message(_transfer, "/ingen/plugin", m);

	} else {
		lo_send_message(_address, "/ingen/plugin", m);
	}
}


/** \page client_osc_namespace
 * <p> \b /ingen/new_patch - Notification of a new patch
 * \arg \b path (const std::string&) - Path of new patch
 * \arg \b poly (int) - Polyphony of new patch (\em not a boolean like new_node) </p> \n \n
 */
void
OSCClientSender::new_patch(const std::string& path, uint32_t poly)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/new_patch", "si", path.c_str(), poly);
	
	/*
	if (p->process())
		patch_enabled(p->path());
	
	// Send metadata
	const map<const std::string&, const std::string&>& data = p->metadata();
	for (map<const std::string&, const std::string&>::const_iterator i = data.begin(); i != data.end(); ++i) {
		metadata_update(p->path(), (*i).first, (*i).second);
	}
	*/
}


/** \page client_osc_namespace
 * <p> \b /ingen/object_renamed - Notification of an object's renaming
 * \arg \b old-path (const std::string&) - Old path of object
 * \arg \b new-path (const std::string&) - New path of object </p> \n \n
 */
void
OSCClientSender::object_renamed(const std::string& old_path, const std::string& new_path)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/object_renamed", "ss", old_path.c_str(), new_path.c_str());
}


/** Sends information about a program associated with a DSSI plugin node.
 */
void
OSCClientSender::program_add(const std::string& node_path, uint32_t bank, uint32_t program, const std::string& name)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/program_add", "siis", 
		node_path.c_str(), bank, program, name.c_str());
}


void
OSCClientSender::program_remove(const std::string& node_path, uint32_t bank, uint32_t program)
{
	if (!_enabled)
		return;

	lo_send(_address, "/ingen/program_remove", "sii", 
		node_path.c_str(), bank, program);
}


} // namespace Ingen