source/import/import.cpp

   1 // import.cpp - Import subroutines.
   2 //
   3 // (c) 2012-2015 Xestia Software Development.
   4 //
   5 // This file is part of Xestia Address Book.
   6 //
   7 // Xestia Address Book is free software: you can redistribute it and/or modify
   8 // it under the terms of the GNU General Public License as published by the
   9 // Free Software Foundation, version 3 of the license.
  10 //
  11 // Xestia Address Book is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License along
  17 // with Xestia Address Book. If not, see <http://www.gnu.org/licenses/>
  18
  19 #include <wx/filedlg.h>
  20 #include <wx/filename.h>
  21 #include <wx/regex.h>
  22 #include <wx/tokenzr.h>
  23 #include <wx/ffile.h>
  24 #include <map>
  25
  26 #include "import-vcard3.h"
  27 #include "import-vcard4.h"
  28 #include "import-struct.h"
  29 #include "../frmMain.h"
  30 #include "frmImportContacts.h"
  31 #include "frmImportResults.h"
  32 #include "../common/uuid.h"
  33 #include "../common/dirs.h"
  34
  35 void ImportRun(frmMain *frmMainPtrInc){
  36
  37         // Bring up a dialog selecting one or multiple
  38         // files for processing based on type.
  39
  40         frmMainPtrInc->PauseAllTimers();
  41
  42         wxString FileTypes;
  43
  44         long ContactsImported = 0;
  45         long ContactsCollected = 0;
  46         int ImportType = 0;
  47         wxString FinalFilename;
  48         wxArrayString SelectedFileList;
  49         wxString SelectedFileDirectory;
  50         wxString AccountName;
  51         wxString AccountType;
  52         int ImportErrorCount = 0;
  53
  54         std::map<int,wxString> *ResultData = new std::map<int,wxString>;
  55
  56         FileTypes.Append(wxT("vCard 4.0 Contact(s) (*.vcf)"));
  57         FileTypes.Append(wxT("|*.vcf|"));
  58         FileTypes.Append(wxT("vCard 3.0 Contact(s) (*.vcf)"));
  59         FileTypes.Append(wxT("|*.vcf"));
  60
  61         // Open up the dialog to import file data.
  62
  63         wxFileDialog ImportDlg(frmMainPtrInc, wxT("Import Contacts"), wxT(""), wxT(""),
  64                 FileTypes, wxFD_OPEN|wxFD_MULTIPLE);
  65
  66         if (ImportDlg.ShowModal() == wxID_CANCEL){
  67
  68                 frmMainPtrInc->ResumeAllTimers();
  69                 return;
  70
  71         }
  72
  73         // Get the list of filenames.
  74
  75         ImportDlg.GetPaths(SelectedFileList);
  76
  77         // Find which data type is being used.
  78
  79         ImportType = ImportDlg.GetFilterIndex();
  80
  81         std::map<int, ImportDataContact> ContactData;
  82
  83         if (ImportType == 0){
  84
  85                 ContactData = ImportVCard4(&SelectedFileList);
  86
  87         }
  88
  89         // vCard 3.0 import
  90
  91         if (ImportType == 1){
  92
  93                 ContactData = ImportVCard3(&SelectedFileList);
  94
  95         }
  96
  97         // Check how many contacts are in the set. If zero, throw an error
  98         // message and exit.
  99
 100         if (ContactData.size() == 0){
 101
 102                 wxMessageBox(_("No valid contacts within the file(s) selected."), _("Unable to import"));
 103
 104                 return;
 105
 106         }
 107
 108         // Show a form to let the user select the
 109         // contacts needed to import and which account
 110         // they go to.
 111
 112         frmImportContacts *frmIC = new frmImportContacts(frmMainPtrInc);
 113
 114         frmIC->SetupList(&ContactData);
 115         frmIC->ShowModal();
 116         frmIC->GetResults(&ContactData);
 117         AccountName = frmIC->GetAccount();
 118         AccountType = frmIC->GetAccountType();
 119
 120         if (frmIC->GetDialogResult() == FALSE){
 121
 122                 // User decided not to import. Clean up the
 123                 // dialog.
 124
 125                 delete frmIC;
 126                 frmIC = NULL;
 127                 frmMainPtrInc->ResumeAllTimers();
 128                 return;
 129
 130         }
 131
 132         delete frmIC;
 133         frmIC = NULL;
 134
 135         // Import the contacts into the selected account.
 136
 137         int ImportCount = 0;
 138         int ImportSeek = 0;
 139
 140         // Setup the directory for writing into.
 141
 142         wxString AccountDirFinal = GetAccountDir(AccountName, FALSE);
 143
 144         wxString GeneratedFilename;
 145         wxString OutputFilename;
 146         wxString UIDValueFinal;
 147         wxString UIDToken;
 148         bool UIDFilterOK = FALSE;
 149         int FilterCount = 0;
 150
 151         GeneratedFilename.Clear();
 152         OutputFilename.Clear();
 153
 154         static wxString BadCharList = wxFileName::GetForbiddenChars(wxPATH_NATIVE);
 155         wxString BadCharListFinal;
 156         BadCharListFinal.Append(wxT("["));
 157         BadCharListFinal.Append(BadCharList);
 158         BadCharListFinal.Append(wxT("]"));
 159
 160         wxRegEx BadCharRex;
 161         BadCharRex.Compile(BadCharListFinal);
 162
 163         for(std::map<int,ImportDataContact>::iterator iter = ContactData.begin();
 164         iter != ContactData.end(); iter++){
 165
 166                 if (iter->second.ContactSelected == TRUE){
 167
 168                         // Workout the file name.
 169                         // Get the UID value and check if it is empty.
 170                         // If it isn't empty, check it can be used for a valid
 171                         // filename other wise generate a file name for the purposes
 172                         // of storage.
 173
 174                         // Filter UID incase writing outside the
 175                         // directory is attempted.
 176
 177 #if defined(__HAIKU__)
 178
 179
 180
 181 #elif defined(__WIN32__)
 182
 183                         wxStringTokenizer UIDValueCheck(iter->second.UIDValue, wxT("\\"));
 184                         while (UIDValueCheck.HasMoreTokens()){
 185                                 UIDValueCheck.GetNextToken();
 186                                 FilterCount++;
 187                         }
 188
 189 #else
 190
 191                         wxStringTokenizer UIDValueCheck(iter->second.UIDValue, wxT("/"));
 192                         while (UIDValueCheck.HasMoreTokens()){
 193                                 UIDValueCheck.GetNextToken();
 194                                 FilterCount++;
 195                         }
 196
 197 #endif
 198
 199                         if (FilterCount == 1){
 200                                 UIDFilterOK = TRUE;
 201                         }
 202
 203                         if (BadCharRex.Matches(iter->second.UIDValue) && UIDFilterOK == TRUE){
 204
 205                                 bool NewFileExists = TRUE;
 206
 207                                 while (NewFileExists){
 208
 209                                         // Generate the UID and check if it exists.
 210
 211                                         wxString CheckFilename;
 212
 213                                         UIDToken = GenerateUUID();
 214
 215                                         CheckFilename.Append(AccountDirFinal);
 216                                         CheckFilename.Append(UIDToken);
 217                                         CheckFilename.Append(wxT(".vcf"));
 218
 219                                         if (!wxFileExists(CheckFilename)){
 220
 221                                                 NewFileExists = FALSE;
 222
 223                                         }
 224
 225                                 }
 226
 227                         } else {
 228
 229                                 UIDToken = iter->second.UIDValue;
 230
 231                         }
 232
 233                         // Check if the file exists in the directory already.
 234
 235                         OutputFilename.Append(AccountDirFinal);
 236                         OutputFilename.Append(wxT("/"));
 237                         OutputFilename.Append(UIDToken);
 238                         OutputFilename.Append(wxT(".vcf"));
 239
 240                         if(wxFileExists(OutputFilename)){
 241
 242                                 // File already exists so mark as an error.
 243
 244                                 ResultData->insert(std::make_pair(ImportSeek, wxString::Format(_("Unable to import '%s'(%s): %s"), iter->second.FriendlyName, UIDToken, _("Contact already exists."))));
 245                                 ImportErrorCount++;
 246                                 ImportSeek++;
 247                                 GeneratedFilename.Clear();
 248                                 OutputFilename.Clear();
 249                                 UIDValueFinal.Clear();
 250                                 UIDFilterOK = FALSE;
 251                                 FilterCount = 0;
 252                                 continue;
 253
 254                         } else {
 255
 256                                 // Write to the file.
 257
 258                                 wxFFile OutputFile;
 259
 260 #if wxABI_VERSION < 20900
 261                                 if(!OutputFile.Open(OutputFilename, wxT("w"))){
 262 #else
 263                                 if(!OutputFile.Open(OutputFilename, wxT("w"))){
 264 #endif
 265                                         ResultData->insert(std::make_pair(ImportSeek, wxString::Format(_("Unable to import '%s'(%s): %s"), iter->second.FriendlyName, UIDToken, _("Error occured whilst trying to create this contact. Please check permissions and storage device."))));
 266
 267                                 }
 268
 269                                 OutputFile.Write(iter->second.ContactData, wxConvAuto());
 270
 271                                 OutputFile.Close();
 272
 273                                 ResultData->insert(std::make_pair(ImportSeek, wxString::Format(_("Contact '%s'(%s) was successfully imported."), iter->second.FriendlyName, UIDToken)));
 274
 275                         }
 276
 277                         // Check if there is a UID value set. If there is not a UID
 278                         // value set then generate a new one.
 279
 280                         ImportCount++;
 281
 282                         ImportSeek++;
 283
 284                         GeneratedFilename.Clear();
 285                         OutputFilename.Clear();
 286                         UIDValueFinal.Clear();
 287                         UIDToken.Clear();
 288                         UIDFilterOK = FALSE;
 289                         FilterCount = 0;
 290
 291                 }
 292
 293         }
 294
 295         // Check if the import error count is more than 0. If it is, then
 296         // display the results dialog.
 297
 298         wxCommandEvent imprres(IMPORT_RESULTSSHOW);
 299         imprres.SetClientData(ResultData);
 300         imprres.SetInt(ImportCount);
 301         imprres.SetExtraLong(ImportErrorCount);
 302         wxPostEvent(frmMainPtrInc, imprres);
 303
 304         // Syncronise the account which the contacts got imported to.
 305         // (If the account is not a local account).
 306
 307         if (AccountType != wxT("Local") && AccountType != wxT("local")){
 308
 309                 wxString AccNamePostEvent;
 310
 311                 AccNamePostEvent.Clear();
 312                 AccNamePostEvent.Append(AccountName);
 313                 AccNamePostEvent.Trim();
 314
 315                 wxCommandEvent accevent(SYNCACCOUNT);
 316                 accevent.SetString(AccNamePostEvent);
 317
 318                 wxPostEvent(frmMainPtrInc, accevent);
 319
 320         }
 321
 322         // Resume the timers.
 323
 324         frmMainPtrInc->ResumeAllTimers();
 325
 326 }