#include "Makefile.h"

#include "Common.h"
#include "FileUtils.h"
#include "IncludeDeps.h"
#include "Utils.h"

#include <cstring>
#include <dirent.h>
#include <fstream>
#include <map>

void Makefile::Save(const ConfigFile& conf, unsigned int flags)
{
  std::set<HFile> hFiles; // hFile, directory
  std::set<std::string> cppFiles;
  if(flags & FLAG_SIMPLE)
    Utils::GetCppFiles(conf, cppFiles);
  else
    Utils::GetCppAndHFiles(conf, hFiles, cppFiles);

  std::ofstream outputFile(conf.configPath + "Makefile");
  outputFile << "# This Makefile was generated using MakeGen "<< MAKEGEN_VERSION<< " made by Tim Håkansson" << std::endl;
  outputFile << "# and is licensed under MIT. Full source of the project can be found at" << std::endl;
  outputFile << "# https://github.com/Thraix/MakeGen" << std::endl;
  outputFile << "CC=@g++" << std::endl;
  if(!conf.executable)
  {
    if(conf.shared)
      outputFile << "CO=@g++ -shared -o" << std::endl;
    else
      outputFile << "CO=@g++ -o" << std::endl;
  }
  else
    outputFile << "CO=@g++ -o" << std::endl;

  outputFile << "MKDIR_P=mkdir -p" << std::endl;
  outputFile << "BIN=" << conf.outputdir << std::endl;
  outputFile << "OBJPATH=$(BIN)intermediates" << std::endl;
  outputFile << "INCLUDES=";
  for(auto it = conf.includedirs.begin();it!=conf.includedirs.end();++it)
  {
    outputFile << "-I./" << *it << " ";
  }
  outputFile << std::endl;
  outputFile << "OBJECTS=";
  for(auto it = cppFiles.begin();it!=cppFiles.end();++it)
  {
    size_t extensionPos = it->find_last_of(".");
    size_t slash = it->find_last_of("/")+1;
    outputFile << "$(OBJPATH)/" << it->substr(slash, extensionPos - slash) << ".o ";
  }
  outputFile << std::endl;
  if(conf.executable || !conf.shared)
  {
    outputFile << "CFLAGS=$(INCLUDES) -std=c++17 -c -w -g3 ";
  }
  else
  {
    outputFile << "CFLAGS=$(INCLUDES) -fPIC -std=c++17 -c -w -g3 ";
  }
  for(auto it = conf.defines.begin();it!=conf.defines.end();++it)
  {
    outputFile << "-D" << *it << " ";
  }
  for(auto it = conf.flags.begin();it!=conf.flags.end();++it)
  {
    outputFile << *it << " ";
  }
  outputFile << std::endl;
  if(conf.executable)
  {
    outputFile << "LIBDIR=";
    for(auto it = conf.libdirs.begin();it!=conf.libdirs.end();++it)
    {
      outputFile << "-L./" << *it << " ";
    }
    outputFile << std::endl;
    outputFile << "LDFLAGS=";
    for(auto it = conf.libdirs.begin();it!=conf.libdirs.end();++it)
    {
      outputFile << "-Wl,-rpath=" << *it << " ";
    }
    outputFile << std::endl;
    outputFile << "LIBS=$(LIBDIR) ";
    for(auto it = conf.libs.begin();it!=conf.libs.end();++it)
    {
      outputFile << "-l" << *it << " ";
    }
    outputFile << std::endl;
    if(!conf.dependencies.empty())
    {
      outputFile << "DEPENDENCIES=";
      for(auto it = conf.dependencies.begin();it!=conf.dependencies.end();++it)
      {
        outputFile << *it << " ";
      }
      outputFile << std::endl;
    }
  }
  outputFile << "OUTPUT=$(BIN)" << conf.outputname << std::endl;
  outputFile << ".PHONY: all directories rebuild clean run" << std::endl;

  // All
  outputFile << "all: directories $(OUTPUT)" << std::endl;

  // Directories
  outputFile << "directories: $(BIN) $(OBJPATH)" << std::endl;

  // Bin path
  outputFile << "$(BIN):" << std::endl;
  outputFile << "\t$(info Creating output directories)" << std::endl;
  outputFile << "\t@$(MKDIR_P) $(BIN)" << std::endl;

  // Object path
  outputFile << "$(OBJPATH):" << std::endl;
  outputFile << "\t@$(MKDIR_P) $(OBJPATH)" << std::endl;

  // Run
  outputFile << "run: all" << std::endl;
  if(conf.executable)
  {
    outputFile << "\t@./$(OUTPUT)" << std::endl;
  }

  // Rebuild
  outputFile << "rebuild: clean all" << std::endl;

  // Clean
  outputFile << "clean:" << std::endl;
  outputFile << "\t$(info Removing intermediates)" << std::endl;
  outputFile << "\trm -rf $(OBJPATH)/*.o" << std::endl;

  // Output file
  outputFile << "$(OUTPUT): $(OBJECTS)" << std::endl;
  outputFile << "\t$(info Generating output file)" << std::endl;
  if(conf.executable)
    outputFile << "\t$(CO) $(OUTPUT) $(OBJECTS) $(LDFLAGS) $(LIBS)" << std::endl;
  else
    outputFile << "\t$(CO) $(OUTPUT) $(OBJECTS)" << std::endl;

  // Install
  outputFile << "install: all" << std::endl;
  outputFile << "\t$(info Installing " << conf.projectname <<" to /usr/bin/)" << std::endl;
  outputFile << "\t@cp $(OUTPUT) /usr/bin/" << conf.outputname << std::endl;

  std::map<std::string, IncludeDeps*> dependencies;
  size_t i = 0;
  for(auto it = cppFiles.begin(); it!= cppFiles.end();++it)
  {
    i++;
    auto itD = dependencies.find(conf.srcdir + *it);
    if(itD == dependencies.end())
    {
      IncludeDeps* deps = new IncludeDeps(*it, conf.configPath+conf.srcdir,hFiles,dependencies);
      size_t extensionPos = it->find_last_of(".");
      size_t slash = it->find_last_of("/")+1;
      std::string oFile = it->substr(slash, extensionPos - slash)+".o ";

      outputFile << "$(OBJPATH)/" << oFile << ": ";
      deps->Output(outputFile, conf);
      outputFile << std::endl;
      outputFile << "\t$(info -[" << (int)(i / (float)cppFiles.size() * 100) << "%]- $<)" << std::endl;
      outputFile << "\t$(CC) $(CFLAGS) -o $@ $<" << std::endl;
    }
  }
}