src/Util/Writer/ViewDefinitionSqlWriter.php

   1 <?php
   2 /**
   3  * @copyright Copyright (C) 2010-2022, the Friendica project
   4  *
   5  * @license GNU AGPL version 3 or any later version
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as
   9  * published by the Free Software Foundation, either version 3 of the
  10  * License, or (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  19  *
  20  */
  21
  22 namespace Friendica\Util\Writer;
  23
  24 use Friendica\Database\Definition\ViewDefinition;
  25
  26 class ViewDefinitionSqlWriter
  27 {
  28         public static function create(ViewDefinition $definition): string
  29         {
  30                 $sqlString = '';
  31
  32                 foreach ($definition->getAll() as $viewName => $viewStructure) {
  33                         $sqlString .= "--\n";
  34                         $sqlString .= "-- VIEW $viewName\n";
  35                         $sqlString .= "--\n";
  36                         $sqlString .= static::dropView($viewName);
  37                         $sqlString .= static::createView($viewName, $viewStructure);
  38                 }
  39
  40                 return $sqlString;
  41         }
  42
  43         public static function dropView(string $viewName): string
  44         {
  45                 return sprintf("DROP VIEW IF EXISTS `%s`", static::escape($viewName)) . ";\n";
  46         }
  47
  48         public static function createView(string $viewName, array $viewStructure): string
  49         {
  50                 $sql_rows = [];
  51                 foreach ($viewStructure['fields'] as $fieldname => $origin) {
  52                         if (is_string($origin)) {
  53                                 $sql_rows[] = $origin . " AS `" . static::escape($fieldname) . "`";
  54                         } elseif (is_array($origin) && (sizeof($origin) == 2)) {
  55                                 $sql_rows[] = "`" . static::escape($origin[0]) . "`.`" . static::escape($origin[1]) . "` AS `" . static::escape($fieldname) . "`";
  56                         }
  57                 }
  58                 return sprintf("CREATE VIEW `%s` AS SELECT \n\t", static::escape($viewName)) .
  59                            implode(",\n\t", $sql_rows) . "\n\t" . $viewStructure['query'] . ";\n\n";
  60         }
  61
  62         public static function escape(string $sqlString): string
  63         {
  64                 return str_replace("'", "\\'", $sqlString);
  65         }
  66 }