Propagate COMPOSER_MEMORY_LIMIT to script executor

Author: PabloKowalczyk

src/ScriptExecutor.php

@@ -127,6 +127,10 @@ private function expandPhpScript(string $cmd): string
             $arguments[] = '--php-ini='.$ini;
         }
 
+        if ($memoryLimit = (string) getenv('COMPOSER_MEMORY_LIMIT')) {
+            $arguments[] = "-d memory_limit={$memoryLimit}";
+        }
+
         $phpArgs = implode(' ', array_map([ProcessExecutor::class, 'escape'], $arguments));
 
         return ProcessExecutor::escape($php).($phpArgs ? ' '.$phpArgs : '').' '.$cmd;

tests/ScriptExecutorTest.php

@@ -0,0 +1,66 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Symfony\Flex\Tests;
+
+use Composer\Composer;
+use Composer\IO\NullIO;
+use Composer\Util\ProcessExecutor;
+use PHPUnit\Framework\TestCase;
+use Symfony\Component\Process\PhpExecutableFinder;
+use Symfony\Flex\Options;
+use Symfony\Flex\ScriptExecutor;
+
+final class ScriptExecutorTest extends TestCase
+{
+    /** @backupGlobals enabled */
+    public function testMemoryLimit(): void
+    {
+        // Arrange
+        $command = './command.php';
+        $memoryLimit = '32M';
+        putenv("COMPOSER_MEMORY_LIMIT={$memoryLimit}");
+        $executorMock = $this->createMock(ProcessExecutor::class);
+        $scriptExecutor = new ScriptExecutor(
+            new Composer(),
+            new NullIO(),
+            new Options(),
+            $executorMock,
+        );
+        $expectedCommand = $this->buildExpectedCommand($command, $memoryLimit);
+
+        // Expect
+        $executorMock
+            ->method('execute')
+            ->with($expectedCommand)
+        ;
+        $this->expectNotToPerformAssertions();
+
+        // Act
+        $scriptExecutor->execute(
+            'php-script',
+            $command
+        );
+    }
+
+    private function buildExpectedCommand(string $command, ?string $memoryLimit = null): string
+    {
+        $phpFinder = new PhpExecutableFinder();
+        if (!$php = $phpFinder->find(false)) {
+            throw new \RuntimeException('The PHP executable could not be found, add it to your PATH and try again.');
+        }
+
+        $arguments = $phpFinder->findArguments();
+        $ini = php_ini_loaded_file();
+        $arguments[] = "--php-ini={$ini}";
+
+        if (null !== $memoryLimit) {
+            $arguments[] = "-d memory_limit={$memoryLimit}";
+        }
+
+        $phpArgs = implode(' ', array_map([ProcessExecutor::class, 'escape'], $arguments));
+
+        return ProcessExecutor::escape($php).($phpArgs ? ' '.$phpArgs : '').' '.$command;
+    }
+}