root/trunk/FileTreeCtrl.cpp

// FileTreeCtrl.cpp : implementation file
//

#include "stdafx.h"
#include "Elephant.h"
#include "FileTreeCtrl.h"

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

/////////////////////////////////////////////////////////////////////////////
// CFileTreeCtrl

CFileTreeCtrl::CFileTreeCtrl()
{
}

CFileTreeCtrl::~CFileTreeCtrl()
{
}

BEGIN_MESSAGE_MAP(CFileTreeCtrl, CXTTreeCtrl)
ON_WM_RBUTTONDOWN() 
ON_WM_LBUTTONDBLCLK()
ON_WM_KEYUP()
//{{AFX_MSG_MAP(CFileTreeCtrl)
ON_WM_CREATE()
ON_NOTIFY_REFLECT(TVN_ENDLABELEDIT, OnEndlabeledit)
//}}AFX_MSG_MAP
    //ON_WM_DROPFILES()

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CFileTreeCtrl message handlers

void CFileTreeCtrl::OnKeyUp(UINT nChar, UINT nRepCnt, UINT nFlags)
{

    if ( nChar == VK_DELETE )
    {


      CMainFrame* mf = static_cast<CMainFrame*>(Globals::theApp.m_pMainWnd );
      if ( mf ) mf->OnFiletreeRemovefile();
    }

    else if (nChar == VK_RETURN )
    {

        OnLButtonDblClk(0,CPoint(0,0 ) );



    }
    else    if ( nChar == VK_ESCAPE ) 
    {

        CMainFrame* m = (CMainFrame*)Globals::theApp.m_pMainWnd;
        if ( m ){
        
        CElephantView* v = m->GetActiveView();
        if ( v )
        {
            CScintillaCtrl& rCtrl = v->GetCtrl();
            rCtrl.SetFocus();
        }

        return;
        }
        


    }



}

BOOL CFileTreeCtrl::PreCreateWindow(CREATESTRUCT& cs) 
{
    if ( !CXTTreeCtrl::PreCreateWindow( cs ) )
        return FALSE;
    
    cs.style |= TVS_HASBUTTONS|TVS_HASLINES|TVS_LINESATROOT|TVS_EDITLABELS;
    
    if (!xtAfxData.bXPMode)
    {
        cs.dwExStyle |= WS_EX_CLIENTEDGE;
    }
    
    
    return TRUE;
}

int CFileTreeCtrl::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
    if (CXTTreeCtrl::OnCreate(lpCreateStruct) == -1)
        return -1;
    
    if ( !m_imageList.Create( 16, 16, ILC_COLOR24 | ILC_MASK, 1, 1 ) )
    {
        TRACE0("Failed to create image list.\n");
        return -1;
    }
    
    m_bitmap.LoadBitmap( IDB_IMGLIST_VIEW );
    m_imageList.Add( &m_bitmap, RGB( 0x00,0xff,0x00 ) );
    
    SetImageList( &m_imageList, TVSIL_NORMAL );
    
    
    
    return 0;
}

void CFileTreeCtrl::OnEndlabeledit(NMHDR* pNMHDR, LRESULT* pResult) 
{
    TV_DISPINFO* pTVDispInfo = (TV_DISPINFO*)pNMHDR;
    
    if ( pTVDispInfo->item.pszText )
    {
        SetItemText( pTVDispInfo->item.hItem,
            pTVDispInfo->item.pszText );
    }
    
    *pResult = 0;
}


void CFileTreeCtrl::OnRButtonDown(UINT nFlags, CPoint point) 
{
    
    CPoint pt = point;
    ClientToScreen(&pt);
    
    CXTMenu menu;
    VERIFY(menu.LoadMenu(IDR_FILETREE_MENU));
    
    CWnd* pWndPopupOwner = this;
    while (pWndPopupOwner->GetStyle() & WS_CHILD)
        pWndPopupOwner = pWndPopupOwner->GetParent();
    
    CXTMenu* pPopup = menu.GetSubMenu(0);
    pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON,
        pt.x, pt.y, pWndPopupOwner);
    
}


void CFileTreeCtrl::OnLButtonDblClk(UINT nFlags, CPoint point)
{

    
    HTREEITEM hCurSel = GetNextItem(TVI_ROOT, TVGN_CARET);
        //string sel = GetSelTreeCtrlItem(this);
    
    char* path = (char*)this->GetItemData(hCurSel);
    
    //AfxMessageBox(path);
    
    if ( Globals::currentProject && path != NULL )
    {
        //ElephantMessageBox(Globals::currentProject->PathForFile(sel).c_str());
            Globals::theApp.OpenFile(path);
    }
}

 
bool CFileTreeCtrl::IsFolder(HTREEITEM item )
{
    int image , selImage = 0;
    this->GetItemImage(item,image,selImage);
    //ephantMessageBox(ITOA(image));  
    
    return image == 2;
}
void CFileTreeCtrl::RemoveFolder()

    {
        HTREEITEM item = NULL;
        string pname = Globals::currentProject->name;

        if ( ( item = GetNextItem(TVI_ROOT, TVGN_CARET) ) != NULL && GetSelTreeCtrlItem(this) != "Work Desktop" && GetSelTreeCtrlItem(this) != pname)
        {
            if ( ItemHasChildren(item))
            {
                Globals::theApp.SetStatusBarText("Remove folder FAILED. Sorry the folder you selected has files underneath it, please ensure the selected folder is empty.");
                
            }
            else {
                Globals::theApp.SetStatusBarText("Remove folder SUCCEEDED : removed folder [ " + GetSelTreeCtrlItem(this) + " ] .");
                project->RemoveFolder(GetSelFolderPath() );
                DeleteItem(item);
            }
            
        }
        else Globals::theApp.SetStatusBarText("Remove folder FAILED. Sorry what you have selected is not a folder.");

    }
void CFileTreeCtrl::AddFile(const string& path, bool create,bool useDefaultDir)
{
    if ( !Globals::currentProject ) return;
    string pname = Globals::currentProject->name;
    
        HTREEITEM hCurSel = NULL;

        // If its not a folder, well just add it to the first listed folder available
        if ( useDefaultDir )
        {
            vector<string> folders;
            Split(Globals::currentProject->GetCache("ProjectFolders"),",",folders );
            if ( !folders.size() ) return;
            hCurSel = projectMap[folders[0] ];

            
        }
        else  hCurSel = GetNextItem(TVI_ROOT, TVGN_CARET);
        
        
        string folder = GetSelFolderPath(hCurSel);      

        CString x = GetItemText(hCurSel);
        string folderName = x.GetBuffer(0);
        

        //ElephantMessageBox(folder.c_str() );
        //AfxMessageBox(path.c_str());
        string file = GetRelativePath(project->path , StripDirPath(path)) + StripFileName(path);
        //ElephantMessageBox(file.c_str());
        //  ElephantMessageBox(folderName.c_str());


        


        Globals::currentProject->AddFile(file,folderName );
        HTREEITEM parent = projectMap[folderName];
        
        int img = 0;
        if ( FileExtension(file) != "d") img = -1;
        
        
        HTREEITEM addedItem = InsertItem(StripFileName(path).c_str(),img,img,parent);
        this->SetItemData(addedItem, (DWORD) strdup(path.c_str()));
        
        if ( create )
        {
            ofstream out(file.c_str());
            out << "";
            out.close();
        }
        Globals::theApp.OpenFile(file);
        Expand(parent,TVE_EXPAND);
        

    
}

string CFileTreeCtrl::GetSelFolderPath(HTREEITEM hCurSel )
{
    string path;


    if ( hCurSel == NULL )
    {
        hCurSel = GetNextItem(TVI_ROOT, TVGN_CARET);
    }


    path = GetItemText(hCurSel).GetBuffer(0);
    if ( ! IsFolder(hCurSel) ) path = "";
    //ElephantMessageBox(path.c_str());
    HTREEITEM sel = NULL;
    while ( ( sel = GetParentItem(hCurSel) ) != NULL )
    {
        if ( GetParentItem(sel) != NULL ) {
            
            string temp = GetItemText(sel).GetBuffer(0);
            path = temp + "//" + path;
            hCurSel = sel;
        }
        else hCurSel = sel;
    }

    if ( path[path.length() - 1 ] == '/' )
    {
        char buffer[100];
        sprintf(buffer,"C : %c", path[path.length() - 1]);
//      ElephantMessageBox(buffer);
    
        path = path.substr(0,path.length() - 2 );
    }

//  ElephantMessageBox(path.c_str());
    return path;
}


void CFileTreeCtrl::RemoveFile()
{
    if ( !Globals::currentProject ) return;
    string pname = Globals::currentProject->name;
    HTREEITEM item = NULL;
    if ( ( item = GetNextItem(TVI_ROOT, TVGN_CARET) ) != NULL && GetSelTreeCtrlItem(this) != "Work Desktop" && GetSelTreeCtrlItem(this) != pname)
    {
        if ( ItemHasChildren(item))
        {
            Globals::theApp.SetStatusBarText("Remove File FAILED : Sorry this looks like a directory.");
            //ElephantMessageBox("This looks like a directory.");
        }
        
        else {
            
            

            string path = Globals::currentProject->PathForFile(GetItemText(item).GetBuffer(0));
            if ( Globals::currentProject->IsBuildTarget(path))
            {
                ElephantMessageBox("Did you know Elephant maintains the build target file so you dont have to?  Unfortunately that means you cannot remove this file sorry!");
                return;
            }
            project->RemoveFile(GetItemText(item).GetBuffer(0));            
            //AfxMessageBox(path.c_str());
            Globals::theApp.RemoveImportToCurrentProject(path);
            DeleteItem(item);
            
            vector<string> myFiles;
            Split(Globals::currentProject->GetCache("files"),",",myFiles);
            string reloadPath = Globals::currentProject->path + "\\" + myFiles[0];
            //AfxMessageBox(reloadPath.c_str());
            CMainFrame* mf = static_cast<CMainFrame*>(Globals::theApp.m_pMainWnd );
            mf->ReloadNamedFile(CString(reloadPath.c_str() ) );
        }
    }
    
}