diff options
Diffstat (limited to 'comm/mailnews/import/src/nsImportFieldMap.cpp')
| -rw-r--r-- | comm/mailnews/import/src/nsImportFieldMap.cpp | 325 |
1 files changed, 325 insertions, 0 deletions
diff --git a/comm/mailnews/import/src/nsImportFieldMap.cpp b/comm/mailnews/import/src/nsImportFieldMap.cpp new file mode 100644 index 0000000000..74e15f11fe --- /dev/null +++ b/comm/mailnews/import/src/nsImportFieldMap.cpp @@ -0,0 +1,325 @@ +/* -*- 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/. */ + +#include "nscore.h" +#include "nsIAbCard.h" +#include "nsIStringBundle.h" +#include "nsImportFieldMap.h" +#include "nsImportStringBundle.h" +#include "nsCRTGlue.h" +#include "ImportDebug.h" +#include "nsCOMPtr.h" + +//////////////////////////////////////////////////////////////////////// + +nsresult nsImportFieldMap::Create(nsIStringBundle* aBundle, REFNSIID aIID, + void** aResult) { + RefPtr<nsImportFieldMap> it = new nsImportFieldMap(aBundle); + return it->QueryInterface(aIID, aResult); +} + +NS_IMPL_ISUPPORTS(nsImportFieldMap, nsIImportFieldMap) + +NS_IMETHODIMP nsImportFieldMap::GetSkipFirstRecord(bool* result) { + NS_ENSURE_ARG_POINTER(result); + *result = m_skipFirstRecord; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::SetSkipFirstRecord(bool aResult) { + m_skipFirstRecord = aResult; + return NS_OK; +} + +nsImportFieldMap::nsImportFieldMap(nsIStringBundle* aBundle) { + m_numFields = 0; + m_pFields = nullptr; + m_pActive = nullptr; + m_allocated = 0; + // need to init the description array + m_mozFieldCount = 0; + m_skipFirstRecord = false; + nsCOMPtr<nsIStringBundle> pBundle = aBundle; + + nsString* pStr; + for (int32_t i = IMPORT_FIELD_DESC_START; i <= IMPORT_FIELD_DESC_END; + i++, m_mozFieldCount++) { + pStr = new nsString(); + if (pBundle) { + nsImportStringBundle::GetStringByID(i, pBundle, *pStr); + } else + pStr->AppendInt(i); + m_descriptions.AppendElement(pStr); + } +} + +nsImportFieldMap::~nsImportFieldMap() { + if (m_pFields) delete[] m_pFields; + if (m_pActive) delete[] m_pActive; + + nsString* pStr; + for (int32_t i = 0; i < m_mozFieldCount; i++) { + pStr = m_descriptions.ElementAt(i); + delete pStr; + } + m_descriptions.Clear(); +} + +NS_IMETHODIMP nsImportFieldMap::GetNumMozFields(int32_t* aNumFields) { + NS_ASSERTION(aNumFields != nullptr, "null ptr"); + if (!aNumFields) return NS_ERROR_NULL_POINTER; + + *aNumFields = m_mozFieldCount; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::GetMapSize(int32_t* aNumFields) { + NS_ASSERTION(aNumFields != nullptr, "null ptr"); + if (!aNumFields) return NS_ERROR_NULL_POINTER; + + *aNumFields = m_numFields; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::GetFieldDescription(int32_t index, + char16_t** _retval) { + NS_ASSERTION(_retval != nullptr, "null ptr"); + if (!_retval) return NS_ERROR_NULL_POINTER; + + *_retval = nullptr; + if ((index < 0) || ((size_t)index >= m_descriptions.Length())) + return NS_ERROR_FAILURE; + + *_retval = ToNewUnicode(*(m_descriptions.ElementAt(index))); + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::SetFieldMapSize(int32_t size) { + nsresult rv = Allocate(size); + if (NS_FAILED(rv)) return rv; + + m_numFields = size; + + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::DefaultFieldMap(int32_t size) { + nsresult rv = SetFieldMapSize(size); + if (NS_FAILED(rv)) return rv; + for (int32_t i = 0; i < size; i++) { + m_pFields[i] = i; + m_pActive[i] = true; + } + + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::GetFieldMap(int32_t index, int32_t* _retval) { + NS_ASSERTION(_retval != nullptr, "null ptr"); + if (!_retval) return NS_ERROR_NULL_POINTER; + + if ((index < 0) || (index >= m_numFields)) return NS_ERROR_FAILURE; + + *_retval = m_pFields[index]; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::SetFieldMap(int32_t index, int32_t fieldNum) { + if (index == -1) { + nsresult rv = Allocate(m_numFields + 1); + if (NS_FAILED(rv)) return rv; + index = m_numFields; + m_numFields++; + } else { + if ((index < 0) || (index >= m_numFields)) return NS_ERROR_FAILURE; + } + + if ((fieldNum != -1) && ((fieldNum < 0) || (fieldNum >= m_mozFieldCount))) + return NS_ERROR_FAILURE; + + m_pFields[index] = fieldNum; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::GetFieldActive(int32_t index, bool* active) { + NS_ASSERTION(active != nullptr, "null ptr"); + if (!active) return NS_ERROR_NULL_POINTER; + if ((index < 0) || (index >= m_numFields)) return NS_ERROR_FAILURE; + + *active = m_pActive[index]; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::SetFieldActive(int32_t index, bool active) { + if ((index < 0) || (index >= m_numFields)) return NS_ERROR_FAILURE; + + m_pActive[index] = active; + return NS_OK; +} + +NS_IMETHODIMP nsImportFieldMap::SetFieldValue(nsIAbDirectory* database, + nsIAbCard* row, int32_t fieldNum, + const nsAString& value) { + // Allow the special value for a null field + if (fieldNum == -1) return NS_OK; + + if ((fieldNum < 0) || (fieldNum >= m_mozFieldCount)) return NS_ERROR_FAILURE; + + // UGGG!!!!! lot's of typing here! + nsresult rv; + + switch (fieldNum) { + case 0: + rv = row->SetFirstName(value); + break; + case 1: + rv = row->SetLastName(value); + break; + case 2: + rv = row->SetDisplayName(value); + break; + case 3: + rv = row->SetPropertyAsAString(kNicknameProperty, value); + break; + case 4: + rv = row->SetPrimaryEmail(value); + break; + case 5: + rv = row->SetPropertyAsAString(k2ndEmailProperty, value); + break; + case 6: + rv = row->SetPropertyAsAString(kWorkPhoneProperty, value); + break; + case 7: + rv = row->SetPropertyAsAString(kHomePhoneProperty, value); + break; + case 8: + rv = row->SetPropertyAsAString(kFaxProperty, value); + break; + case 9: + rv = row->SetPropertyAsAString(kPagerProperty, value); + break; + case 10: + rv = row->SetPropertyAsAString(kCellularProperty, value); + break; + case 11: + rv = row->SetPropertyAsAString(kHomeAddressProperty, value); + break; + case 12: + rv = row->SetPropertyAsAString(kHomeAddress2Property, value); + break; + case 13: + rv = row->SetPropertyAsAString(kHomeCityProperty, value); + break; + case 14: + rv = row->SetPropertyAsAString(kHomeStateProperty, value); + break; + case 15: + rv = row->SetPropertyAsAString(kHomeZipCodeProperty, value); + break; + case 16: + rv = row->SetPropertyAsAString(kHomeCountryProperty, value); + break; + case 17: + rv = row->SetPropertyAsAString(kWorkAddressProperty, value); + break; + case 18: + rv = row->SetPropertyAsAString(kWorkAddress2Property, value); + break; + case 19: + rv = row->SetPropertyAsAString(kWorkCityProperty, value); + break; + case 20: + rv = row->SetPropertyAsAString(kWorkStateProperty, value); + break; + case 21: + rv = row->SetPropertyAsAString(kWorkZipCodeProperty, value); + break; + case 22: + rv = row->SetPropertyAsAString(kWorkCountryProperty, value); + break; + case 23: + rv = row->SetPropertyAsAString(kJobTitleProperty, value); + break; + case 24: + rv = row->SetPropertyAsAString(kDepartmentProperty, value); + break; + case 25: + rv = row->SetPropertyAsAString(kCompanyProperty, value); + break; + case 26: + rv = row->SetPropertyAsAString(kWorkWebPageProperty, value); + break; + case 27: + rv = row->SetPropertyAsAString(kHomeWebPageProperty, value); + break; + case 28: + rv = row->SetPropertyAsAString(kBirthYearProperty, value); + break; + case 29: + rv = row->SetPropertyAsAString(kBirthMonthProperty, value); + break; + case 30: + rv = row->SetPropertyAsAString(kBirthDayProperty, value); + break; + case 31: + rv = row->SetPropertyAsAString(kCustom1Property, value); + break; + case 32: + rv = row->SetPropertyAsAString(kCustom2Property, value); + break; + case 33: + rv = row->SetPropertyAsAString(kCustom3Property, value); + break; + case 34: + rv = row->SetPropertyAsAString(kCustom4Property, value); + break; + case 35: + rv = row->SetPropertyAsAString(kNotesProperty, value); + break; + case 36: + rv = row->SetPropertyAsAString(kAIMProperty, value); + break; + default: + /* Get the field description, and add it as an anonymous attr? */ + /* OR WHAT???? */ + { rv = NS_ERROR_FAILURE; } + } + + return rv; +} + +nsresult nsImportFieldMap::Allocate(int32_t newSize) { + if (newSize <= m_allocated) return NS_OK; + + int32_t sz = m_allocated; + while (sz < newSize) sz += 30; + + int32_t* pData = new int32_t[sz]; + if (!pData) return NS_ERROR_OUT_OF_MEMORY; + bool* pActive = new bool[sz]; + if (!pActive) { + delete[] pData; + return NS_ERROR_OUT_OF_MEMORY; + } + + int32_t i; + for (i = 0; i < sz; i++) { + pData[i] = -1; + pActive[i] = true; + } + if (m_numFields) { + for (i = 0; i < m_numFields; i++) { + pData[i] = m_pFields[i]; + pActive[i] = m_pActive[i]; + } + delete[] m_pFields; + delete[] m_pActive; + } + m_allocated = sz; + m_pFields = pData; + m_pActive = pActive; + return NS_OK; +} |