Add interface for specifying brancher labels and settings

Author: tdgroot

src/BrancherServer.php

@@ -0,0 +1,38 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Hypernode\DeployConfiguration;
+
+class BrancherServer extends Server
+{
+    public function getLabels(): array
+    {
+        return $this->getOptions()[self::OPTION_HN_BRANCHER_LABELS] ?? [];
+    }
+
+    /**
+     * @param string[] $labels Labels to be applied to the brancher node
+     * @return $this
+     */
+    public function setLabels(array $labels): self
+    {
+        $this->setOption(self::OPTION_HN_BRANCHER_LABELS, $labels);
+        return $this;
+    }
+
+    public function getSettings(): array
+    {
+        return $this->getOptions()[self::OPTION_HN_BRANCHER_SETTINGS] ?? [];
+    }
+
+    /**
+     * @param array $settings Settings to be applied to the brancher node
+     * @return $this
+     */
+    public function setSettings(array $settings): self
+    {
+        $this->setOption(self::OPTION_HN_BRANCHER_SETTINGS, $settings);
+        return $this;
+    }
+}

src/Server.php

@@ -8,6 +8,8 @@
 class Server
 {
     public const OPTION_HN_BRANCHER = 'hn_brancher';
+    public const OPTION_HN_BRANCHER_LABELS = 'hn_brancher_labels';
+    public const OPTION_HN_BRANCHER_SETTINGS = 'hn_brancher_settings';
     public const OPTION_HN_PARENT_APP = 'hn_parent_app';
 
     /**
@@ -62,9 +64,6 @@ public function getRoles(): array
         return $this->roles;
     }
 
-    /**
-     * @return string[]
-     */
     public function getOptions(): array
     {
         return $this->options;
@@ -78,6 +77,16 @@ public function getSshOptions(): array
         return $this->sshOptions;
     }
 
+    /**
+     * @param string $option
+     * @param mixed $value
+     * @return void
+     */
+    protected function setOption(string $option, $value)
+    {
+        $this->options[$option] = $value;
+    }
+
     /**
      * @param $options
      */

src/Stage.php

@@ -75,16 +75,18 @@ public function addServer(
      * @param array|null $roles Roles for the server to be applied
      * @param array $options Extra host options for Deployer
      * @see ServerRole
-     * @return Server
+     * @return BrancherServer
      */
-    public function addBrancherServer(string $appName, array $roles = null, array $options = []): Server
+    public function addBrancherServer(string $appName, array $roles = null, array $options = []): BrancherServer
     {
         $brancherOptions = [
             Server::OPTION_HN_BRANCHER => true,
             Server::OPTION_HN_PARENT_APP => $appName,
+            Server::OPTION_HN_BRANCHER_LABELS => [],
+            Server::OPTION_HN_BRANCHER_SETTINGS => [],
         ];
         $options = array_merge($brancherOptions, $options);
-        $server = new Server('', $roles, $options);
+        $server = new BrancherServer('', $roles, $options);
         $this->servers[] = $server;
         return $server;
     }

templates/deploy.magento2.php

@@ -14,6 +14,8 @@
 $productionStage->addServer('appname.hypernode.io');
 
 $testStage = $configuration->addStage('test', 'example.com');
-$testStage->addBrancherServer('appname');
+$testStage->addBrancherServer('appname')
+    ->setLabels(['stage=test', 'ci_ref=' . \getenv('GITHUB_RUN_ID') ?: 'none'])
+    ->setSettings(['cron_enabled' => false, 'supervisor_enabled' => false]);
 
 return $configuration;