src/generic/dirdlgg.cpp

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