Add custom transport to allow wrapping with any command

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+### Unreleased
+
+* Adds a new option to site aliases which lets users specify any command to wrap invoked processes (#61).
+
 ### 5.0.0 - 2022/Feb/18
 
 * Support symfony/process ^6

src/Factory/CustomTransportFactory.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace Consolidation\SiteProcess\Factory;
+
+use Consolidation\SiteAlias\SiteAliasInterface;
+use Consolidation\SiteProcess\Transport\CustomTransport;
+use Consolidation\Config\ConfigInterface;
+
+/**
+ * CustomTransportFactory will create an CustomTransport for applicable site aliases.
+ */
+class CustomTransportFactory implements TransportFactoryInterface
+{
+    /**
+     * @inheritdoc
+     */
+    public function check(SiteAliasInterface $siteAlias)
+    {
+        return $siteAlias->has('command');
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function create(SiteAliasInterface $siteAlias)
+    {
+        return new CustomTransport($siteAlias);
+    }
+}

src/ProcessManager.php

@@ -6,6 +6,7 @@
 use Consolidation\SiteProcess\Factory\VagrantTransportFactory;
 use Psr\Log\LoggerInterface;
 use Consolidation\SiteAlias\SiteAliasInterface;
+use Consolidation\SiteProcess\Factory\CustomTransportFactory;
 use Consolidation\SiteProcess\Factory\SshTransportFactory;
 use Consolidation\SiteProcess\Factory\DockerComposeTransportFactory;
 use Consolidation\SiteProcess\Factory\TransportFactoryInterface;
@@ -72,6 +73,7 @@ public static function addTransports(ProcessManager $processManager)
         $processManager->add(new KubectlTransportFactory());
         $processManager->add(new DockerComposeTransportFactory());
         $processManager->add(new VagrantTransportFactory());
+        $processManager->add(new CustomTransportFactory());
 
         return $processManager;
     }

src/Transport/CustomTransport.php

@@ -0,0 +1,71 @@
+<?php
+
+namespace Consolidation\SiteProcess\Transport;
+
+use Consolidation\SiteProcess\SiteProcess;
+use Consolidation\SiteProcess\Util\Escape;
+use Consolidation\SiteAlias\SiteAliasInterface;
+use Consolidation\SiteProcess\Util\Shell;
+use Consolidation\Config\ConfigInterface;
+
+/**
+ * CustomTransport knows how to wrap a command such that it runs within
+ * any cli.
+ */
+class CustomTransport implements TransportInterface
+{
+    protected $tty;
+    protected $siteAlias;
+
+    public function __construct(SiteAliasInterface $siteAlias)
+    {
+        $this->siteAlias = $siteAlias;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function configure(SiteProcess $process)
+    {
+        $this->tty = $process->isTty();
+    }
+
+    /**
+     * inheritdoc
+     */
+    public function wrap($args)
+    {
+        $cmd = $this->siteAlias->get('custom.command', '');
+        $transport = $cmd ? [Shell::preEscaped($cmd)] : [];
+        $commandToExecute = $this->getCommandToExecute($args);
+
+        return array_filter(array_merge(
+            $transport,
+            $commandToExecute
+        ));
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function addChdir($cd_remote, $args)
+    {
+        // Make no assumptions about the CLI and what it can support.
+        // The CLI itself should handle this with the options specified
+        // in the custom command.
+        return [];
+    }
+
+    /**
+     * getCommandToExecute processes the arguments for the command to
+     * be executed such that they are appropriate for the transport mechanism.
+     */
+    protected function getCommandToExecute($args)
+    {
+        // Escape each argument for the target system and then join
+        $args = Escape::argsForSite($this->siteAlias, $args);
+        $commandToExecute = implode(' ', $args);
+
+        return [$commandToExecute];
+    }
+}

tests/Transport/CustomTransportTest.php

@@ -0,0 +1,54 @@
+<?php
+
+namespace Consolidation\SiteProcess;
+
+use Consolidation\SiteProcess\Transport\CustomTransport;
+use PHPUnit\Framework\TestCase;
+use Consolidation\SiteAlias\SiteAlias;
+
+class CustomTransportTest extends TestCase
+{
+    /**
+     * Data provider for testWrap.
+     */
+    public function wrapTestValues()
+    {
+        return [
+            [
+                'ls',
+                [
+                    'custom' => [
+                        'command' => '',
+                    ],
+                ],
+            ],
+            [
+                'platform ls',
+                [
+                    'custom' => [
+                        'command' => 'platform',
+                    ],
+                ],
+            ],
+            [
+                'platform -e dev ls',
+                [
+                    'custom' => [
+                        'command' => 'platform -e dev',
+                    ],
+                ],
+            ],
+        ];
+    }
+
+    /**
+     * @dataProvider wrapTestValues
+     */
+    public function testWrap($expected, $siteAliasData)
+    {
+        $siteAlias = new SiteAlias($siteAliasData, '@alias.dev');
+        $customTransport = new CustomTransport($siteAlias);
+        $actual = $customTransport->wrap(['ls']);
+        $this->assertEquals($expected, implode(' ', $actual));
+    }
+}