Merge pull request #2084 from nextcloud/backport/2074/stable30

[stable30] fix(push): Make testing the push server easier
nextcloud · Oct 22, 2024 · 327cd58 · 327cd58
2 parents 45515cd + 7fb1011
commit 327cd58
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 6 deletions.
diff --git a/lib/Push.php b/lib/Push.php
@@ -30,6 +30,7 @@
 use OCP\Notification\IManager as INotificationManager;
 use OCP\Notification\IncompleteParsedNotificationException;
 use OCP\Notification\INotification;
+use OCP\Security\ISecureRandom;
 use OCP\UserStatus\IManager as IUserStatusManager;
 use OCP\UserStatus\IUserStatus;
 use OCP\Util;
@@ -116,6 +117,7 @@ public function __construct(
 		IUserStatusManager $userStatusManager,
 		IFactory $l10nFactory,
 		protected ITimeFactory $timeFactory,
+		protected ISecureRandom $random,
 		LoggerInterface $log,
 	) {
 		$this->db = $connection;
@@ -470,6 +472,17 @@ protected function sendNotificationsToProxies(): void {
 			return;
 		}
 
+		$subscriptionAwareServer = rtrim($this->config->getAppValue(Application::APP_ID, 'subscription_aware_server', 'https://push-notifications.nextcloud.com'), '/');
+		if ($subscriptionAwareServer === 'https://push-notifications.nextcloud.com') {
+			$subscriptionKey = $this->config->getAppValue('support', 'subscription_key');
+		} else {
+			$subscriptionKey = $this->config->getAppValue(Application::APP_ID, 'push_subscription_key');
+			if ($subscriptionKey === '') {
+				$subscriptionKey = $this->createPushSubscriptionKey();
+				$this->config->setAppValue(Application::APP_ID, 'push_subscription_key', $subscriptionKey);
+			}
+		}
+
 		$client = $this->clientService->newClient();
 		foreach ($pushNotifications as $proxyServer => $notifications) {
 			try {
@@ -479,11 +492,8 @@ protected function sendNotificationsToProxies(): void {
 					],
 				];
 
-				if ($proxyServer === 'https://push-notifications.nextcloud.com') {
-					$subscriptionKey = $this->config->getAppValue('support', 'subscription_key');
-					if ($subscriptionKey) {
-						$requestData['headers']['X-Nextcloud-Subscription-Key'] = $subscriptionKey;
-					}
+				if ($subscriptionKey !== '' && $proxyServer === $subscriptionAwareServer) {
+					$requestData['headers']['X-Nextcloud-Subscription-Key'] = $subscriptionKey;
 				}
 
 				$response = $client->post($proxyServer . '/notifications', $requestData);
@@ -777,4 +787,9 @@ protected function deletePushTokenByDeviceIdentifier(string $deviceIdentifier):
 	protected function createFakeUserObject(string $userId): IUser {
 		return new FakeUser($userId);
 	}
+
+	protected function createPushSubscriptionKey(): string {
+		$key = $this->random->generate(25, ISecureRandom::CHAR_ALPHANUMERIC);
+		return implode('-', str_split($key, 5));
+	}
 }
diff --git a/tests/Unit/PushTest.php b/tests/Unit/PushTest.php
@@ -26,6 +26,7 @@
 use OCP\L10N\IFactory;
 use OCP\Notification\IManager as INotificationManager;
 use OCP\Notification\INotification;
+use OCP\Security\ISecureRandom;
 use OCP\UserStatus\IManager as IUserStatusManager;
 use PHPUnit\Framework\MockObject\MockObject;
 use Psr\Http\Message\ResponseInterface;
@@ -61,6 +62,8 @@ class PushTest extends TestCase {
 	protected $l10nFactory;
 	/** @var ITimeFactory|MockObject */
 	protected $timeFactory;
+	/** @var ISecureRandom|MockObject */
+	protected $random;
 	/** @var LoggerInterface|MockObject */
 	protected $logger;
 
@@ -78,6 +81,7 @@ protected function setUp(): void {
 		$this->userStatusManager = $this->createMock(IUserStatusManager::class);
 		$this->l10nFactory = $this->createMock(IFactory::class);
 		$this->timeFactory = $this->createMock(ITimeFactory::class);
+		$this->random = $this->createMock(ISecureRandom::class);
 		$this->logger = $this->createMock(LoggerInterface::class);
 
 		$this->cacheFactory->method('createDistributed')
@@ -103,6 +107,7 @@ protected function getPush(array $methods = []) {
 					$this->userStatusManager,
 					$this->l10nFactory,
 					$this->timeFactory,
+					$this->random,
 					$this->logger,
 				])
 				->setMethods($methods)
@@ -119,7 +124,9 @@ protected function getPush(array $methods = []) {
 			$this->cacheFactory,
 			$this->userStatusManager,
 			$this->l10nFactory,
-			$this->logger
+			$this->timeFactory,
+			$this->random,
+			$this->logger,
 		);
 	}
 
@@ -497,6 +504,13 @@ public function testPushToDeviceSending($isDebug) {
 			->with('debug', false)
 			->willReturn($isDebug);
 
+		$this->config
+			->method('getAppValue')
+			->willReturnMap([
+				['notifications', 'subscription_aware_server', 'https://push-notifications.nextcloud.com', 'https://push-notifications.nextcloud.com'],
+				['support', 'subscription_key', '', ''],
+			]);
+
 		$this->l10nFactory
 			->method('getUserLanguage')
 			->with($user)
@@ -805,6 +819,13 @@ public function testPushToDeviceTalkNotification(array $deviceTypes, $isTalkNoti
 			->with('has_internet_connection', true)
 			->willReturn(true);
 
+		$this->config
+			->method('getAppValue')
+			->willReturnMap([
+				['notifications', 'subscription_aware_server', 'https://push-notifications.nextcloud.com', 'https://push-notifications.nextcloud.com'],
+				['support', 'subscription_key', '', ''],
+			]);
+
 		$this->notificationManager->method('isFairUseOfFreePushService')
 			->willReturn(true);