1 files changed, 52 insertions, 0 deletions
diff --git a/apps/files_sharing/lib/Middleware/OCSShareAPIMiddleware.php b/apps/files_sharing/lib/Middleware/OCSShareAPIMiddleware.php
new file mode 100644
index 00000000000..04a6545c9a7
--- /dev/null
+++ b/apps/files_sharing/lib/Middleware/OCSShareAPIMiddleware.php
@@ -0,0 +1,52 @@
+<?php
+
+namespace OCA\Files_Sharing\Middleware;
+
+use OCA\Files_Sharing\API\Share20OCS;
+use OCP\AppFramework\Http\Response;
+use OCP\AppFramework\Middleware;
+use OCP\AppFramework\OCS\OCSNotFoundException;
+use OCP\IL10N;
+use OCP\Share\IManager;
+
+class OCSShareAPIMiddleware extends Middleware {
+	/** @var IManager */
+	private $shareManager;
+	/** @var IL10N */
+	private $l;
+
+	public function __construct(IManager $shareManager,
+								IL10N $l) {
+		$this->shareManager = $shareManager;
+		$this->l = $l;
+	}
+
+	/**
+	 * @param \OCP\AppFramework\Controller $controller
+	 * @param string $methodName
+	 *
+	 * @throws OCSNotFoundException
+	 */
+	public function beforeController($controller, $methodName) {
+		if ($controller instanceof Share20OCS) {
+			if (!$this->shareManager->shareApiEnabled()) {
+				throw new OCSNotFoundException($this->l->t('Share API is disabled'));
+			}
+		}
+	}
+
+	/**
+	 * @param \OCP\AppFramework\Controller $controller
+	 * @param string $methodName
+	 * @param Response $response
+	 * @return Response
+	 */
+	public function afterController($controller, $methodName, Response $response) {
+		if ($controller instanceof Share20OCS) {
+			/** @var Share20OCS $controller */
+			$controller->cleanup();
+		}
+
+		return $response;
+	}
+}