src/generic/dirdlgg.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        dirdlg.cpp
// Purpose:     wxDirDialog
// Author:      Harm van der Heijden and Robert Roebling
// Modified by:
// Created:     12/12/98
// RCS-ID:      $Id$
// Copyright:   (c) Harm van der Heijden and Robert Roebling
// Licence:     wxWindows licence
/////////////////////////////////////////////////////////////////////////////

#ifdef __GNUG__
#pragma implementation "dirdlgg.h"
#endif

// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#include "wx/defs.h"

#if wxUSE_DIRDLG

#include "wx/utils.h"
#include "wx/dialog.h"
#include "wx/button.h"
#include "wx/layout.h"
#include "wx/msgdlg.h"
#include "wx/textdlg.h"
#include "wx/filefn.h"
#include "wx/cmndata.h"
#include "wx/gdicmn.h"
#include "wx/intl.h"
#include "wx/imaglist.h"
#include "wx/icon.h"
#include "wx/log.h"
#include "wx/sizer.h"
#include "wx/tokenzr.h"

#if wxUSE_STATLINE
    #include "wx/statline.h"
#endif

#include "wx/generic/dirdlgg.h"

// If compiled under Windows, this macro can cause problems
#ifdef GetFirstChild
#undef GetFirstChild
#endif

#ifndef __WXMSW__
/* XPM */
static char * icon1_xpm[] = {
/* width height ncolors chars_per_pixel */
"16 16 6 1",
/* colors */
"   s None  c None",
".  c #000000",
"+  c #c0c0c0",
"@  c #808080",
"#  c #ffff00",
"$  c #ffffff",
/* pixels */
"                ",
"   @@@@@        ",
"  @#+#+#@       ",
" @#+#+#+#@@@@@@ ",
" @$$$$$$$$$$$$@.",
" @$#+#+#+#+#+#@.",
" @$+#+#+#+#+#+@.",
" @$#+#+#+#+#+#@.",
" @$+#+#+#+#+#+@.",
" @$#+#+#+#+#+#@.",
" @$+#+#+#+#+#+@.",
" @$#+#+#+#+#+#@.",
" @@@@@@@@@@@@@@.",
"  ..............",
"                ",
"                "};

/* XPM */
static char * icon2_xpm[] = {
/* width height ncolors chars_per_pixel */
"16 16 6 1",
/* colors */
"   s None  c None",
".  c #000000",
"+  c #c0c0c0",
"@  c #808080",
"#  c #ffff00",
"$  c #ffffff",
/* pixels */
"                ",
"   @@@@@        ",
"  @$$$$$@       ",
" @$#+#+#$@@@@@@ ",
" @$+#+#+$$$$$$@.",
" @$#+#+#+#+#+#@.",
"@@@@@@@@@@@@@#@.",
"@$$$$$$$$$$@@+@.",
"@$#+#+#+#+##.@@.",
" @$#+#+#+#+#+.@.",
" @$+#+#+#+#+#.@.",
"  @$+#+#+#+##@..",
"  @@@@@@@@@@@@@.",
"   .............",
"                ",
"                "};

#endif // !wxMSW

static const int ID_DIRCTRL = 1000;
static const int ID_TEXTCTRL = 1001;
static const int ID_OK = 1002;
static const int ID_CANCEL = 1003;
static const int ID_NEW = 1004;
//static const int ID_CHECK = 1005;

//-----------------------------------------------------------------------------
// wxDirItemData
//-----------------------------------------------------------------------------

wxDirItemData::wxDirItemData(wxString& path, wxString& name)
{
    m_path = path;
    m_name = name;
    /* Insert logic to detect hidden files here
     * In UnixLand we just check whether the first char is a dot
     * For FileNameFromPath read LastDirNameInThisPath ;-) */
    // m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
    m_isHidden = FALSE;
    m_hasSubDirs = HasSubDirs();
}

wxDirItemData::~wxDirItemData()
{
}

void wxDirItemData::SetNewDirName( wxString path )
{
    m_path = path;
    m_name = wxFileNameFromPath( path );
}

bool wxDirItemData::HasSubDirs()
{
    wxString search = m_path + wxT("/*");
    wxLogNull log;
    wxString path = wxFindFirstFile( search, wxDIR );
    return (bool)(!path.IsNull());
}

//-----------------------------------------------------------------------------
// wxDirCtrl
//-----------------------------------------------------------------------------

IMPLEMENT_DYNAMIC_CLASS(wxDirCtrl,wxTreeCtrl)

BEGIN_EVENT_TABLE(wxDirCtrl,wxTreeCtrl)
  EVT_TREE_ITEM_EXPANDING     (-1, wxDirCtrl::OnExpandItem)
  EVT_TREE_ITEM_COLLAPSED     (-1, wxDirCtrl::OnCollapseItem)
  EVT_TREE_BEGIN_LABEL_EDIT   (-1, wxDirCtrl::OnBeginEditItem)
  EVT_TREE_END_LABEL_EDIT     (-1, wxDirCtrl::OnEndEditItem)
END_EVENT_TABLE()

wxDirCtrl::wxDirCtrl(void)
{
    m_showHidden = FALSE;
}

wxDirCtrl::wxDirCtrl(wxWindow *parent,
                     const wxWindowID id,
                     const wxString &WXUNUSED(dir),
                     const wxPoint& pos,
                     const wxSize& size,
                     const long style,
                     const wxString& name )
         : wxTreeCtrl( parent, id, pos, size, style, wxDefaultValidator, name )
{
#ifndef __WXMSW__
    m_imageListNormal = new wxImageList(16, 16, TRUE);
    m_imageListNormal->Add(wxICON(icon1));
    m_imageListNormal->Add(wxICON(icon2));
    SetImageList(m_imageListNormal);
#endif // !MSW

    m_showHidden = FALSE;
    m_rootId = AddRoot( _("Sections") );
    SetItemHasChildren(m_rootId);
    Expand(m_rootId); // automatically expand first level
}

/* Quick macro. Don't worry, I'll #undef it later */
#define ADD_SECTION(a,b) \
  if (wxPathExists((a))) { m_paths.Add( (a) ); m_names.Add( (b) ); };

void wxDirCtrl::SetupSections()
{
  wxString home;

  m_paths.Clear();
  m_names.Clear();
#ifdef __WXMSW__
  // better than nothing
  ADD_SECTION(wxT("c:\\"), _("My Harddisk") )
#else
  ADD_SECTION(wxT("/"), _("The Computer") )
  wxGetHomeDir(&home);
  ADD_SECTION(home, _("My Home") )
  ADD_SECTION(wxT("/mnt"), _("Mounted Devices") )
  ADD_SECTION(wxT("/usr/local"), _("User Local") )
  ADD_SECTION(wxT("/usr"), _("User") )
  ADD_SECTION(wxT("/var"), _("Variables") )
  ADD_SECTION(wxT("/etc"), _("Etcetera") )
  ADD_SECTION(wxT("/tmp"), _("Temporary") )
#endif
}
#undef ADD_SECTION

void wxDirCtrl::CreateItems(const wxTreeItemId &parent)
{
    wxTreeItemId id;
    wxDirItemData *dir_item;

//  wxASSERT(m_paths.Count() == m_names.Count());  ?

    size_t count = m_paths.GetCount();
    for ( size_t i=0; i<count; i++)
    {
        dir_item = new wxDirItemData(m_paths[i],m_names[i]);
#ifdef __WXMSW__
        id = AppendItem( parent, m_names[i], -1, -1, dir_item);
#else
        id = AppendItem( parent, m_names[i], 0, -1, dir_item);
        SetItemImage( id, 1, wxTreeItemIcon_Expanded );
#endif
        if (dir_item->m_hasSubDirs) SetItemHasChildren(id);
    }
}

void wxDirCtrl::OnBeginEditItem(wxTreeEvent &event)
{
    // don't rename the main entry "Sections"
    if (event.GetItem() == m_rootId)
    {
        event.Veto();
        return;
    }

    // don't rename the individual sections
    if (GetParent( event.GetItem() ) == m_rootId)
    {
        event.Veto();
        return;
    }
}

void wxDirCtrl::OnEndEditItem(wxTreeEvent &event)
{
    if ((event.GetLabel().IsEmpty()) ||
        (event.GetLabel() == _(".")) ||
        (event.GetLabel() == _("..")) ||
        (event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
    {
        wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK | wxICON_ERROR );
        dialog.ShowModal();
        event.Veto();
        return;
    }

    wxTreeItemId id = event.GetItem();
    wxDirItemData *data = (wxDirItemData*)GetItemData( id );
    wxASSERT( data );

    wxString new_name( wxPathOnly( data->m_path ) );
    new_name += wxT("/");
    new_name += event.GetLabel();

    wxLogNull log;

    if (wxFileExists(new_name))
    {
        wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK | wxICON_ERROR );
        dialog.ShowModal();
        event.Veto();
    }

    if (wxRenameFile(data->m_path,new_name))
    {
        data->SetNewDirName( new_name );
    }
    else
    {
        wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
        dialog.ShowModal();
        event.Veto();
    }
}

void wxDirCtrl::OnExpandItem(wxTreeEvent &event)
{
    if (event.GetItem() == m_rootId)
    {
        SetupSections();
        CreateItems(m_rootId);
        return;
    }

    // This may take a longish time. Go to busy cursor
    wxBeginBusyCursor();

    wxDirItemData *data = (wxDirItemData *)GetItemData(event.GetItem());
    wxASSERT(data);

    wxString search,path,filename;

    m_paths.Clear();
    m_names.Clear();
#ifdef __WXMSW__
    search = data->m_path + _T("\\*.*");
#else
    search = data->m_path + _T("/*");
#endif
    for (path = wxFindFirstFile( search, wxDIR ); !path.IsNull();
       path=wxFindNextFile() )
    {
        filename = wxFileNameFromPath( path );
        /* Don't add "." and ".." to the tree. I think wxFindNextFile
         * also checks this, but I don't quite understand what happens
         * there. Also wxFindNextFile seems to swallow hidden dirs */
        if ( (filename != _T(".")) && (filename != _T("..")) )
        {
            m_paths.Add(path);
            m_names.Add(filename);
        }
    }

    CreateItems( event.GetItem() );
    SortChildren( event.GetItem() );

    wxEndBusyCursor();
}

void wxDirCtrl::OnCollapseItem(wxTreeEvent &event )
{
    wxTreeItemId child, parent = event.GetItem();
    long cookie;
    /* Workaround because DeleteChildren has disapeared (why?) and
     * CollapseAndReset doesn't work as advertised (deletes parent too) */
    child = GetFirstChild(parent, cookie);
    while (child.IsOk())
    {
        Delete(child);
        /* Not GetNextChild below, because the cookie mechanism can't
         * handle disappearing children! */
        child = GetFirstChild(parent, cookie);
    }
}

//-----------------------------------------------------------------------------
// wxDirDialog
//-----------------------------------------------------------------------------


IMPLEMENT_DYNAMIC_CLASS( wxDirDialog, wxDialog )

BEGIN_EVENT_TABLE( wxDirDialog, wxDialog )
  EVT_TREE_KEY_DOWN        (ID_DIRCTRL,   wxDirDialog::OnTreeKeyDown)
  EVT_TREE_SEL_CHANGED     (ID_DIRCTRL,   wxDirDialog::OnTreeSelected)
  EVT_SIZE                 (              wxDirDialog::OnSize)
  EVT_BUTTON               (ID_OK,        wxDirDialog::OnOK)
  EVT_BUTTON               (ID_CANCEL,    wxDirDialog::OnCancel)
  EVT_BUTTON               (ID_NEW,       wxDirDialog::OnNew)
  EVT_TEXT_ENTER           (ID_TEXTCTRL,  wxDirDialog::OnOK)
  //  EVT_CHECKBOX             (ID_CHECK,     wxDirDialog::OnCheck)
END_EVENT_TABLE()

wxDirDialog::wxDirDialog(wxWindow *parent,
                         const wxString& message,
                         const wxString& defaultPath,
                         long style,
                         const wxPoint& pos)
           : wxDialog(parent, -1, message, pos, wxSize(300,300),
                      wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER)
{
    m_message = message;
    m_dialogStyle = style;
    m_parent = parent;

    m_path = defaultPath;

    wxBeginBusyCursor();

    wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );

    // 1) dir ctrl
    m_dir = new wxDirCtrl( this, ID_DIRCTRL, _T("/"),
                           wxDefaultPosition,
                           wxSize(200,200),
                           wxTR_HAS_BUTTONS |
                           wxSUNKEN_BORDER |
                           wxTR_EDIT_LABELS );
    topsizer->Add( m_dir, 1, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, 10 );

    // 2) text ctrl
    m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition );
    topsizer->Add( m_input, 0, wxTOP|wxLEFT|wxRIGHT | wxEXPAND, 10 );

    // m_check = new wxCheckBox( this, ID_CHECK, _("Show hidden") );
    // m_check->SetValue(TRUE);

#if wxUSE_STATLINE
    // 3) static line
    topsizer->Add( new wxStaticLine( this, -1 ), 0, wxEXPAND | wxLEFT|wxRIGHT|wxTOP, 10 );
#endif

    // 4) buttons
    wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
    m_ok = new wxButton( this, ID_OK, _("OK") );
    buttonsizer->Add( m_ok, 0, wxLEFT|wxRIGHT, 10 );
    m_cancel = new wxButton( this, ID_CANCEL, _("Cancel") );
    buttonsizer->Add( m_cancel, 0, wxLEFT|wxRIGHT, 10 );
    m_new = new wxButton( this, ID_NEW, _("New...") );
    buttonsizer->Add( m_new, 0, wxLEFT|wxRIGHT, 10 );

    topsizer->Add( buttonsizer, 0, wxALL | wxCENTER, 10 );

    m_ok->SetDefault();
    m_dir->SetFocus();

    SetAutoLayout( TRUE );
    SetSizer( topsizer );

    topsizer->SetSizeHints( this );
    topsizer->Fit( this );

    Centre( wxBOTH );

    if (m_path == wxT("~"))
        wxGetHomeDir( &m_path );

    // choose the directory corresponding to defaultPath in the tree
    // VZ: using wxStringTokenizer is probably unsafe here (escaped slashes
    //     will not be processed correctly...)
    wxStringTokenizer tk(m_path, wxFILE_SEP_PATH, wxTOKEN_STRTOK);

    wxString path;

    long cookie = 0;
    // default to root dir
    wxTreeItemId item = m_dir->GetFirstChild(m_dir->GetRootItem(), cookie);

    if (!m_path.IsEmpty() && (m_path != wxT("/")) && (m_dir->m_paths.Count() > 1))
    {
        size_t count = m_dir->m_paths.GetCount();
        for ( size_t i=1; i<count; i++)
        {
            if (m_path.Find( m_dir->m_paths[i] ) == 0)
            {
                path = m_dir->m_paths[i];

                for (size_t j = 0; j < i; j++)
                   item = m_dir->GetNextChild(m_dir->GetRootItem(), cookie);

                wxStringTokenizer tk2(path, wxFILE_SEP_PATH, wxTOKEN_STRTOK);
                for (size_t h = 0; h < tk2.CountTokens(); h++)
                   tk.GetNextToken();

                break;
            }
        }
    }
    while ( tk.HasMoreTokens() && item.IsOk() )
    {
        path << wxFILE_SEP_PATH << tk.GetNextToken();

        m_dir->Expand(item);

        wxTreeItemId child = m_dir->GetFirstChild(item, cookie);
        while ( child.IsOk() )
        {
            wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(child);
            if ( data->m_path == path )
                break;

            child = m_dir->GetNextChild(item, cookie);
        }

        item = child;
    }

    if ( item.IsOk() )
    {
        m_dir->Expand(item);
        m_dir->SelectItem(item);
        m_dir->EnsureVisible(item);
    }

    wxEndBusyCursor();
}

int wxDirDialog::ShowModal()
{
    m_input->SetValue( m_path );
    return wxDialog::ShowModal();
}

void wxDirDialog::OnTreeSelected( wxTreeEvent &event )
{
    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(event.GetItem());
    if (data)
       m_input->SetValue( data->m_path );
};

void wxDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
{
    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData(m_dir->GetSelection());
    if (data)
        m_input->SetValue( data->m_path );
};

void wxDirDialog::OnOK( wxCommandEvent& WXUNUSED(event) )
{
  m_path = m_input->GetValue();
  // Does the path exist? (User may have typed anything in m_input)
  if (wxPathExists(m_path)) {
    // OK, path exists, we're done.
    EndModal(wxID_OK);
    return;
  }
  // Interact with user, find out if the dir is a typo or to be created
  wxString msg( _("The directory ") );
  msg = msg + m_path;
  msg = msg + _("\ndoes not exist\nCreate it now?") ;
  wxMessageDialog dialog(this, msg, _("Directory does not exist"), wxYES_NO | wxICON_WARNING );
  if ( dialog.ShowModal() == wxID_YES ) {
    // Okay, let's make it
    wxLogNull log;
    if (wxMkdir(m_path)) {
      // The new dir was created okay.
      EndModal(wxID_OK);
      return;
    }
    else {
      // Trouble...
      msg = _("Failed to create directory ")+m_path+
  _("\n(Do you have the required permissions?)");
      wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK | wxICON_ERROR);
      errmsg.ShowModal();
      // We still don't have a valid dir. Back to the main dialog.
    }
  }
  // User has answered NO to create dir.
}

void wxDirDialog::OnCancel( wxCommandEvent& WXUNUSED(event) )
{
  EndModal(wxID_CANCEL);
}

void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
{
    wxTreeItemId id = m_dir->GetSelection();
    if ((id == m_dir->GetRootItem()) ||
        (m_dir->GetParent(id) == m_dir->GetRootItem()))
    {
        wxMessageDialog msg(this, _("You cannot add a new directory to this section."),
                            _("Create directory"), wxOK | wxICON_INFORMATION );
        msg.ShowModal();
        return;
    }

    wxTreeItemId parent = m_dir->GetParent( id );
    wxDirItemData *data = (wxDirItemData*)m_dir->GetItemData( parent );
    wxASSERT( data );

    wxString new_name( wxT("NewName") );
    wxString path( data->m_path );
    path += wxT("/");
    path += new_name;
    if (wxFileExists(path))
    {
        // try NewName0, NewName1 etc.
        int i = 0;
        do {
            new_name = wxT("NewName");
            wxString num;
            num.Printf( wxT("%d"), i );
            new_name += num;

            path = data->m_path;
            path += wxT("/");
            path += new_name;
            i++;
        } while (wxFileExists(path));
    }

    wxLogNull log;
    if (!wxMkdir(path))
    {
        wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK | wxICON_ERROR );
        dialog.ShowModal();
        return;
    }

    wxDirItemData *new_data = new wxDirItemData( path, new_name );
    wxTreeItemId new_id = m_dir->AppendItem( parent, new_name, 0, 1, new_data );
    m_dir->EnsureVisible( new_id );
    m_dir->EditLabel( new_id );
}

/*
void wxDirDialog::OnCheck( wxCommandEvent& WXUNUSED(event) )
{
  printf("Checkbox clicked: %s\n", ( m_check->GetValue() ? "on" : "off" ) );
}
*/

#endif // wxUSE_DIRDLG
Commit	Line	Data
	1	/////////////////////////////////////////////////////////////////////////////
	2	// Name: dirdlg.cpp
	3	// Purpose: wxDirDialog
	4	// Author: Harm van der Heijden and Robert Roebling
	5	// Modified by:
	6	// Created: 12/12/98
	7	// RCS-ID: $Id$
	8	// Copyright: (c) Harm van der Heijden and Robert Roebling
	9	// Licence: wxWindows licence
	10	/////////////////////////////////////////////////////////////////////////////
	11
	12	#ifdef __GNUG__
	13	#pragma implementation "dirdlgg.h"
	14	#endif
	15
	16	// For compilers that support precompilation, includes "wx.h".
	17	#include "wx/wxprec.h"
	18
	19	#ifdef __BORLANDC__
	20	#pragma hdrstop
	21	#endif
	22
	23	#include "wx/defs.h"
	24
	25	#if wxUSE_DIRDLG
	26
	27	#include "wx/utils.h"
	28	#include "wx/dialog.h"
	29	#include "wx/button.h"
	30	#include "wx/layout.h"
	31	#include "wx/msgdlg.h"
	32	#include "wx/textdlg.h"
	33	#include "wx/filefn.h"
	34	#include "wx/cmndata.h"
	35	#include "wx/gdicmn.h"
	36	#include "wx/intl.h"
	37	#include "wx/imaglist.h"
	38	#include "wx/icon.h"
	39	#include "wx/log.h"
	40	#include "wx/sizer.h"
	41	#include "wx/tokenzr.h"
	42
	43	#if wxUSE_STATLINE
	44	#include "wx/statline.h"
	45	#endif
	46
	47	#include "wx/generic/dirdlgg.h"
	48
	49	// If compiled under Windows, this macro can cause problems
	50	#ifdef GetFirstChild
	51	#undef GetFirstChild
	52	#endif
	53
	54	#ifndef __WXMSW__
	55	/* XPM */
	56	static char * icon1_xpm[] = {
	57	/* width height ncolors chars_per_pixel */
	58	"16 16 6 1",
	59	/* colors */
	60	" s None c None",
	61	". c #000000",
	62	"+ c #c0c0c0",
	63	"@ c #808080",
	64	"# c #ffff00",
	65	"$ c #ffffff",
	66	/* pixels */
	67	" ",
	68	" @@@@@ ",
	69	" @#+#+#@ ",
	70	" @#+#+#+#@@@@@@ ",
	71	" @$$$$$$$$$$$$@.",
	72	" @$#+#+#+#+#+#@.",
	73	" @$+#+#+#+#+#+@.",
	74	" @$#+#+#+#+#+#@.",
	75	" @$+#+#+#+#+#+@.",
	76	" @$#+#+#+#+#+#@.",
	77	" @$+#+#+#+#+#+@.",
	78	" @$#+#+#+#+#+#@.",
	79	" @@@@@@@@@@@@@@.",
	80	" ..............",
	81	" ",
	82	" "};
	83
	84	/* XPM */
	85	static char * icon2_xpm[] = {
	86	/* width height ncolors chars_per_pixel */
	87	"16 16 6 1",
	88	/* colors */
	89	" s None c None",
	90	". c #000000",
	91	"+ c #c0c0c0",
	92	"@ c #808080",
	93	"# c #ffff00",
	94	"$ c #ffffff",
	95	/* pixels */
	96	" ",
	97	" @@@@@ ",
	98	" @$$$$$@ ",
	99	" @$#+#+#$@@@@@@ ",
	100	" @$+#+#+$$$$$$@.",
	101	" @$#+#+#+#+#+#@.",
	102	"@@@@@@@@@@@@@#@.",
	103	"@$$$$$$$$$$@@+@.",
	104	"@$#+#+#+#+##.@@.",
	105	" @$#+#+#+#+#+.@.",
	106	" @$+#+#+#+#+#.@.",
	107	" @$+#+#+#+##@..",
	108	" @@@@@@@@@@@@@.",
	109	" .............",
	110	" ",
	111	" "};
	112
	113	#endif // !wxMSW
	114
	115	static const int ID_DIRCTRL = 1000;
	116	static const int ID_TEXTCTRL = 1001;
	117	static const int ID_OK = 1002;
	118	static const int ID_CANCEL = 1003;
	119	static const int ID_NEW = 1004;
	120	//static const int ID_CHECK = 1005;
	121
	122	//-----------------------------------------------------------------------------
	123	// wxDirItemData
	124	//-----------------------------------------------------------------------------
	125
	126	wxDirItemData::wxDirItemData(wxString& path, wxString& name)
	127	{
	128	m_path = path;
	129	m_name = name;
	130	/* Insert logic to detect hidden files here
	131	* In UnixLand we just check whether the first char is a dot
	132	* For FileNameFromPath read LastDirNameInThisPath ;-) */
	133	// m_isHidden = (bool)(wxFileNameFromPath(*m_path)[0] == '.');
	134	m_isHidden = FALSE;
	135	m_hasSubDirs = HasSubDirs();
	136	}
	137
	138	wxDirItemData::~wxDirItemData()
	139	{
	140	}
	141
	142	void wxDirItemData::SetNewDirName( wxString path )
	143	{
	144	m_path = path;
	145	m_name = wxFileNameFromPath( path );
	146	}
	147
	148	bool wxDirItemData::HasSubDirs()
	149	{
	150	wxString search = m_path + wxT("/*");
	151	wxLogNull log;
	152	wxString path = wxFindFirstFile( search, wxDIR );
	153	return (bool)(!path.IsNull());
	154	}
	155
	156	//-----------------------------------------------------------------------------
	157	// wxDirCtrl
	158	//-----------------------------------------------------------------------------
	159
	160	IMPLEMENT_DYNAMIC_CLASS(wxDirCtrl,wxTreeCtrl)
	161
	162	BEGIN_EVENT_TABLE(wxDirCtrl,wxTreeCtrl)
	163	EVT_TREE_ITEM_EXPANDING (-1, wxDirCtrl::OnExpandItem)
	164	EVT_TREE_ITEM_COLLAPSED (-1, wxDirCtrl::OnCollapseItem)
	165	EVT_TREE_BEGIN_LABEL_EDIT (-1, wxDirCtrl::OnBeginEditItem)
	166	EVT_TREE_END_LABEL_EDIT (-1, wxDirCtrl::OnEndEditItem)
	167	END_EVENT_TABLE()
	168
	169	wxDirCtrl::wxDirCtrl(void)
	170	{
	171	m_showHidden = FALSE;
	172	}
	173
	174	wxDirCtrl::wxDirCtrl(wxWindow *parent,
	175	const wxWindowID id,
	176	const wxString &WXUNUSED(dir),
	177	const wxPoint& pos,
	178	const wxSize& size,
	179	const long style,
	180	const wxString& name )
	181	: wxTreeCtrl( parent, id, pos, size, style, wxDefaultValidator, name )
	182	{
	183	#ifndef __WXMSW__
	184	m_imageListNormal = new wxImageList(16, 16, TRUE);
	185	m_imageListNormal->Add(wxICON(icon1));
	186	m_imageListNormal->Add(wxICON(icon2));
	187	SetImageList(m_imageListNormal);
	188	#endif // !MSW
	189
	190	m_showHidden = FALSE;
	191	m_rootId = AddRoot( _("Sections") );
	192	SetItemHasChildren(m_rootId);
	193	Expand(m_rootId); // automatically expand first level
	194	}
	195
	196	/* Quick macro. Don't worry, I'll #undef it later */
	197	#define ADD_SECTION(a,b) \
	198	if (wxPathExists((a))) { m_paths.Add( (a) ); m_names.Add( (b) ); };
	199
	200	void wxDirCtrl::SetupSections()
	201	{
	202	wxString home;
	203
	204	m_paths.Clear();
	205	m_names.Clear();
	206	#ifdef __WXMSW__
	207	// better than nothing
	208	ADD_SECTION(wxT("c:\\"), _("My Harddisk") )
	209	#else
	210	ADD_SECTION(wxT("/"), _("The Computer") )
	211	wxGetHomeDir(&home);
	212	ADD_SECTION(home, _("My Home") )
	213	ADD_SECTION(wxT("/mnt"), _("Mounted Devices") )
	214	ADD_SECTION(wxT("/usr/local"), _("User Local") )
	215	ADD_SECTION(wxT("/usr"), _("User") )
	216	ADD_SECTION(wxT("/var"), _("Variables") )
	217	ADD_SECTION(wxT("/etc"), _("Etcetera") )
	218	ADD_SECTION(wxT("/tmp"), _("Temporary") )
	219	#endif
	220	}
	221	#undef ADD_SECTION
	222
	223	void wxDirCtrl::CreateItems(const wxTreeItemId &parent)
	224	{
	225	wxTreeItemId id;
	226	wxDirItemData *dir_item;
	227
	228	// wxASSERT(m_paths.Count() == m_names.Count()); ?
	229
	230	size_t count = m_paths.GetCount();
	231	for ( size_t i=0; i<count; i++)
	232	{
	233	dir_item = new wxDirItemData(m_paths[i],m_names[i]);
	234	#ifdef __WXMSW__
	235	id = AppendItem( parent, m_names[i], -1, -1, dir_item);
	236	#else
	237	id = AppendItem( parent, m_names[i], 0, -1, dir_item);
	238	SetItemImage( id, 1, wxTreeItemIcon_Expanded );
	239	#endif
	240	if (dir_item->m_hasSubDirs) SetItemHasChildren(id);
	241	}
	242	}
	243
	244	void wxDirCtrl::OnBeginEditItem(wxTreeEvent &event)
	245	{
	246	// don't rename the main entry "Sections"
	247	if (event.GetItem() == m_rootId)
	248	{
	249	event.Veto();
	250	return;
	251	}
	252
	253	// don't rename the individual sections
	254	if (GetParent( event.GetItem() ) == m_rootId)
	255	{
	256	event.Veto();
	257	return;
	258	}
	259	}
	260
	261	void wxDirCtrl::OnEndEditItem(wxTreeEvent &event)
	262	{
	263	if ((event.GetLabel().IsEmpty()) \|\|
	264	(event.GetLabel() == _(".")) \|\|
	265	(event.GetLabel() == _("..")) \|\|
	266	(event.GetLabel().First( wxT("/") ) != wxNOT_FOUND))
	267	{
	268	wxMessageDialog dialog(this, _("Illegal directory name."), _("Error"), wxOK \| wxICON_ERROR );
	269	dialog.ShowModal();
	270	event.Veto();
	271	return;
	272	}
	273
	274	wxTreeItemId id = event.GetItem();
	275	wxDirItemData data = (wxDirItemData)GetItemData( id );
	276	wxASSERT( data );
	277
	278	wxString new_name( wxPathOnly( data->m_path ) );
	279	new_name += wxT("/");
	280	new_name += event.GetLabel();
	281
	282	wxLogNull log;
	283
	284	if (wxFileExists(new_name))
	285	{
	286	wxMessageDialog dialog(this, _("File name exists already."), _("Error"), wxOK \| wxICON_ERROR );
	287	dialog.ShowModal();
	288	event.Veto();
	289	}
	290
	291	if (wxRenameFile(data->m_path,new_name))
	292	{
	293	data->SetNewDirName( new_name );
	294	}
	295	else
	296	{
	297	wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK \| wxICON_ERROR );
	298	dialog.ShowModal();
	299	event.Veto();
	300	}
	301	}
	302
	303	void wxDirCtrl::OnExpandItem(wxTreeEvent &event)
	304	{
	305	if (event.GetItem() == m_rootId)
	306	{
	307	SetupSections();
	308	CreateItems(m_rootId);
	309	return;
	310	}
	311
	312	// This may take a longish time. Go to busy cursor
	313	wxBeginBusyCursor();
	314
	315	wxDirItemData data = (wxDirItemData )GetItemData(event.GetItem());
	316	wxASSERT(data);
	317
	318	wxString search,path,filename;
	319
	320	m_paths.Clear();
	321	m_names.Clear();
	322	#ifdef __WXMSW__
	323	search = data->m_path + _T("\\.");
	324	#else
	325	search = data->m_path + _T("/*");
	326	#endif
	327	for (path = wxFindFirstFile( search, wxDIR ); !path.IsNull();
	328	path=wxFindNextFile() )
	329	{
	330	filename = wxFileNameFromPath( path );
	331	/* Don't add "." and ".." to the tree. I think wxFindNextFile
	332	* also checks this, but I don't quite understand what happens
	333	* there. Also wxFindNextFile seems to swallow hidden dirs */
	334	if ( (filename != _T(".")) && (filename != _T("..")) )
	335	{
	336	m_paths.Add(path);
	337	m_names.Add(filename);
	338	}
	339	}
	340
	341	CreateItems( event.GetItem() );
	342	SortChildren( event.GetItem() );
	343
	344	wxEndBusyCursor();
	345	}
	346
	347	void wxDirCtrl::OnCollapseItem(wxTreeEvent &event )
	348	{
	349	wxTreeItemId child, parent = event.GetItem();
	350	long cookie;
	351	/* Workaround because DeleteChildren has disapeared (why?) and
	352	* CollapseAndReset doesn't work as advertised (deletes parent too) */
	353	child = GetFirstChild(parent, cookie);
	354	while (child.IsOk())
	355	{
	356	Delete(child);
	357	/* Not GetNextChild below, because the cookie mechanism can't
	358	* handle disappearing children! */
	359	child = GetFirstChild(parent, cookie);
	360	}
	361	}
	362
	363	//-----------------------------------------------------------------------------
	364	// wxDirDialog
	365	//-----------------------------------------------------------------------------
	366
	367
	368	IMPLEMENT_DYNAMIC_CLASS( wxDirDialog, wxDialog )
	369
	370	BEGIN_EVENT_TABLE( wxDirDialog, wxDialog )
	371	EVT_TREE_KEY_DOWN (ID_DIRCTRL, wxDirDialog::OnTreeKeyDown)
	372	EVT_TREE_SEL_CHANGED (ID_DIRCTRL, wxDirDialog::OnTreeSelected)
	373	EVT_SIZE ( wxDirDialog::OnSize)
	374	EVT_BUTTON (ID_OK, wxDirDialog::OnOK)
	375	EVT_BUTTON (ID_CANCEL, wxDirDialog::OnCancel)
	376	EVT_BUTTON (ID_NEW, wxDirDialog::OnNew)
	377	EVT_TEXT_ENTER (ID_TEXTCTRL, wxDirDialog::OnOK)
	378	// EVT_CHECKBOX (ID_CHECK, wxDirDialog::OnCheck)
	379	END_EVENT_TABLE()
	380
	381	wxDirDialog::wxDirDialog(wxWindow *parent,
	382	const wxString& message,
	383	const wxString& defaultPath,
	384	long style,
	385	const wxPoint& pos)
	386	: wxDialog(parent, -1, message, pos, wxSize(300,300),
	387	wxDEFAULT_DIALOG_STYLE \| wxRESIZE_BORDER)
	388	{
	389	m_message = message;
	390	m_dialogStyle = style;
	391	m_parent = parent;
	392
	393	m_path = defaultPath;
	394
	395	wxBeginBusyCursor();
	396
	397	wxBoxSizer *topsizer = new wxBoxSizer( wxVERTICAL );
	398
	399	// 1) dir ctrl
	400	m_dir = new wxDirCtrl( this, ID_DIRCTRL, _T("/"),
	401	wxDefaultPosition,
	402	wxSize(200,200),
	403	wxTR_HAS_BUTTONS \|
	404	wxSUNKEN_BORDER \|
	405	wxTR_EDIT_LABELS );
	406	topsizer->Add( m_dir, 1, wxTOP\|wxLEFT\|wxRIGHT \| wxEXPAND, 10 );
	407
	408	// 2) text ctrl
	409	m_input = new wxTextCtrl( this, ID_TEXTCTRL, m_path, wxDefaultPosition );
	410	topsizer->Add( m_input, 0, wxTOP\|wxLEFT\|wxRIGHT \| wxEXPAND, 10 );
	411
	412	// m_check = new wxCheckBox( this, ID_CHECK, _("Show hidden") );
	413	// m_check->SetValue(TRUE);
	414
	415	#if wxUSE_STATLINE
	416	// 3) static line
	417	topsizer->Add( new wxStaticLine( this, -1 ), 0, wxEXPAND \| wxLEFT\|wxRIGHT\|wxTOP, 10 );
	418	#endif
	419
	420	// 4) buttons
	421	wxSizer* buttonsizer = new wxBoxSizer( wxHORIZONTAL );
	422	m_ok = new wxButton( this, ID_OK, _("OK") );
	423	buttonsizer->Add( m_ok, 0, wxLEFT\|wxRIGHT, 10 );
	424	m_cancel = new wxButton( this, ID_CANCEL, _("Cancel") );
	425	buttonsizer->Add( m_cancel, 0, wxLEFT\|wxRIGHT, 10 );
	426	m_new = new wxButton( this, ID_NEW, _("New...") );
	427	buttonsizer->Add( m_new, 0, wxLEFT\|wxRIGHT, 10 );
	428
	429	topsizer->Add( buttonsizer, 0, wxALL \| wxCENTER, 10 );
	430
	431	m_ok->SetDefault();
	432	m_dir->SetFocus();
	433
	434	SetAutoLayout( TRUE );
	435	SetSizer( topsizer );
	436
	437	topsizer->SetSizeHints( this );
	438	topsizer->Fit( this );
	439
	440	Centre( wxBOTH );
	441
	442	if (m_path == wxT("~"))
	443	wxGetHomeDir( &m_path );
	444
	445	// choose the directory corresponding to defaultPath in the tree
	446	// VZ: using wxStringTokenizer is probably unsafe here (escaped slashes
	447	// will not be processed correctly...)
	448	wxStringTokenizer tk(m_path, wxFILE_SEP_PATH, wxTOKEN_STRTOK);
	449
	450	wxString path;
	451
	452	long cookie = 0;
	453	// default to root dir
	454	wxTreeItemId item = m_dir->GetFirstChild(m_dir->GetRootItem(), cookie);
	455
	456	if (!m_path.IsEmpty() && (m_path != wxT("/")) && (m_dir->m_paths.Count() > 1))
	457	{
	458	size_t count = m_dir->m_paths.GetCount();
	459	for ( size_t i=1; i<count; i++)
	460	{
	461	if (m_path.Find( m_dir->m_paths[i] ) == 0)
	462	{
	463	path = m_dir->m_paths[i];
	464
	465	for (size_t j = 0; j < i; j++)
	466	item = m_dir->GetNextChild(m_dir->GetRootItem(), cookie);
	467
	468	wxStringTokenizer tk2(path, wxFILE_SEP_PATH, wxTOKEN_STRTOK);
	469	for (size_t h = 0; h < tk2.CountTokens(); h++)
	470	tk.GetNextToken();
	471
	472	break;
	473	}
	474	}
	475	}
	476	while ( tk.HasMoreTokens() && item.IsOk() )
	477	{
	478	path << wxFILE_SEP_PATH << tk.GetNextToken();
	479
	480	m_dir->Expand(item);
	481
	482	wxTreeItemId child = m_dir->GetFirstChild(item, cookie);
	483	while ( child.IsOk() )
	484	{
	485	wxDirItemData data = (wxDirItemData)m_dir->GetItemData(child);
	486	if ( data->m_path == path )
	487	break;
	488
	489	child = m_dir->GetNextChild(item, cookie);
	490	}
	491
	492	item = child;
	493	}
	494
	495	if ( item.IsOk() )
	496	{
	497	m_dir->Expand(item);
	498	m_dir->SelectItem(item);
	499	m_dir->EnsureVisible(item);
	500	}
	501
	502	wxEndBusyCursor();
	503	}
	504
	505	int wxDirDialog::ShowModal()
	506	{
	507	m_input->SetValue( m_path );
	508	return wxDialog::ShowModal();
	509	}
	510
	511	void wxDirDialog::OnTreeSelected( wxTreeEvent &event )
	512	{
	513	wxDirItemData data = (wxDirItemData)m_dir->GetItemData(event.GetItem());
	514	if (data)
	515	m_input->SetValue( data->m_path );
	516	};
	517
	518	void wxDirDialog::OnTreeKeyDown( wxTreeEvent &WXUNUSED(event) )
	519	{
	520	wxDirItemData data = (wxDirItemData)m_dir->GetItemData(m_dir->GetSelection());
	521	if (data)
	522	m_input->SetValue( data->m_path );
	523	};
	524
	525	void wxDirDialog::OnOK( wxCommandEvent& WXUNUSED(event) )
	526	{
	527	m_path = m_input->GetValue();
	528	// Does the path exist? (User may have typed anything in m_input)
	529	if (wxPathExists(m_path)) {
	530	// OK, path exists, we're done.
	531	EndModal(wxID_OK);
	532	return;
	533	}
	534	// Interact with user, find out if the dir is a typo or to be created
	535	wxString msg( _("The directory ") );
	536	msg = msg + m_path;
	537	msg = msg + _("\ndoes not exist\nCreate it now?") ;
	538	wxMessageDialog dialog(this, msg, _("Directory does not exist"), wxYES_NO \| wxICON_WARNING );
	539	if ( dialog.ShowModal() == wxID_YES ) {
	540	// Okay, let's make it
	541	wxLogNull log;
	542	if (wxMkdir(m_path)) {
	543	// The new dir was created okay.
	544	EndModal(wxID_OK);
	545	return;
	546	}
	547	else {
	548	// Trouble...
	549	msg = _("Failed to create directory ")+m_path+
	550	_("\n(Do you have the required permissions?)");
	551	wxMessageDialog errmsg(this, msg, _("Error creating directory"), wxOK \| wxICON_ERROR);
	552	errmsg.ShowModal();
	553	// We still don't have a valid dir. Back to the main dialog.
	554	}
	555	}
	556	// User has answered NO to create dir.
	557	}
	558
	559	void wxDirDialog::OnCancel( wxCommandEvent& WXUNUSED(event) )
	560	{
	561	EndModal(wxID_CANCEL);
	562	}
	563
	564	void wxDirDialog::OnNew( wxCommandEvent& WXUNUSED(event) )
	565	{
	566	wxTreeItemId id = m_dir->GetSelection();
	567	if ((id == m_dir->GetRootItem()) \|\|
	568	(m_dir->GetParent(id) == m_dir->GetRootItem()))
	569	{
	570	wxMessageDialog msg(this, _("You cannot add a new directory to this section."),
	571	_("Create directory"), wxOK \| wxICON_INFORMATION );
	572	msg.ShowModal();
	573	return;
	574	}
	575
	576	wxTreeItemId parent = m_dir->GetParent( id );
	577	wxDirItemData data = (wxDirItemData)m_dir->GetItemData( parent );
	578	wxASSERT( data );
	579
	580	wxString new_name( wxT("NewName") );
	581	wxString path( data->m_path );
	582	path += wxT("/");
	583	path += new_name;
	584	if (wxFileExists(path))
	585	{
	586	// try NewName0, NewName1 etc.
	587	int i = 0;
	588	do {
	589	new_name = wxT("NewName");
	590	wxString num;
	591	num.Printf( wxT("%d"), i );
	592	new_name += num;
	593
	594	path = data->m_path;
	595	path += wxT("/");
	596	path += new_name;
	597	i++;
	598	} while (wxFileExists(path));
	599	}
	600
	601	wxLogNull log;
	602	if (!wxMkdir(path))
	603	{
	604	wxMessageDialog dialog(this, _("Operation not permitted."), _("Error"), wxOK \| wxICON_ERROR );
	605	dialog.ShowModal();
	606	return;
	607	}
	608
	609	wxDirItemData *new_data = new wxDirItemData( path, new_name );
	610	wxTreeItemId new_id = m_dir->AppendItem( parent, new_name, 0, 1, new_data );
	611	m_dir->EnsureVisible( new_id );
	612	m_dir->EditLabel( new_id );
	613	}
	614
	615	/*
	616	void wxDirDialog::OnCheck( wxCommandEvent& WXUNUSED(event) )
	617	{
	618	printf("Checkbox clicked: %s\n", ( m_check->GetValue() ? "on" : "off" ) );
	619	}
	620	*/
	621
	622	#endif // wxUSE_DIRDLG