Feature #1711 CreaDevManager application implementation

[crea.git] / lib / creaDevManagerLib / modelCDMApplication.cpp

diff --git a/lib/creaDevManagerLib/modelCDMApplication.cpp b/lib/creaDevManagerLib/modelCDMApplication.cpp
index 0ee18a913fa037ed71e1065488b54f6cb5785d57..f7f7431bdd3af8cec850e6974767f13031f28ac3 100644 (file)
--- a/lib/creaDevManagerLib/modelCDMApplication.cpp
+++ b/lib/creaDevManagerLib/modelCDMApplication.cpp
@@ -37,6 +37,7 @@
 #include <fstream>
 #include <sstream>
 #include <algorithm>
+#include <boost/regex.hpp>
 
 #include "CDMUtilities.h"
 #include "creaWx.h"
@@ -117,7 +118,6 @@ modelCDMApplication::modelCDMApplication(modelCDMIProjectTreeNode* parent, const
       while (cont)
         {
           std::string stdfileName = crea::wx2std(fileName);
-
           //if CMakeLists, create CMakeLists
           if(stdfileName == "CMakeLists.txt")
             {
@@ -129,23 +129,46 @@ modelCDMApplication::modelCDMApplication(modelCDMIProjectTreeNode* parent, const
             {
               modelCDMFile* file = new modelCDMFile(this, this->path + CDMUtilities::SLASH + stdfileName, stdfileName, this->level + 1);
               std::string extension = stdfileName.substr(stdfileName.size()-4);
+              //if is cxx or cpp check if is the main file.
               if (mainFile == NULL && (extension == ".cxx" || extension == ".cpp"))
                 {
-                  std::ifstream fileStream;
-                  std::string word;
-                  fileStream.open((this->path + CDMUtilities::SLASH + stdfileName).c_str());
-                  while (fileStream.is_open() && !fileStream.eof())
+                  std::ifstream fileStream((this->path + CDMUtilities::SLASH + stdfileName).c_str());
+
+                  if (fileStream.is_open())
                     {
-                      //get sets
-                      std::getline(fileStream,word,'(');
-                      std::vector<std::string> wordBits;
-                      CDMUtilities::splitter::split(wordBits,word," \n",CDMUtilities::splitter::no_empties);
-                      if (wordBits[wordBits.size() - 1] == "main" || wordBits[wordBits.size() - 1] == "IMPLEMENT_APP")
+                      std::string fileContent = "";
+                      char ch = fileStream.get();
+                      while(!fileStream.eof())
+                        {
+                          fileContent.push_back(ch);
+                          ch = fileStream.get();
+                        }
+                      fileStream.close();
+
+                      boost::regex expression("^\\h*IMPLEMENT_APP[#\\s\\(]");
+                      std::string::const_iterator start, end;
+                      start = fileContent.begin();
+                      end = fileContent.end();
+                      boost::match_results<std::string::const_iterator> what;
+                      boost::match_flag_type flags = boost::match_default;
+                      if(boost::regex_search(start, end, what, expression, flags))
                         {
+                          std::cout << "found main wxwidgets file: " << stdfileName << std::endl;
                           this->mainFile = file;
                         }
+                      else
+                        {
+                          expression = boost::regex("^\\h*int\\h+main[#\\s\\(]");
+                          start = fileContent.begin();
+                          end = fileContent.end();
+                          if(boost::regex_search(start, end, what, expression, flags))
+                            {
+                              std::cout << "found main console file: " << stdfileName << std::endl;
+                              this->mainFile = file;
+                            }
+                        }
                     }
-                  fileStream.close();
+
                 }
               this->children.push_back(file);
             }