123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141 |
- <?php
- 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 $request;
-
- private $ocsVersion;
-
- public function __construct(IRequest $request) {
- $this->request = $request;
- }
-
- public function beforeController($controller, $methodName) {
- 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);
- }
- }
-
- public function afterException($controller, $methodName, \Exception $exception) {
- if ($controller instanceof OCSController && $exception instanceof OCSException) {
- $code = $exception->getCode();
- if ($code === 0) {
- $code = \OCP\AppFramework\OCSController::RESPOND_UNKNOWN_ERROR;
- }
- return $this->buildNewResponse($controller, $code, $exception->getMessage());
- }
- throw $exception;
- }
-
- public function afterController($controller, $methodName, Response $response) {
-
- if ($controller instanceof OCSController && !($response instanceof BaseResponse)) {
- if ($response->getStatus() === Http::STATUS_UNAUTHORIZED) {
- $message = '';
- if ($response instanceof JSONResponse) {
-
- $message = $response->getData()['message'];
- }
- return $this->buildNewResponse($controller, OCSController::RESPOND_UNAUTHORISED, $message);
- }
- if ($response->getStatus() === Http::STATUS_FORBIDDEN) {
- $message = '';
- if ($response instanceof JSONResponse) {
-
- $message = $response->getData()['message'];
- }
- return $this->buildNewResponse($controller, Http::STATUS_FORBIDDEN, $message);
- }
- }
- return $response;
- }
-
- private function buildNewResponse(Controller $controller, $code, $message) {
- $format = $this->getFormat($controller);
- $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;
- }
-
- private function getFormat(Controller $controller) {
-
- $format = $this->request->getParam('format');
-
- if ($format === null) {
- $headers = $this->request->getHeader('Accept');
- $format = $controller->getResponderByHTTPHeader($headers, 'xml');
- }
- return $format;
- }
- }
|