Check that provided class implements IStorage in StorageManager::setBackend

[friendica.git] / src / Core / StorageManager.php

diff --git a/src/Core/StorageManager.php b/src/Core/StorageManager.php
index 5c14fdbf3388646c954441b30d8aadce96ebe753..3bb1cc4511161960dcbc47aec33634eba7c8895f 100644 (file)
--- a/src/Core/StorageManager.php
+++ b/src/Core/StorageManager.php
@@ -2,8 +2,8 @@
 
 namespace Friendica\Core;
 
-use Friendica\Core\Config;
-
+use Friendica\Database\DBA;
+use Friendica\Model\Storage\IStorage;
 
 
 /**
@@ -23,7 +23,7 @@ class StorageManager
 
        private static function setup()
        {
-               if (count(self::$backends)==0) {
+               if (count(self::$backends) == 0) {
                        self::$backends = Config::get('storage', 'backends', self::$default_backends);
                }
        }
@@ -31,6 +31,7 @@ class StorageManager
        /**
         * @brief Return current storage backend class
         * @return string
+        * @throws \Friendica\Network\HTTPException\InternalServerErrorException
         */
        public static function getBackend()
        {
@@ -52,12 +53,19 @@ class StorageManager
        /**
         * @brief Set current storage backend class
         *
-        * @param string  $class  Backend class name
+        * @param string $class Backend class name
+        * @return bool
+        * @throws \Friendica\Network\HTTPException\InternalServerErrorException
         */
        public static function setBackend($class)
        {
-               /// @todo Check that $class implements IStorage
+               if (!in_array('Friendica\Model\Storage\IStorage', class_implements($class))) {
+                       return false;
+               }
+
                Config::set('storage', 'class', $class);
+
+               return true;
        }
 
        /**
@@ -72,12 +80,12 @@ class StorageManager
        }
 
 
-
        /**
         * @brief Register a storage backend class
         *
-        * @param string  $name   User readable backend name
-        * @param string  $class  Backend class name
+        * @param string $name  User readable backend name
+        * @param string $class Backend class name
+        * @throws \Friendica\Network\HTTPException\InternalServerErrorException
         */
        public static function register($name, $class)
        {
@@ -91,12 +99,82 @@ class StorageManager
        /**
         * @brief Unregister a storage backend class
         *
-        * @param string  $name   User readable backend name
+        * @param string $name User readable backend name
+        * @throws \Friendica\Network\HTTPException\InternalServerErrorException
         */
-       public static function unregister($class)
+       public static function unregister($name)
        {
                self::setup();
                unset(self::$backends[$name]);
                Config::set('storage', 'backends', self::$backends);
        }
-}
\ No newline at end of file
+
+
+       /**
+        * @brief Move resources to storage $dest
+        *
+        * Copy existing data to destination storage and delete from source.
+        * This method cannot move to legacy in-table `data` field.
+        *
+        * @param string     $dest    Destination storage class name
+        * @param array|null $tables  Tables to look in for resources. Optional, defaults to ['photo', 'attach']
+        *
+        * @throws \Exception
+        * @return int Number of moved resources
+        */
+       public static function move($dest, $tables = null)
+       {
+               if (is_null($dest) || empty($dest)) {
+                       throw new \Exception('Can\'t move to NULL storage backend');
+               }
+               
+               if (is_null($tables)) {
+                       $tables = ['photo', 'attach'];
+               }
+
+               $moved = 0;
+               foreach ($tables as $table) {
+                       // Get the rows where backend class is not the destination backend class
+                       $rr = DBA::select(
+                               $table, 
+                               ['id', 'data', 'backend-class', 'backend-ref'],
+                               ['`backend-class` IS NULL or `backend-class` != ?' , $dest ]
+                       );
+
+                       if (DBA::isResult($rr)) {
+                               while($r = DBA::fetch($rr)) {
+                                       $id = $r['id'];
+                                       $data = $r['data'];
+                                       /** @var IStorage $backendClass */
+                                       $backendClass = $r['backend-class'];
+                                       $backendRef = $r['backend-ref'];
+                                       if (!is_null($backendClass) && $backendClass !== '') {
+                                               Logger::log("get data from old backend " .  $backendClass . " : " . $backendRef);
+                                               $data = $backendClass::get($backendRef);
+                                       }
+                                       
+                                       Logger::log("save data to new backend " . $dest);
+                                       /** @var IStorage $dest */
+                                       $ref = $dest::put($data);
+                                       Logger::log("saved data as " . $ref);
+
+                                       if ($ref !== '') {
+                                               Logger::log("update row");
+                                               $ru = DBA::update($table, ['backend-class' => $dest, 'backend-ref' => $ref, 'data' => ''], ['id' => $id]);
+                                               
+                                               if ($ru) {
+                                                       if (!is_null($backendClass) && $backendClass !== '') {
+                                                               Logger::log("delete data from old backend " . $backendClass . " : " . $backendRef);
+                                                               $backendClass::delete($backendRef);
+                                                       }
+                                                       $moved++;
+                                               }
+                                       }
+                               }
+                       }
+               }
+
+               return $moved;
+       }
+}
+