c@7: 
c@7: #include "VampJson.h"
c@7: 
c@7: #include <iostream>
c@7: #include <sstream>
c@7: #include <stdexcept>
c@7: 
c@7: #include <map>
c@7: #include <set>
c@7: 
c@7: using namespace std;
c@7: using namespace Vamp;
c@7: using namespace Vamp::HostExt;
c@7: using namespace json11;
c@10: using namespace vampipe;
c@7: 
c@11: class Mapper : public PluginHandleMapper
c@11: {
c@11: public:
c@11:     Mapper() : m_nextHandle(1) { }
c@7: 
c@11:     void addPlugin(Plugin *p) {
c@11: 	if (m_rplugins.find(p) == m_rplugins.end()) {
c@11: 	    int32_t h = m_nextHandle++;
c@11: 	    m_plugins[h] = p;
c@11: 	    m_rplugins[p] = h;
c@11: 	}
c@11:     }
c@11:     
c@11:     int32_t pluginToHandle(Plugin *p) {
c@11: 	if (m_rplugins.find(p) == m_rplugins.end()) {
c@11: 	    throw NotFound();
c@11: 	}
c@11: 	return m_rplugins[p];
c@11:     }
c@11:     
c@11:     Plugin *handleToPlugin(int32_t h) {
c@11: 	if (m_plugins.find(h) == m_plugins.end()) {
c@11: 	    throw NotFound();
c@11: 	}
c@11: 	return m_plugins[h];
c@11:     }
c@11: 
c@11:     bool isInitialised(int32_t h) {
c@11: 	return m_initialisedPlugins.find(h) != m_initialisedPlugins.end();
c@11:     }
c@11: 
c@11:     void markInitialised(int32_t h) {
c@11: 	m_initialisedPlugins.insert(h);
c@11:     }
c@11:     
c@11: private:
c@11: //!!! + mutex
c@11:     int32_t m_nextHandle; // plugin handle type must fit in JSON number
c@11:     map<uint32_t, Plugin *> m_plugins;
c@11:     map<Plugin *, uint32_t> m_rplugins;
c@11:     set<uint32_t> m_initialisedPlugins;
c@11: };
c@11: 
c@11: static Mapper mapper;
c@7: 
c@7: Vamp::HostExt::LoadResponse
c@7: loadPlugin(json11::Json j) {
c@7: 
c@7:     auto req = VampJson::toLoadRequest(j);
c@7:     auto loader = Vamp::HostExt::PluginLoader::getInstance();
c@7:     auto response = loader->loadPlugin(req);
c@7:     
c@7:     if (!response.plugin) {
c@7: 	throw VampJson::Failure("plugin load failed");
c@7:     }
c@7:     
c@7:     return response;
c@7: }
c@7:     
c@12: Vamp::HostExt::ConfigurationResponse
c@7: configurePlugin(Vamp::Plugin *plugin, json11::Json j) {
c@7:     
c@7:     auto config = VampJson::toPluginConfiguration(j);
c@7:     auto loader = Vamp::HostExt::PluginLoader::getInstance();
c@7:     auto outputs = loader->configurePlugin(plugin, config);
c@7: 
c@7:     if (outputs.empty()) {
c@7: 	throw VampJson::Failure("plugin initialisation failed (invalid channelCount, stepSize, blockSize?)");
c@7:     }
c@12: 
c@12:     Vamp::HostExt::ConfigurationResponse response;
c@12:     response.outputs = outputs;
c@12:     return response;
c@7: }
c@7: 
c@7: Json
c@7: handle_list(Json content)
c@7: {
c@7:     if (content != Json()) {
c@7: 	throw VampJson::Failure("no content expected for list request");
c@7:     }
c@7:     
c@7:     auto loader = PluginLoader::getInstance();
c@7:     auto pluginData = loader->listPluginData();
c@7: 
c@7:     Json::array j;
c@7:     for (const auto &pd: pluginData) {
c@7: 	j.push_back(VampJson::fromPluginStaticData(pd));
c@7:     }
c@7:     return Json(j);
c@7: }
c@7: 
c@7: Json
c@7: handle_load(Json j)
c@7: {
c@7:     auto loadResponse = loadPlugin(j);
c@7: 
c@7:     if (!loadResponse.plugin) {
c@7: 	throw VampJson::Failure("plugin load failed");
c@7:     }
c@11: 
c@11:     mapper.addPlugin(loadResponse.plugin);
c@7:     
c@11:     return VampJson::fromLoadResponse(loadResponse, mapper);
c@7: }
c@7: 
c@7: Json
c@7: handle_configure(Json j)
c@7: {
c@7:     string err;
c@7: 
c@7:     if (!j.has_shape({
c@7: 		{ "pluginHandle", Json::NUMBER },
c@7: 		{ "configuration", Json::OBJECT }}, err)) {
c@7: 	throw VampJson::Failure("malformed configuration request: " + err);
c@7:     }
c@7: 
c@11:     int32_t handle = j["pluginHandle"].int_value();
c@7: 
c@11:     if (mapper.isInitialised(handle)) {
c@7: 	throw VampJson::Failure("plugin has already been initialised");
c@7:     }
c@7: 
c@11:     Plugin *plugin = mapper.handleToPlugin(handle);
c@7:     
c@7:     Json config = j["configuration"];
c@7: 
c@12:     auto response = configurePlugin(plugin, config);
c@7: 
c@11:     mapper.markInitialised(handle);
c@7: 
c@7:     cerr << "Configured and initialised plugin " << handle << endl;
c@7: 
c@12:     return VampJson::fromConfigurationResponse(response);
c@7: }
c@7: 
c@7: Json
c@7: handle(string input)
c@7: {
c@7:     string err;
c@7:     Json j = Json::parse(input, err);
c@7: 
c@7:     if (err != "") {
c@7: 	throw VampJson::Failure("invalid request: " + err);
c@7:     }
c@7: 
c@7:     if (!j["verb"].is_string()) {
c@7: 	throw VampJson::Failure("verb expected in request");
c@7:     }
c@7: 
c@7:     if (!j["content"].is_null() &&
c@7: 	!j["content"].is_object()) {
c@7: 	throw VampJson::Failure("object expected for content");
c@7:     }
c@7: 
c@7:     string verb = j["verb"].string_value();
c@7:     Json content = j["content"];
c@7:     Json result;
c@7: 
c@7:     if (verb == "list") {
c@7: 	result = handle_list(content);
c@7:     } else if (verb == "load") {
c@7: 	result = handle_load(content);
c@7:     } else if (verb == "configure") {
c@7: 	result = handle_configure(content);
c@7:     } else {
c@7: 	throw VampJson::Failure("unknown verb: " + verb +
c@7: 				" (known verbs are: list load configure)");
c@7:     }
c@7: 
c@7:     return result;
c@7: }
c@7: 
c@7: Json
c@7: success_response(Json payload)
c@7: {
c@7:     Json::object obj;
c@7:     obj["success"] = true;
c@7:     obj["response"] = payload;
c@7:     return Json(obj);
c@7: }
c@7: 
c@7: Json
c@7: error_response(string text)
c@7: {
c@7:     Json::object obj;
c@7:     obj["success"] = false;
c@7:     obj["errorText"] = text;
c@7:     return Json(obj);
c@7: }
c@7: 
c@7: template<typename T>
c@7: T &getline(T &in, string prompt, string &out)
c@7: {
c@7:     cerr << prompt;
c@7:     return getline(in, out);
c@7: }
c@7: 
c@7: int main(int, char **)
c@7: {
c@7:     string line;
c@7: 
c@7:     while (getline(cin, "> ", line)) {
c@7: 	try {
c@7: 	    Json result = handle(line);
c@7: 	    cout << success_response(result).dump() << endl;
c@7: 	} catch (const VampJson::Failure &e) {
c@7: 	    cout << error_response(e.what()).dump() << endl;
c@11: 	} catch (const PluginHandleMapper::NotFound &e) {
c@11: 	    cout << error_response(e.what()).dump() << endl;
c@7: 	}
c@7:     }
c@7: 
c@7:     return 0;
c@7: }
c@7: 
c@7: