From 3927acdde547a2d908f91358360bd99c20c94a04 Mon Sep 17 00:00:00 2001
From: Mikael Nordfeldth <mmn@hethane.se>
Date: Mon, 12 Jan 2015 19:22:10 +0100
Subject: [PATCH] getUrl() fixed for File and File_thumbnail to correct
 http/https stuff and other

---
 classes/File.php           | 31 +++++++++++++++++++++++++++++++
 classes/File_thumbnail.php | 30 ++++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+)

diff --git a/classes/File.php b/classes/File.php
index 1edc89d5d7..3bbdcbe990 100644
--- a/classes/File.php
+++ b/classes/File.php
@@ -453,11 +453,42 @@ class File extends Managed_DataObject
     {
         return self::path($this->filename);
     }
+
     public function getUrl()
     {
+        if (!empty($this->filename)) {
+            // A locally stored file, so let's generate a URL for our instance.
+            $url = self::url($this->filename);
+            if ($url != $this->url) {
+                // For indexing purposes, in case we do a lookup on the 'url' field.
+                // also we're fixing possible changes from http to https, or paths
+                $this->updateUrl($url);
+            }
+            return $url;
+        }
+
+        // No local filename available, return the URL we have stored
         return $this->url;
     }
 
+    public function updateUrl($url)
+    {
+        $file = File::getKV('url', $url);
+        if ($file instanceof File) {
+            throw new ServerException('URL already exists in DB');
+        }
+        $sql = 'UPDATE %1$s SET url=%2$s WHERE url=%3$s;';
+        $result = $this->query(sprintf($sql, $this->__table,
+                                             $this->_quote((string)$url),
+                                             $this->_quote((string)$this->url)));
+        if ($result === false) {
+            common_log_db_error($this, 'UPDATE', __FILE__);
+            throw new ServerException("Could not UPDATE {$this->__table}.url");
+        }
+
+        return $result;
+    }
+
     /**
      * Blow the cache of notices that link to this URL
      *
diff --git a/classes/File_thumbnail.php b/classes/File_thumbnail.php
index 10135f3d2f..4d76b7dc55 100644
--- a/classes/File_thumbnail.php
+++ b/classes/File_thumbnail.php
@@ -110,9 +110,39 @@ class File_thumbnail extends Managed_DataObject
 
     public function getUrl()
     {
+        if (!empty($this->filename)) {
+            // A locally stored file, so let's generate a URL for our instance.
+            $url = File::url($this->filename);
+            if ($url != $this->url) {
+                // For indexing purposes, in case we do a lookup on the 'url' field.
+                // also we're fixing possible changes from http to https, or paths
+                $this->updateUrl($url);
+            }
+            return $url;
+        }
+
+        // No local filename available, return the URL we have stored
         return $this->url;
     }
 
+    public function updateUrl($url)
+    {
+        $file = File_thumbnail::getKV('url', $url);
+        if ($file instanceof File_thumbnail) {
+            throw new ServerException('URL already exists in DB');
+        }
+        $sql = 'UPDATE %1$s SET url=%2$s WHERE url=%3$s;';
+        $result = $this->query(sprintf($sql, $this->__table,
+                                             $this->_quote((string)$url),
+                                             $this->_quote((string)$this->url)));
+        if ($result === false) {
+            common_log_db_error($this, 'UPDATE', __FILE__);
+            throw new ServerException("Could not UPDATE {$this->__table}.url");
+        }
+
+        return $result;
+    }
+
     public function delete($useWhere=false)
     {
         if (!empty($this->filename) && file_exists(File_thumbnail::path($this->filename))) {
-- 
2.39.2