captured stdout from system command, fixed bug in trim.seqs added error checks for...

[mothur.git] / systemcommand.cpp

diff --git a/systemcommand.cpp b/systemcommand.cpp
index 6b0a6d06474809f16fae1a0c17338ef1f7458dc7..e5aeb3e8b73dfb59008f6ddccaa3b0703b25e696 100644 (file)
--- a/systemcommand.cpp
+++ b/systemcommand.cpp
@@ -9,7 +9,20 @@
 
 #include "systemcommand.h"
 
-
+//**********************************************************************************************************************
+vector<string> SystemCommand::setParameters(){ 
+       try {
+               CommandParameter pcommand("command", "String", "", "", "", "", "",false,false); parameters.push_back(pcommand);
+                               
+               vector<string> myArray;
+               for (int i = 0; i < parameters.size(); i++) {   myArray.push_back(parameters[i].name);          }
+               return myArray;
+       }
+       catch(exception& e) {
+               m->errorOut(e, "SystemCommand", "setParameters");
+               exit(1);
+       }
+}
 //**********************************************************************************************************************
 SystemCommand::SystemCommand(string option)  {
        try {
@@ -17,10 +30,25 @@ SystemCommand::SystemCommand(string option)  {
                
                //allow user to run help
                if(option == "help") { help(); abort = true; calledHelp = true; }
+               else if(option == "citation") { citation(); abort = true; calledHelp = true;}
                
                else {
-                       if (option == "") { m->mothurOut("You must enter a command to run."); m->mothurOutEndLine(); abort = true; }
-                       else { 
+                       vector<string> myArray = setParameters();
+                       
+                       OptionParser parser(option);
+                       map<string, string> parameters = parser.getParameters();
+                       map<string, string>::iterator it;
+                       
+                       ValidParameters validParameter;
+                       
+                       //check for optional parameter and set defaults
+                       // ...at some point should added some additional type checking...
+                       string commandOption = validParameter.validFile(parameters, "command", false);                  
+                       if (commandOption == "not found") { commandOption = ""; }
+                       else { command = commandOption; }
+                       
+                       if ((option == "") && (commandOption == "")) { m->mothurOut("You must enter a command to run."); m->mothurOutEndLine(); abort = true; }
+                       else if (commandOption == "") { 
                                //check for outputdir and inputdir parameters
                                int commaPos = option.find_first_of(',');
                                
@@ -63,7 +91,19 @@ int SystemCommand::execute(){
                
                if (abort == true) { if (calledHelp) { return 0; }  return 2;   }
                
-               system(command.c_str());
+               //system(command.c_str());
+               FILE *lsofFile_p = popen(command.c_str(), "r");
+               
+               if (!lsofFile_p) { return 0; }
+               
+               char buffer[1024];
+               while ( fgets(buffer, 1024, lsofFile_p) != NULL ) {
+                       string temp = buffer;
+                       m->mothurOut(temp); 
+               }
+               m->mothurOutEndLine();
+               
+               pclose(lsofFile_p);
                
                return 0;               
        }