Detect PSR-4 prefixes

Author: dunglas

src/Command/GenerateTypesCommand.php

@@ -35,15 +35,32 @@ final class GenerateTypesCommand extends Command
 {
     private const DEFAULT_CONFIG_FILE = 'schema.yaml';
 
+    private $namespacePrefix;
+    private $defaultOutput;
+
     /**
      * {@inheritdoc}
      */
     protected function configure(): void
     {
+        if (file_exists('composer.json') && is_file('composer.json') && is_readable('composer.json')) {
+            $composer = json_decode(file_get_contents('composer.json'), true);
+            foreach ($composer['autoload']['psr-4'] ?? [] as $prefix => $output) {
+                if ('' === $prefix) {
+                    continue;
+                }
+
+                $this->namespacePrefix = $prefix;
+                $this->defaultOutput = $output;
+
+                break;
+            }
+        }
+
         $this
             ->setName('generate-types')
             ->setDescription('Generate types')
-            ->addArgument('output', InputArgument::REQUIRED, 'The output directory')
+            ->addArgument('output', $this->defaultOutput ? InputArgument::OPTIONAL : InputArgument::REQUIRED, 'The output directory', $this->defaultOutput)
             ->addArgument('config', InputArgument::OPTIONAL, 'The config file to use (default to "schema.yaml" in the current directory, will generate all types if no config file exists)');
     }
 
@@ -52,10 +69,18 @@ protected function configure(): void
      */
     protected function execute(InputInterface $input, OutputInterface $output): void
     {
+        $defaultOutput = $this->defaultOutput ? realpath($this->defaultOutput) : null;
         $outputDir = $input->getArgument('output');
+        $configArgument = $input->getArgument('config');
+
         if ($dir = realpath($input->getArgument('output'))) {
             if (!is_dir($dir)) {
-                throw new \InvalidArgumentException(sprintf('The file "%s" is not a directory.', $dir));
+                if (!$this->defaultOutput) {
+                    throw new \InvalidArgumentException(sprintf('The file "%s" is not a directory.', $dir));
+                }
+
+                $dir = $defaultOutput;
+                $configArgument = $outputDir;
             }
 
             if (!is_writable($dir)) {
@@ -69,7 +94,6 @@ protected function execute(InputInterface $input, OutputInterface $output): void
             $outputDir = realpath($outputDir);
         }
 
-        $configArgument = $input->getArgument('config');
         if ($configArgument) {
             if (!file_exists($configArgument)) {
                 throw new \InvalidArgumentException(sprintf('The file "%s" doesn\'t exist.', $configArgument));
@@ -95,7 +119,7 @@ protected function execute(InputInterface $input, OutputInterface $output): void
         }
 
         $processor = new Processor();
-        $configuration = new TypesGeneratorConfiguration();
+        $configuration = new TypesGeneratorConfiguration($dir === $defaultOutput ? $this->namespacePrefix : null);
         $processedConfiguration = $processor->processConfiguration($configuration, [$config]);
         $processedConfiguration['output'] = $outputDir;
         if (!$processedConfiguration['output']) {

src/TypesGeneratorConfiguration.php

@@ -39,11 +39,20 @@ final class TypesGeneratorConfiguration implements ConfigurationInterface
     public const GOOD_RELATIONS_OWL_URL = __DIR__.'/../data/v1.owl';
     public const SCHEMA_ORG_NAMESPACE = 'http://schema.org/';
 
+    private $defaultPrefix;
+
+    public function __construct(?string $defaultPrefix = null)
+    {
+        $this->defaultPrefix = $defaultPrefix;
+    }
+
     /**
      * {@inheritdoc}
      */
     public function getConfigTreeBuilder(): TreeBuilder
     {
+        $namespacePrefix = $this->defaultPrefix ?? 'AppBundle\\';
+
         $treeBuilder = new TreeBuilder();
         $rootNode = $treeBuilder->root('config');
         $rootNode
@@ -92,10 +101,10 @@ function ($rdfa) {
                     ->addDefaultsIfNotSet()
                     ->info('PHP namespaces')
                     ->children()
-                        ->scalarNode('prefix')->defaultNull()->info('The global namespace\'s prefix')->example('App\\')->end()
-                        ->scalarNode('entity')->defaultValue('AppBundle\Entity')->info('The namespace of the generated entities')->example('App\Entity')->end()
-                        ->scalarNode('enum')->defaultValue('AppBundle\Enum')->info('The namespace of the generated enumerations')->example('App\Enum')->end()
-                        ->scalarNode('interface')->defaultValue('AppBundle\Model')->info('The namespace of the generated interfaces')->example('App\Model')->end()
+                        ->scalarNode('prefix')->defaultValue($this->defaultPrefix)->info('The global namespace\'s prefix')->example('App\\')->end()
+                        ->scalarNode('entity')->defaultValue("{$namespacePrefix}Entity")->info('The namespace of the generated entities')->example('App\Entity')->end()
+                        ->scalarNode('enum')->defaultValue("{$namespacePrefix}Enum")->info('The namespace of the generated enumerations')->example('App\Enum')->end()
+                        ->scalarNode('interface')->defaultValue("{$namespacePrefix}Model")->info('The namespace of the generated interfaces')->example('App\Model')->end()
                     ->end()
                 ->end()
                 ->arrayNode('doctrine')

tests/Command/GenerateTypesCommandTest.php

@@ -323,4 +323,26 @@ public function testNamespacesPrefix()
 
         $this->assertContains('namespace App\Entity;', $person);
     }
+
+    public function testNamespacesPrefixAutodetect()
+    {
+        $outputDir = __DIR__.'/../../build/namespaces-prefix-autodetect/';
+
+        $this->fs->mkdir($outputDir);
+        $this->fs->copy(__DIR__.'/../config/namespaces-prefix-autodetect/composer.json', "$outputDir/composer.json");
+        $this->fs->copy(__DIR__.'/../config/namespaces-prefix-autodetect/schema.yaml', "$outputDir/schema.yaml");
+
+        $currentDir = getcwd();
+        chdir($outputDir);
+        try {
+            $commandTester = new CommandTester(new GenerateTypesCommand());
+            $this->assertEquals(0, $commandTester->execute([]));
+
+            $person = file_get_contents("$outputDir/src/Entity/Person.php");
+
+            $this->assertContains('namespace App\Entity;', $person);
+        } finally {
+            chdir($currentDir);
+        }
+    }
 }

tests/config/namespaces-prefix-autodetect/composer.json

@@ -0,0 +1,5 @@
+{
+  "autoload": {
+    "psr-4": { "App\\": "src/" }
+  }
+}

tests/config/namespaces-prefix-autodetect/schema.yaml

@@ -0,0 +1,4 @@
+types:
+  Person:
+    properties:
+      name: ~