From 470971cf4747f2eec548891769c340295a45cb22 Mon Sep 17 00:00:00 2001
From: Mikael Nordfeldth <mmn@hethane.se>
Date: Sun, 25 Jan 2015 13:13:01 +0100
Subject: [PATCH] File_thumbnail uses file_id as PRI

---
 classes/Managed_DataObject.php  | 21 +++++++++++++++------
 plugins/Oembed/OembedPlugin.php |  2 +-
 2 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/classes/Managed_DataObject.php b/classes/Managed_DataObject.php
index a221f96633..5ed0e525ea 100644
--- a/classes/Managed_DataObject.php
+++ b/classes/Managed_DataObject.php
@@ -323,7 +323,12 @@ abstract class Managed_DataObject extends Memcached_DataObject
 
     // 'update' won't write key columns, so we have to do it ourselves.
     // This also automatically calls "update" _before_ it sets the keys.
-    public function updateWithKeys(&$orig)
+    // 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.');
@@ -346,14 +351,17 @@ abstract class Managed_DataObject extends Memcached_DataObject
                 $this->query('ROLLBACK');
                 throw new ServerException("Could not UPDATE non-keys for {$this->__table}");
             }
+            $orig->decache();
+            $this->encache();
             return true;
         }
-        $toupdate = implode(', ', $parts);
 
-        $table = common_database_tablename($this->tableName());
-        $qry = 'UPDATE ' . $table . ' SET ' . $toupdate .
-          ' WHERE id = ' . $this->getID();
-        $orig->decache();
+        $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__);
@@ -371,6 +379,7 @@ abstract class Managed_DataObject extends Memcached_DataObject
             $this->query('ROLLBACK');
             throw new ServerException("Could not UPDATE non-keys for {$this->__table}");
         }
+        $orig->decache();
         $this->encache();
 
         // commit our db transaction
diff --git a/plugins/Oembed/OembedPlugin.php b/plugins/Oembed/OembedPlugin.php
index 29c6f08dd2..305dee614f 100644
--- a/plugins/Oembed/OembedPlugin.php
+++ b/plugins/Oembed/OembedPlugin.php
@@ -289,7 +289,7 @@ class OembedPlugin extends Plugin
         $thumbnail->width = $info[0];    // array indexes documented on php.net:
         $thumbnail->height = $info[1];   // https://php.net/manual/en/function.getimagesize.php
         // Throws exception on failure.
-        $thumbnail->updateWithKeys($orig);
+        $thumbnail->updateWithKeys($orig, 'file_id');
     }
 
     public function onPluginVersion(array &$versions)