1 files changed, 84 insertions, 0 deletions
diff --git a/comm/mail/components/migration/src/nsSeamonkeyProfileMigrator.h b/comm/mail/components/migration/src/nsSeamonkeyProfileMigrator.h
new file mode 100644
index 0000000000..e085e1b2cd
--- /dev/null
+++ b/comm/mail/components/migration/src/nsSeamonkeyProfileMigrator.h
@@ -0,0 +1,84 @@
+/* -*- 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