Add code for parsing globs file in /usr/local as well
[wxWidgets.git] / src / unix / mimetype.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/unix/mimetype.cpp
3 // Purpose: classes and functions to manage MIME types
4 // Author: Vadim Zeitlin
5 // Modified by:
6 // Created: 23.09.98
7 // RCS-ID: $Id$
8 // Copyright: (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
9 // Licence: wxWindows licence (part of wxExtra library)
10 /////////////////////////////////////////////////////////////////////////////
11
12 // for compilers that support precompilation, includes "wx.h".
13 #include "wx/wxprec.h"
14
15 #ifdef __BORLANDC__
16 #pragma hdrstop
17 #endif
18
19 #if wxUSE_MIMETYPE && wxUSE_FILE
20
21 #include "wx/unix/mimetype.h"
22
23 #ifndef WX_PRECOMP
24 #include "wx/dynarray.h"
25 #include "wx/string.h"
26 #include "wx/intl.h"
27 #include "wx/log.h"
28 #include "wx/utils.h"
29 #endif
30
31 #include "wx/file.h"
32 #include "wx/confbase.h"
33
34 #include "wx/ffile.h"
35 #include "wx/dir.h"
36 #include "wx/tokenzr.h"
37 #include "wx/iconloc.h"
38 #include "wx/filename.h"
39 #include "wx/app.h"
40 #include "wx/apptrait.h"
41
42 // other standard headers
43 #include <ctype.h>
44
45 class wxMimeTextFile
46 {
47 public:
48 wxMimeTextFile()
49 {
50 }
51
52 wxMimeTextFile(const wxString& fname)
53 {
54 m_fname = fname;
55 }
56
57 bool Open()
58 {
59 wxFFile file( m_fname );
60 if (!file.IsOpened())
61 return false;
62
63 size_t size = file.Length();
64 wxCharBuffer buffer( size );
65 file.Read( (void*) (const char*) buffer, size );
66
67 // Check for valid UTF-8 here?
68 wxString all = wxString::FromUTF8( buffer, size );
69
70 wxStringTokenizer tok( all, "\n" );
71 while (tok.HasMoreTokens())
72 {
73 wxString t = tok.GetNextToken();
74 t.MakeLower();
75 if ((!!t) && (t.Find( "comment" ) != 0) && (t.Find( "#" ) != 0) && (t.Find( "generic" ) != 0))
76 m_text.Add( t );
77 }
78 return true;
79 }
80
81 unsigned int GetLineCount() const { return m_text.GetCount(); }
82 wxString &GetLine( unsigned int line ) { return m_text[line]; }
83
84 int pIndexOf(const wxString& sSearch,
85 bool bIncludeComments = false,
86 int iStart = 0)
87 {
88 wxString sTest = sSearch;
89 sTest.MakeLower();
90 for(size_t i = iStart; i < GetLineCount(); i++)
91 {
92 wxString sLine = GetLine(i);
93 if(bIncludeComments || ! sLine.StartsWith(wxT("#")))
94 {
95 if(sLine.StartsWith(sTest))
96 return (int)i;
97 }
98 }
99 return wxNOT_FOUND;
100 }
101
102 wxString GetVerb(size_t i)
103 {
104 if (i > GetLineCount() )
105 return wxEmptyString;
106
107 wxString sTmp = GetLine(i).BeforeFirst(wxT('='));
108 return sTmp;
109 }
110
111 wxString GetCmd(size_t i)
112 {
113 if (i > GetLineCount() )
114 return wxEmptyString;
115
116 wxString sTmp = GetLine(i).AfterFirst(wxT('='));
117 return sTmp;
118 }
119
120 private:
121 wxArrayString m_text;
122 wxString m_fname;
123 };
124
125 // ----------------------------------------------------------------------------
126 // constants
127 // ----------------------------------------------------------------------------
128
129 // MIME code tracing mask
130 #define TRACE_MIME wxT("mime")
131
132
133 // Read a XDG *.desktop file of type 'Application'
134 void wxMimeTypesManagerImpl::LoadXDGApp(const wxString& filename)
135 {
136 wxLogTrace(TRACE_MIME, wxT("loading XDG file %s"), filename.c_str());
137
138 wxMimeTextFile file(filename);
139 if ( !file.Open() )
140 return;
141
142 // Here, only type 'Application' should be considered.
143 int nIndex = file.pIndexOf( "Type=" );
144 if (nIndex != wxNOT_FOUND && file.GetCmd(nIndex) != "application")
145 return;
146
147 // The hidden entry specifies a file to be ignored.
148 nIndex = file.pIndexOf( "Hidden=" );
149 if (nIndex != wxNOT_FOUND && file.GetCmd(nIndex) == "true")
150 return;
151
152 // Semicolon separated list of mime types handled by the application.
153 nIndex = file.pIndexOf( wxT("MimeType=") );
154 if (nIndex == wxNOT_FOUND)
155 return;
156 wxString mimetypes = file.GetCmd (nIndex);
157
158 // Name of the application
159 wxString nameapp;
160 nIndex = wxNOT_FOUND;
161 #if wxUSE_INTL // try "Name[locale name]" first
162 wxLocale *locale = wxGetLocale();
163 if ( locale )
164 nIndex = file.pIndexOf(_T("Name[")+locale->GetName()+_T("]="));
165 #endif // wxUSE_INTL
166 if(nIndex == wxNOT_FOUND)
167 nIndex = file.pIndexOf( wxT("Name=") );
168 if(nIndex != wxNOT_FOUND)
169 nameapp = file.GetCmd(nIndex);
170
171 // Icon of the application.
172 wxString nameicon, namemini;
173 nIndex = wxNOT_FOUND;
174 #if wxUSE_INTL // try "Icon[locale name]" first
175 if ( locale )
176 nIndex = file.pIndexOf(_T("Icon[")+locale->GetName()+_T("]="));
177 #endif // wxUSE_INTL
178 if(nIndex == wxNOT_FOUND)
179 nIndex = file.pIndexOf( wxT("Icon=") );
180 if(nIndex != wxNOT_FOUND) {
181 nameicon = wxString(wxT("--icon ")) + file.GetCmd(nIndex);
182 namemini = wxString(wxT("--miniicon ")) + file.GetCmd(nIndex);
183 }
184
185 // Replace some of the field code in the 'Exec' entry.
186 // TODO: deal with %d, %D, %n, %N, %k and %v (but last one is deprecated)
187 nIndex = file.pIndexOf( wxT("Exec=") );
188 if (nIndex == wxNOT_FOUND)
189 return;
190 wxString sCmd = file.GetCmd(nIndex);
191 // we expect %f; others including %F and %U and %u are possible
192 sCmd.Replace(wxT("%F"), wxT("%f"));
193 sCmd.Replace(wxT("%U"), wxT("%f"));
194 sCmd.Replace(wxT("%u"), wxT("%f"));
195 if (0 == sCmd.Replace ( wxT("%f"), wxT("%s") ))
196 sCmd = sCmd + wxT(" %s");
197 sCmd.Replace(wxT("%c"), nameapp);
198 sCmd.Replace(wxT("%i"), nameicon);
199 sCmd.Replace(wxT("%m"), namemini);
200
201 wxStringTokenizer tokenizer(mimetypes, _T(";"));
202 while(tokenizer.HasMoreTokens()) {
203 wxString mimetype = tokenizer.GetNextToken().Lower();
204 nIndex = m_aTypes.Index(mimetype);
205 if(nIndex != wxNOT_FOUND) { // is this a known MIME type?
206 wxMimeTypeCommands* entry = m_aEntries[nIndex];
207 entry->AddOrReplaceVerb(wxT("open"), sCmd);
208 }
209 }
210 }
211
212 void wxMimeTypesManagerImpl::LoadXDGAppsFilesFromDir(const wxString& dirname)
213 {
214 // Don't complain if we don't have permissions to read - it confuses users
215 wxLogNull logNull;
216
217 if(! wxDir::Exists(dirname))
218 return;
219 wxDir dir(dirname);
220 if ( !dir.IsOpened() )
221 return;
222
223 wxString filename;
224 // Look into .desktop files
225 bool cont = dir.GetFirst(&filename, _T("*.desktop"), wxDIR_FILES);
226 while (cont)
227 {
228 wxFileName p(dirname, filename);
229 LoadXDGApp( p.GetFullPath() );
230 cont = dir.GetNext(&filename);
231 }
232
233 #if 0
234 // RR: I'm not sure this makes any sense. On my system we'll just
235 // scan the YAST2 and other useless directories
236
237 // Look recursively into subdirs
238 cont = dir.GetFirst(&filename, wxEmptyString, wxDIR_DIRS);
239 while (cont)
240 {
241 wxFileName p(dirname, wxEmptyString);
242 p.AppendDir(filename);
243 LoadXDGAppsFilesFromDir( p.GetPath() );
244 cont = dir.GetNext(&filename);
245 }
246 #endif
247 }
248
249
250 void wxMimeTypesManagerImpl::LoadXDGGlobs(const wxString& filename)
251 {
252 wxLogTrace(TRACE_MIME, wxT("loading XDG globs file from %s"), filename.c_str());
253
254 wxMimeTextFile file(filename);
255 if ( !file.Open() )
256 return;
257
258 size_t i;
259 for (i = 0; i < file.GetLineCount(); i++)
260 {
261 wxStringTokenizer tok( file.GetLine(i), ":" );
262 wxString mime = tok.GetNextToken();
263 wxString ext = tok.GetNextToken();
264 ext.Remove( 0, 2 );
265 wxArrayString exts;
266 exts.Add( ext );
267
268 AddToMimeData(mime, wxEmptyString, NULL, exts, wxEmptyString, true );
269 }
270 }
271
272 // ----------------------------------------------------------------------------
273 // wxFileTypeImpl (Unix)
274 // ----------------------------------------------------------------------------
275
276 wxString wxFileTypeImpl::GetExpandedCommand(const wxString & verb, const wxFileType::MessageParameters& params) const
277 {
278 wxString sTmp;
279 size_t i = 0;
280 while ( (i < m_index.GetCount() ) && sTmp.empty() )
281 {
282 sTmp = m_manager->GetCommand( verb, m_index[i] );
283 i++;
284 }
285
286 return wxFileType::ExpandCommand(sTmp, params);
287 }
288
289 bool wxFileTypeImpl::GetIcon(wxIconLocation *iconLoc) const
290 {
291 wxString sTmp;
292 size_t i = 0;
293 while ( (i < m_index.GetCount() ) && sTmp.empty() )
294 {
295 sTmp = m_manager->m_aIcons[m_index[i]];
296 i++;
297 }
298
299 if ( sTmp.empty() )
300 return false;
301
302 if ( iconLoc )
303 {
304 iconLoc->SetFileName(sTmp);
305 }
306
307 return true;
308 }
309
310 bool wxFileTypeImpl::GetMimeTypes(wxArrayString& mimeTypes) const
311 {
312 mimeTypes.Clear();
313 size_t nCount = m_index.GetCount();
314 for (size_t i = 0; i < nCount; i++)
315 mimeTypes.Add(m_manager->m_aTypes[m_index[i]]);
316
317 return true;
318 }
319
320 size_t wxFileTypeImpl::GetAllCommands(wxArrayString *verbs,
321 wxArrayString *commands,
322 const wxFileType::MessageParameters& params) const
323 {
324 wxString vrb, cmd, sTmp;
325 size_t count = 0;
326 wxMimeTypeCommands * sPairs;
327
328 // verbs and commands have been cleared already in mimecmn.cpp...
329 // if we find no entries in the exact match, try the inexact match
330 for (size_t n = 0; ((count == 0) && (n < m_index.GetCount())); n++)
331 {
332 // list of verb = command pairs for this mimetype
333 sPairs = m_manager->m_aEntries [m_index[n]];
334 size_t i;
335 for ( i = 0; i < sPairs->GetCount(); i++ )
336 {
337 vrb = sPairs->GetVerb(i);
338 // some gnome entries have "." inside
339 vrb = vrb.AfterLast(wxT('.'));
340 cmd = sPairs->GetCmd(i);
341 if (! cmd.empty() )
342 {
343 cmd = wxFileType::ExpandCommand(cmd, params);
344 count++;
345 if ( vrb.IsSameAs(wxT("open")))
346 {
347 if ( verbs )
348 verbs->Insert(vrb, 0u);
349 if ( commands )
350 commands ->Insert(cmd, 0u);
351 }
352 else
353 {
354 if ( verbs )
355 verbs->Add(vrb);
356 if ( commands )
357 commands->Add(cmd);
358 }
359 }
360 }
361 }
362
363 return count;
364 }
365
366 bool wxFileTypeImpl::GetExtensions(wxArrayString& extensions)
367 {
368 const wxString strExtensions = m_manager->GetExtension(m_index[0]);
369 extensions.Empty();
370
371 // one extension in the space or comma-delimited list
372 wxString strExt;
373 wxString::const_iterator end = strExtensions.end();
374 for ( wxString::const_iterator p = strExtensions.begin(); /* nothing */; ++p )
375 {
376 if ( p == end || *p == wxT(' ') || *p == wxT(',') )
377 {
378 if ( !strExt.empty() )
379 {
380 extensions.Add(strExt);
381 strExt.Empty();
382 }
383 //else: repeated spaces
384 // (shouldn't happen, but it's not that important if it does happen)
385
386 if ( p == end )
387 break;
388 }
389 else if ( *p == wxT('.') )
390 {
391 // remove the dot from extension (but only if it's the first char)
392 if ( !strExt.empty() )
393 {
394 strExt += wxT('.');
395 }
396 //else: no, don't append it
397 }
398 else
399 {
400 strExt += *p;
401 }
402 }
403
404 return true;
405 }
406
407 // set an arbitrary command:
408 // could adjust the code to ask confirmation if it already exists and
409 // overwriteprompt is true, but this is currently ignored as *Associate* has
410 // no overwrite prompt
411 bool
412 wxFileTypeImpl::SetCommand(const wxString& cmd,
413 const wxString& verb,
414 bool WXUNUSED(overwriteprompt))
415 {
416 wxArrayString strExtensions;
417 wxString strDesc, strIcon;
418
419 wxArrayString strTypes;
420 GetMimeTypes(strTypes);
421 if ( strTypes.IsEmpty() )
422 return false;
423
424 wxMimeTypeCommands *entry = new wxMimeTypeCommands();
425 entry->Add(verb + wxT("=") + cmd + wxT(" %s "));
426
427 bool ok = false;
428 size_t nCount = strTypes.GetCount();
429 for ( size_t i = 0; i < nCount; i++ )
430 {
431 if ( m_manager->DoAssociation
432 (
433 strTypes[i],
434 strIcon,
435 entry,
436 strExtensions,
437 strDesc
438 ) )
439 {
440 // DoAssociation() took ownership of entry, don't delete it below
441 ok = true;
442 }
443 }
444
445 if ( !ok )
446 delete entry;
447
448 return ok;
449 }
450
451 // ignore index on the grounds that we only have one icon in a Unix file
452 bool wxFileTypeImpl::SetDefaultIcon(const wxString& strIcon, int WXUNUSED(index))
453 {
454 if (strIcon.empty())
455 return false;
456
457 wxArrayString strExtensions;
458 wxString strDesc;
459
460 wxArrayString strTypes;
461 GetMimeTypes(strTypes);
462 if ( strTypes.IsEmpty() )
463 return false;
464
465 wxMimeTypeCommands *entry = new wxMimeTypeCommands();
466 bool ok = false;
467 size_t nCount = strTypes.GetCount();
468 for ( size_t i = 0; i < nCount; i++ )
469 {
470 if ( m_manager->DoAssociation
471 (
472 strTypes[i],
473 strIcon,
474 entry,
475 strExtensions,
476 strDesc
477 ) )
478 {
479 // we don't need to free entry now, DoAssociation() took ownership
480 // of it
481 ok = true;
482 }
483 }
484
485 if ( !ok )
486 delete entry;
487
488 return ok;
489 }
490
491 // ----------------------------------------------------------------------------
492 // wxMimeTypesManagerImpl (Unix)
493 // ----------------------------------------------------------------------------
494
495 wxMimeTypesManagerImpl::wxMimeTypesManagerImpl()
496 {
497 m_initialized = false;
498 }
499
500 void wxMimeTypesManagerImpl::InitIfNeeded()
501 {
502 if ( !m_initialized )
503 {
504 // set the flag first to prevent recursion
505 m_initialized = true;
506
507 wxString wm = wxTheApp->GetTraits()->GetDesktopEnvironment();
508
509 if (wm == wxT("KDE"))
510 Initialize( wxMAILCAP_KDE );
511 else if (wm == wxT("GNOME"))
512 Initialize( wxMAILCAP_GNOME );
513 else
514 Initialize();
515 }
516 }
517
518
519
520 // read system and user mailcaps and other files
521 void wxMimeTypesManagerImpl::Initialize(int mailcapStyles,
522 const wxString& sExtraDir)
523 {
524 #ifdef __VMS
525 // XDG tables are never installed on OpenVMS
526 return;
527 #endif
528
529 // Read MIME type - extension associations
530 LoadXDGGlobs( "/usr/share/mime/globs" );
531 LoadXDGGlobs( "/usr/local/share/mime/globs" );
532
533 // Load desktop files for XDG, and then override them with the defaults.
534 // We will override them one desktop file at a time, rather
535 // than one mime type at a time, but it should be a reasonable
536 // heuristic.
537 {
538 wxString xdgDataHome = wxGetenv("XDG_DATA_HOME");
539 if ( xdgDataHome.empty() )
540 xdgDataHome = wxGetHomeDir() + "/.local/share";
541 wxString xdgDataDirs = wxGetenv("XDG_DATA_DIRS");
542 if ( xdgDataDirs.empty() )
543 {
544 xdgDataDirs = "/usr/local/share:/usr/share";
545 if (mailcapStyles & wxMAILCAP_GNOME)
546 xdgDataDirs += ":/usr/share/gnome:/opt/gnome/share";
547 if (mailcapStyles & wxMAILCAP_KDE)
548 xdgDataDirs += ":/usr/share/kde3:/opt/kde3/share";
549 }
550 if ( !sExtraDir.empty() )
551 {
552 xdgDataDirs += ':';
553 xdgDataDirs += sExtraDir;
554 }
555
556 wxArrayString dirs;
557 wxStringTokenizer tokenizer(xdgDataDirs, ":");
558 while ( tokenizer.HasMoreTokens() )
559 {
560 wxString p = tokenizer.GetNextToken();
561 dirs.Add(p);
562 }
563 dirs.insert(dirs.begin(), xdgDataHome);
564
565 wxString defaultsList;
566 size_t i;
567 for (i = 0; i < dirs.GetCount(); i++)
568 {
569 wxString f = dirs[i];
570 if (f.Last() != '/') f += '/';
571 f += "applications/defaults.list";
572 if (wxFileExists(f))
573 {
574 defaultsList = f;
575 break;
576 }
577 }
578
579 // Load application files and associate them to corresponding mime types.
580 size_t nDirs = dirs.GetCount();
581 for (size_t nDir = 0; nDir < nDirs; nDir++)
582 {
583 wxString dirStr = dirs[nDir];
584 if (dirStr.Last() != '/') dirStr += '/';
585 dirStr += "applications";
586 LoadXDGAppsFilesFromDir(dirStr);
587 }
588
589 if (!defaultsList.IsEmpty())
590 {
591 wxArrayString deskTopFilesSeen;
592
593 wxMimeTextFile textfile(defaultsList);
594 if ( textfile.Open() )
595 {
596 int nIndex = textfile.pIndexOf( wxT("[Default Applications]") );
597 if (nIndex != wxNOT_FOUND)
598 {
599 for (i = nIndex+1; i < textfile.GetLineCount(); i++)
600 {
601 if (textfile.GetLine(i).Find(wxT("=")) != wxNOT_FOUND)
602 {
603 wxString mimeType = textfile.GetVerb(i);
604 wxString desktopFile = textfile.GetCmd(i);
605
606 if (deskTopFilesSeen.Index(desktopFile) == wxNOT_FOUND)
607 {
608 deskTopFilesSeen.Add(desktopFile);
609 size_t j;
610 for (j = 0; j < dirs.GetCount(); j++)
611 {
612 wxString desktopPath = dirs[j];
613 if (desktopPath.Last() != '/') desktopPath += '/';
614 desktopPath += "applications/";
615 desktopPath += desktopFile;
616
617 if (wxFileExists(desktopPath))
618 LoadXDGApp(desktopPath);
619 }
620 }
621 }
622 }
623 }
624 }
625 }
626 }
627 }
628
629 // clear data so you can read another group of WM files
630 void wxMimeTypesManagerImpl::ClearData()
631 {
632 m_aTypes.Clear();
633 m_aIcons.Clear();
634 m_aExtensions.Clear();
635 m_aDescriptions.Clear();
636
637 WX_CLEAR_ARRAY(m_aEntries);
638 m_aEntries.Empty();
639 }
640
641 wxMimeTypesManagerImpl::~wxMimeTypesManagerImpl()
642 {
643 ClearData();
644 }
645
646 wxFileType * wxMimeTypesManagerImpl::Associate(const wxFileTypeInfo& ftInfo)
647 {
648 InitIfNeeded();
649
650 wxString strType = ftInfo.GetMimeType();
651 wxString strDesc = ftInfo.GetDescription();
652 wxString strIcon = ftInfo.GetIconFile();
653
654 wxMimeTypeCommands *entry = new wxMimeTypeCommands();
655
656 if ( ! ftInfo.GetOpenCommand().empty())
657 entry->Add(wxT("open=") + ftInfo.GetOpenCommand() + wxT(" %s "));
658 if ( ! ftInfo.GetPrintCommand().empty())
659 entry->Add(wxT("print=") + ftInfo.GetPrintCommand() + wxT(" %s "));
660
661 // now find where these extensions are in the data store and remove them
662 wxArrayString sA_Exts = ftInfo.GetExtensions();
663 wxString sExt, sExtStore;
664 size_t i, nIndex;
665 size_t nExtCount = sA_Exts.GetCount();
666 for (i=0; i < nExtCount; i++)
667 {
668 sExt = sA_Exts.Item(i);
669
670 // clean up to just a space before and after
671 sExt.Trim().Trim(false);
672 sExt = wxT(' ') + sExt + wxT(' ');
673 size_t nCount = m_aExtensions.GetCount();
674 for (nIndex = 0; nIndex < nCount; nIndex++)
675 {
676 sExtStore = m_aExtensions.Item(nIndex);
677 if (sExtStore.Replace(sExt, wxT(" ") ) > 0)
678 m_aExtensions.Item(nIndex) = sExtStore;
679 }
680 }
681
682 if ( !DoAssociation(strType, strIcon, entry, sA_Exts, strDesc) )
683 return NULL;
684
685 return GetFileTypeFromMimeType(strType);
686 }
687
688 bool wxMimeTypesManagerImpl::DoAssociation(const wxString& strType,
689 const wxString& strIcon,
690 wxMimeTypeCommands *entry,
691 const wxArrayString& strExtensions,
692 const wxString& strDesc)
693 {
694 int nIndex = AddToMimeData(strType, strIcon, entry, strExtensions, strDesc, true);
695
696 if ( nIndex == wxNOT_FOUND )
697 return false;
698
699 return true;
700 }
701
702 int wxMimeTypesManagerImpl::AddToMimeData(const wxString& strType,
703 const wxString& strIcon,
704 wxMimeTypeCommands *entry,
705 const wxArrayString& strExtensions,
706 const wxString& strDesc,
707 bool replaceExisting)
708 {
709 InitIfNeeded();
710
711 // ensure mimetype is always lower case
712 wxString mimeType = strType.Lower();
713
714 // is this a known MIME type?
715 int nIndex = m_aTypes.Index(mimeType);
716 if ( nIndex == wxNOT_FOUND )
717 {
718 // new file type
719 m_aTypes.Add(mimeType);
720 m_aIcons.Add(strIcon);
721 m_aEntries.Add(entry ? entry : new wxMimeTypeCommands);
722
723 // change nIndex so we can use it below to add the extensions
724 m_aExtensions.Add(wxEmptyString);
725 nIndex = m_aExtensions.size() - 1;
726
727 m_aDescriptions.Add(strDesc);
728 }
729 else // yes, we already have it
730 {
731 if ( replaceExisting )
732 {
733 // if new description change it
734 if ( !strDesc.empty())
735 m_aDescriptions[nIndex] = strDesc;
736
737 // if new icon change it
738 if ( !strIcon.empty())
739 m_aIcons[nIndex] = strIcon;
740
741 if ( entry )
742 {
743 delete m_aEntries[nIndex];
744 m_aEntries[nIndex] = entry;
745 }
746 }
747 else // add data we don't already have ...
748 {
749 // if new description add only if none
750 if ( m_aDescriptions[nIndex].empty() )
751 m_aDescriptions[nIndex] = strDesc;
752
753 // if new icon and no existing icon
754 if ( m_aIcons[nIndex].empty() )
755 m_aIcons[nIndex] = strIcon;
756
757 // add any new entries...
758 if ( entry )
759 {
760 wxMimeTypeCommands *entryOld = m_aEntries[nIndex];
761
762 size_t count = entry->GetCount();
763 for ( size_t i = 0; i < count; i++ )
764 {
765 const wxString& verb = entry->GetVerb(i);
766 if ( !entryOld->HasVerb(verb) )
767 {
768 entryOld->AddOrReplaceVerb(verb, entry->GetCmd(i));
769 }
770 }
771
772 // as we don't store it anywhere, it won't be deleted later as
773 // usual -- do it immediately instead
774 delete entry;
775 }
776 }
777 }
778
779 // always add the extensions to this mimetype
780 wxString& exts = m_aExtensions[nIndex];
781
782 // add all extensions we don't have yet
783 wxString ext;
784 size_t count = strExtensions.GetCount();
785 for ( size_t i = 0; i < count; i++ )
786 {
787 ext = strExtensions[i];
788 ext += wxT(' ');
789
790 if ( exts.Find(ext) == wxNOT_FOUND )
791 {
792 exts += ext;
793 }
794 }
795
796 // check data integrity
797 wxASSERT( m_aTypes.GetCount() == m_aEntries.GetCount() &&
798 m_aTypes.GetCount() == m_aExtensions.GetCount() &&
799 m_aTypes.GetCount() == m_aIcons.GetCount() &&
800 m_aTypes.GetCount() == m_aDescriptions.GetCount() );
801
802 return nIndex;
803 }
804
805 wxFileType * wxMimeTypesManagerImpl::GetFileTypeFromExtension(const wxString& ext)
806 {
807 if (ext.empty() )
808 return NULL;
809
810 InitIfNeeded();
811
812 size_t count = m_aExtensions.GetCount();
813 for ( size_t n = 0; n < count; n++ )
814 {
815 wxStringTokenizer tk(m_aExtensions[n], wxT(' '));
816
817 while ( tk.HasMoreTokens() )
818 {
819 // consider extensions as not being case-sensitive
820 if ( tk.GetNextToken().IsSameAs(ext, false /* no case */) )
821 {
822 // found
823 wxFileType *fileType = new wxFileType;
824 fileType->m_impl->Init(this, n);
825
826 return fileType;
827 }
828 }
829 }
830
831 return NULL;
832 }
833
834 wxFileType * wxMimeTypesManagerImpl::GetFileTypeFromMimeType(const wxString& mimeType)
835 {
836 InitIfNeeded();
837
838 wxFileType * fileType = NULL;
839 // mime types are not case-sensitive
840 wxString mimetype(mimeType);
841 mimetype.MakeLower();
842
843 // first look for an exact match
844 int index = m_aTypes.Index(mimetype);
845
846 if ( index != wxNOT_FOUND )
847 {
848 fileType = new wxFileType;
849 fileType->m_impl->Init(this, index);
850 }
851
852 // then try to find "text/*" as match for "text/plain" (for example)
853 // NB: if mimeType doesn't contain '/' at all, BeforeFirst() will return
854 // the whole string - ok.
855
856 index = wxNOT_FOUND;
857 wxString strCategory = mimetype.BeforeFirst(wxT('/'));
858
859 size_t nCount = m_aTypes.GetCount();
860 for ( size_t n = 0; n < nCount; n++ )
861 {
862 if ( (m_aTypes[n].BeforeFirst(wxT('/')) == strCategory ) &&
863 m_aTypes[n].AfterFirst(wxT('/')) == wxT("*") )
864 {
865 index = n;
866 break;
867 }
868 }
869
870 if ( index != wxNOT_FOUND )
871 {
872 // don't throw away fileType that was already found
873 if (!fileType)
874 fileType = new wxFileType;
875 fileType->m_impl->Init(this, index);
876 }
877
878 return fileType;
879 }
880
881 wxString wxMimeTypesManagerImpl::GetCommand(const wxString & verb, size_t nIndex) const
882 {
883 wxString command, testcmd, sV, sTmp;
884 sV = verb + wxT("=");
885
886 // list of verb = command pairs for this mimetype
887 wxMimeTypeCommands * sPairs = m_aEntries [nIndex];
888
889 size_t i;
890 size_t nCount = sPairs->GetCount();
891 for ( i = 0; i < nCount; i++ )
892 {
893 sTmp = sPairs->GetVerbCmd (i);
894 if ( sTmp.Contains(sV) )
895 command = sTmp.AfterFirst(wxT('='));
896 }
897
898 return command;
899 }
900
901 void wxMimeTypesManagerImpl::AddFallback(const wxFileTypeInfo& filetype)
902 {
903 InitIfNeeded();
904
905 wxString extensions;
906 const wxArrayString& exts = filetype.GetExtensions();
907 size_t nExts = exts.GetCount();
908 for ( size_t nExt = 0; nExt < nExts; nExt++ )
909 {
910 if ( nExt > 0 )
911 extensions += wxT(' ');
912
913 extensions += exts[nExt];
914 }
915
916 AddMimeTypeInfo(filetype.GetMimeType(),
917 extensions,
918 filetype.GetDescription());
919 }
920
921 void wxMimeTypesManagerImpl::AddMimeTypeInfo(const wxString& strMimeType,
922 const wxString& strExtensions,
923 const wxString& strDesc)
924 {
925 // reading mailcap may find image/* , while
926 // reading mime.types finds image/gif and no match is made
927 // this means all the get functions don't work fix this
928 wxString strIcon;
929 wxString sTmp = strExtensions;
930
931 wxArrayString sExts;
932 sTmp.Trim().Trim(false);
933
934 while (!sTmp.empty())
935 {
936 sExts.Add(sTmp.AfterLast(wxT(' ')));
937 sTmp = sTmp.BeforeLast(wxT(' '));
938 }
939
940 AddToMimeData(strMimeType, strIcon, NULL, sExts, strDesc, true);
941 }
942
943 size_t wxMimeTypesManagerImpl::EnumAllFileTypes(wxArrayString& mimetypes)
944 {
945 InitIfNeeded();
946
947 mimetypes.Empty();
948
949 size_t count = m_aTypes.GetCount();
950 for ( size_t n = 0; n < count; n++ )
951 {
952 // don't return template types from here (i.e. anything containg '*')
953 const wxString &type = m_aTypes[n];
954 if ( type.Find(wxT('*')) == wxNOT_FOUND )
955 {
956 mimetypes.Add(type);
957 }
958 }
959
960 return mimetypes.GetCount();
961 }
962
963 // ----------------------------------------------------------------------------
964 // writing to MIME type files
965 // ----------------------------------------------------------------------------
966
967 bool wxMimeTypesManagerImpl::Unassociate(wxFileType *ft)
968 {
969 InitIfNeeded();
970
971 wxArrayString sMimeTypes;
972 ft->GetMimeTypes(sMimeTypes);
973
974 size_t i;
975 size_t nCount = sMimeTypes.GetCount();
976 for (i = 0; i < nCount; i ++)
977 {
978 const wxString &sMime = sMimeTypes.Item(i);
979 int nIndex = m_aTypes.Index(sMime);
980 if ( nIndex == wxNOT_FOUND)
981 {
982 // error if we get here ??
983 return false;
984 }
985 else
986 {
987 m_aTypes.RemoveAt(nIndex);
988 m_aEntries.RemoveAt(nIndex);
989 m_aExtensions.RemoveAt(nIndex);
990 m_aDescriptions.RemoveAt(nIndex);
991 m_aIcons.RemoveAt(nIndex);
992 }
993 }
994 // check data integrity
995 wxASSERT( m_aTypes.GetCount() == m_aEntries.GetCount() &&
996 m_aTypes.GetCount() == m_aExtensions.GetCount() &&
997 m_aTypes.GetCount() == m_aIcons.GetCount() &&
998 m_aTypes.GetCount() == m_aDescriptions.GetCount() );
999
1000 return true;
1001 }
1002
1003 #endif
1004 // wxUSE_MIMETYPE && wxUSE_FILE