getAliases for Profile and Notice

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

diff --git a/classes/Managed_DataObject.php b/classes/Managed_DataObject.php
index c02685e55e30adbdbfe208c605770f94417e400e..cab7edd5c71750a014cd9463dc0110dd09ca33ab 100644 (file)
--- a/classes/Managed_DataObject.php
+++ b/classes/Managed_DataObject.php
@@ -409,8 +409,56 @@ abstract class Managed_DataObject extends Memcached_DataObject
             throw new Exception('Empty ID for object! (not inserted yet?).');
         }
 
-        // FIXME: How about forcing to return an int? Or will that overflow eventually?
-        return $this->id;
+        return intval($this->id);
+    }
+
+    /**
+     * WARNING: Only use this on Profile and Notice. We should probably do
+     * this with traits/"implements" or whatever, but that's over the top
+     * right now, I'm just throwing this in here to avoid code duplication
+     * in Profile and Notice classes.
+     */
+    public function getAliases()
+    {
+        $aliases = array();
+        $aliases[$this->getUri()] = $this->getID();
+
+        try {
+            $aliases[$this->getUrl()] = $this->getID();
+        } catch (InvalidUrlException $e) {
+            // getUrl failed because no valid URL could be returned, just ignore it
+        }
+
+        if (common_config('fix', 'fancyurls')) {
+            /**
+             * Here we add some hacky hotfixes for remote lookups that have been taught the
+             * (at least now) wrong URI but it's still obviously the same user. Such as:
+             * - https://site.example/user/1 even if the client requests https://site.example/index.php/user/1
+             * - https://site.example/user/1 even if the client requests https://site.example//index.php/user/1
+             * - https://site.example/index.php/user/1 even if the client requests https://site.example/user/1
+             * - https://site.example/index.php/user/1 even if the client requests https://site.example///index.php/user/1
+             */
+            foreach ($aliases as $alias=>$id) {
+                try {
+                    // get a "fancy url" version of the alias, even without index.php/
+                    $alt_url = common_fake_local_fancy_url($alias);
+                    // store this as well so remote sites can be sure we really are the same profile
+                    $aliases[$alt_url] = $id;
+                } catch (Exception $e) {
+                    // Apparently we couldn't rewrite that, the $alias was as the function wanted it to be
+                }
+
+                try {
+                    // get a non-"fancy url" version of the alias, i.e. add index.php/
+                    $alt_url = common_fake_local_nonfancy_url($alias);
+                    // store this as well so remote sites can be sure we really are the same profile
+                    $aliases[$alt_url] = $id;
+                } catch (Exception $e) {
+                    // Apparently we couldn't rewrite that, the $alias was as the function wanted it to be
+                }
+            }
+        }
+        return $aliases;
     }
 
     // 'update' won't write key columns, so we have to do it ourselves.
@@ -516,4 +564,26 @@ abstract class Managed_DataObject extends Memcached_DataObject
                                         $object->getID(),
                                         common_date_iso8601($created));
     }
+
+    protected function onInsert()
+    {
+        // NOOP by default
+    }
+
+    protected function onUpdate($dataObject=false)
+    {
+        // NOOP by default
+    }
+
+    public function insert()
+    {
+        $this->onInsert();
+        return parent::insert();
+    }
+
+    public function update($dataObject=false)
+    {
+        $this->onUpdate($dataObject);
+        return parent::update($dataObject);
+    }
 }