feat(config): allow to override some config values by hostname

Signed-off-by: Benjamin Gaussorgues <[email protected]>

Author: Altahrim

build/psalm-baseline.xml

@@ -3538,10 +3538,6 @@
       <code><![CDATA[$this->delete($key)]]></code>
       <code><![CDATA[$this->set($key, $value)]]></code>
     </InvalidOperand>
-    <UndefinedVariable>
-      <code><![CDATA[$CONFIG]]></code>
-      <code><![CDATA[$CONFIG]]></code>
-    </UndefinedVariable>
   </file>
   <file src="lib/private/Console/Application.php">
     <NoInterfaceProperties>

config/config.sample.php

@@ -45,6 +45,17 @@
 	 */
 	'instanceid' => '',
 
+	/**
+	 * This is a unique identifier for your server.
+	 * It is useful when your Nextcloud instance is spread between different servers.
+	 * Once it's set it shouldn't be changed.
+	 *
+	 * Value must be an integer, comprised between 0 and 1023.
+	 *
+	 * This value should be overriden by hostname in $CONFIG_HOSTNAME
+	 */
+	'serverid' => -1,
+
 	/**
 	 * The salt used to hash all passwords, auto-generated by the Nextcloud
 	 * installer. (There are also per-user salts.) If you lose this salt, you lose
@@ -2811,3 +2822,18 @@
 	 */
 	'enable_lazy_objects' => true,
 ];
+
+/**
+ * CONFIG_HOSTNAME allows to set specific configuration value on specific hosts.
+ * It should be used when configuration is stored on a shared filesystem between several servers.
+ *
+ * Only options listed in \OC\Config::HOST_OVERRIDE_CONFIG can be defined here.
+ */
+$CONFIG_HOSTNAME = [
+	'hostname_a' => [
+		'serverid' => 42,
+	],
+	'hostname_b' => [
+		'serverid' => 43,
+	],
+];

lib/private/Config.php

@@ -15,6 +15,10 @@
  */
 class Config {
 	public const ENV_PREFIX = 'NC_';
+	// List configurations that can be overriden based on server hostname
+	private const HOST_OVERRIDE_CONFIG = [
+		'serverid',
+	];
 
 	/** @var array Associative array ($key => $value) */
 	protected $cache = [];
@@ -199,7 +203,7 @@ private function readData() {
 
 		// Include file and merge config
 		foreach ($configFiles as $file) {
-			unset($CONFIG);
+			$CONFIG = $CONFIG_HOSTNAME = null;
 
 			// Invalidate opcache (only if the timestamp changed)
 			if (function_exists('opcache_invalidate')) {
@@ -226,6 +230,10 @@ private function readData() {
 			}
 
 			try {
+				/**
+				 * @var ?array $CONFIG
+				 * @var ?array $CONFIG_HOSTNAME
+				 */
 				include $file;
 			} finally {
 				// Close the file pointer and release the lock
@@ -241,9 +249,20 @@ private function readData() {
 				}
 				throw new \Exception($errorMessage);
 			}
-			if (isset($CONFIG) && is_array($CONFIG)) {
+			if (is_array($CONFIG)) {
 				$this->cache = array_merge($this->cache, $CONFIG);
 			}
+			if (is_array($CONFIG_HOSTNAME) && !empty($CONFIG_HOSTNAME)) {
+				$hostname = gethostname();
+				if (isset($CONFIG_HOSTNAME[$hostname]) && is_array($CONFIG_HOSTNAME[$hostname])) {
+					$filteredConfig = array_filter(
+						$CONFIG_HOSTNAME[$hostname],
+						fn ($key) => in_array($key, self::HOST_OVERRIDE_CONFIG),
+						ARRAY_FILTER_USE_KEY,
+					);
+					$this->cache = array_merge($this->cache, $filteredConfig);
+				}
+			}
 		}
 
 		// grab any "NC_" environment variables

lib/private/SnowflakeIdGenerator.php

@@ -11,6 +11,7 @@
 
 use OCP\AppFramework\Utility\ITimeFactory;
 use OCP\ICacheFactory;
+use OCP\IConfig;
 
 /**
  * Nextcloud Snowflake ID generator
@@ -22,6 +23,7 @@
 class SnowflakeIdGenerator {
 	public function __construct(
 		private readonly ITimeFactory $timeFactory,
+		private readonly IConfig $config,
 		private readonly ICacheFactory $cacheFactory,
 	) {
 	}
@@ -100,7 +102,10 @@ private function getCurrentTime(): array {
 	}
 
 	private function getServerId(): int {
-		return crc32(gethostname() ?: random_bytes(8));
+		$serverid = $this->config->getSystemValueInt('serverid', -1);
+		return $serverid > 0
+			? $serverid
+			: crc32(gethostname() ?: random_bytes(8));
 	}
 
 	private function isCli(): bool {

tests/lib/SnowflakeIdGeneratorTest.php

@@ -12,6 +12,7 @@
 use OC\SnowflakeIdGenerator;
 use OCP\AppFramework\Utility\ITimeFactory;
 use OCP\ICacheFactory;
+use OCP\IConfig;
 use PHPUnit\Framework\Attributes\DataProvider;
 use PHPUnit\Framework\MockObject\MockObject;
 
@@ -20,13 +21,18 @@
  */
 class SnowflakeIdGeneratorTest extends TestCase {
 	private ICacheFactory|MockObject $cacheFactory;
+	private IConfig|MockObject $config;
 
 	public function setUp(): void {
 		$this->cacheFactory = $this->createMock(ICacheFactory::class);
+		$this->config = $this->createMock(IConfig::class);
+		$this->config->method('getSystemValueInt')
+			->with('serverid')
+			->willReturn(42);
 	}
 
 	public function testGenerator(): void {
-		$generator = new SnowflakeIdGenerator(new TimeFactory(), $this->cacheFactory);
+		$generator = new SnowflakeIdGenerator(new TimeFactory(), $this->config, $this->cacheFactory);
 		$snowflakeId = ($generator)();
 		$this->assertGreaterThan(0x100000000, $snowflakeId);
 		if (PHP_INT_SIZE === 8) {
@@ -42,11 +48,13 @@ public function testGeneratorWithFixedTime(string $date, int $expectedSeconds, i
 		$timeFactory = $this->createMock(ITimeFactory::class);
 		$timeFactory->method('now')->willReturn($dt);
 
-		$generator = new SnowflakeIdGenerator($timeFactory, $this->cacheFactory);
+		$generator = new SnowflakeIdGenerator($timeFactory, $this->config, $this->cacheFactory);
 		$snowflakeId = new SnowflakeId($generator());
 
 		$this->assertEquals($expectedSeconds, $snowflakeId->seconds());
 		$this->assertEquals($expectedMilliseconds, $snowflakeId->milliseconds());
+		$this->assertTrue($snowflakeId->isCli());
+		$this->assertEquals(42, $snowflakeId->serverId());
 	}
 
 	public static function provideSnowflakeData(): array {