+ for (CFIndex i = 0; i < GetCount(); ++i)
+ {
+ //must be an array, dictionary, string, bool, or int and cannot be null
+ //and dictionaries can only contain cfstring keys
+ CFTypeRef cfRef = (*this)[i];
+ if (!cfRef)
+ {
+ Remove(i);
+ --i;
+ }
+ else if (CFGetTypeID(cfRef) == CFArrayGetTypeID())
+ {
+ CFRetain(cfRef);
+ wxCFArray cfaCurrent(cfRef);
+ cfaCurrent.MakeMutable();
+ cfaCurrent.MakeValidXML();
+ Set(i, cfaCurrent);
+ }
+ else if (CFGetTypeID(cfRef) == CFDictionaryGetTypeID())
+ {
+ CFRetain(cfRef);
+ wxCFDictionary cfdCurrent(cfRef);
+ cfdCurrent.MakeMutable();
+ cfdCurrent.MakeValidXML();
+ Set(i, cfdCurrent);
+ }
+ else if ( CFGetTypeID(cfRef) != CFStringGetTypeID() &&
+ CFGetTypeID(cfRef) != CFNumberGetTypeID() &&
+ CFGetTypeID(cfRef) != CFBooleanGetTypeID() )
+ {
+ Remove(i);
+ --i;
+ }
+ }
+}
+
+//
+//
+//
+// END TODO
+//
+//
+//
+
+wxFileType* wxMimeTypesManagerImpl::Associate(const wxFileTypeInfo& ftInfo)
+{
+ bool bInfoSuccess = false;
+
+ const wxArrayString& asExtensions = ftInfo.GetExtensions();
+ size_t dwFoundIndex = 0;
+ if (!asExtensions.GetCount())
+ {
+ wxLogDebug(wxT("Must have extension to associate with"));
+ }
+
+ // Find and write to Info.plist in main bundle (note that some other
+ // apps have theirs named differently, i.e. IE's is named Info-macos.plist
+ // some apps (non-wx) use the 'plst' resource instead
+ CFBundleRef cfbMain = CFBundleGetMainBundle();
+ if (cfbMain)
+ {
+ UInt32 dwBundleType, dwBundleCreator;
+ CFBundleGetPackageInfo(cfbMain, &dwBundleType, &dwBundleCreator);
+
+ // if launching terminal non-app, version will be 'BNDL' (generic bundle, maybe in other cases too),
+ // which will give us the incorrect info.plist path
+ // otherwise it will be 'APPL', or in the case of a framework, 'FMWK'
+ if (dwBundleType == 'APPL')
+ {
+ wxCFURL cfurlBundleLoc((CFTypeRef)CFBundleCopyBundleURL(cfbMain));
+// wxCFURL cfurlBundleLoc((CFTypeRef)CFBundleCopyExecutableURL(cfbMain));
+ wxString sInfoPath;
+// sInfoPath << wxT("file://");
+ sInfoPath << cfurlBundleLoc.BuildWXString();
+ sInfoPath << wxT("Contents/Info.plist");
+
+// wxCFDictionary cfdInfo( CFBundleGetInfoDictionary(cfbMain), wxCF_RETAIN );
+ wxCFDictionary cfdInfo;
+ bool bInfoOpenSuccess = false;
+ wxFile indictfile;
+ if (indictfile.Open(sInfoPath, wxFile::read))
+ {
+ CFIndex cfiBufLen = (CFIndex) indictfile.Length();
+ const UInt8* pBuffer = new UInt8[cfiBufLen];
+ indictfile.Read((void*)pBuffer, cfiBufLen);
+ wxCFData cfdaInDict(pBuffer, cfiBufLen);
+ wxString sError;
+ bInfoOpenSuccess = cfdInfo.ReadAsXML(cfdaInDict, &sError);
+ if (!bInfoOpenSuccess)
+ wxLogDebug(sError);
+ indictfile.Close();
+ }
+
+ if (bInfoOpenSuccess)
+ {
+ cfdInfo.MakeMutable( cfdInfo.GetCount() + 1 );
+
+ wxCFArray cfaDocTypes( wxCFRetain( cfdInfo[ wxCFStringRef(wxT("CFBundleDocumentTypes")) ] ) );
+
+ bool bAddDocTypesArrayToDictionary = !cfaDocTypes.IsOk();
+ if (bAddDocTypesArrayToDictionary)
+ cfaDocTypes.Create();
+ else
+ cfaDocTypes.MakeMutable( cfaDocTypes.GetCount() + 1 );
+
+ bool bEntryFound = false;
+
+ // search for duplicates
+ CFIndex i;
+ for (i = 0; i < cfaDocTypes.GetCount(); ++i)
+ {
+ wxCFDictionary cfdDocTypeEntry( wxCFRetain( cfaDocTypes[i] ) );
+
+ // A lot of apps don't support MIME types for some reason
+ // so we go by extensions only
+ wxCFArray cfaExtensions( wxCFRetain( cfdDocTypeEntry[ wxCFStringRef(wxT("CFBundleTypeExtensions")) ] ) );
+
+ if (!cfaExtensions.IsOk())
+ continue;
+
+ for (CFIndex iExt = 0; iExt < cfaExtensions.GetCount(); ++iExt)
+ {
+ for (size_t iWXExt = 0; iWXExt < asExtensions.GetCount(); ++iWXExt)
+ {
+ if (asExtensions[iWXExt] ==
+ wxCFStringRef( wxCFRetain( (CFStringRef) cfaExtensions[iExt] ) ).AsString())
+ {
+ bEntryFound = true;
+ dwFoundIndex = iWXExt;
+
+ break;
+ }
+ } //end of wxstring array
+
+ if (bEntryFound)
+ break;
+ } //end for cf array
+
+ if (bEntryFound)
+ break;
+ } //end for doctypes
+
+ wxCFDictionary cfdNewEntry;
+
+ if (!ftInfo.GetDescription().empty())
+ {
+ cfdNewEntry.Add( wxCFStringRef(wxT("CFBundleTypeName")),
+ wxCFStringRef(ftInfo.GetDescription()) );
+ }
+
+ if (!ftInfo.GetIconFile().empty())
+ {
+ cfdNewEntry.Add( wxCFStringRef(wxT("CFBundleTypeIconFile")),
+ wxCFStringRef(ftInfo.GetIconFile()) );
+ }
+
+ wxCFArray cfaOSTypes;
+ wxCFArray cfaExtensions;
+ wxCFArray cfaMimeTypes;
+
+ //OSTypes is a cfarray of four-char-codes - '****' for unrestricted
+ cfaOSTypes.Add( wxCFStringRef(wxT("****")) );
+ cfdNewEntry.Add( wxCFStringRef(wxT("CFBundleTypeOSTypes")), cfaOSTypes );
+
+ //'*' for unrestricted
+ if (ftInfo.GetExtensionsCount() != 0)
+ {
+ for (size_t iExtension = 0; iExtension < ftInfo.GetExtensionsCount(); ++iExtension)
+ {
+ cfaExtensions.Add( wxCFStringRef( asExtensions[iExtension] ) );
+ }
+
+ cfdNewEntry.Add( wxCFStringRef(wxT("CFBundleTypeExtensions")), cfaExtensions );
+ }
+
+ if (!ftInfo.GetMimeType().empty())
+ {
+ cfaMimeTypes.Add( wxCFStringRef(ftInfo.GetMimeType()) );
+ cfdNewEntry.Add( wxCFStringRef(wxT("CFBundleTypeMIMETypes")), cfaMimeTypes );
+ }
+
+ // Editor - can perform all actions
+ // Viewer - all actions except manipulation/saving
+ // None - can perform no actions
+ cfdNewEntry.Add( wxCFStringRef(wxT("CFBundleTypeRole")), wxCFStringRef(wxT("Editor")) );
+
+ // Is application bundled?
+ cfdNewEntry.Add( wxCFStringRef(wxT("LSTypeIsPackage")), kCFBooleanTrue );
+
+ if (bEntryFound)
+ cfaDocTypes.Set(i, cfdNewEntry);
+ else
+ cfaDocTypes.Add(cfdNewEntry);
+
+ // set the doc types array in the muted dictionary
+ if (bAddDocTypesArrayToDictionary)
+ cfdInfo.Add(wxCFStringRef(wxT("CFBundleDocumentTypes")), cfaDocTypes);
+ else
+ cfdInfo.Set(wxCFStringRef(wxT("CFBundleDocumentTypes")), cfaDocTypes);
+
+ cfdInfo.MakeValidXML();
+
+ wxFile outdictfile;
+ if (outdictfile.Open(sInfoPath, wxFile::write))
+ {
+ wxCFData cfdaInfo(cfdInfo.WriteAsXML());
+ if (cfdaInfo.IsOk())
+ {
+ if (outdictfile.Write(cfdaInfo.GetValue(), cfdaInfo.GetCount()) !=
+ (wxFileOffset)cfdaInfo.GetCount())
+ {
+ wxLogDebug(wxT("error in writing to file"));
+ }
+ else
+ {
+ bInfoSuccess = true;
+
+//#if defined(__DARWIN__)
+// //force launch services to update its database for the finder
+// OSStatus status = LSRegisterURL((CFURLRef)(CFTypeRef)cfurlBundleLoc, true);
+// if (status != noErr)
+// {
+// wxLogDebug(wxT("LSRegisterURL Failed."));
+// }
+//#endif
+ }
+ outdictfile.Close();
+ }
+ else
+ {
+ outdictfile.Close();
+ wxLogDebug(wxT("Could not read in new dictionary"));
+ }
+ }
+ else
+ {
+ wxLogDebug(wxString(wxT("Could not open [")) +
+ sInfoPath + wxT("] for writing."));
+ }
+ }
+ else
+ {
+ wxLogDebug(wxT("No info dictionary in main bundle"));
+ }
+ }
+ else
+ {
+ wxLogDebug(wxT("Can only call associate from bundled app within XXX.app"));
+ }
+ }
+ else
+ {
+ wxLogDebug(wxT("No main bundle"));
+ }
+
+ if (!bInfoSuccess)
+ return NULL;
+
+ // on mac you have to embed it into the mac's file reference resource ('FREF' I believe)
+ // or, alternately, you could just add an entry to m_hDatabase, but you'd need to get
+ // the app's signature somehow...
+
+ OSType processType, creator;
+ OSStatus status = MoreProcGetProcessTypeSignature(NULL, &processType, &creator);
+
+ if (status == noErr)
+ {
+ Str255 psCreatorName;
+#ifndef __LP64__
+ FSSpec dummySpec;
+ status = FindApplication(creator, false, psCreatorName, &dummySpec);
+#else
+ FSRef fsref;
+ status = LSFindApplicationForInfo( creator, NULL, NULL, &fsref ,NULL);
+ HFSUniStr255 name;
+ status = FSGetCatalogInfo(&fsref, kFSCatInfoNone, NULL, &name, NULL, NULL);
+ CFStringRef str = FSCreateStringFromHFSUniStr( 0 , &name );
+ CFStringGetPascalString(str, psCreatorName, 256, CFStringGetSystemEncoding());
+ CFRelease( str );
+#endif
+
+ if (status == noErr)
+ {
+ //get the file type if it exists -
+ //if it really does then modify the database then save it,
+ //otherwise we need to create a whole new entry
+ wxFileType* pFileType = GetFileTypeFromExtension(asExtensions[dwFoundIndex]);
+ if (pFileType)
+ {
+ ICMapEntry entry;
+ ICGetMapEntry( (ICInstance) m_hIC, (Handle) m_hDatabase,
+ pFileType->m_impl->m_lIndex, &entry );
+
+ memcpy(entry.creatorAppName, psCreatorName, sizeof(Str255));
+ entry.fileCreator = creator;
+
+ status = ICSetMapEntry( (ICInstance) m_hIC, (Handle) m_hDatabase,
+ pFileType->m_impl->m_lIndex, &entry );
+
+ //success
+ if (status == noErr)
+ {
+ return pFileType;
+
+ //kICAttrNoChange means we don't care about attributes such as
+ //locking in the database
+// status = ICSetPrefHandle((ICInstance) m_hIC, kICMapping,
+// kICAttrNoChange, (Handle) m_hDatabase);
+// if (status == noErr)
+// return pFileType;
+// else
+// {
+// wxLogDebug(wxString::Format(wxT("%i - %s"), (int)status, wxT("ICSetPrefHandle failed.")));
+// }
+ }
+ else
+ {
+ wxLogDebug(wxString::Format(wxT("%i - %s"), __LINE__, wxT("ICSetMapEntry failed.")));
+ }
+
+ // failure - cleanup
+ delete pFileType;
+ }
+ else
+ {
+ // TODO: Maybe force all 3 of these to be non-empty?
+ Str255 psExtension, psMimeType, psDescription;
+
+ wxMacStringToPascal(wxString(wxT(".")) + ftInfo.GetExtensions()[0], psExtension);
+ wxMacStringToPascal(ftInfo.GetMimeType(), psMimeType);
+ wxMacStringToPascal(ftInfo.GetDescription(), psDescription);
+
+ Str255 psPostCreatorName;
+ wxMacStringToPascal(wxEmptyString, psPostCreatorName);
+
+ //add the entry to the database
+ ICMapEntry entry;
+ entry.totalLength = sizeof(ICMapEntry);
+ entry.fixedLength = kICMapFixedLength;
+ entry.version = 0;
+ entry.fileType = 0; //TODO: File type?
+ entry.fileCreator = creator;
+ entry.postCreator = 0;
+ entry.flags = kICMapDataForkBit; //TODO: Maybe resource is valid by default too?
+ PLstrcpy( entry.extension , psExtension ) ;
+ memcpy(entry.creatorAppName, psCreatorName, sizeof(Str255));
+ memcpy(entry.postAppName, psPostCreatorName, sizeof(Str255));
+ memcpy(entry.MIMEType, psMimeType, sizeof(Str255));
+ memcpy(entry.entryName, psDescription, sizeof(Str255));
+
+ status = ICAddMapEntry( (ICInstance) m_hIC, (Handle) m_hDatabase, &entry);
+ if (status == noErr)
+ {
+ return GetFileTypeFromExtension(ftInfo.GetMimeType());
+
+// kICAttrNoChange means we don't care about attributes such as
+// locking in the database
+// status = ICSetPrefHandle((ICInstance) m_hIC, kICMapping,
+// kICAttrNoChange, (Handle) m_hDatabase);
+
+ // return the entry in the database if successful
+// if (status == noErr)
+// return GetFileTypeFromExtension(ftInfo.GetMimeType());
+// else
+// {
+// wxLogDebug(wxString::Format(wxT("%i - %s"), __LINE__, wxT("ICSetPrefHandle failed.")));
+ // }
+ }
+ else
+ {
+ wxLogDebug(wxString::Format(wxT("%i - %s"), __LINE__, wxT("ICAppMapEntry failed.")));
+ }
+ }
+ } // end if FindApplcation was successful
+ else
+ {
+ wxLogDebug(wxString::Format(wxT("%i - %s"), __LINE__, wxT("FindApplication failed.")));
+ }
+ } // end if it could obtain app's signature
+ else
+ {
+ wxLogDebug(wxString::Format(wxT("%i - %s"), __LINE__, wxT("GetProcessSignature failed.")));
+ }
+