root/trunk/PCRE/filegrep.h

#ifndef FILEGREP_H
#define FILEGREP_H

#include <vector>
#include <string>
#include "pme-1.0.4/pme.h"
#include "MainFrm.h"

using namespace std;

struct Listing
{
    string file;
    string match;
    int lineNumber;

};

typedef vector < vector< string > > str_vec_2d;

class FileGrep
{
public:
    
    int nestingLevel;
    int nestingLimit;

    str_vec_2d undoFileContents;
    str_vec_2d undoFileNames;



    FileGrep(int limit = 10 ) : nestingLimit(limit), nestingLevel(0) {} // this is for how many undo operations you want to support

    vector<Listing> GrepFiles(const vector<string> &files, const string& regex, const string& options = "" )
    {
        

        PME reg(regex, options);

        vector<Listing> listings;
        Listing listing;
        


        for ( int i = 0 ; i < files.size(); i++ )
        {

            int lineCount = 1;          
            string line;
            ifstream in(files[i].c_str() );
            while ( !in.eof () )
            {
                std::getline(in,line);
                //AfxMessageBox(string("line [ " + ITOA(lineCount) + "] " + line).c_str());
                if ( reg.match(line) > 0 )
                {
                    listing.match = line;
                    listing.file = files[i];
                    listing.lineNumber = lineCount;
                    listings.push_back(listing);
                }
                lineCount++;

            }

            
        }

        return listings;


    }

    vector<Listing> GrepReplaceFiles(const vector<string> &files, const string& regex, const string& options , const string& replaceRegex)
    {

        int i = 0;

        vector<string> fileContents;
        vector<string> backupContents;
        vector<string> fileNames;
        

        vector<Listing> listings;
        Listing listing;
        
        for ( i = 0 ; i < files.size(); i ++ )
        {
            //AfxMessageBox(ReadFile(files[i]).c_str() );
            backupContents.push_back(ReadFile(files[i]));       
        }

        PME reg(regex, options);
        //--- SAVE ALL FILES --- //



        for ( i = 0 ; i < files.size(); i ++ )
        {
            int lineCount = 1;          
            string line = "";
            string data = "";
            
            ifstream in(files[i].c_str() );
            while ( !in.eof () )
            {
                std::getline(in,line);
                //AfxMessageBox(line.c_str());
                
                
                if ( reg.match(line) > 0 )
                {
                    
                    listing.file = files[i];
                    listing.lineNumber = lineCount;
                    
                    //AfxMessageBox(string("line before : " + line).c_str());
                    line = reg.sub( line, replaceRegex);
                    //AfxMessageBox(string("line after : " + line).c_str());
                    listing.match = line;

                    listings.push_back(listing);

                    data += line + "\n";

                    
                }
                else {
                    data += line + "\n";
                }
                lineCount++;

            }
            in.close();
            fileNames.push_back(files[i]);
            fileContents.push_back(data);

            //AfxMessageBox(data.c_str());
        }

        for ( i = 0 ; i < fileContents.size(); i ++ )
        {
            string fileName = files[i];
            ofstream out(fileName.c_str());
            out << fileContents[i];
            //AfxMessageBox(fileContents[i].c_str())
                ;
            out.close();
        }

        undoFileContents.push_back(backupContents);
        undoFileNames.push_back(fileNames);
        nestingLevel++;


        return listings;
    }


    bool Undo()
    {
        
        int level = --nestingLevel;
        if ( level >= 0)
        {
            
            for ( int i = 0 ; i < undoFileNames[level].size();i++)
            {
                //AfxMessageBox(undoFileContents[level][i].c_str() );
                //AfxMessageBox(undoFileNames[level][i].c_str() );
                //log(string("undo file name" + undoFileNames[level][i] + " has contents " + undoFileNames[level][i]),__FILE__,__LINE__ );
                
                ofstream out(undoFileNames[level][i].c_str() );
                out << undoFileContents[level][i];
                out.close();
            }
            
            undoFileNames.pop_back();
            undoFileContents.pop_back();
            return true;
            
        }
        else {
            // resset the undo's
            nestingLevel = 0;
            undoFileContents.clear();
            undoFileNames.clear();
            return false;
        }
        
        
    }





};


#endif