Fixed importing clients without projects in new import.
[timetracker.git] / WEB-INF / lib / ttOrgImportHelper.class.php
index cc632f1..77458cf 100644 (file)
@@ -34,19 +34,19 @@ import('ttClientHelper');
 import('ttInvoiceHelper');
 import('ttCustomFieldHelper');
 import('ttExpenseHelper');
+import('ttFavReportHelper');
 
-// ttOrgImportHelper - this class is a future replacement for ttImportHelper.
-// Currently, it is work in progress.
-// When done, it should handle import of complex groups consisting of other groups.
+// ttOrgImportHelper class is used to import organization data from an XML file
+// prepared by ttOrgExportHelper and consisting of nested groups with their info.
 class ttOrgImportHelper {
   var $errors               = null; // Errors go here. Set in constructor by reference.
-  var $conflicting_entities = null; // A comma-separated list of entity names we cannot import.
+  var $schema_version       = null; // Database schema version from XML file we import from.
+  var $conflicting_logins   = null; // A comma-separated list of logins we cannot import.
   var $canImport      = true;    // False if we cannot import data due to a conflict such as login collision.
   var $firstPass      = true;    // True during first pass through the file.
   var $org_id         = null;    // Organization id (same as top group_id).
   var $current_group_id        = null; // Current group id during parsing.
   var $current_parent_group_id = null; // Current parent group id during parsing.
-                                       // Set when we create a new group.
   var $top_role_id    = 0;       // Top role id.
 
   // Entity maps for current group. They map XML ids with database ids.
@@ -66,17 +66,34 @@ class ttOrgImportHelper {
     $this->top_role_id = ttRoleHelper::getRoleByRank(512, 0);
   }
 
-  // startElement - callback handler for opening tag of an XML element in the file.
+  // startElement - callback handler for opening tags in XML.
   function startElement($parser, $name, $attrs) {
     global $i18n;
 
-    // First pass. We only check user logins for potential collisions with existing.
+    // First pass through the file determines if we can import data.
+    // We require 2 things:
+    //   1) Database schema version must be set. This ensures we have a compatible file.
+    //   2) No login coillisions are allowed.
     if ($this->firstPass) {
+      if ($name == 'ORG' && $this->canImport) {
+         if ($attrs['SCHEMA'] == null) {
+           // We need (database) schema attribute to be available for import to work.
+           // Old Time Tracker export files don't have this.
+           // Current import code does not work with old format because we had to
+           // restructure data in export files for subgroup support.
+           $this->canImport = false;
+           $this->errors->add($i18n->get('error.format'));
+           return;
+         }
+      }
+
+      // In first pass we check user logins for potential collisions with existing.
       if ($name == 'USER' && $this->canImport) {
         $login = $attrs['LOGIN'];
         if ('' != $attrs['STATUS'] && ttUserHelper::getUserByLogin($login)) {
           // We have a login collision. Append colliding login to a list of things we cannot import.
-          $this->conflicting_entities .= ($this->conflicting_entities ? ", $login" : $login);
+          $this->conflicting_logins .= ($this->conflicting_logins ? ", $login" : $login);
+          // The above is printed in error message with all found colliding logins.
         }
       }
     }
@@ -119,6 +136,7 @@ class ttOrgImportHelper {
         }
         // Set parent group to create subgroups with this group as parent at next entry here.
         $this->current_parent_group_id = $this->current_group_id;
+        return;
       }
 
       if ($name == 'ROLES') {
@@ -126,22 +144,24 @@ class ttOrgImportHelper {
         unset($this->currentGroupRoleMap);
         $this->currentGroupRoleMap = array();
         // Role map is reconstructed after processing <role> elements in XML. See below.
+        return;
       }
 
       if ($name == 'ROLE') {
         // We get here when processing <role> tags for the current group.
         $role_id = ttRoleHelper::insert(array(
-              'group_id' => $this->current_group_id,
-              'org_id' => $this->org_id,
-              'name' => $attrs['NAME'],
-              'description' => $attrs['DESCRIPTION'],
-              'rank' => $attrs['RANK'],
-              'rights' => $attrs['RIGHTS'],
-              'status' => $attrs['STATUS']));
+          'group_id' => $this->current_group_id,
+          'org_id' => $this->org_id,
+          'name' => $attrs['NAME'],
+          'description' => $attrs['DESCRIPTION'],
+          'rank' => $attrs['RANK'],
+          'rights' => $attrs['RIGHTS'],
+          'status' => $attrs['STATUS']));
         if ($role_id) {
           // Add a mapping.
           $this->currentGroupRoleMap[$attrs['ID']] = $role_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'TASKS') {
@@ -149,6 +169,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupTaskMap);
         $this->currentGroupTaskMap = array();
         // Task map is reconstructed after processing <task> elements in XML. See below.
+        return;
       }
 
       if ($name == 'TASK') {
@@ -163,6 +184,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupTaskMap[$attrs['ID']] = $task_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'PROJECTS') {
@@ -170,6 +192,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupProjectMap);
         $this->currentGroupProjectMap = array();
         // Project map is reconstructed after processing <project> elements in XML. See below.
+        return;
       }
 
       if ($name == 'PROJECT') {
@@ -191,6 +214,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupProjectMap[$attrs['ID']] = $project_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'CLIENTS') {
@@ -198,15 +222,18 @@ class ttOrgImportHelper {
         unset($this->currentGroupClientMap);
         $this->currentGroupClientMap = array();
         // Client map is reconstructed after processing <client> elements in XML. See below.
+        return;
       }
 
       if ($name == 'CLIENT') {
         // We get here when processing <client> tags for the current group.
 
         // Prepare a list of project ids.
-        $projects = explode(',', $attrs['PROJECTS']);
-        foreach ($projects as $id)
-          $mapped_projects[] = $this->currentGroupProjectMap[$id];
+        if ($attrs['PROJECTS']) {
+          $projects = explode(',', $attrs['PROJECTS']);
+          foreach ($projects as $id)
+            $mapped_projects[] = $this->currentGroupProjectMap[$id];
+        }
 
         $client_id = ttClientHelper::insert(array(
           'group_id' => $this->current_group_id,
@@ -220,6 +247,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupClientMap[$attrs['ID']] = $client_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'USERS') {
@@ -227,6 +255,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupUserMap);
         $this->currentGroupUserMap = array();
         // User map is reconstructed after processing <user> elements in XML. See below.
+        return;
       }
 
       if ($name == 'USER') {
@@ -249,6 +278,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupUserMap[$attrs['ID']] = $user_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'USER_PROJECT_BIND') {
@@ -261,6 +291,7 @@ class ttOrgImportHelper {
           'status' => $attrs['STATUS']))) {
           $this->errors->add($i18n->get('error.db'));
         }
+        return;
       }
 
       if ($name == 'INVOICES') {
@@ -268,6 +299,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupInvoiceMap);
         $this->currentGroupInvoiceMap = array();
         // Invoice map is reconstructed after processing <invoice> elements in XML. See below.
+        return;
       }
 
       if ($name == 'INVOICE') {
@@ -283,6 +315,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupInvoiceMap[$attrs['ID']] = $invoice_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'LOG') {
@@ -290,6 +323,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupLogMap);
         $this->currentGroupLogMap = array();
         // Log map is reconstructed after processing <log_item> elements in XML. See below.
+        return;
       }
 
       if ($name == 'LOG_ITEM') {
@@ -314,6 +348,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupLogMap[$attrs['ID']] = $log_item_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'CUSTOM_FIELDS') {
@@ -321,6 +356,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupCustomFieldMap);
         $this->currentGroupCustomFieldMap = array();
         // Custom field map is reconstructed after processing <custom_field> elements in XML. See below.
+        return;
       }
 
       if ($name == 'CUSTOM_FIELD') {
@@ -336,6 +372,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupCustomFieldMap[$attrs['ID']] = $custom_field_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'CUSTOM_FIELD_OPTIONS') {
@@ -343,6 +380,7 @@ class ttOrgImportHelper {
         unset($this->currentGroupCustomFieldOptionMap);
         $this->currentGroupCustomFieldOptionMap = array();
         // Custom field option map is reconstructed after processing <custom_field_option> elements in XML. See below.
+        return;
       }
 
       if ($name == 'CUSTOM_FIELD_OPTION') {
@@ -356,6 +394,7 @@ class ttOrgImportHelper {
           // Add a mapping.
           $this->currentGroupCustomFieldOptionMap[$attrs['ID']] = $custom_field_option_id;
         } else $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'CUSTOM_FIELD_LOG_ENTRY') {
@@ -370,6 +409,7 @@ class ttOrgImportHelper {
           'status' => $attrs['STATUS']))) {
           $this->errors->add($i18n->get('error.db'));
         }
+        return;
       }
 
       if ($name == 'EXPENSE_ITEM') {
@@ -387,6 +427,7 @@ class ttOrgImportHelper {
           'paid' => $attrs['PAID'],
           'status' => $attrs['STATUS']));
         if (!$expense_item_id) $this->errors->add($i18n->get('error.db'));
+        return;
       }
 
       if ($name == 'MONTHLY_QUOTA') {
@@ -397,6 +438,47 @@ class ttOrgImportHelper {
           $attrs['MINUTES'])) {
           $this->errors->add($i18n->get('error.db'));
         }
+        return;
+      }
+
+      if ($name == 'FAV_REPORT') {
+        $user_list = '';
+        if (strlen($attrs['USERS']) > 0) {
+          $arr = explode(',', $attrs['USERS']);
+          foreach ($arr as $v)
+            $user_list .= (strlen($user_list) == 0 ? '' : ',').$this->currentGroupUserMap[$v];
+        }
+        $fav_report_id = ttFavReportHelper::insertReport(array(
+          'name' => $attrs['NAME'],
+          'user_id' => $this->currentGroupUserMap[$attrs['USER_ID']],
+          'client' => $this->currentGroupClientMap[$attrs['CLIENT_ID']],
+          'option' => $this->currentGroupCustomFieldOptionMap[$attrs['CF_1_OPTION_ID']],
+          'project' => $this->currentGroupProjectMap[$attrs['PROJECT_ID']],
+          'task' => $this->currentGroupTaskMap[$attrs['TASK_ID']],
+          'billable' => $attrs['BILLABLE'],
+          'users' => $user_list,
+          'period' => $attrs['PERIOD'],
+          'from' => $attrs['PERIOD_START'],
+          'to' => $attrs['PERIOD_END'],
+          'chclient' => (int) $attrs['SHOW_CLIENT'],
+          'chinvoice' => (int) $attrs['SHOW_INVOICE'],
+          'chpaid' => (int) $attrs['SHOW_PAID'],
+          'chip' => (int) $attrs['SHOW_IP'],
+          'chproject' => (int) $attrs['SHOW_PROJECT'],
+          'chstart' => (int) $attrs['SHOW_START'],
+          'chduration' => (int) $attrs['SHOW_DURATION'],
+          'chcost' => (int) $attrs['SHOW_COST'],
+          'chtask' => (int) $attrs['SHOW_TASK'],
+          'chfinish' => (int) $attrs['SHOW_END'],
+          'chnote' => (int) $attrs['SHOW_NOTE'],
+          'chcf_1' => (int) $attrs['SHOW_CUSTOM_FIELD_1'],
+          'chunits' => (int) $attrs['SHOW_WORK_UNITS'],
+          'group_by1' => $attrs['GROUP_BY1'],
+          'group_by2' => $attrs['GROUP_BY2'],
+          'group_by3' => $attrs['GROUP_BY3'],
+          'chtotalsonly' => (int) $attrs['SHOW_TOTALS_ONLY']));
+         if (!$fav_report_id) $this->errors->add($i18n->get('error.db'));
+         return;
       }
     }
   }
@@ -438,13 +520,10 @@ class ttOrgImportHelper {
     xml_set_element_handler($parser, 'startElement', false);
 
     // We need to parse the file 2 times:
-    //   1) First pass: determine if import is possible - there must be no login collisions.
-    //   2) Second pass: if we can import, then do import in a second pass.
-    // This is different from earlier approach for single group import, where we could
-    // do both things in one pass because user info was in the beginning of XML file.
-    // Now, with subgroups, users can be located anywhere in the file.
+    //   1) First pass: determine if import is possible.
+    //   2) Second pass: import data, one tag at a time.
 
-    // Read and parse the content of the file. During parsing, startElement, endElement, and dataElement functions are called.
+    // Read and parse the content of the file. During parsing, startElement is called back for each tag.
     $file = fopen($filename, 'r');
     while ($data = fread($file, 4096)) {
       if (!xml_parse($parser, $data, feof($file))) {
@@ -453,10 +532,10 @@ class ttOrgImportHelper {
           xml_error_string(xml_get_error_code($parser))));
       }
     }
-    if ($this->conflicting_entities) {
+    if ($this->conflicting_logins) {
       $this->canImport = false;
       $this->errors->add($i18n->get('error.user_exists'));
-      $this->errors->add(sprintf($i18n->get('error.cannot_import'), $this->conflicting_entities));
+      $this->errors->add(sprintf($i18n->get('error.cannot_import'), $this->conflicting_logins));
     }
 
     $this->firstPass = false; // We are done with 1st pass.
@@ -473,7 +552,7 @@ class ttOrgImportHelper {
     xml_set_object($parser, $this);
     xml_set_element_handler($parser, 'startElement', false);
 
-    // Read and parse the content of the file. During parsing, startElement, endElement, and dataElement functions are called.
+    // Read and parse the content of the file. During parsing, startElement is called back for each tag.
     $file = fopen($filename, 'r');
     while ($data = fread($file, 4096)) {
       if (!xml_parse($parser, $data, feof($file))) {