nsSeamonkeyProfileMigrator.h

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef seamonkeyprofilemigrator___h___

#define seamonkeyprofilemigrator___h___

#include "nsTHashMap.h"

#include "nsIMailProfileMigrator.h"

#include "nsIMsgAccountManager.h"

#include "nsNetscapeProfileMigratorBase.h"

class nsIPrefBranch;

class nsIPrefService;

class nsSeamonkeyProfileMigrator : public nsNetscapeProfileMigratorBase {

 public:

  NS_DECL_ISUPPORTS_INHERITED

  nsSeamonkeyProfileMigrator();

  // nsIMailProfileMigrator methods

  NS_IMETHOD Migrate(uint16_t aItems, nsIProfileStartup* aStartup,

                     const char16_t* aProfile) override;

  NS_IMETHOD GetMigrateData(const char16_t* aProfile, bool aReplace,

                            uint16_t* aResult) override;

  NS_IMETHOD GetSourceProfiles(nsTArray<nsString>& aResult) override;

  NS_IMETHOD GetSourceProfileLocations(

      nsTArray<RefPtr<nsIFile>>& aResult) override;

 protected:

  virtual ~nsSeamonkeyProfileMigrator();

  nsresult FillProfileDataFromSeamonkeyRegistry();

  nsresult GetSourceProfile(const char16_t* aProfile);

  nsresult MigrateMABFile(const nsCString& aDirPrefId,

                          const nsCString& aSourceFileName);

  nsresult CopyPreferences(bool aReplace);

  nsresult ImportPreferences(uint16_t aItems);

  nsresult TransformPreferences(const nsAString& aSourcePrefFileName,

                                const nsAString& aTargetPrefFileName);

  nsresult DummyCopyRoutine(bool aReplace);

  nsresult CopyJunkTraining(bool aReplace);

  nsresult CopyPasswords(bool aReplace);

  nsresult CopyMailFolders(PBStructArray& aMailServers,

                           nsIPrefService* aPrefBranch);

  nsresult CopySignatureFiles(PBStructArray& aIdentities,

                              nsIPrefService* aPrefBranch);

  typedef nsTHashMap<nsCStringHashKey, nsCString> PrefKeyHashTable;

  nsresult TransformIdentitiesForImport(

      PBStructArray& aIdentities, nsIMsgAccountManager* accountManager,

      PrefKeyHashTable& smtpServerKeyHashTable, PrefKeyHashTable& keyHashTable);

  nsresult TransformMailAccountsForImport(

      nsIPrefService* aPrefService, PBStructArray& aAccounts,

      nsIMsgAccountManager* accountManager,

      PrefKeyHashTable& identityKeyHashTable,

      PrefKeyHashTable& serverKeyHashTable);

  nsresult TransformMailServersForImport(const char* branchName,

                                         nsIPrefService* aPrefService,

                                         PBStructArray& aMailServers,

                                         nsIMsgAccountManager* accountManager,

                                         PrefKeyHashTable& keyHashTable);

  nsresult TransformSmtpServersForImport(PBStructArray& aServers,

                                         PrefKeyHashTable& keyHashTable);

  nsresult TransformAddressbooksForImport(nsIPrefService* aPrefService,

                                          PBStructArray& aAddressbooks,

                                          bool aReplace);

  void ReadBranch(const char* branchName, nsIPrefService* aPrefService,

                  PBStructArray& aPrefs);

  void WriteBranch(const char* branchName, nsIPrefService* aPrefService,

                   PBStructArray& aPrefs, bool deallocate = true);

 private:

  nsTArray<nsString> mProfileNames;

  nsTArray<RefPtr<nsIFile>> mProfileLocations;

};

#endif