Add instance category while checking new updates

Signed-off-by: Benjamin Gaussorgues <benjamin.gaussorgues@nextcloud.com>

Author: Altahrim

apps/updatenotification/lib/Notification/BackgroundJob.php

@@ -40,44 +40,22 @@
 class BackgroundJob extends TimedJob {
 	protected $connectionNotifications = [3, 7, 14, 30];
 
-	/** @var IConfig */
-	protected $config;
-
-	/** @var IManager */
-	protected $notificationManager;
-
-	/** @var IGroupManager */
-	protected $groupManager;
-
-	/** @var IAppManager */
-	protected $appManager;
-
-	/** @var IClientService */
-	protected $client;
-
-	/** @var Installer */
-	protected $installer;
-
 	/** @var string[] */
 	protected $users;
 
-	public function __construct(ITimeFactory $timeFactory,
-								IConfig $config,
-								IManager $notificationManager,
-								IGroupManager $groupManager,
-								IAppManager $appManager,
-								IClientService $client,
-								Installer $installer) {
+	public function __construct(
+		ITimeFactory $timeFactory,
+		protected IConfig $config,
+		protected IManager $notificationManager,
+		protected IUserManager $userManager,
+		protected IGroupManager $groupManager,
+		protected IAppManager $appManager,
+		protected IClientService $client,
+		protected Installer $installer
+	) {
 		parent::__construct($timeFactory);
 		// Run once a day
 		$this->setInterval(60 * 60 * 24);
-
-		$this->config = $config;
-		$this->notificationManager = $notificationManager;
-		$this->groupManager = $groupManager;
-		$this->appManager = $appManager;
-		$this->client = $client;
-		$this->installer = $installer;
 	}
 
 	protected function run($argument) {
@@ -264,7 +242,8 @@ protected function deleteOutdatedNotifications($app, $version) {
 	protected function createVersionCheck(): VersionCheck {
 		return new VersionCheck(
 			$this->client,
-			$this->config
+			$this->config,
+			$this->userManager,
 		);
 	}
 

apps/updatenotification/tests/Notification/BackgroundJobTest.php

@@ -63,6 +63,7 @@ protected function setUp(): void {
 
 		$this->config = $this->createMock(IConfig::class);
 		$this->notificationManager = $this->createMock(IManager::class);
+		$this->userManager = $this->createMock(IUserManager::class);
 		$this->groupManager = $this->createMock(IGroupManager::class);
 		$this->appManager = $this->createMock(IAppManager::class);
 		$this->client = $this->createMock(IClientService::class);
@@ -80,6 +81,7 @@ protected function getJob(array $methods = []) {
 				$this->timeFactory,
 				$this->config,
 				$this->notificationManager,
+				$this->userManager,
 				$this->groupManager,
 				$this->appManager,
 				$this->client,

lib/private/Updater/VersionCheck.php

@@ -28,23 +28,15 @@
 
 use OCP\Http\Client\IClientService;
 use OCP\IConfig;
+use OCP\IUserManager;
 use OCP\Util;
 
 class VersionCheck {
-	/** @var IClientService */
-	private $clientService;
-
-	/** @var IConfig */
-	private $config;
-
-	/**
-	 * @param IClientService $clientService
-	 * @param IConfig $config
-	 */
-	public function __construct(IClientService $clientService,
-								IConfig $config) {
-		$this->clientService = $clientService;
-		$this->config = $config;
+	public function __construct(
+		private IClientService $clientService,
+		private IConfig $config,
+		private IUserManager $userManager,
+	) {
 	}
 
 
@@ -81,6 +73,8 @@ public function check() {
 		$version['php_major'] = PHP_MAJOR_VERSION;
 		$version['php_minor'] = PHP_MINOR_VERSION;
 		$version['php_release'] = PHP_RELEASE_VERSION;
+		$version['category'] = $this->computeCategory();
+		$version['isSubscriber'] = (int) $this->config->getAppValue('support', 'subscription_key', '0');
 		$versionString = implode('x', $version);
 
 		//fetch xml data from updater
@@ -130,4 +124,25 @@ protected function getUrlContent($url) {
 		$response = $client->get($url);
 		return $response->getBody();
 	}
+
+	private function computeCategory() {
+		$categoryBoundaries = [
+			100,
+			500,
+			1000,
+			5000,
+			10000,
+			100000,
+			1000000,
+		];
+
+		$nbUsers = array_sum($this->userManager->countUsers());
+		foreach ($categoryBoundaries as $categoryId => $boundary) {
+			if ($nbUsers <= $boundary) {
+				return $categoryId;
+			}
+		}
+
+		return $categoryId + 1;
+	}
 }