root/trunk/AdvancedFindAndReplaceDlg.cpp

// AdvancedFindAndReplaceDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Elephant.h"
#include "AdvancedFindAndReplaceDlg.h"
#include "pcre/filegrep.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAdvancedFindAndReplaceDlg dialog


CAdvancedFindAndReplaceDlg::CAdvancedFindAndReplaceDlg(CWnd* pParent /*=NULL*/)
: CXTResizeDialog(CAdvancedFindAndReplaceDlg::IDD, pParent) 
{
    //{{AFX_DATA_INIT(CAdvancedFindAndReplaceDlg)
    m_csFind = _T("");
    m_csReplace = _T("");
    m_csOptions = _T("");
    m_found = _T("");
    //}}AFX_DATA_INIT
    grep = new FileGrep(3);
    m_filter = "";
}


void CAdvancedFindAndReplaceDlg::DoDataExchange(CDataExchange* pDX)
{
    CXTResizeDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CAdvancedFindAndReplaceDlg)
    DDX_Control(pDX, IDC_STATIC2, m_searchdataTitle);
    DDX_Control(pDX, IDC_BUTTON11, m_findBtn);
    DDX_Control(pDX, IDC_STATIC1, m_static);
    DDX_Control(pDX, IDC_LIST2, m_matches);
    DDX_Control(pDX, IDC_LIST1, m_files);
    DDX_Control(pDX, IDC_EDIT8, m_replace);
    DDX_Control(pDX, IDC_EDIT4, m_options);
    DDX_Control(pDX, IDC_EDIT1, m_find);
    DDX_Control(pDX, IDC_BUTTON6, m_close);
    DDX_Control(pDX, IDC_BUTTON3, m_replaceSelected);
    DDX_Control(pDX, IDC_BUTTON2, m_replaceAll);
    DDX_Text(pDX, IDC_EDIT1, m_csFind);
    DDX_Text(pDX, IDC_EDIT8, m_csReplace);
    DDX_Text(pDX, IDC_EDIT4, m_csOptions);
    DDX_Text(pDX, IDC_STATIC3, m_found);
    //}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CAdvancedFindAndReplaceDlg, CXTResizeDialog)
    //{{AFX_MSG_MAP(CAdvancedFindAndReplaceDlg)
    ON_WM_DESTROY()
    ON_BN_CLICKED(IDC_BUTTON11, OnFind) 
    ON_COMMAND(ID_FINDMENU_ADDFILES, OnAddFiles)
    ON_BN_CLICKED(IDC_BUTTON6, OnClose)
    ON_COMMAND(ID_FINDMENU_REMOVEFILE, OnFindmenuRemovefile)
    ON_NOTIFY(NM_DBLCLK, IDC_LIST2, OnDblclkList2)
    ON_BN_CLICKED(IDC_BUTTON2, OnReplceAll)
    ON_COMMAND(ID_FILEMATCHESMENU_UNDO, OnFilematchesmenuUndo)
    ON_BN_CLICKED(IDC_BUTTON3, OnRelpaceSelected)
    ON_COMMAND(ID_FINDMENU_SETFILTER, OnFindmenuSetfilter)
    ON_COMMAND(ID_FINDMENU_ADDDIR, OnFindmenuAdddir)
    ON_COMMAND(ID_FINDMENU_CLEARFILES, OnFindmenuClearfiles)
    //}}AFX_MSG_MAP
    ON_WM_RBUTTONDOWN() 
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CAdvancedFindAndReplaceDlg message handlers

BOOL CAdvancedFindAndReplaceDlg::OnInitDialog() 
{
    CXTResizeDialog::OnInitDialog();
    
    SetupButton(&m_replaceAll);
    SetupButton(&m_replaceSelected);
    SetupButton(&m_close);
    SetupButton(&m_findBtn);


    SetupTitle(&m_static);
    SetupTitle(&m_searchdataTitle);

    m_files.InsertColumn(0,"Files to Search",LVCFMT_LEFT,135);
    
    
    m_matches.InsertColumn(0,"Line No.",LVCFMT_LEFT,60);
    m_matches.InsertColumn(1,"File",LVCFMT_LEFT,150);
    m_matches.InsertColumn(2,"Line",LVCFMT_LEFT,200);
    m_matches.InsertColumn(3,"Path",LVCFMT_LEFT,200);

    SetupListCtrl((CListCtrl*)&m_files);
    SetupListCtrl((CListCtrl*)&m_matches);
    
    m_find.SetFocus();

    /*
    m_replace.SetText("Replace text.");
    m_options.SetText("Options [gimsx].");
    m_find.SetText("Find text.");
    */

    SetResize(IDC_EDIT1 ,SZ_TOP_LEFT,SZ_TOP_RIGHT);
    SetResize(IDC_EDIT8,SZ_TOP_LEFT,SZ_TOP_RIGHT);
    SetResize(IDC_EDIT4,SZ_TOP_LEFT,SZ_TOP_RIGHT);
    
    SetResize(IDC_LIST2,SZ_TOP_LEFT,SZ_BOTTOM_RIGHT);
    SetResize(IDC_LIST1,SZ_TOP_LEFT,SZ_BOTTOM_LEFT);

    SetResize(IDC_BUTTON2,SZ_BOTTOM_LEFT,SZ_BOTTOM_LEFT);
    SetResize(IDC_BUTTON3,SZ_BOTTOM_LEFT,SZ_BOTTOM_LEFT);
    SetResize(IDC_BUTTON6,SZ_BOTTOM_LEFT,SZ_BOTTOM_LEFT);
    SetResize(IDC_BUTTON11,SZ_BOTTOM_LEFT,SZ_BOTTOM_LEFT);
    
    SetResize(IDC_STATIC1,SZ_BOTTOM_LEFT,SZ_BOTTOM_LEFT);   
    SetResize(IDC_STATIC3,SZ_BOTTOM_LEFT,SZ_BOTTOM_LEFT);



SetIcon(AfxGetApp()->LoadIcon(IDR_ELEPHANT) ,true);

    return TRUE;  // return TRUE unless you set the focus to a control
                  // EXCEPTION: OCX Property Pages should return FALSE
}

void CAdvancedFindAndReplaceDlg::OnDestroy() 
{
    CXTResizeDialog::OnDestroy();
    
    SavePlacement("CAdvancedFindAndReplaceDlg");

    CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
    m->SetForegroundWindow();

    
}

void CAdvancedFindAndReplaceDlg::OnFind() 
{
    UpdateData();
    CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
    
    
    m_matches.DeleteAllItems();
    m->m_wndOutputBar.m_findSheet.m_findList.DeleteAllItems();

    string options = m_csOptions.GetBuffer(0);
    if ( options == m_options.GetText() ) options = "";

    grep;
    vector<Listing> matches = grep->GrepFiles(m_searchfiles,m_csFind.GetBuffer(0),options);
    for ( int i = 0 ; i < matches.size(); i ++ )
    {
        if ( m )
        {
            m->m_wndOutputBar.m_findSheet.m_findList.InsertItem(i,ITOA(matches[i].lineNumber).c_str())      ;
            m->m_wndOutputBar.m_findSheet.m_findList.SetItemText(i,1,StripFileName(matches[i].file).c_str());
            m->m_wndOutputBar.m_findSheet.m_findList.SetItemText(i, 2 , WhiteSpace::ReplaceWhiteSpaceWithBlanks(matches[i].match).c_str() );
            m->m_wndOutputBar.m_findSheet.m_findList.SetItemText(i, 3 , matches[i].file.c_str() );
        }
        m_matches.InsertItem(i,ITOA(matches[i].lineNumber).c_str())     ;
        m_matches.SetItemText(i,1,StripFileName(matches[i].file).c_str());
        m_matches.SetItemText(i, 2 , WhiteSpace::ReplaceWhiteSpaceWithBlanks(matches[i].match).c_str() );
        m_matches.SetItemText(i, 3 , matches[i].file.c_str() );

    }
    
    string display = ITOA(matches.size() ) + " Total Matches Found";

    m_found = display.c_str();
    UpdateData(FALSE);
    
}

void CAdvancedFindAndReplaceDlg::OnRButtonDown(UINT nFlags, CPoint point)
{
    

}

void CAdvancedFindAndReplaceDlg::OnAddFiles() 
{
    vector<string> files;
    GetFilesFromCFileDialog(Globals::lastDir, files,"d","All Files (*.*)|*.*||",this);

    int count = m_files.GetItemCount();

    for ( int i = 0 ; i < files.size(); i++ )
    {
        m_searchfiles.push_back(files[i]);
        m_files.InsertItem(count++,StripFileName(files[i] ).c_str() );
    }
    
}

void CAdvancedFindAndReplaceDlg::OnClose() 
{
    this->ShowWindow(SW_HIDE);

    
    CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
    m->SetForegroundWindow();

}

void CAdvancedFindAndReplaceDlg::OnFindmenuRemovefile() 
{
    vector<string> removeFiles;
    GetSelListCtrlItems(&m_files,&removeFiles);
    vector<int> removes;


    for ( int i = 0 ; i < removeFiles.size(); i++ )
    {
        for ( int j = 0 ; j < m_searchfiles.size(); j++ )
        {
            if ( StripFileName(m_searchfiles[j]) == removeFiles[i] ) removes.push_back(j);
        }


    }

    

    for ( int k = 0; k < removes.size(); k++ )
    {
        int removeStart ;
        
        removeStart = removes[k] - k;
        m_searchfiles.erase(m_searchfiles.begin() + removeStart);


    }

    DeleteAllSelectedItems(&m_files);
    
}

void CAdvancedFindAndReplaceDlg::OnDblclkList2(NMHDR* pNMHDR, LRESULT* pResult) 
{
    vector<string> files;
    vector<string> lineNumbers;
    GetSelListCtrlItems(&m_matches,&files,3 );
    GetSelListCtrlItems(&m_matches,&lineNumbers,0 );

    if ( files.size() )
    {
        CMainFrame* m = ( CMainFrame* ) Globals::theApp.m_pMainWnd;
        m->GotoFileAndLineNumber(files[0],atoi(lineNumbers[0].c_str()));

    }

    this->SetFocus();
    
    *pResult = 0;
}


void CAdvancedFindAndReplaceDlg::Populate()
{

    m_files.DeleteAllItems();
    m_searchfiles.clear();
    
    vector<string> files;
    if ( Globals::currentProject )
        Split(Globals::currentProject->GetCache("files"),",",files);
    
    for ( int i = 0 ; i < files.size(); i ++ )
    {
        m_files.InsertItem(i,StripFileName(files[i] ).c_str() );
        m_searchfiles.push_back(files[i] );
    }

}

void CAdvancedFindAndReplaceDlg::OnReplceAll() 
{
    UpdateData();

    m_matches.DeleteAllItems();
    
    string options = m_csOptions.GetBuffer(0);
    string replace = m_csReplace.GetBuffer(0);
    if ( replace == m_replace.GetText() ) replace = "";
    if ( options == m_options.GetText() ) options = "";
    
    CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
    m->m_wndOutputBar.m_buildReport.m_buildList.DeleteAllItems();

    //vector<string> files = m->CloseAllNamedFiles();

    vector<Listing> matches = grep->GrepReplaceFiles(m_searchfiles,m_csFind.GetBuffer(0),options, replace);
    
    //1m->ReopenFiles(files);
    m->ReloadAllNamedFiles();

    for ( int i = 0 ; i < matches.size(); i ++ )
    {
        
        m_matches.InsertItem(i,ITOA(matches[i].lineNumber).c_str())     ;
        m_matches.SetItemText(i,1,string(StripFileName(matches[i].file) + " [ altered ] ").c_str() );
        m_matches.SetItemText(i, 2 , string( WhiteSpace::ReplaceWhiteSpaceWithBlanks(matches[i].match) + " [altered] " ).c_str() );
        m_matches.SetItemText(i, 3 , matches[i].file.c_str() );
        
    }

    CString totals;

    totals.Format("%d Total Matches Replaced [ %s -> %s ]",i,m_csFind.GetBuffer(0),m_csReplace.GetBuffer(0));

    m_found = totals;
    UpdateData(FALSE);
 
    
    
    
}

void CAdvancedFindAndReplaceDlg::OnFilematchesmenuUndo() 
{

    CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
    //vector<string> files = m->CloseAllNamedFiles();



    if ( ! grep->Undo() )
    {
        m_found = "Sorry, 'undos' are available only after 'replace alls'.";
        
    }
    else
    {
    
    
    OnFind();

//  m->ReopenFiles(files);
    m->ReloadAllNamedFiles();
    
    CString totals;
    totals.Format("%d Total Files Undone ",m_searchfiles.size() );
    m_found = totals;


    
    }

    UpdateData(FALSE);
}

void CAdvancedFindAndReplaceDlg::OnRelpaceSelected() 
{

    UpdateData();
    
    CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
    //vector<string> files = m->CloseAllNamedFiles();


    vector<string> matches, matchLineNos;
    GetSelListCtrlItems(&m_matches,&matchLineNos);
    GetSelListCtrlItems(&m_matches,&matches,3);

    if ( !matches.size()) return;
    
    if ( matches.size() >= 2 ) 
    {
        if ( matches[0] == "" || matches[1] == "" ) {
        
            ElephantMessageBox("Please select a line to replace.",this);
            return;

        }
    }
    
    string options = m_csOptions.GetBuffer(0);
    if ( options == m_options.GetText() ) options = "";
    string replace = m_csReplace.GetBuffer(0);
    if ( replace == m_replace.GetText() ) replace = "";

    
    string regex = m_csFind.GetBuffer(0), replaceRegex  = replace;
    
    for ( int i = 0 ; i < matches.size();i++ )
    {
        
        int lineNo = atoi(matchLineNos[i].c_str() );
        string path = matches[i];
        
        if ( !ReplaceLine(path,lineNo,regex,replaceRegex,options) )
        {
            ElephantMessageBox(string("Could not replace " + path + " :: line number :: " + matches[0]).c_str() ,this);
            return;
        }
        
    }
    m->ReloadAllNamedFiles();
    //m->ReopenFiles(files);

    //OnFind();

    CString totals;
    totals.Format("%d Total Matches Replaced [ %s -> %s ] ",matches.size() ,m_csFind.GetBuffer(0) , m_csReplace.GetBuffer(0) );
    m_found = totals;

    UpdateData(FALSE);


}

void CAdvancedFindAndReplaceDlg::OnFindmenuSetfilter() 
{
CGetInputDlg dlg("Please enter the filter. Example: *.d","Example: *.d",this );
if ( dlg.DoModal() )
{
    
    m_filter = dlg.m_csValue.GetBuffer(0) ;
}

}

void CAdvancedFindAndReplaceDlg::OnFindmenuAdddir() 
{

    CDirDialog d;
    string dir = "";
    if ( d.DoBrowse() == IDOK )
    {

        dir = d.m_strPath.GetBuffer(0);


    }

    if ( dir != "" )
    {
        string filter = m_filter != "" ? "*" + m_filter : "**.d";
        vector<string> dirs; dirs.push_back(dir);
        ElephantFileGlob glob;
        glob.GetFilesRecursive(dirs,filter );

        vector<string> files = glob.matches;

        for ( int i = 0 ; i < files.size(); i ++ )
        {
            m_files.InsertItem(i,StripFileName(files[i] ).c_str() );
            m_searchfiles.push_back(files[i] );
        }

    }


}

void CAdvancedFindAndReplaceDlg::OnFindmenuClearfiles() 
{

    m_searchfiles.clear();
    m_filter = "";
    m_files.DeleteAllItems();
}