Merge pull request #81 from craftcms/cloud/commands-worker

Commands via SQS

Author: timkelty

composer.json

@@ -1,6 +1,5 @@
 {
   "minimum-stability": "dev",
-  "prefer-stable": true,
   "name": "craftcms/cloud",
   "type": "yii2-extension",
   "require": {

src/bref/craft/CraftCliEntrypoint.php

@@ -0,0 +1,37 @@
+<?php
+
+namespace craft\cloud\bref\craft;
+
+use Symfony\Component\Process\Process;
+
+final class CraftCliEntrypoint
+{
+    /**
+     * We leave a 10-second buffer for Lambda to start up and shutdown gracefully.
+     */
+    private const LAMBDA_EXECUTION_LIMIT = 890;
+
+    private function command(string $command, array $environment, int $timeout): array
+    {
+        $php = PHP_BINARY;
+
+        $shellCommand = escapeshellcmd("$php /var/task/craft $command");
+
+        $process = Process::fromShellCommandline($shellCommand, null, $environment, null, $timeout);
+
+        $process->run(function($type, $buffer): void {
+            echo $buffer;
+        });
+
+        return [
+            'exit_code' => $process->getExitCode(),
+            'output' => $process->getErrorOutput() . $process->getOutput(),
+            'duration' => microtime(true) - $process->getStartTime(),
+        ];
+    }
+
+    public function lambdaCommand(string $command, array $environment)
+    {
+        return $this->command($command, $environment, self::LAMBDA_EXECUTION_LIMIT);
+    }
+}

src/bref/handlers/CraftCommandHandler.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace craft\cloud\bref\handlers;
+
+use Bref\Context\Context;
+use Bref\Event\Handler;
+use craft\cloud\bref\craft\CraftCliEntrypoint;
+use InvalidArgumentException;
+
+/**
+ * @internal
+ */
+final class CraftCommandHandler implements Handler
+{
+    /**
+     * @inheritDoc
+     */
+    public function handle(mixed $event, Context $context): array
+    {
+        if (!isset($event['command'])) {
+            throw new InvalidArgumentException('No command found.');
+        }
+
+        $command = $event['command'];
+
+        $environment = ['LAMBDA_INVOCATION_CONTEXT' => json_encode($context, JSON_THROW_ON_ERROR)];
+
+        $entrypoint = new CraftCliEntrypoint();
+
+        return $entrypoint->lambdaCommand($command, $environment);
+    }
+}

src/bref/handlers/CraftCommandSqsHandler.php

@@ -0,0 +1,44 @@
+<?php
+
+namespace craft\cloud\bref\handlers;
+
+use Bref\Context\Context;
+use Bref\Event\Sqs\SqsEvent;
+use Bref\Event\Sqs\SqsHandler;
+use Craft;
+use craft\cloud\bref\craft\CraftCliEntrypoint;
+use RuntimeException;
+
+/**
+ * @internal
+ */
+final class CraftCommandSqsHandler extends SqsHandler
+{
+    public function handleSqs(SqsEvent $event, Context $context): void
+    {
+        $entrypoint = new CraftCliEntrypoint();
+
+        foreach ($event->getRecords() as $record) {
+            $record->getBody();
+
+            $body = json_decode($record->getBody(), flags: JSON_THROW_ON_ERROR);
+
+            $command = $body['command'] ?? throw new RuntimeException('Command not found');
+
+            $callback = $body['callback'] ?? throw new RuntimeException('Callback URL not found');
+
+            $environment = ['LAMBDA_INVOCATION_CONTEXT' => json_encode($context, JSON_THROW_ON_ERROR)];
+
+            $result = $entrypoint->lambdaCommand($command, $environment);
+
+            $this->sendResultBack($callback, $result);
+        }
+    }
+
+    private function sendResultBack(string $url, array $body): void
+    {
+        $client = Craft::createGuzzleClient();
+
+        $client->post($url, ['body' => json_encode($body)]);
+    }
+}