fanategorius
102 lines
3.4 KiB
PHP
102 lines
3.4 KiB
PHP
<?php
|
|
|
|
namespace OCA\OCCWeb\Controller;
|
|
|
|
use OCP\AppFramework\Controller;
|
|
use OCP\IRequest;
|
|
use OCP\IDBConnection;
|
|
use OCP\AppFramework\Http\JSONResponse;
|
|
use OCP\IGroupManager;
|
|
use OCP\IUserSession;
|
|
|
|
class DbController extends Controller
|
|
{
|
|
private $db;
|
|
private $groupManager;
|
|
private $userSession;
|
|
|
|
public function __construct(
|
|
$AppName,
|
|
IRequest $request,
|
|
IDBConnection $db,
|
|
IGroupManager $groupManager,
|
|
IUserSession $userSession
|
|
) {
|
|
parent::__construct($AppName, $request);
|
|
$this->db = $db;
|
|
$this->groupManager = $groupManager;
|
|
$this->userSession = $userSession;
|
|
}
|
|
|
|
/**
|
|
* @NoCSRFRequired
|
|
*/
|
|
public function query()
|
|
{
|
|
// Проверка прав администратора
|
|
$user = $this->userSession->getUser();
|
|
if (!$user) {
|
|
return new JSONResponse(['error' => 'Not authenticated'], 401);
|
|
}
|
|
|
|
if (!$this->groupManager->isAdmin($user->getUID())) {
|
|
return new JSONResponse(['error' => 'Admin privileges required'], 403);
|
|
}
|
|
|
|
$sql = $this->request->getParam('sql', '');
|
|
|
|
if (empty(trim($sql))) {
|
|
return new JSONResponse(['success' => false, 'error' => 'Empty query']);
|
|
}
|
|
|
|
// Разделяем запросы по точке с запятой. Все запросы выполняются
|
|
// последовательно на одном и том же соединении с БД (в рамках
|
|
// одного HTTP-запроса), поэтому SET сохраняет своё значение
|
|
// для current_setting() в последующих запросах пачки.
|
|
$queries = array_values(array_filter(array_map('trim', explode(';', $sql)), function ($q) {
|
|
return $q !== '';
|
|
}));
|
|
|
|
$results = [];
|
|
|
|
foreach ($queries as $query) {
|
|
$isSelect = stripos($query, 'SELECT') === 0;
|
|
$isSet = stripos($query, 'SET ') === 0;
|
|
|
|
try {
|
|
$stmt = $this->db->prepare($query);
|
|
$stmt->execute();
|
|
|
|
if ($isSelect) {
|
|
$rows = $stmt->fetchAll();
|
|
$results[] = [
|
|
'query' => $query,
|
|
'type' => 'select',
|
|
'count' => count($rows),
|
|
'data' => $rows
|
|
];
|
|
} else {
|
|
$affected = $stmt->rowCount();
|
|
$results[] = [
|
|
'query' => $query,
|
|
'type' => $isSet ? 'set' : 'write',
|
|
'affected_rows' => $affected
|
|
];
|
|
}
|
|
} catch (\Exception $e) {
|
|
$results[] = [
|
|
'query' => $query,
|
|
'type' => 'error',
|
|
'error' => $e->getMessage()
|
|
];
|
|
// Останавливаемся на первой ошибке: не продолжаем выполнять
|
|
// оставшиеся запросы пачки (например, серию DELETE),
|
|
// если один из предыдущих шагов не выполнился.
|
|
break;
|
|
}
|
|
}
|
|
|
|
return new JSONResponse(['success' => true, 'results' => $results]);
|
|
}
|
|
}
|