X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs-website;a=blobdiff_plain;f=vendor%2Fdrush%2Fdrush%2Fsrc%2FRuntime%2FRedispatchHook.php;fp=vendor%2Fdrush%2Fdrush%2Fsrc%2FRuntime%2FRedispatchHook.php;h=84d7890bd2bf664fdba306650a4a6fd03ad668cb;hp=0000000000000000000000000000000000000000;hb=af6d1fb995500ae68849458ee10d66abbdcfb252;hpb=680c79a86e3ed402f263faeac92e89fb6d9edcc0

diff --git a/vendor/drush/drush/src/Runtime/RedispatchHook.php b/vendor/drush/drush/src/Runtime/RedispatchHook.php
new file mode 100644
index 000000000..84d7890bd
--- /dev/null
+++ b/vendor/drush/drush/src/Runtime/RedispatchHook.php
@@ -0,0 +1,162 @@
+<?php
+
+namespace Drush\Runtime;
+
+use Consolidation\AnnotatedCommand\Hooks\InitializeHookInterface;
+use Drush\Drush;
+use Robo\Contract\ConfigAwareInterface;
+use Symfony\Component\Console\Input\InputInterface;
+use Consolidation\AnnotatedCommand\AnnotationData;
+use Drush\Log\LogLevel;
+use Robo\Common\ConfigAwareTrait;
+
+/**
+ * The RedispatchHook is installed as an init hook that runs before
+ * all commands. If the commandline contains an alias or a site specification
+ * that points at a remote machine, then we will stop execution of the
+ * current command and instead run the command remotely.
+ */
+class RedispatchHook implements InitializeHookInterface, ConfigAwareInterface
+{
+    use ConfigAwareTrait;
+
+    /**
+     * Check to see if it is necessary to redispatch to a remote site.
+     * We do not redispatch to local sites here; usually, local sites may
+     * simply be selected and require no redispatch. When a local redispatch
+     * is needed, it happens in the RedispatchToSiteLocal class.
+     *
+     * @param InputInterface $input
+     * @param AnnotationData $annotationData
+     */
+    public function initialize(InputInterface $input, AnnotationData $annotationData)
+    {
+        // See drush_preflight_command_dispatch; also needed are:
+        //   - redispatch to a different site-local Drush on same system
+        //   - site-list handling (REMOVED)
+        // These redispatches need to be done regardless of the presence
+        // of a @handle-remote-commands annotation.
+
+        // If the command has the @handle-remote-commands annotation, then
+        // short-circuit redispatches to remote hosts.
+        if ($annotationData->has('handle-remote-commands')) {
+            return;
+        }
+        return $this->redispatchIfRemote($input);
+    }
+
+    /**
+     * Check to see if the target of the command is remote. Call redispatch
+     * if it is.
+     *
+     * @param InputInterface $input
+     */
+    public function redispatchIfRemote(InputInterface $input)
+    {
+        // Determine if this is a remote command.
+        // n.b. 'hasOption' only means that the option definition exists, so don't use that here.
+        $root = $input->getOption('remote-host');
+        if (!empty($root)) {
+            return $this->redispatch($input);
+        }
+    }
+
+    /**
+     * Called from RemoteCommandProxy::execute() to run remote commands.
+     *
+     * @param InputInterface $input
+     */
+    public function redispatch(InputInterface $input)
+    {
+        $remote_host = $input->getOption('remote-host');
+        $remote_user = $input->getOption('remote-user');
+
+        // Get the command arguments, and shift off the Drush command.
+        $redispatchArgs = Drush::config()->get('runtime.argv');
+        $drush_path = array_shift($redispatchArgs);
+        $command_name = array_shift($redispatchArgs);
+
+        Drush::logger()->debug('Redispatch hook {command}', ['command' => $command_name]);
+
+        // Remove argument patterns that should not be propagated
+        $redispatchArgs = $this->alterArgsForRedispatch($redispatchArgs);
+
+        // The options the user provided on the commandline will be included
+        // in $redispatchArgs.
+        $redispatchOptions = [];
+
+        // n.b. Defining the 'backend' flag here causes failed execution in the
+        // non-interactive case, even if 'backend' is set to 'false'.
+        $backend_options = [
+            'drush-script' => $this->getConfig()->get('paths.drush-script', null),
+            'remote-host' => $remote_host,
+            'remote-user' => $remote_user,
+            'additional-global-options' => [],
+            'interactive' => true,
+        ];
+        if ($input->isInteractive()) {
+            $backend_options['#tty'] = true;
+            $backend_options['interactive'] = true;
+        }
+
+        $invocations = [
+            [
+                'command' => $command_name,
+                'args' => $redispatchArgs,
+            ],
+        ];
+        $common_backend_options = [];
+        $default_command = null;
+        $default_site = [
+            'remote-host' => $remote_host,
+            'remote-user' => $remote_user,
+            'root' => $input->getOption('root'),
+            'uri' => $input->getOption('uri'),
+        ];
+        $context = null;
+
+        $values = drush_backend_invoke_concurrent(
+            $invocations,
+            $redispatchOptions,
+            $backend_options,
+            $default_command,
+            $default_site,
+            $context
+        );
+
+        return $this->exitEarly($values);
+    }
+
+    /**
+     * Remove anything that is not necessary for the remote side.
+     * At the moment this is limited to configuration options
+     * provided via -D.
+     *
+     * @param array $redispatchArgs
+     */
+    protected function alterArgsForRedispatch($redispatchArgs)
+    {
+        return array_filter($redispatchArgs, function ($item) {
+            return strpos($item, '-D') !== 0;
+        });
+    }
+
+    /**
+     * Abort the current execution without causing distress to our
+     * shutdown handler.
+     *
+     * @param array $values The results from backend invoke.
+     */
+    protected function exitEarly($values)
+    {
+        Drush::logger()->log(LogLevel::DEBUG, 'Redispatch hook exit early');
+
+        // TODO: This is how Drush exits from redispatch commands today;
+        // perhaps this could be somewhat improved, though.
+        // Note that RemoteCommandProxy::execute() is expecting that
+        // the redispatch() method will not return, so that will need
+        // to be altered if this behavior is changed.
+        drush_set_context('DRUSH_EXECUTION_COMPLETED', true);
+        exit($values['error_status']);
+    }
+}