2020-05-13 13:10:24 +01:00
|
|
|
<?php
|
|
|
|
|
2020-05-20 17:53:53 +01:00
|
|
|
// {{{ License
|
2020-09-10 23:25:47 +01:00
|
|
|
|
2020-05-20 17:53:53 +01:00
|
|
|
// This file is part of GNU social - https://www.gnu.org/software/social
|
|
|
|
//
|
|
|
|
// GNU social is free software: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
|
|
|
// GNU social is distributed in the hope that it will be useful,
|
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
// GNU Affero General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
|
|
// along with GNU social. If not, see <http://www.gnu.org/licenses/>.
|
2020-09-10 23:25:47 +01:00
|
|
|
|
2020-05-20 17:53:53 +01:00
|
|
|
// }}}
|
|
|
|
|
2020-05-13 13:10:24 +01:00
|
|
|
/**
|
|
|
|
* Doctrine entity manager static wrapper
|
|
|
|
*
|
|
|
|
* @package GNUsocial
|
|
|
|
* @category DB
|
|
|
|
*
|
|
|
|
* @author Hugo Sales <hugo@fc.up.pt>
|
|
|
|
* @copyright 2020 Free Software Foundation, Inc http://www.fsf.org
|
|
|
|
* @license https://www.gnu.org/licenses/agpl.html GNU AGPL v3 or later
|
|
|
|
*/
|
|
|
|
|
2020-06-04 22:00:05 +01:00
|
|
|
namespace App\Core\DB;
|
2020-05-13 13:10:24 +01:00
|
|
|
|
2020-09-10 23:25:47 +01:00
|
|
|
use App\Util\Exception\NotFoundException;
|
2020-07-21 22:32:21 +01:00
|
|
|
use App\Util\Formatting;
|
|
|
|
use Doctrine\Common\Collections\Criteria;
|
|
|
|
use Doctrine\Common\Collections\ExpressionBuilder;
|
2020-05-14 22:55:04 +01:00
|
|
|
use Doctrine\ORM\EntityManagerInterface;
|
2020-08-14 23:36:08 +01:00
|
|
|
use Doctrine\ORM\Query;
|
2020-09-04 19:45:28 +01:00
|
|
|
use Doctrine\ORM\Query\ResultSetMappingBuilder;
|
2020-05-13 13:10:24 +01:00
|
|
|
|
|
|
|
abstract class DB
|
|
|
|
{
|
2020-05-14 22:55:04 +01:00
|
|
|
private static ?EntityManagerInterface $em;
|
|
|
|
public static function setManager($m): void
|
2020-05-13 13:10:24 +01:00
|
|
|
{
|
|
|
|
self::$em = $m;
|
|
|
|
}
|
|
|
|
|
2020-08-28 21:16:26 +01:00
|
|
|
public static function dql(string $query, array $params = [])
|
2020-08-14 23:36:08 +01:00
|
|
|
{
|
|
|
|
$q = new Query(self::$em);
|
|
|
|
$q->setDQL($query);
|
|
|
|
foreach ($params as $k => $v) {
|
|
|
|
$q->setParameter($k, $v);
|
|
|
|
}
|
|
|
|
return $q->getResult();
|
|
|
|
}
|
|
|
|
|
2020-09-04 19:45:28 +01:00
|
|
|
public static function sql(string $query, array $entities, array $params = [])
|
|
|
|
{
|
|
|
|
$rsm = new ResultSetMappingBuilder(self::$em);
|
|
|
|
foreach ($entities as $alias => $entity) {
|
|
|
|
$rsm->addRootEntityFromClassMetadata($entity, $alias);
|
|
|
|
}
|
|
|
|
$query = preg_replace('/{select}/', $rsm->generateSelectClause(), $query);
|
|
|
|
$q = self::$em->createNativeQuery($query, $rsm);
|
|
|
|
foreach ($params as $k => $v) {
|
|
|
|
$q->setParameter($k, $v);
|
|
|
|
}
|
|
|
|
// dump($q);
|
|
|
|
// die();
|
|
|
|
return $q->getResult();
|
|
|
|
}
|
|
|
|
|
2020-07-21 22:32:21 +01:00
|
|
|
private static array $find_by_ops = ['or', 'and', 'eq', 'neq', 'lt', 'lte',
|
2020-07-25 02:55:39 +01:00
|
|
|
'gt', 'gte', 'is_null', 'in', 'not_in',
|
|
|
|
'contains', 'member_of', 'starts_with', 'ends_with', ];
|
2020-07-21 22:32:21 +01:00
|
|
|
|
|
|
|
private static function buildExpression(ExpressionBuilder $eb, array $criteria)
|
|
|
|
{
|
|
|
|
$expressions = [];
|
|
|
|
foreach ($criteria as $op => $exp) {
|
|
|
|
if ($op == 'or' || $op == 'and') {
|
|
|
|
$method = "{$op}X";
|
|
|
|
return $eb->{$method}(...self::buildExpression($eb, $exp));
|
|
|
|
} elseif ($op == 'is_null') {
|
|
|
|
$expressions[] = $eb->isNull($exp);
|
|
|
|
} else {
|
|
|
|
if (in_array($op, self::$find_by_ops)) {
|
|
|
|
$method = Formatting::snakeCaseToCamelCase($op);
|
|
|
|
$expressions[] = $eb->{$method}(...$exp);
|
|
|
|
} else {
|
|
|
|
$expressions[] = $eb->eq($op, $exp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $expressions;
|
|
|
|
}
|
|
|
|
|
2020-07-25 18:49:57 +01:00
|
|
|
public static function findBy(string $table, array $criteria, ?array $orderBy = null, ?int $limit = null, ?int $offset = null): array
|
2020-07-21 22:32:21 +01:00
|
|
|
{
|
|
|
|
$criteria = array_change_key_case($criteria);
|
|
|
|
$ops = array_intersect(array_keys($criteria), self::$find_by_ops);
|
2020-07-25 18:49:57 +01:00
|
|
|
$repo = self::getRepository($table);
|
2020-07-21 22:32:21 +01:00
|
|
|
if (empty($ops)) {
|
|
|
|
return $repo->findBy($criteria, $orderBy, $limit, $offset);
|
|
|
|
} else {
|
|
|
|
$criteria = new Criteria(self::buildExpression(Criteria::expr(), $criteria), $orderBy, $offset, $limit);
|
2020-07-25 18:49:57 +01:00
|
|
|
return $repo->matching($criteria)->toArray(); // Always work with array or it becomes really complicated
|
2020-07-21 22:32:21 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
public static function findOneBy(string $table, array $criteria, ?array $orderBy = null, ?int $offset = null)
|
|
|
|
{
|
2020-07-25 02:55:39 +01:00
|
|
|
$res = self::findBy($table, $criteria, $orderBy, 1, $offset);
|
2020-07-25 18:49:57 +01:00
|
|
|
if (count($res) == 1) {
|
|
|
|
return $res[0];
|
2020-07-25 02:55:39 +01:00
|
|
|
} else {
|
2020-09-10 23:25:47 +01:00
|
|
|
throw new NotFoundException("No value in table {$table} matches the requested criteria");
|
2020-07-25 02:55:39 +01:00
|
|
|
}
|
2020-07-21 22:32:21 +01:00
|
|
|
}
|
|
|
|
|
2020-05-14 22:55:04 +01:00
|
|
|
public static function __callStatic(string $name, array $args)
|
2020-05-13 13:10:24 +01:00
|
|
|
{
|
2020-07-18 03:16:18 +01:00
|
|
|
foreach (['find', 'getReference', 'getPartialReference', 'getRepository'] as $m) {
|
2020-09-04 15:15:03 +01:00
|
|
|
$pref = '\App\Entity\\';
|
|
|
|
if ($name == $m && Formatting::startsWith($name, $pref) === false) {
|
|
|
|
$args[0] = $pref . ucfirst(Formatting::snakeCaseToCamelCase($args[0]));
|
2020-07-18 03:16:18 +01:00
|
|
|
}
|
|
|
|
}
|
2020-08-14 23:36:08 +01:00
|
|
|
|
|
|
|
if (isset($args[0]) && is_string($args[0])) {
|
|
|
|
$args[0] = preg_replace('/Gsactor/', 'GSActor', $args[0] ?? '');
|
2020-08-13 02:23:22 +01:00
|
|
|
}
|
2020-07-18 03:16:18 +01:00
|
|
|
|
2020-05-14 22:55:04 +01:00
|
|
|
return self::$em->{$name}(...$args);
|
2020-05-13 13:10:24 +01:00
|
|
|
}
|
|
|
|
}
|