]> wagnertech.de Git - timetracker.git/blobdiff - WEB-INF/lib/ttOrgImportHelper.class.php
A bit of refactoring in ttOrgImportHelper class.
[timetracker.git] / WEB-INF / lib / ttOrgImportHelper.class.php
index cce763da6b2991628a22715a47fa6785cbfd55df..f3a2d8eaa49b9dd6393d910e38718b797e74a524 100644 (file)
@@ -50,6 +50,21 @@ class ttOrgImportHelper {
   var $canImport      = true;    // False if we cannot import data due to a 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_parent_group_id = null; // Current parent group id as we parse the file.
+                                       // Set when we create a new group.
+  // Entities for current group.
+  var $currentGroupRoles = array(); // Array of arrays of role properties.
+  // var $currentGroupUsers = array(); // Array of arrays of user properties.
+
+  // Entity maps for current group. They map XML ids with database ids.
+  var $currentGroupRoleMap = array(); // Maps role ids from XML to their database ids.
+  //var $userMap       = array(); // User ids.
+  //var $projectMap    = array(); // Project ids.
+  //var $taskMap       = array(); // Task ids.
+  //var $clientMap     = array(); // Client ids.
+  //var $invoiceMap    = array(); // Invoice ids.
+
   // Constructor.
   function __construct(&$errors) {
     $this->errors = &$errors;
@@ -58,17 +73,75 @@ class ttOrgImportHelper {
   // startElement - callback handler for opening tag of an XML element.
   // In this function we assign passed in attributes to currentElement.
   function startElement($parser, $name, $attrs) {
+/*
     if ($name == 'GROUP'
       || $name == 'USER') {
       $this->currentElement = $attrs;
     }
     $this->currentTag = $name;
+*/
+    // First pass. We only check user logins for potential collisions with existing.
+    if ($this->firstPass) {
+      if ($name == 'USER' && $this->canImport) {
+        if ('' != $attrs['STATUS'] && ttUserHelper::getUserByLogin($attrs['LOGIN'])) {
+          // We have a login collision, cannot import any data.
+          $this->canImport = false;
+        }
+      }
+      //$this->currentTag = '';
+    }
+
+    // Second pass processing. We import data here, one tag at a time.
+    if (!$this->firstPass && $this->canImport) {
+      $mdb2 = getConnection();
+
+      // We are in second pass through the XML file and can import data.
+      if ($name == 'GROUP') {
+        // Create a new group.
+        $group_id = $this->createGroup(array(
+          'parent_id' => $this->current_parent_group_id,
+          'org_id' => $this->org_id,
+          'name' => $this->currentElement['NAME'],
+          'currency' => $this->currentElement['CURRENCY'],
+          'lang' => $this->currentElement['LANG']));
+        // We only have 3 properties in export at the moment, while work is ongoing...
+
+        // Special handling for top group.
+        if (!$this->org_id) {
+          $this->org_id = $group_id;
+          $sql = "update tt_groups set org_id = $group_id where org_id is NULL and id = $group_id";
+          $affected = $mdb2->exec($sql);
+          // TODO: design a better error handling approach for the entire import process.
+        }
+        // Set current parent group.
+        $this->current_parent_group_id = $group_id;
+      }
+
+      if ($name == 'ROLES') {
+        // If we get here, we have to recycle both $currentGroupRoles and $currentGroupRoleMap.
+        unset($this->currentGroupRoles);
+        unset($this->currentGroupRoleMap);
+        $this->currentGroupRoles = array();
+        $this->currentGroupRoleMap = array();
+        // Both arrays are now empty.
+        // They will get reconstructed after processing of <role> elements in XML. See below.
+      }
+
+      if ($name == 'ROLE') {
+        // We get here when processing a <role> tag for the current group.
+        // Add new role to $this->currentGroupRoles and a mapping to $this->currentGroupRoleMap.
+        $this->currentGroupRoles[$this->currentElement['ID']] = $this->currentElement;
+      $this->currentElement = array();
+      }
+    }
   }
 
   // endElement - callback handler for the closing tag of an XML element.
   // When we are here, currentElement is an array of the element attributes (as set in startElement).
   // Here we do the actual import of data into the database.
   function endElement($parser, $name) {
+    // Do nothing here. Everything is done in startElement to keep things simple.
+    /*
     // During first pass we only check user logins.
     if ($this->firstPass) {
       if ($name == 'USER' && $this->canImport) {
@@ -82,12 +155,14 @@ class ttOrgImportHelper {
 
     // During second pass we import data.
     if (!$this->firstPass && $this->canImport) {
-      // TODO: write code here. Nothing is imported currently.
-    }
+      // Nothing is done here, see startElement for second pass.
+    }*/
   }
 
   // dataElement - callback handler for text data fragments. It builds up currentElement array with text pieces from XML.
   function dataElement($parser, $data) {
+    // New approach is to do nothing here. Everything is now done when processing start tag (startElement).
+      /*
     if ($this->currentTag == 'NAME'
       || $this->currentTag == 'DESCRIPTION'
       || $this->currentTag == 'LABEL'
@@ -101,6 +176,7 @@ class ttOrgImportHelper {
       else
         $this->currentElement[$this->currentTag] = trim($data);
     }
+       * */
   }
 
   // importXml - uncompresses the file, reads and parses its content. During parsing,
@@ -213,4 +289,49 @@ class ttOrgImportHelper {
     fclose ($out_file);
     return true;
   }
+
+  // createGroup function creates a new group.
+  private function createGroup($fields) {
+
+    global $user;
+    $mdb2 = getConnection();
+
+    $columns = '(parent_id, org_id, name, currency, lang)';
+
+//    $columns = '(name, currency, decimal_mark, lang, date_format, time_format, week_start, tracking_mode'.
+//      ', project_required, task_required, record_type, bcc_email, allow_ip, password_complexity, plugins'.
+//      ', lock_spec, workday_minutes, config, created, created_ip, created_by)';
+
+    $values = ' values (';
+    $values .= $mdb2->quote($fields['parent_id']);
+    $values .= ', '.$mdb2->quote($fields['org_id']);
+    $values .= ', '.$mdb2->quote(trim($fields['name']));
+    $values .= ', '.$mdb2->quote(trim($fields['currency']));
+    //$values .= ', '.$mdb2->quote($fields['decimal_mark']);
+    $values .= ', '.$mdb2->quote($fields['lang']);
+/*
+    $values .= ', '.$mdb2->quote($fields['date_format']);
+    $values .= ', '.$mdb2->quote($fields['time_format']);
+    $values .= ', '.(int)$fields['week_start'];
+    $values .= ', '.(int)$fields['tracking_mode'];
+    $values .= ', '.(int)$fields['project_required'];
+    $values .= ', '.(int)$fields['task_required'];
+    $values .= ', '.(int)$fields['record_type'];
+    $values .= ', '.$mdb2->quote($fields['bcc_email']);
+    $values .= ', '.$mdb2->quote($fields['allow_ip']);
+    $values .= ', '.$mdb2->quote($fields['password_complexity']);
+    $values .= ', '.$mdb2->quote($fields['plugins']);
+    $values .= ', '.$mdb2->quote($fields['lock_spec']);
+    $values .= ', '.(int)$fields['workday_minutes'];
+    $values .= ', '.$mdb2->quote($fields['config']);
+    $values .= ', now(), '.$mdb2->quote($_SERVER['REMOTE_ADDR']).', '.$mdb2->quote($user->id); */
+    $values .= ')';
+
+    $sql = 'insert into tt_groups '.$columns.$values;
+    $affected = $mdb2->exec($sql);
+    if (is_a($affected, 'PEAR_Error')) return false;
+
+    $group_id = $mdb2->lastInsertID('tt_groups', 'id');
+    return $group_id;
+  }
 }