embedding/qa/testembed/ProfileMgr.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  *
   3  * ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is the Mozilla browser.
  17  *
  18  * The Initial Developer of the Original Code is
  19  * Netscape Communications, Inc.
  20  * Portions created by the Initial Developer are Copyright (C) 1999
  21  * the Initial Developer. All Rights Reserved.
  22  *
  23  * Contributor(s):
  24  *   Conrad Carlen <ccarlen@netscape.com>
  25  *
  26  * Alternatively, the contents of this file may be used under the terms of
  27  * either the GNU General Public License Version 2 or later (the "GPL"), or
  28  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  29  * in which case the provisions of the GPL or the LGPL are applicable instead
  30  * of those above. If you wish to allow use of your version of this file only
  31  * under the terms of either the GPL or the LGPL, and not to allow others to
  32  * use your version of this file under the terms of the MPL, indicate your
  33  * decision by deleting the provisions above and replace them with the notice
  34  * and other provisions required by the GPL or the LGPL. If you do not delete
  35  * the provisions above, a recipient may use your version of this file under
  36  * the terms of any one of the MPL, the GPL or the LGPL.
  37  *
  38  * ***** END LICENSE BLOCK ***** */
  39
  40 // Local Includes
  41 #include "stdafx.h"
  42 #include "Testembed.h"
  43 #include "ProfileMgr.h"
  44 #include "ProfilesDlg.h"
  45
  46 // Mozilla Includes
  47 #include "nsString.h"
  48 #include "nsXPIDLString.h"
  49 #include "nsIRegistry.h"
  50 #include "nsIProfile.h"
  51
  52 // Constants
  53 #define kRegistryGlobalPrefsSubtreeString (NS_LITERAL_STRING("global-prefs"))
  54 #define kRegistryShowProfilesAtStartup "start-show-dialog"
  55
  56 //*****************************************************************************
  57 //***    CProfileMgr: Object Management
  58 //*****************************************************************************
  59
  60 CProfileMgr::CProfileMgr()
  61 {
  62 }
  63
  64 CProfileMgr::~CProfileMgr()
  65 {
  66 }
  67
  68
  69 //*****************************************************************************
  70 //***    CProfileMgr: Public Methods
  71 //*****************************************************************************
  72
  73 nsresult CProfileMgr::StartUp()
  74 {
  75     nsresult rv;
  76
  77     //NS_WITH_SERVICE(nsIProfile, profileService, NS_PROFILE_CONTRACTID, &rv);
  78     nsCOMPtr<nsIProfile> profileService(do_GetService(NS_PROFILE_CONTRACTID,&rv));
  79     if (NS_FAILED(rv)) return rv;
  80
  81     PRInt32 profileCount;
  82     rv = profileService->GetProfileCount(&profileCount);
  83     if (NS_FAILED(rv)) return rv;
  84     if (profileCount == 0)
  85     {
  86         // Make a new default profile
  87         NS_NAMED_LITERAL_STRING(newProfileName, "default");
  88
  89         rv = profileService->CreateNewProfile(newProfileName.get(), nsnull, nsnull, PR_FALSE);
  90         if (NS_FAILED(rv)) return rv;
  91         rv = profileService->SetCurrentProfile(newProfileName.get());
  92         if (NS_FAILED(rv)) return rv;
  93     }
  94     else
  95     {
  96         // Use our flag here to chek for whether to show profile mgr UI. Ifc the flag
  97         // says don't show it, just start with the last used profile.
  98
  99         PRBool showIt;
 100         rv = GetShowDialogOnStart(&showIt);
 101
 102         if (NS_FAILED(rv) || (profileCount > 1 && showIt))
 103         {
 104             DoManageProfilesDialog(TRUE);
 105         }
 106         else
 107         {
 108             // GetCurrentProfile returns the profile which was last used but is not nescesarily
 109             // active. Call SetCurrentProfile to make it installed and active.
 110
 111             nsXPIDLString   currProfileName;
 112             rv = profileService->GetCurrentProfile(getter_Copies(currProfileName));
 113             if (NS_FAILED(rv)) return rv;
 114             rv = profileService->SetCurrentProfile(currProfileName);
 115             if (NS_FAILED(rv)) return rv;
 116         }
 117     }
 118
 119     return NS_OK;
 120 }
 121
 122 nsresult CProfileMgr::DoManageProfilesDialog(PRBool bAtStartUp)
 123 {
 124     CProfilesDlg    dialog;
 125     nsresult        rv;
 126     PRBool          showIt;
 127
 128     rv = GetShowDialogOnStart(&showIt);
 129     dialog.m_bAtStartUp = bAtStartUp;
 130     dialog.m_bAskAtStartUp = NS_SUCCEEDED(rv) ? showIt : TRUE;
 131
 132     if (dialog.DoModal() == IDOK)
 133     {
 134         SetShowDialogOnStart(dialog.m_bAskAtStartUp);
 135
 136         //NS_WITH_SERVICE(nsIProfile, profileService, NS_PROFILE_CONTRACTID, &rv);
 137             nsCOMPtr<nsIProfile> profileService(do_GetService(NS_PROFILE_CONTRACTID,&rv));
 138         if (NS_FAILED(rv))
 139                rv = profileService->SetCurrentProfile(dialog.m_SelectedProfile.get());
 140     }
 141     return NS_OK;
 142 }
 143
 144
 145 //*****************************************************************************
 146 //***    CProfileMgr: Protected Methods
 147 //*****************************************************************************
 148
 149 nsresult CProfileMgr::GetShowDialogOnStart(PRBool* showIt)
 150 {
 151     nsresult rv = NS_OK;
 152
 153     *showIt = PR_TRUE;
 154
 155     nsCOMPtr<nsIRegistry> registry(do_CreateInstance(NS_REGISTRY_CONTRACTID, &rv));
 156     rv = registry->OpenWellKnownRegistry(nsIRegistry::ApplicationRegistry);
 157     if (NS_FAILED(rv)) return rv;
 158
 159     nsRegistryKey profilesTreeKey;
 160
 161     rv = registry->GetKey(nsIRegistry::Common,
 162                           kRegistryGlobalPrefsSubtreeString.get(),
 163                           &profilesTreeKey);
 164
 165     if (NS_SUCCEEDED(rv))
 166     {
 167         PRInt32 flagValue;
 168         rv = registry->GetInt(profilesTreeKey,
 169                               kRegistryShowProfilesAtStartup,
 170                               &flagValue);
 171
 172         if (NS_SUCCEEDED(rv))
 173             *showIt = (flagValue != 0);
 174     }
 175     return rv;
 176 }
 177
 178 nsresult CProfileMgr::SetShowDialogOnStart(PRBool showIt)
 179 {
 180     nsresult rv = NS_OK;
 181
 182     nsCOMPtr<nsIRegistry> registry(do_CreateInstance(NS_REGISTRY_CONTRACTID, &rv));
 183     rv = registry->OpenWellKnownRegistry(nsIRegistry::ApplicationRegistry);
 184     if (NS_FAILED(rv)) return rv;
 185
 186     nsRegistryKey profilesTreeKey;
 187
 188     rv = registry->GetKey(nsIRegistry::Common,
 189                           kRegistryGlobalPrefsSubtreeString.get(),
 190                           &profilesTreeKey);
 191
 192     if (NS_FAILED(rv))
 193     {
 194         rv = registry->AddKey(nsIRegistry::Common,
 195                               kRegistryGlobalPrefsSubtreeString.get(),
 196                               &profilesTreeKey);
 197     }
 198     if (NS_SUCCEEDED(rv))
 199     {
 200
 201         rv = registry->SetInt(profilesTreeKey,
 202                               kRegistryShowProfilesAtStartup,
 203                               showIt);
 204     }
 205
 206     return rv;
 207 }
 208