Improved new export by adding clients in output.
[timetracker.git] / WEB-INF / lib / ttGroupExportHelper.class.php
index af49e19..054c435 100644 (file)
@@ -40,9 +40,11 @@ class ttGroupExportHelper {
 
   // The following arrays are maps between entity ids in the file versus the database.
   // We write to the file sequentially (1,2,3...) while in the database the entities have different ids.
-  var $userMap   = array(); // User ids.
-  var $roleMap   = array(); // Role ids.
-  var $clientMap = array(); // Client ids.
+  var $userMap    = array(); // User ids.
+  var $roleMap    = array(); // Role ids.
+  var $taskMap    = array(); // Task ids.
+  var $projectMap = array(); // Project ids.
+  var $clientMap  = array(); // Client ids.
 
   // Constructor.
   function __construct($group_id, $file, $indentation) {
@@ -114,6 +116,60 @@ class ttGroupExportHelper {
     return false;
   }
 
+  // getTasks - obtains all tasks defined for group.
+  function getTasks() {
+    global $user;
+    $mdb2 = getConnection();
+
+    $result = array();
+    $sql = "select * from tt_tasks where group_id = $this->group_id and org_id = $user->org_id";
+    $res = $mdb2->query($sql);
+    $result = array();
+    if (!is_a($res, 'PEAR_Error')) {
+      while ($val = $res->fetchRow()) {
+        $result[] = $val;
+      }
+      return $result;
+    }
+    return false;
+  }
+
+  // getProjects - obtains all projects defined for group.
+  function getProjects() {
+    global $user;
+    $mdb2 = getConnection();
+
+    $result = array();
+    $sql = "select * from tt_projects where group_id = $this->group_id and org_id = $user->org_id";
+    $res = $mdb2->query($sql);
+    $result = array();
+    if (!is_a($res, 'PEAR_Error')) {
+      while ($val = $res->fetchRow()) {
+        $result[] = $val;
+      }
+      return $result;
+    }
+    return false;
+  }
+
+  // getClients - obtains all clients defined for group.
+  function getClients() {
+    global $user;
+    $mdb2 = getConnection();
+
+    $result = array();
+    $sql = "select * from tt_clients where group_id = $this->group_id and org_id = $user->org_id";
+    $res = $mdb2->query($sql);
+    $result = array();
+    if (!is_a($res, 'PEAR_Error')) {
+      while ($val = $res->fetchRow()) {
+        $result[] = $val;
+      }
+      return $result;
+    }
+    return false;
+  }
+
   // writeData writes group data into file.
   function writeData() {
 
@@ -138,15 +194,25 @@ class ttGroupExportHelper {
     foreach ($roles as $key=>$role_item)
       $this->roleMap[$role_item['id']] = $key + 1;
 
+    // Prepare task map.
+    $tasks = $this->getTasks();
+    foreach ($tasks as $key=>$task_item)
+      $this->taskMap[$task_item['id']] = $key + 1;
+
+    // Prepare project map.
+    $projects = $this->getProjects();
+    foreach ($projects as $key=>$project_item)
+      $this->projectMap[$project_item['id']] = $key + 1;
+
     // Prepare client map.
-    $clients = ttTeamHelper::getAllClients($this->group_id, true);
+    $clients = $this->getClients();
     foreach ($clients as $key=>$client_item)
       $this->clientMap[$client_item['id']] = $key + 1;
 
     // Write roles.
-    fwrite($this->file, $this->indentation."<roles>\n");
+    fwrite($this->file, $this->indentation."  <roles>\n");
     foreach ($roles as $role) {
-      $role_part = $this->indentation.'  '."<role id=\"".$this->roleMap[$role['id']]."\"";
+      $role_part = $this->indentation.'    '."<role id=\"".$this->roleMap[$role['id']]."\"";
       $role_part .= " name=\"".htmlentities($role['name'])."\"";
       $role_part .= " description=\"".htmlentities($role['description'])."\"";
       $role_part .= " rank=\"".$role['rank']."\"";
@@ -155,13 +221,66 @@ class ttGroupExportHelper {
       $role_part .= "></role>\n";
       fwrite($this->file, $role_part);
     }
-    fwrite($this->file, $this->indentation."</roles>\n");
+    fwrite($this->file, $this->indentation."  </roles>\n");
+
+    // Write tasks.
+    fwrite($this->file, $this->indentation."  <tasks>\n");
+    foreach ($tasks as $task) {
+      $task_part = $this->indentation.'    '."<task id=\"".$this->taskMap[$task['id']]."\"";
+      $task_part .= " name=\"".htmlentities($task['name'])."\"";
+      $task_part .= " description=\"".htmlentities($task['description'])."\"";
+      $task_part .= " status=\"".$task['status']."\"";
+      $task_part .= "></task>\n";
+      fwrite($this->file, $task_part);
+    }
+    fwrite($this->file, $this->indentation."  </tasks>\n");
+
+    // Write projects.
+    fwrite($this->file, $this->indentation."  <projects>\n");
+    foreach ($projects as $project_item) {
+      if($project_item['tasks']){
+        $tasks = explode(',', $project_item['tasks']);
+        $tasks_mapped = array();
+        foreach ($tasks as $item)
+          $tasks_mapped[] = $this->taskMap[$item];
+        $tasks_str = implode(',', $tasks_mapped);
+      }
+      $project_part = $this->indentation.'    '."<project id=\"".$this->projectMap[$project_item['id']]."\"";
+      $project_part .= " name=\"".htmlentities($project_item['name'])."\"";
+      $project_part .= " description=\"".htmlentities($project_item['description'])."\"";
+      $project_part .= " tasks=\"".$tasks_str."\"";
+      $project_part .= " status=\"".$project_item['status']."\"";
+      $project_part .= "></project>\n";
+      fwrite($this->file, $project_part);
+    }
+    fwrite($this->file, $this->indentation."  </projects>\n");
+
+    // Write clients.
+    fwrite($this->file, $this->indentation."  <clients>\n");
+    foreach ($clients as $client_item) {
+      if($client_item['projects']){
+        $projects_db = explode(',', $client_item['projects']);
+        $projects_mapped = array();
+        foreach ($projects_db as $item)
+          $projects_mapped[] = $this->projectMap[$item];
+        $projects_str = implode(',', $projects_mapped);
+      }
+      $client_part = $this->indentation.'    '."<client id=\"".$this->clientMap[$client_item['id']]."\"";
+      $client_part .= " name=\"".htmlentities($client_item['name'])."\"";
+      $client_part .= " address=\"".htmlentities($client_item['address'])."\"";
+      $client_part .= " tax=\"".$client_item['tax']."\"";
+      $client_part .= " projects=\"".$projects_str."\"";
+      $client_part .= " status=\"".$client_item['status']."\"";
+      $client_part .= "></client>\n";
+      fwrite($this->file, $client_part);
+    }
+    fwrite($this->file, $this->indentation."  </clients>\n");
 
     // Write users.
-    fwrite($this->file, $this->indentation."<users>\n");
+    fwrite($this->file, $this->indentation."  <users>\n");
     foreach ($users as $user_item) {
       $role_id = $user_item['rank'] == 512 ? 0 : $this->roleMap[$user_item['role_id']]; // Special role_id 0 (not null) for top manager.
-      $user_part = $this->indentation.'  '."<user id=\"".$this->userMap[$user_item['id']]."\"";
+      $user_part = $this->indentation.'    '."<user id=\"".$this->userMap[$user_item['id']]."\"";
       $user_part .= " name=\"".htmlentities($user_item['name'])."\"";
       $user_part .= " login=\"".htmlentities($user_item['login'])."\"";
       $user_part .= " password=\"".$user_item['password']."\"";
@@ -173,7 +292,7 @@ class ttGroupExportHelper {
       $user_part .= "></user>\n";
       fwrite($this->file, $user_part);
     }
-    fwrite($this->file, $this->indentation."</users>\n");
+    fwrite($this->file, $this->indentation."  </users>\n");
 
     // Call self recursively for all subgroups.
     foreach ($this->subgroups as $subgroup) {