ferencd@0: #include "dictionary.h"
ferencd@0: 
ferencd@0: #include <tntdb.h>
ferencd@0: #include <boost/algorithm/string.hpp>
ferencd@0: 
ferencd@0: const std::vector<std::string> dictionary::supported_languages = {"gb", "no"};
ferencd@0: std::map<std::string, std::map<std::string, std::string>> dictionary::m_inMemoryTranslations;
ferencd@0: 
ferencd@0: template<typename T>
ferencd@0: std::string join(const std::vector<T> in,
ferencd@0:                  const std::string & separator =", ",  // see 1.
ferencd@0:                  const std::string & concluder =" ")    // see 1.
ferencd@0: {
ferencd@0:     std::ostringstream ss;
ferencd@0:     std::size_t i = 0;
ferencd@0:     for(; i<in.size() - 1; i++)
ferencd@0:     {
ferencd@0:         ss << in[i] << separator;
ferencd@0:     }
ferencd@0: 
ferencd@0:     ss << in[i] << concluder;
ferencd@0:     return ss.str();
ferencd@0: }
ferencd@0: 
ferencd@0: std::string dictionary::translate(const std::string & what, const std::string &target_language, bool other_languages_too, std::map<std::string, std::string> &translations)
ferencd@0: {
ferencd@0:     std::string result_translation;
ferencd@0: 	bool found = false;
ferencd@0: 
ferencd@0:     try {
ferencd@0:         tntdb::Connection conn = tntdb::connect("sqlite:lang.db");
ferencd@0: 
ferencd@0:         // TODO: prepared statement
ferencd@0: 
ferencd@0:         std::string v = "select " + join(supported_languages) + " from translations where source='" + what + "'";
ferencd@0:         tntdb::Result result = conn.select(v);
ferencd@0:         for (tntdb::Result::const_iterator it = result.begin(); it != result.end(); ++it)
ferencd@0:         {
ferencd@0:             tntdb::Row row = *it;
ferencd@0: 
ferencd@0:             for(size_t i=0; i<row.size(); i++)
ferencd@0:             {
ferencd@0:                 std::string row_name = boost::to_lower_copy(row.getName(i));
ferencd@0: 
ferencd@0:                 if(other_languages_too)
ferencd@0:                 {
ferencd@0:                     std::string temp;
ferencd@0:                     row[i].get(temp);
ferencd@0:                     translations[row_name] = temp;
ferencd@0:                 }
ferencd@0: 
ferencd@0:                 if(row_name == boost::to_lower_copy(target_language))
ferencd@0:                 {
ferencd@0:                     row[i].get(result_translation);
ferencd@0: 					found = true;
ferencd@0:                 }
ferencd@0:             }
ferencd@0:         }
ferencd@0: 
ferencd@0:     }
ferencd@0:     catch (std::exception& ex)
ferencd@0:     {
ferencd@0:         std::cerr << ex.what();
ferencd@0: 		return "";
ferencd@0:     }
ferencd@0: 
ferencd@0: 	if(!found)
ferencd@0: 	{
ferencd@0: 		if(m_inMemoryTranslations.count(what))
ferencd@0: 		{
ferencd@0: 			for(const auto& l : supported_languages)
ferencd@0: 			{
ferencd@0: 				if(m_inMemoryTranslations[what].count(l))
ferencd@0: 				{
ferencd@0: 					translations[l] = m_inMemoryTranslations[what][l];
ferencd@0: 				}
ferencd@0: 			}
ferencd@0: 			if(m_inMemoryTranslations[what].count(target_language))
ferencd@0: 			{
ferencd@0: 				result_translation = m_inMemoryTranslations[what][target_language];
ferencd@0: 			}
ferencd@0: 		}
ferencd@0: 	}
ferencd@0: 
ferencd@0: 	return result_translation;
ferencd@0: }
ferencd@0: 
ferencd@0: std::string dictionary::translate(const std::string &what, const std::string &target_language)
ferencd@0: {
ferencd@0: 	std::map<std::string, std::string> translations;
ferencd@0: 	return translate(what, target_language, false, translations);
ferencd@0: }
ferencd@0: 
ferencd@0: void dictionary::add_translation(const std::string &key, const std::string &language, const std::string &translated)
ferencd@0: {
ferencd@0: 	m_inMemoryTranslations[key][language] = translated;
ferencd@0: }