Source code
Revision control
Copy as Markdown
Other Tools
/* -*- 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
#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;
}