1 files changed, 104 insertions, 0 deletions
diff --git a/lib/private/Profiler/Profiler.php b/lib/private/Profiler/Profiler.php
new file mode 100644
index 00000000000..84a4e3eff34
--- /dev/null
+++ b/lib/private/Profiler/Profiler.php
@@ -0,0 +1,104 @@
+<?php
+
+declare(strict_types = 1);
+
+/**
+ * SPDX-FileCopyrightText: 2021 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+namespace OC\Profiler;
+
+use OC\AppFramework\Http\Request;
+use OC\SystemConfig;
+use OCP\AppFramework\Http\Response;
+use OCP\DataCollector\IDataCollector;
+use OCP\Profiler\IProfile;
+use OCP\Profiler\IProfiler;
+
+class Profiler implements IProfiler {
+	/** @var array<string, IDataCollector> */
+	private array $dataCollectors = [];
+
+	private ?FileProfilerStorage $storage = null;
+
+	private bool $enabled = false;
+
+	public function __construct(SystemConfig $config) {
+		$this->enabled = $config->getValue('profiler', false);
+		if ($this->enabled) {
+			$this->storage = new FileProfilerStorage($config->getValue('datadirectory', \OC::$SERVERROOT . '/data') . '/__profiler');
+		}
+	}
+
+	public function add(IDataCollector $dataCollector): void {
+		$this->dataCollectors[$dataCollector->getName()] = $dataCollector;
+	}
+
+	public function loadProfileFromResponse(Response $response): ?IProfile {
+		if (!$token = $response->getHeaders()['X-Debug-Token']) {
+			return null;
+		}
+
+		return $this->loadProfile($token);
+	}
+
+	public function loadProfile(string $token): ?IProfile {
+		if ($this->storage) {
+			return $this->storage->read($token);
+		} else {
+			return null;
+		}
+	}
+
+	public function saveProfile(IProfile $profile): bool {
+		if ($this->storage) {
+			return $this->storage->write($profile);
+		} else {
+			return false;
+		}
+	}
+
+	public function collect(Request $request, Response $response): IProfile {
+		$profile = new Profile($request->getId());
+		$profile->setTime(time());
+		$profile->setUrl($request->getRequestUri());
+		$profile->setMethod($request->getMethod());
+		$profile->setStatusCode($response->getStatus());
+		foreach ($this->dataCollectors as $dataCollector) {
+			$dataCollector->collect($request, $response, null);
+
+			// We clone for subrequests
+			$profile->addCollector(clone $dataCollector);
+		}
+		return $profile;
+	}
+
+	/**
+	 * @return array[]
+	 */
+	public function find(?string $url, ?int $limit, ?string $method, ?int $start, ?int $end,
+		?string $statusCode = null): array {
+		if ($this->storage) {
+			return $this->storage->find($url, $limit, $method, $start, $end, $statusCode);
+		} else {
+			return [];
+		}
+	}
+
+	public function dataProviders(): array {
+		return array_keys($this->dataCollectors);
+	}
+
+	public function isEnabled(): bool {
+		return $this->enabled;
+	}
+
+	public function setEnabled(bool $enabled): void {
+		$this->enabled = $enabled;
+	}
+
+	public function clear(): void {
+		$this->storage->purge();
+	}
+}