michael@0: /******* BEGIN LICENSE BLOCK *******
michael@0:  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
michael@0:  * 
michael@0:  * The contents of this file are subject to the Mozilla Public License Version
michael@0:  * 1.1 (the "License"); you may not use this file except in compliance with
michael@0:  * the License. You may obtain a copy of the License at
michael@0:  * http://www.mozilla.org/MPL/
michael@0:  * 
michael@0:  * Software distributed under the License is distributed on an "AS IS" basis,
michael@0:  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
michael@0:  * for the specific language governing rights and limitations under the
michael@0:  * License.
michael@0:  * 
michael@0:  * The Initial Developers of the Original Code are Kevin Hendricks (MySpell)
michael@0:  * and László Németh (Hunspell). Portions created by the Initial Developers
michael@0:  * are Copyright (C) 2002-2005 the Initial Developers. All Rights Reserved.
michael@0:  * 
michael@0:  * Contributor(s): Benjamin Smedberg (benjamin@smedbergs.us) (Original Code)
michael@0:  *                 László Németh (nemethl@gyorsposta.hu)
michael@0:  *                 Ryan VanderMeulen (ryanvm@gmail.com)
michael@0:  * 
michael@0:  * Alternatively, the contents of this file may be used under the terms of
michael@0:  * either the GNU General Public License Version 2 or later (the "GPL"), or
michael@0:  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
michael@0:  * in which case the provisions of the GPL or the LGPL are applicable instead
michael@0:  * of those above. If you wish to allow use of your version of this file only
michael@0:  * under the terms of either the GPL or the LGPL, and not to allow others to
michael@0:  * use your version of this file under the terms of the MPL, indicate your
michael@0:  * decision by deleting the provisions above and replace them with the notice
michael@0:  * and other provisions required by the GPL or the LGPL. If you do not delete
michael@0:  * the provisions above, a recipient may use your version of this file under
michael@0:  * the terms of any one of the MPL, the GPL or the LGPL.
michael@0:  *
michael@0:  ******* END LICENSE BLOCK *******/
michael@0: 
michael@0: #include "mozHunspellDirProvider.h"
michael@0: #include "nsXULAppAPI.h"
michael@0: #include "nsString.h"
michael@0: 
michael@0: #include "mozISpellCheckingEngine.h"
michael@0: #include "nsICategoryManager.h"
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(mozHunspellDirProvider,
michael@0: 		   nsIDirectoryServiceProvider,
michael@0: 		   nsIDirectoryServiceProvider2)
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: mozHunspellDirProvider::GetFile(const char *aKey, bool *aPersist,
michael@0: 			       nsIFile* *aResult)
michael@0: {
michael@0:   return NS_ERROR_FAILURE;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: mozHunspellDirProvider::GetFiles(const char *aKey,
michael@0: 				nsISimpleEnumerator* *aResult)
michael@0: {
michael@0:   if (strcmp(aKey, DICTIONARY_SEARCH_DIRECTORY_LIST) != 0) {
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   nsCOMPtr<nsIProperties> dirSvc =
michael@0:     do_GetService(NS_DIRECTORY_SERVICE_CONTRACTID);
michael@0:   if (!dirSvc)
michael@0:     return NS_ERROR_FAILURE;
michael@0: 
michael@0:   nsCOMPtr<nsISimpleEnumerator> list;
michael@0:   nsresult rv = dirSvc->Get(XRE_EXTENSIONS_DIR_LIST,
michael@0: 			    NS_GET_IID(nsISimpleEnumerator),
michael@0: 			    getter_AddRefs(list));
michael@0:   if (NS_FAILED(rv))
michael@0:     return rv;
michael@0: 
michael@0:   nsCOMPtr<nsISimpleEnumerator> e = new AppendingEnumerator(list);
michael@0:   if (!e)
michael@0:     return NS_ERROR_OUT_OF_MEMORY;
michael@0: 
michael@0:   *aResult = nullptr;
michael@0:   e.swap(*aResult);
michael@0:   return NS_SUCCESS_AGGREGATE_RESULT;
michael@0: }
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(mozHunspellDirProvider::AppendingEnumerator,
michael@0: 		   nsISimpleEnumerator)
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: mozHunspellDirProvider::AppendingEnumerator::HasMoreElements(bool *aResult)
michael@0: {
michael@0:   *aResult = mNext ? true : false;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: mozHunspellDirProvider::AppendingEnumerator::GetNext(nsISupports* *aResult)
michael@0: {
michael@0:   if (aResult)
michael@0:     NS_ADDREF(*aResult = mNext);
michael@0: 
michael@0:   mNext = nullptr;
michael@0: 
michael@0:   nsresult rv;
michael@0: 
michael@0:   // Ignore all errors
michael@0: 
michael@0:   bool more;
michael@0:   while (NS_SUCCEEDED(mBase->HasMoreElements(&more)) && more) {
michael@0:     nsCOMPtr<nsISupports> nextbasesupp;
michael@0:     mBase->GetNext(getter_AddRefs(nextbasesupp));
michael@0: 
michael@0:     nsCOMPtr<nsIFile> nextbase(do_QueryInterface(nextbasesupp));
michael@0:     if (!nextbase)
michael@0:       continue;
michael@0: 
michael@0:     nextbase->Clone(getter_AddRefs(mNext));
michael@0:     if (!mNext)
michael@0:       continue;
michael@0: 
michael@0:     mNext->AppendNative(NS_LITERAL_CSTRING("dictionaries"));
michael@0: 
michael@0:     bool exists;
michael@0:     rv = mNext->Exists(&exists);
michael@0:     if (NS_SUCCEEDED(rv) && exists)
michael@0:       break;
michael@0: 
michael@0:     mNext = nullptr;
michael@0:   }
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: mozHunspellDirProvider::AppendingEnumerator::AppendingEnumerator
michael@0:     (nsISimpleEnumerator* aBase) :
michael@0:   mBase(aBase)
michael@0: {
michael@0:   // Initialize mNext to begin
michael@0:   GetNext(nullptr);
michael@0: }
michael@0: 
michael@0: char const *const
michael@0: mozHunspellDirProvider::kContractID = "@mozilla.org/spellcheck/dir-provider;1";