mirror of
https://github.com/nextcloud/server.git
synced 2026-10-19 06:19:36 +02:00
104 lines
3.0 KiB
PHP
104 lines
3.0 KiB
PHP
<?php
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2016 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace OC\AppFramework\Middleware;
|
|
|
|
use OC\AppFramework\Http;
|
|
use OC\AppFramework\OCS\BaseResponse;
|
|
use OC\AppFramework\OCS\V1Response;
|
|
use OC\AppFramework\OCS\V2Response;
|
|
use OCP\AppFramework\Controller;
|
|
use OCP\AppFramework\Http\DataResponse;
|
|
use OCP\AppFramework\Http\JSONResponse;
|
|
use OCP\AppFramework\Http\Response;
|
|
use OCP\AppFramework\Middleware;
|
|
use OCP\AppFramework\OCS\OCSException;
|
|
use OCP\AppFramework\OCSController;
|
|
use OCP\IRequest;
|
|
|
|
class OCSMiddleware extends Middleware {
|
|
private int $ocsVersion = 2;
|
|
|
|
public function __construct(
|
|
private readonly IRequest $request,
|
|
) {
|
|
}
|
|
|
|
#[\Override]
|
|
public function beforeController(Controller $controller, string $methodName): void {
|
|
if ($controller instanceof OCSController) {
|
|
if (substr_compare($this->request->getScriptName(), '/ocs/v2.php', -strlen('/ocs/v2.php')) === 0) {
|
|
$this->ocsVersion = 2;
|
|
} else {
|
|
$this->ocsVersion = 1;
|
|
}
|
|
$controller->setOCSVersion($this->ocsVersion);
|
|
}
|
|
}
|
|
|
|
#[\Override]
|
|
public function afterException(Controller $controller, string $methodName, \Exception $exception): Response {
|
|
if ($controller instanceof OCSController && $exception instanceof OCSException) {
|
|
$code = $exception->getCode();
|
|
if ($code === 0) {
|
|
$code = OCSController::RESPOND_UNKNOWN_ERROR;
|
|
}
|
|
|
|
return $this->buildNewResponse($controller, $code, $exception->getMessage());
|
|
}
|
|
|
|
throw $exception;
|
|
}
|
|
|
|
#[\Override]
|
|
public function afterController(Controller $controller, string $methodName, Response $response): Response {
|
|
/*
|
|
* If a different middleware has detected that a request unauthorized or forbidden
|
|
* we need to catch the response and convert it to a proper OCS response.
|
|
*/
|
|
if ($controller instanceof OCSController && !($response instanceof BaseResponse)) {
|
|
if ($response->getStatus() === Http::STATUS_UNAUTHORIZED) {
|
|
$message = '';
|
|
if ($response instanceof JSONResponse) {
|
|
/** @var DataResponse $response */
|
|
$message = $response->getData()['message'];
|
|
}
|
|
|
|
return $this->buildNewResponse($controller, OCSController::RESPOND_UNAUTHORISED, $message);
|
|
}
|
|
if ($response->getStatus() === Http::STATUS_FORBIDDEN) {
|
|
$message = '';
|
|
if ($response instanceof JSONResponse) {
|
|
/** @var DataResponse $response */
|
|
$message = $response->getData()['message'];
|
|
}
|
|
|
|
return $this->buildNewResponse($controller, Http::STATUS_FORBIDDEN, $message);
|
|
}
|
|
}
|
|
|
|
return $response;
|
|
}
|
|
|
|
private function buildNewResponse(Controller $controller, int $code, string $message): V1Response|V2Response {
|
|
$format = $this->request->getFormat();
|
|
if ($format === null || !$controller->isResponderRegistered($format)) {
|
|
$format = 'xml';
|
|
}
|
|
|
|
$data = new DataResponse();
|
|
$data->setStatus($code);
|
|
if ($this->ocsVersion === 1) {
|
|
$response = new V1Response($data, $format, $message);
|
|
} else {
|
|
$response = new V2Response($data, $format, $message);
|
|
}
|
|
|
|
return $response;
|
|
}
|
|
}
|