intl/locale/nsLanguageAtomService.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "nsLanguageAtomService.h"
   7 #include "nsILocaleService.h"
   8 #include "nsUnicharUtils.h"
   9 #include "nsIAtom.h"
  10 #include "mozilla/Services.h"
  11 #include "nsServiceManagerUtils.h"
  12 #include "mozilla/dom/EncodingUtils.h"
  13
  14 NS_IMPL_ISUPPORTS(nsLanguageAtomService, nsILanguageAtomService)
  15
  16 nsLanguageAtomService::nsLanguageAtomService()
  17 {
  18 }
  19
  20 nsresult
  21 nsLanguageAtomService::InitLangGroupTable()
  22 {
  23   if (mLangGroups)
  24     return NS_OK;
  25
  26   nsCOMPtr<nsIStringBundleService> bundleService =
  27     mozilla::services::GetStringBundleService();
  28   if (!bundleService)
  29     return NS_ERROR_FAILURE;
  30
  31   return bundleService->CreateBundle("resource://gre/res/langGroups.properties",
  32                                      getter_AddRefs(mLangGroups));
  33 }
  34
  35 nsIAtom*
  36 nsLanguageAtomService::LookupLanguage(const nsACString &aLanguage,
  37                                       nsresult *aError)
  38 {
  39   nsAutoCString lowered(aLanguage);
  40   ToLowerCase(lowered);
  41
  42   nsCOMPtr<nsIAtom> lang = do_GetAtom(lowered);
  43   return GetLanguageGroup(lang, aError);
  44 }
  45
  46 already_AddRefed<nsIAtom>
  47 nsLanguageAtomService::LookupCharSet(const nsACString& aCharSet)
  48 {
  49   nsAutoCString group;
  50   mozilla::dom::EncodingUtils::LangGroupForEncoding(aCharSet, group);
  51   return do_GetAtom(group);
  52 }
  53
  54 nsIAtom*
  55 nsLanguageAtomService::GetLocaleLanguage(nsresult *aError)
  56 {
  57   nsresult res = NS_OK;
  58
  59   do {
  60     if (!mLocaleLanguage) {
  61       nsCOMPtr<nsILocaleService> localeService;
  62       localeService = do_GetService(NS_LOCALESERVICE_CONTRACTID);
  63       if (!localeService) {
  64         res = NS_ERROR_FAILURE;
  65         break;
  66       }
  67
  68       nsCOMPtr<nsILocale> locale;
  69       res = localeService->GetApplicationLocale(getter_AddRefs(locale));
  70       if (NS_FAILED(res))
  71         break;
  72
  73       nsAutoString loc;
  74       res = locale->GetCategory(NS_LITERAL_STRING(NSILOCALE_MESSAGE), loc);
  75       if (NS_FAILED(res))
  76         break;
  77
  78       ToLowerCase(loc); // use lowercase for all language atoms
  79       mLocaleLanguage = do_GetAtom(loc);
  80     }
  81   } while (0);
  82
  83   if (aError)
  84     *aError = res;
  85
  86   return mLocaleLanguage;
  87 }
  88
  89 nsIAtom*
  90 nsLanguageAtomService::GetLanguageGroup(nsIAtom *aLanguage,
  91                                         nsresult *aError)
  92 {
  93   nsIAtom *retVal;
  94   nsresult res = NS_OK;
  95
  96   retVal = mLangToGroup.GetWeak(aLanguage);
  97
  98   if (!retVal) {
  99     if (!mLangGroups) {
 100       if (NS_FAILED(InitLangGroupTable())) {
 101         if (aError) {
 102           *aError = NS_ERROR_FAILURE;
 103         }
 104         return nullptr;
 105       }
 106     }
 107
 108     nsString langStr;
 109     aLanguage->ToString(langStr);
 110
 111     nsXPIDLString langGroupStr;
 112     res = mLangGroups->GetStringFromName(langStr.get(),
 113                                          getter_Copies(langGroupStr));
 114     if (NS_FAILED(res)) {
 115       int32_t hyphen = langStr.FindChar('-');
 116       if (hyphen >= 0) {
 117         nsAutoString truncated(langStr);
 118         truncated.Truncate(hyphen);
 119         res = mLangGroups->GetStringFromName(truncated.get(),
 120                                              getter_Copies(langGroupStr));
 121         if (NS_FAILED(res)) {
 122           langGroupStr.AssignLiteral("x-unicode");
 123         }
 124       } else {
 125         langGroupStr.AssignLiteral("x-unicode");
 126       }
 127     }
 128
 129     nsCOMPtr<nsIAtom> langGroup = do_GetAtom(langGroupStr);
 130
 131     // The hashtable will keep an owning reference to the atom
 132     mLangToGroup.Put(aLanguage, langGroup);
 133     retVal = langGroup.get();
 134   }
 135
 136   if (aError) {
 137     *aError = res;
 138   }
 139
 140   return retVal;
 141 }