Only serve tagprofile HTML if we aren't POSTing via ajax

[quix0rs-gnu-social.git] / classes / Managed_DataObject.php

diff --git a/classes/Managed_DataObject.php b/classes/Managed_DataObject.php
index 531232332dd24e0889bf1f61250db7ac79e12976..b324984b7f8e7512d4c8b77606654ca6b10b8ae2 100644 (file)
--- a/classes/Managed_DataObject.php
+++ b/classes/Managed_DataObject.php
@@ -299,6 +299,11 @@ abstract class Managed_DataObject extends Memcached_DataObject
         return $ckeys;
     }
 
+    public function escapedTableName()
+    {
+        return common_database_tablename($this->tableName());
+    }
+
     /**
      * Returns an ID, checked that it is set and reasonably valid
      *
@@ -320,4 +325,80 @@ abstract class Managed_DataObject extends Memcached_DataObject
         // FIXME: How about forcing to return an int? Or will that overflow eventually?
         return $this->id;
     }
+
+    // 'update' won't write key columns, so we have to do it ourselves.
+    // This also automatically calls "update" _before_ it sets the keys.
+    // FIXME: This only works with single-column primary keys so far! Beware!
+    /**
+     * @param DB_DataObject &$orig  Must be "instanceof" $this
+     * @param string         $pid   Primary ID column (no escaping is done on column name!)
+     */
+    public function updateWithKeys(&$orig, $pid='id')
+    {
+        if (!$orig instanceof $this) {
+            throw new ServerException('Tried updating a DataObject with a different class than itself.');
+        }
+
+        // do it in a transaction
+        $this->query('BEGIN');
+
+        $parts = array();
+        foreach ($this->keys() as $k) {
+            if (strcmp($this->$k, $orig->$k) != 0) {
+                $parts[] = $k . ' = ' . $this->_quote($this->$k);
+            }
+        }
+        if (count($parts) == 0) {
+            // No changes to keys, it's safe to run ->update(...)
+            if ($this->update($orig) === false) {
+                common_log_db_error($this, 'UPDATE', __FILE__);
+                // rollback as something bad occurred
+                $this->query('ROLLBACK');
+                throw new ServerException("Could not UPDATE non-keys for {$this->__table}");
+            }
+            $orig->decache();
+            $this->encache();
+
+            // commit our db transaction since we won't reach the COMMIT below
+            $this->query('COMMIT');
+            // @FIXME return true only if something changed (otherwise 0)
+            return true;
+        }
+
+        $qry = sprintf('UPDATE %1$s SET %2$s WHERE %3$s = %4$s',
+                            common_database_tablename($this->tableName()),
+                            implode(', ', $parts),
+                            $pid,
+                            $this->_quote($this->$pid));
+
+        $result = $this->query($qry);
+        if ($result === false) {
+            common_log_db_error($this, 'UPDATE', __FILE__);
+            // rollback as something bad occurred
+            $this->query('ROLLBACK');
+            throw new ServerException("Could not UPDATE key fields for {$this->__table}");
+        }
+
+        // Update non-keys too, if the previous endeavour worked.
+        // The ->update call uses "$this" values for keys, that's why we can't do this until
+        // the keys are updated (because they might differ from $orig and update the wrong entries).
+        if ($this->update($orig) === false) {
+            common_log_db_error($this, 'UPDATE', __FILE__);
+            // rollback as something bad occurred
+            $this->query('ROLLBACK');
+            throw new ServerException("Could not UPDATE non-keys for {$this->__table}");
+        }
+        $orig->decache();
+        $this->encache();
+
+        // commit our db transaction
+        $this->query('COMMIT');
+        // @FIXME return true only if something changed (otherwise 0)
+        return $result;
+    }
+
+    static public function beforeSchemaUpdate()
+    {
+        // NOOP
+    }
 }