#include <GameData.h>
#include <File.h>
#include <DatCat.h>
#include <ExdData.h>
#include <ExdCat.h>
#include <Exd.h>
#include <Exh.h>
#include <iostream>
#include <cctype>
#include <set>
#include <src/servers/Server_Common/Exd/ExdData.h>
#include <src/servers/Server_Common/Logging/Logger.h>
#include <boost/range/algorithm/remove_if.hpp>
#include <boost/algorithm/string/classification.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/make_shared.hpp>
#include <boost/property_tree/json_parser.hpp>
#include <boost/property_tree/xml_parser.hpp>
#include <boost/property_tree/ptree.hpp>
#include <boost/foreach.hpp>
#include <boost/algorithm/string/replace.hpp>
#include <boost/lexical_cast.hpp>

#include <fstream>
#include <streambuf>
#include <regex>


Core::Logger g_log;
Core::Data::ExdData g_exdData;
bool skipUnmapped = true;

//const std::string datLocation( "/opt/sapphire_3_15_0/bin/sqpack" );
const std::string datLocation( "C:\\SquareEnix\\FINAL FANTASY XIV - A Realm Reborn\\game\\sqpack\\ffxiv" );
std::map< uint8_t, std::string > g_typeMap;


std::string generateDatAccessDecl( const std::string &exd )
{
   return "     xiv::exd::Exd m_" + exd + "Dat;\n";
}

std::string generateDirectGetters( const std::string& exd )
{
   return "     boost::shared_ptr< " + exd + " > get" + exd + "( uint32_t " + exd + "Id );";
}

std::string generateSetDatAccessCall( const std::string &exd )
{
   auto& cat = g_exdData.m_exd_data->get_category( exd );
   auto exh = cat.get_header();

   std::string lang = "xiv::exd::Language::none";
   auto langs = exh.get_languages();
   if( langs.size() > 1 )
      lang = "xiv::exd::Language::en";

   return "      m_" + exd + "Dat = setupDatAccess( \"" + exd + "\", " + lang + " );";
}

std::string generateDirectGetterDef( const std::string& exd )
{
   std::string result = "";
   result =
      "boost::shared_ptr< Core::Data::" + exd + " >\n"
      "   Core::Data::ExdDataGenerated::get" + exd + "( uint32_t " + exd + "Id )\n"
      "{\n"
      "   try\n"
      "   {\n"
      "      auto row = m_" + exd + "Dat.get_row( " + exd + "Id );\n"
      "      auto info = boost::make_shared< " + exd + " >( " + exd + "Id, this );\n"

      "      return info;\n"
      "   }\n"
      "   catch( ... )\n"
      "   {\n"
      "      return nullptr;\n"
      "   }\n"

      "   return nullptr;\n"

      "}\n";
   return result;
}
std::map< uint32_t, std::string > indexToNameMap;
std::map< uint32_t, std::string > indexToTypeMap;

std::string generateStruct( const std::string &exd )
{
   auto& cat = g_exdData.m_exd_data->get_category( exd );
   auto exh = cat.get_header();
   auto exhMem = exh.get_exh_members();

   int count = 0;

   using boost::property_tree::ptree;
   ptree m_propTree;
   boost::property_tree::read_json( "ex.json", m_propTree );

   BOOST_FOREACH( boost::property_tree::ptree::value_type &sheet, m_propTree.get_child( "sheets" ) )
   {
      std::string name = sheet.second.get< std::string >( "sheet" );
      if( name != exd )
         continue;
//      g_log.info( name + " ---- " );

      BOOST_FOREACH( boost::property_tree::ptree::value_type &show, sheet.second.get_child( "definitions" ) )
      {
         uint32_t index;
         try
         {
            index = show.second.get< uint32_t >("index");
         }
         catch( ... )
         {
            index = 0;
         }
         try {
            std::string fieldName = show.second.get< std::string >( "name" );
            indexToNameMap[index] = fieldName;
         }
         catch(...){}
//         g_log.info( std::to_string( index ) );
         
      }
   }

   std::string result = "struct " + exd +"\n{\n";


   for( auto member : exhMem )
   {
      auto typei = static_cast< uint8_t >( member.type );
      auto it = g_typeMap.find( typei );

      std::string type;
      if( it != g_typeMap.end() )
         type = it->second; 
      else
         type = "bool";

      std::string fieldName = "field" + std::to_string( count );
      if( indexToNameMap.find( count ) == indexToNameMap.end() )
      {
         if( skipUnmapped )
         {
           count++;
           continue;
         }
         indexToNameMap[count] = fieldName;
      }
      else
      {
         fieldName = indexToNameMap[count];
      }
      fieldName[0] = std::tolower( fieldName[0] );
      fieldName.erase( boost::remove_if( fieldName, boost::is_any_of(",-':![](){}<>% \x02\x1f\x01\x03") ), fieldName.end() );   
      indexToNameMap[count] = fieldName;
      indexToTypeMap[count] = type;
     
      result += "   " + type + " " + fieldName + ";\n";
   
      count++;
   }

   result += "\n   " + exd + "( uint32_t row_id, Core::Data::ExdDataGenerated* exdData );\n";
   result += "};\n\n";
   
   return result;
}

std::string generateConstructorsDecl( const std::string& exd )
{
   std::string result;

   auto& cat = g_exdData.m_exd_data->get_category( exd );
   auto exh = cat.get_header();
   auto exhMem = exh.get_exh_members();

   int count = 0;


   result += "\n      Core::Data::" + exd + "::" + exd + "( uint32_t row_id, Core::Data::ExdDataGenerated* exdData )\n";
   result += "      {\n";
   std::string indent = "         ";
   result += indent + "auto row = exdData->m_" + exd + "Dat.get_row( row_id );\n";
   for( auto member : exhMem )
   {  
      if( indexToNameMap.find( count ) == indexToNameMap.end() )
      { count++; continue; }
      result += indent + indexToNameMap[count] + " = exdData->getField< " + indexToTypeMap[count] + " >( row, " + std::to_string( count) + " );\n";
      count++;
   }
   result += "      }\n";

   indexToNameMap.clear();
   indexToTypeMap.clear();
   return result;
}

int main()
{
   g_typeMap[0] = "std::string";
   g_typeMap[1] = "bool";
   g_typeMap[2] = "int8_t";
   g_typeMap[3] = "uint8_t";
   g_typeMap[4] = "int16_t";
   g_typeMap[5] = "uint16_t";
   g_typeMap[6] = "int32_t";
   g_typeMap[7] = "uint32_t";
   g_typeMap[9] = "float";
   g_typeMap[11] = "uint64_t";

   std::ifstream t( "ExdData.h.tmpl" );
   std::string exdH( ( std::istreambuf_iterator<char>( t ) ),
                       std::istreambuf_iterator<char>() );

   std::ifstream s( "ExdData.cpp.tmpl" );
   std::string exdC( ( std::istreambuf_iterator<char>( s ) ),
                       std::istreambuf_iterator<char>() );


   using boost::property_tree::ptree;
   ptree m_propTree;
   boost::property_tree::read_json( "ex.json", m_propTree );
   g_log.init();

   g_log.info( "Setting up EXD data" );
   if( !g_exdData.init( datLocation ) )
   {
      g_log.fatal( "Error setting up EXD data " );
      return 0;
   }
  
   std::string structDefs;
   std::string dataDecl;
   std::string getterDecl;
   std::string datAccCall;
   std::string getterDef;
   std::string constructorDecl;


   BOOST_FOREACH(boost::property_tree::ptree::value_type &sheet, m_propTree.get_child("sheets"))
   {
      std::string name = sheet.second.get<std::string>("sheet");
    
      structDefs += generateStruct( name );
      dataDecl += generateDatAccessDecl( name );
      getterDecl += generateDirectGetters( name );
      datAccCall += generateSetDatAccessCall( name );
      getterDef += generateDirectGetterDef( name );
      constructorDecl += generateConstructorsDecl( name );
   }

   // for all sheets in the json i guess....

   std::string result;
   result = std::regex_replace( exdH, std::regex( "\\STRUCTS" ), structDefs );
   result = std::regex_replace( result, std::regex( "\\DATACCESS" ), dataDecl );
   result = std::regex_replace( result, std::regex( "\\DIRECTGETTERS" ), getterDecl );

//   g_log.info( result );

   std::ofstream outH("ExdDataGenerated.h");
   outH << result;
   outH.close();

   result = std::regex_replace( exdC, std::regex( "\\SETUPDATACCESS" ), datAccCall );
   result = std::regex_replace( result, std::regex( "\\DIRECTGETTERS" ), getterDef );
   result = std::regex_replace( result, std::regex( "\\CONSTRUCTORS" ), constructorDecl );
   
   std::ofstream outC("ExdDataGenerated.cpp");
   outC << result;
   outC.close();
   
//   g_log.info( result );

   return 0;
}