Work in progress on attachment delete.

[timetracker.git] / WEB-INF / lib / ttFileHelper.class.php

diff --git a/WEB-INF/lib/ttFileHelper.class.php b/WEB-INF/lib/ttFileHelper.class.php
index 68bb38a5707294078ac2ff91a092d64f7d95c944..80c9fd31461bccdd916a5601cb368b8b422faf66 100644 (file)
--- a/WEB-INF/lib/ttFileHelper.class.php
+++ b/WEB-INF/lib/ttFileHelper.class.php
@@ -85,8 +85,14 @@ class ttFileHelper {
       curl_close($ch);
 
       $result_array = json_decode($result, true);
       curl_close($ch);
 
       $result_array = json_decode($result, true);

-      if ($result_array && $result_array['id'] && $result_array['key']) {
-
+      if (!$result_array) {
+        $this->errors->add($i18n->get('error.file_storage'));
+      }
+      else if ($result_array['error']) {
+        // Add an error from file storage facility if we have it.
+        $this->errors->add($result_array['error']);
+      }
+      else if ($result_array['id'] && $result_array['key']) {

         $this->site_id = $result_array['id'];
         $this->site_key = $result_array['key'];
 
         $this->site_id = $result_array['id'];
         $this->site_key = $result_array['key'];
 


@@ -99,7 +105,7 @@ class ttFileHelper {
         $this->errors->add($i18n->get('error.file_storage'));
       }
     } else {
         $this->errors->add($i18n->get('error.file_storage'));
       }
     } else {

-      // Site id found, need to update site attributes.
+      // Site id found.

       $this->site_id = $val['id'];
 
       // Obtain site key.
       $this->site_id = $val['id'];
 
       // Obtain site key.


@@ -112,8 +118,6 @@ class ttFileHelper {
 
   // putFile - puts uploaded file in remote storage.
   function putFile($fields) {
 
   // putFile - puts uploaded file in remote storage.
   function putFile($fields) {

-    // if (!$this->site_id || !$this->site_key) return false;
-

     global $i18n;
     global $user;
     $mdb2 = getConnection();
     global $i18n;
     global $user;
     $mdb2 = getConnection();


@@ -121,22 +125,21 @@ class ttFileHelper {
     $group_id = $user->getGroup();
     $org_id = $user->org_id;
 
     $group_id = $user->getGroup();
     $org_id = $user->org_id;
 

-    $fields = array('site_id' => urlencode($this->site_id),
+    $curl_fields = array('site_id' => urlencode($this->site_id),

       'site_key' => urlencode($this->site_key),
       'org_id' => urlencode($org_id),
       'site_key' => urlencode($this->site_key),
       'org_id' => urlencode($org_id),

-      //'org_key' => urlencode($this->org_key),     // TODO: obtain this properly.
+      'org_key' => urlencode($this->getOrgKey()),

       'group_id' => urlencode($group_id),
       'group_id' => urlencode($group_id),

-      //'group_key' => urlencode($this->group_key), // TODO: obtain this properly.
-      //'user_id' => urlencode($this->user_id),     // TODO: obtain this properly.
-      //'user_key' => urlencode($this->user_key),   // TODO: obtain this properly.
+      'group_key' => urlencode($this->getGroupKey()),
+      'user_id' => urlencode($fields['user_id']),   // May be null.
+      'user_key' => urlencode($fields['user_key']), // May be null.

       'file_name' => urlencode($fields['file_name']),
       'description' => urlencode($fields['description']),
       'file_name' => urlencode($fields['file_name']),
       'description' => urlencode($fields['description']),

-      // TODO: add file content here, too. Will this work for large files?
-      //
+      'content' => urlencode(file_get_contents($_FILES['newfile']['tmp_name']))

     );
 
     // url-ify the data for the POST.
     );
 
     // url-ify the data for the POST.

-    foreach($fields as $key=>$value) { $fields_string .= $key.'='.$value.'&'; }
+    foreach($curl_fields as $key=>$value) { $fields_string .= $key.'='.$value.'&'; }

     $fields_string = rtrim($fields_string, '&');
 
     // Open connection.
     $fields_string = rtrim($fields_string, '&');
 
     // Open connection.


@@ -162,12 +165,12 @@ class ttFileHelper {
     $result_array = json_decode($result, true);
     $file_id = (int) $result_array['file_id'];
     $file_key = $result_array['file_key'];
     $result_array = json_decode($result, true);
     $file_id = (int) $result_array['file_id'];
     $file_key = $result_array['file_key'];

-    $file_error = $result_array['file_error'];
+    $error = $result_array['error'];

 
 

-    if (!$file_id || !$file_key) {
-      if ($file_error) {
+    if ($error || !$file_id || !$file_key) {
+      if ($error) {

         // Add an error from file storage facility if we have it.
         // Add an error from file storage facility if we have it.

-        $this->errors->add($file_error);
+        $this->errors->add($error);

       }
       return false;
     }
       }
       return false;
     }


@@ -187,4 +190,70 @@ class ttFileHelper {
     $affected = $mdb2->exec($sql);
     return (!is_a($affected, 'PEAR_Error'));
   }
     $affected = $mdb2->exec($sql);
     return (!is_a($affected, 'PEAR_Error'));
   }

+
+  // getOrgKey obtains organization key from the database.
+  private function getOrgKey() {
+    global $user;
+    $mdb2 = getConnection();
+
+    $org_id = $user->org_id;
+    $sql = "select group_key from tt_groups where id = $org_id and status = 1";
+    $res = $mdb2->query($sql);
+    $val = $res->fetchRow();
+    return $val['group_key'];
+  }
+
+  // getGrtoupKey obtains group key from the database.
+  private function getGroupKey() {
+    global $user;
+    $mdb2 = getConnection();
+
+    $group_id = $user->getGroup();
+    $org_id = $user->org_id;
+
+    $sql = "select group_key from tt_groups where id = $group_id and org_id = $org_id and status = 1";
+    $res = $mdb2->query($sql);
+    $val = $res->fetchRow();
+    return $val['group_key'];
+  }
+
+  // getProjectFiles obtains a list of files for a project.
+  function getProjectFiles($project_id) {
+    global $user;
+    $mdb2 = getConnection();
+
+    $group_id = $user->getGroup();
+    $org_id = $user->org_id;
+
+    $result = array();
+    $sql = "select id, remote_id, file_name as name, description from tt_files".
+      " where entity_type = 'project' and entity_id = $project_id".
+      " and group_id = $group_id and org_id = $org_id and status = 1 order by id";
+    $res = $mdb2->query($sql);
+    if (!is_a($res, 'PEAR_Error')) {
+      while ($val = $res->fetchRow()) {
+        $result[] = $val;
+      }
+    }
+    return $result;
+  }
+
+  // get - obtains file details from local database. 
+  static function get($id) {
+    global $user;
+    $mdb2 = getConnection();
+
+    $group_id = $user->getGroup();
+    $org_id = $user->org_id;
+
+    $sql = "select id, remote_id, file_key, entity_type, entity_id, file_name, description, status from tt_files".
+      " where id = $id and group_id = $group_id and org_id = $org_id and (status = 0 or status = 1)";
+    $res = $mdb2->query($sql);
+    if (!is_a($res, 'PEAR_Error')) {
+      $val = $res->fetchRow();
+      if ($val && $val['id'])
+        return $val;
+    }
+    return false;
+  }

 }
 }