#include <creaImageIOGimmick.h>

#include <creaMessageManager.h>
#include <creaImageIOImageFinder.h>

#include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp>

namespace creaImageIO
{
  
  //==============================================================
  Gimmick::Gimmick()
  {    
    crea::MessageManager::RegisterMessageType("Gimmick!",
					      "Gimmick",1);
  }
  //==============================================================



  //==============================================================
  Gimmick::~Gimmick()
  {
    
  }
  //==============================================================
  

  //==============================================================
  bool Gimmick::Initialize()
  {
    // Create the UserSettings dir if does not exist
    if (!CreateUserSettingsDirectory()) return false;
    // Sets the current directory to the home dir
    mCurrentDirectory =  GetHomeDirectory();

    // Create local database handler
    mLocalDatabase = new SQLiteTreeHandler(GetLocalDatabasePath());
    // Create or open local database
    if (! boost::filesystem::exists( GetLocalDatabasePath() ) )
      {
	creaMessage("Gimmick!",1,
		    "[Gimmick!] Local database '"<<GetLocalDatabasePath()<<"' "
		    << "does not exist : creating it"<<std::endl);

	// CREATING DEFAULT DB STRUCTURE
	mLocalDatabase->GetTree().GetDescriptor().CreateDefault();
	
	if ( ! mLocalDatabase->Create(true) )
	  {
	    creaMessage("Gimmick!",1,
			"[Gimmick!] !! ERROR CREATING '"<<GetLocalDatabasePath()<<"'");
	    return false;
	  }
	mLocalDatabase->SetAttribute(0,"Name","Local database");
      }
    else 
      {
	/// Open and test it
	creaMessage("Gimmick!",1,
		    "[Gimmick!] Opening local database '"
		    <<GetLocalDatabasePath()<<"' "
		    <<std::endl);
	if ( ! mLocalDatabase->Open(true) )
	  {
	    creaMessage("Gimmick!",1,
			"[Gimmick!] !! ERROR OPENING '"<<GetLocalDatabasePath()<<"'");
	    return false;
	  }
	
      }
    return true;

  }
  //================================================================


  //==============================================================
  bool Gimmick::Finalize()
  {
    delete mLocalDatabase;
  }
  //==============================================================

  //================================================================
  // file separator
#if defined(_WIN32)
#define VALID_FILE_SEPARATOR "\\"
#define INVALID_FILE_SEPARATOR "/"
#else
#define INVALID_FILE_SEPARATOR "\\"
#define VALID_FILE_SEPARATOR "/"
#endif
  //================================================================

  //================================================================
  const std::string& Gimmick::GetHomeDirectory()
  {
    if (mHomeDirectory.size()==0) 
      {
#if defined(__GNUC__)
	mHomeDirectory = getenv("HOME");
#elif defined(_WIN32)
	mHomeDirectory = getenv("USERPROFILE");
#endif
      }
    return mHomeDirectory;
  }
  //================================================================
  const std::string& Gimmick::GetUserSettingsDirectory()
  {
    if (mUserSettingsDirectory.size()==0) 
      {
	mUserSettingsDirectory = GetHomeDirectory();
	mUserSettingsDirectory += "/.gimmick/";
	boost::algorithm::replace_all( mUserSettingsDirectory, 
				       INVALID_FILE_SEPARATOR , 
				       VALID_FILE_SEPARATOR);
      }
    return mUserSettingsDirectory;
  }
  //================================================================

  //================================================================
  const std::string& Gimmick::GetLocalDatabasePath()
  {
    if (mLocalDatabasePath.size()==0) 
      {
	mLocalDatabasePath = GetUserSettingsDirectory();
	mLocalDatabasePath += "local_database.sqlite3";
	boost::algorithm::replace_all( mLocalDatabasePath,
				       INVALID_FILE_SEPARATOR , 
				       VALID_FILE_SEPARATOR);
      }
    return mLocalDatabasePath;    
  }
  //========================================================================

  //========================================================================
  bool Gimmick::CreateUserSettingsDirectory()
  {
    if (! boost::filesystem::is_directory( GetUserSettingsDirectory() ) )
      {
	creaMessage("Gimmick!",1,
		    "[Gimmick!] Directory '"<<GetUserSettingsDirectory()<<"' "
		    << "does not exist : creating it"<<std::endl);
	
	if ( ! boost::filesystem::create_directory( GetUserSettingsDirectory() ) )
	  {
	    creaMessage("Gimmick!",1,
			"[Gimmick!] !! ERROR CREATING '"<<GetUserSettingsDirectory()<<"'");
	    return false;
	  }
      }
    return true;
  }
  //========================================================================


  //========================================================================
  /// Sets message level
  void Gimmick::SetMessageLevel(int l)
  {
    crea::MessageManager::SetMessageLevel("Gimmick!",l);
  }
  //========================================================================

  //========================================================================
  /// Add a file to the local database
  void Gimmick::AddFileToLocalDatabase(const std::string& f)
  {
    ImageFinder finder(mLocalDatabase);
    if (finder.IsHandledFile(f)) finder.AddFile(f);
    
  }
  //========================================================================

  //========================================================================
  /// Add a dir to the local database
  void Gimmick::AddDirToLocalDatabase(const std::string& f, bool recurse)
  {
    ImageFinder finder(mLocalDatabase);
    
    finder.AddDirectory(f,recurse);
    
  }
  //========================================================================

  //========================================================================
  /// 
  void Gimmick::PrintLocalDatabase()
  {
    mLocalDatabase->GetTree().Print();
  }
  //========================================================================


}