/* 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 <iostream>
#include <set>
#include <locale.h>
#include <glibmm/ustring.h>
#include <redlandmm/Model.hpp>
#include <redlandmm/Query.hpp>
#include <raul/TableImpl.hpp>
#include <raul/Atom.hpp>
#include <raul/AtomRDF.hpp>
#include "interface/EngineInterface.hpp"
#include "Loader.hpp"

using namespace std;
using namespace Raul;
using namespace Ingen::Shared;

namespace Ingen {
namespace Serialisation {


/** Load (create) a patch from RDF into the engine.
 *
 * @param document_uri URI of file to load objects from.
 * @param parent Path of parent under which to load objects.
 * @return whether or not load was successful.
 */
bool
Loader::load(Ingen::Shared::World*       world,
             const Glib::ustring&        document_uri,
	         boost::optional<Raul::Path> parent,
	         std::string                 patch_name,
	         Glib::ustring               patch_uri,
             GraphObject::Variables      data)
{
	setlocale(LC_NUMERIC, "C");

	// FIXME: this whole thing is a mess

	std::set<Path> created;

	Redland::Model model(*world->rdf_world, document_uri, document_uri);

	if (patch_uri == "")
		patch_uri = string("<") + document_uri + ">";
	else
		patch_uri = string("<") + patch_uri + ">";

	cout << "[Loader] Loading " << patch_uri << endl;

	size_t patch_poly = 1;
	
	/* Use parameter overridden polyphony, if given */
	GraphObject::Variables::iterator poly_param = data.find("ingen:polyphony");
	if (poly_param != data.end() && poly_param->second.type() == Atom::INT) {
		patch_poly = poly_param->second.get_int32();
	
	/* Get polyphony from file (mandatory if not specified in parameters) */
	} else {
		// FIXME: polyphony datatype?
		Redland::Query query(*world->rdf_world, Glib::ustring(
			"SELECT DISTINCT ?poly WHERE {\n") +
			patch_uri + " ingen:polyphony ?poly\n }");

		Redland::Query::Results results = query.run(*world->rdf_world, model);

		if (results.size() == 0) {
			cerr << "[Loader] ERROR: No polyphony found!" << endl;
			return false;
		}

		const Redland::Node& poly_node = (*results.begin())["poly"];
		assert(poly_node.is_int());
		patch_poly = static_cast<size_t>(poly_node.to_int());
	}

	/* Get name (if available/necessary) */

	if (patch_name == "") {	
		patch_name = string(document_uri.substr(document_uri.find_last_of("/")+1));
		if (patch_name.substr(patch_name.length()-10) == ".ingen.ttl")
			patch_name = patch_name.substr(0, patch_name.length()-10);

		Redland::Query query(*world->rdf_world, Glib::ustring(
			"SELECT DISTINCT ?name WHERE {\n") +
			patch_uri + " ingen:symbol ?name\n}");

		Redland::Query::Results results = query.run(*world->rdf_world, model);

		if (results.size() > 0)
			patch_name = (*results.begin())["name"].to_string();
	}

	const Path patch_path = ( parent ? (parent.get().base() + patch_name) : Path("/") );
	
	cout << " as " << patch_path << endl;
	
	if (patch_path != "/")
		world->engine->new_patch(patch_path, patch_poly);
	
	/* Set document metadata (so File->Save doesn't prompt)
	 * FIXME: This needs some thinking for multiple clients... */
	world->engine->set_variable(patch_path, "ingen:document", Atom(document_uri.c_str()));

	/* Load (plugin) nodes */

	Redland::Query query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?name ?plugin ?varkey ?varval ?poly WHERE {\n") +
		patch_uri + " ingen:node       ?node .\n"
		"?node        ingen:symbol     ?name ;\n"
		"             ingen:plugin     ?plugin ;\n"
		"             ingen:polyphonic ?poly .\n"
		"OPTIONAL { ?node     ingen:variable ?variable .\n"
		"           ?variable ingen:key      ?varkey ;\n"
		"                     ingen:value    ?varval .\n"
		"         }"
		"}");

	Redland::Query::Results results = query.run(*world->rdf_world, model);

	map<const string, const Atom> variable;

	world->rdf_world->mutex().lock();
	
	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {

		const string node_name = (*i)["name"].to_string();
		const Path   node_path = patch_path.base() + node_name;

		if (created.find(node_path) == created.end()) {
			const string node_plugin     = world->rdf_world->qualify((*i)["plugin"].to_string());
			bool         node_polyphonic = false;

			Redland::Node poly_node = (*i)["poly"];
			if (poly_node.is_bool() && poly_node.to_bool() == true)
				node_polyphonic = true;
		
			world->engine->new_node(node_path, node_plugin, node_polyphonic);
			created.insert(node_path);
		}

		const string key = world->rdf_world->prefixes().qualify((*i)["varkey"].to_string());
		Redland::Node val_node = (*i)["varval"];

		if (key != "")
			world->engine->set_variable(node_path, key, AtomRDF::node_to_atom(val_node));
	}
	
	world->rdf_world->mutex().unlock();
		

	/* Load subpatches */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?patch ?name WHERE {\n") +
		patch_uri + " ingen:node   ?patch .\n"
		"?patch       a            ingen:Patch ;\n"
		"             ingen:symbol ?name .\n"
		"}");

	results = query.run(*world->rdf_world, model);
	
	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {

		const string name  = (*i)["name"].to_string();
		const string patch = (*i)["patch"].to_string();

		const Path subpatch_path = patch_path.base() + (string)name;

		if (created.find(subpatch_path) == created.end()) {
			created.insert(subpatch_path);
			load(world, document_uri, patch_path, name, patch);
		}
	}

	//created.clear();


	/* Set node port control values */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?nodename ?portname ?portval WHERE {\n") +
		patch_uri + " ingen:node   ?node .\n"
		"?node        ingen:symbol ?nodename ;\n"
		"             ingen:port   ?port .\n"
		"?port        ingen:symbol ?portname ;\n"
		"             ingen:value  ?portval .\n"
		"FILTER ( datatype(?portval) = xsd:decimal )\n"
		"}\n");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {

		const string node_name = (*i)["nodename"].to_string();
		const string port_name = (*i)["portname"].to_string();

		assert(Path::is_valid_name(node_name));
		assert(Path::is_valid_name(port_name));
		const Path port_path = patch_path.base() + node_name + "/" + port_name;

		world->engine->set_port_value(port_path, AtomRDF::node_to_atom((*i)["portval"]));
	}


	/* Load this patch's ports */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?port ?type ?name ?datatype ?varkey ?varval ?portval WHERE {\n") +
		patch_uri + " ingen:port     ?port .\n"
		"?port        a              ?type ;\n"
		"             a              ?datatype ;\n"
		"             ingen:symbol   ?name .\n"
		" FILTER (?type != ?datatype && ((?type = ingen:InputPort) || (?type = ingen:OutputPort)))\n"
		"OPTIONAL { ?port ingen:value ?portval . \n"
		"           FILTER ( datatype(?portval) = xsd:decimal ) }\n"
		"OPTIONAL { ?port     ingen:variable ?variable .\n"
		"           ?variable ingen:key      ?varkey ;\n"
		"                     ingen:value    ?varval .\n"
		"         }"
		"}");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {
		const string name     = (*i)["name"].to_string();
		const string type     = world->rdf_world->qualify((*i)["type"].to_string());
		const string datatype = world->rdf_world->qualify((*i)["datatype"].to_string());

		assert(Path::is_valid_name(name));
		const Path port_path = patch_path.base() + name;
			
		if (created.find(port_path) == created.end()) {
			bool is_output = (type == "ingen:OutputPort"); // FIXME: check validity
			world->engine->new_port(port_path, datatype, is_output);
			created.insert(port_path);
		}

		const Redland::Node val_node = (*i)["portval"];
		world->engine->set_port_value(patch_path.base() + name, AtomRDF::node_to_atom(val_node));

		const string key = world->rdf_world->prefixes().qualify((*i)["varkey"].to_string());
		const Redland::Node var_val_node = (*i)["varval"];

		if (key != "")
			world->engine->set_variable(patch_path.base() + name, key, AtomRDF::node_to_atom(var_val_node));
	}

	created.clear();


	/* Node -> Node connections */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?srcnodename ?srcname ?dstnodename ?dstname WHERE {\n") +
		patch_uri +  "ingen:node ?srcnode ;\n"
		"             ingen:node ?dstnode .\n"
		"?srcnode     ingen:port ?src ;\n"
		"             ingen:symbol ?srcnodename .\n"
		"?dstnode     ingen:port ?dst ;\n"
		"             ingen:symbol ?dstnodename .\n"
		"?src         ingen:symbol ?srcname .\n"
		"?dst         ingen:connectedTo ?src ;\n"
		"             ingen:symbol ?dstname .\n" 
		"}\n");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {
		Path src_node = patch_path.base() + Path::nameify((*i)["srcnodename"].to_string());
		Path src_port = src_node.base() + Path::nameify((*i)["srcname"].to_string());
		Path dst_node = patch_path.base() + Path::nameify((*i)["dstnodename"].to_string());
		Path dst_port = dst_node.base() + Path::nameify((*i)["dstname"].to_string());

		//cerr << patch_path << " 1 CONNECTION: " << src_port << " -> " << dst_port << endl;

		world->engine->connect(src_port, dst_port);
	}


	/* This Patch -> Node connections */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?srcname ?dstnodename ?dstname WHERE {\n") +
		patch_uri + " ingen:port ?src ;\n"
		"             ingen:node ?dstnode .\n"
		"?dstnode     ingen:port ?dst ;\n"
		"             ingen:symbol ?dstnodename .\n"
		"?dst         ingen:connectedTo ?src ;\n"
		"             ingen:symbol ?dstname .\n" 
		"?src         ingen:symbol ?srcname .\n"
		"}\n");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {
		Path src_port = patch_path.base() + Path::nameify((*i)["srcname"].to_string());
		Path dst_node = patch_path.base() + Path::nameify((*i)["dstnodename"].to_string());
		Path dst_port = dst_node.base() + Path::nameify((*i)["dstname"].to_string());

		//cerr << patch_path << " 2 CONNECTION: " << src_port << " -> " << dst_port << endl;

		world->engine->connect(src_port, dst_port);
	}


	/* Node -> This Patch connections */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?srcnodename ?srcname ?dstname WHERE {\n") +
		patch_uri + " ingen:port ?dst ;\n"
		"             ingen:node ?srcnode .\n"
		"?srcnode     ingen:port ?src ;\n"
		"             ingen:symbol ?srcnodename .\n"
		"?dst         ingen:connectedTo ?src ;\n"
		"             ingen:symbol ?dstname .\n" 
		"?src         ingen:symbol ?srcname .\n"
		"}\n");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {
		Path dst_port = patch_path.base() + Path::nameify((*i)["dstname"].to_string());
		Path src_node = patch_path.base() + Path::nameify((*i)["srcnodename"].to_string());
		Path src_port = src_node.base() + Path::nameify((*i)["srcname"].to_string());

		//cerr << patch_path << " 3 CONNECTION: " << src_port << " -> " << dst_port << endl;

		world->engine->connect(src_port, dst_port);
	}
	
	
	/* This Patch -> This Patch connections */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?srcname ?dstname WHERE {\n") +
		patch_uri + " ingen:port ?src ;\n"
		"             ingen:port ?dst .\n"
		"?dst         ingen:connectedTo ?src ;\n"
		"             ingen:symbol ?dstname .\n" 
		"?src         ingen:symbol ?srcname .\n"
		"}\n");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {
		Path src_port = patch_path.base() + Path::nameify((*i)["srcname"].to_string());
		Path dst_port = patch_path.base() + Path::nameify((*i)["dstname"].to_string());

		//cerr << patch_path << " 4 CONNECTION: " << src_port << " -> " << dst_port << endl;

		world->engine->connect(src_port, dst_port);
	}


	/* Load variables */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?varkey ?varval WHERE {\n") +
		patch_uri + " ingen:variable ?variable .\n"
		"?variable    ingen:key      ?varkey ;\n"
		"             ingen:value    ?varval .\n"
		"}");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {

		const string key = world->rdf_world->prefixes().qualify(string((*i)["varkey"]));
		Redland::Node val_node = (*i)["varval"];

		if (key != "")
			world->engine->set_variable(patch_path, key, AtomRDF::node_to_atom(val_node));
	}


	// Set passed variables last to override any loaded values
	for (GraphObject::Variables::const_iterator i = data.begin(); i != data.end(); ++i)
		world->engine->set_variable(patch_path, i->first, i->second);


	/* Enable */

	query = Redland::Query(*world->rdf_world, Glib::ustring(
		"SELECT DISTINCT ?enabled WHERE {\n") +
		patch_uri + " ingen:enabled ?enabled .\n"
		"}");

	results = query.run(*world->rdf_world, model);

	for (Redland::Query::Results::iterator i = results.begin(); i != results.end(); ++i) {

		Redland::Node enabled_node = (*i)["enabled"];

		if (enabled_node.is_bool() && enabled_node) {
			world->engine->enable_patch(patch_path);
			break;
		}
	}

	return true;
}

} // namespace Serialisation
} // namespace Ingen