add authenticator in security.yaml the simplest way

Author: nikophil

src/Maker/MakeAuthenticator.php

@@ -13,8 +13,12 @@
 
 use Symfony\Bundle\MakerBundle\ConsoleStyle;
 use Symfony\Bundle\MakerBundle\DependencyBuilder;
+use Symfony\Bundle\MakerBundle\FileManager;
 use Symfony\Bundle\MakerBundle\Generator;
 use Symfony\Bundle\MakerBundle\InputConfiguration;
+use Symfony\Bundle\MakerBundle\Security\SecurityConfigUpdater;
+use Symfony\Bundle\MakerBundle\Util\YamlManipulationFailedException;
+use Symfony\Bundle\MakerBundle\Util\YamlSourceManipulator;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Bundle\SecurityBundle\SecurityBundle;
@@ -25,6 +29,16 @@
  */
 final class MakeAuthenticator extends AbstractMaker
 {
+    private $fileManager;
+
+    private $configUpdater;
+
+    public function __construct(FileManager $fileManager, SecurityConfigUpdater $configUpdater)
+    {
+        $this->fileManager = $fileManager;
+        $this->configUpdater = $configUpdater;
+    }
+
     public static function getCommandName(): string
     {
         return 'make:auth';
@@ -51,6 +65,19 @@ public function generate(InputInterface $input, ConsoleStyle $io, Generator $gen
             'authenticator/Empty.tpl.php',
             []
         );
+
+        $path = 'config/packages/security.yaml';
+        if ($this->fileManager->fileExists($path)) {
+            try {
+                $newYaml = $this->configUpdater->updateForAuthenticator(
+                    $this->fileManager->getFileContents($path),
+                    $classNameDetails->getFullName()
+                );
+                $generator->dumpFile($path, $newYaml);
+            } catch (YamlManipulationFailedException $e) {
+            }
+        }
+
         $generator->writeChanges();
 
         $this->writeSuccessMessage($io);

src/Resources/config/makers.xml

@@ -8,6 +8,8 @@
             <defaults public="false" />
 
             <service id="maker.maker.make_authenticator" class="Symfony\Bundle\MakerBundle\Maker\MakeAuthenticator">
+                <argument type="service" id="maker.file_manager" />
+                <argument type="service" id="maker.security_config_updater" />
                 <tag name="maker.command" />
             </service>
 

src/Security/SecurityConfigUpdater.php

@@ -34,12 +34,7 @@ public function updateForUserClass(string $yamlSource, UserClassConfiguration $u
     {
         $this->manipulator = new YamlSourceManipulator($yamlSource);
 
-        // normalize the top level, just in case
-        if (!isset($this->manipulator->getData()['security'])) {
-            $newData = $this->manipulator->getData();
-            $newData['security'] = [];
-            $this->manipulator->setData($newData);
-        }
+        $this->normalizeSecurityYamlFile();
 
         $this->updateProviders($userConfig, $userClass);
 
@@ -53,6 +48,57 @@ public function updateForUserClass(string $yamlSource, UserClassConfiguration $u
         return $contents;
     }
 
+    public function updateForAuthenticator(string $yamlSource, string $authenticatorFQCN)
+    {
+        $this->manipulator = new YamlSourceManipulator($yamlSource);
+
+        $this->normalizeSecurityYamlFile();
+
+        $newData = $this->manipulator->getData();
+
+        if (!isset($newData['security']['firewalls'])) {
+            $newData['security']['firewalls'] = [];
+        }
+
+        $firewalls = array_filter(
+            $newData['security']['firewalls'],
+            function ($item) {
+                return !isset($item['security']) || true === $item['security'];
+            }
+        );
+
+        if (!$firewalls) {
+            $firewalls['main'] = ['anonymous' => true];
+        }
+
+        $firewall = $firewalls['main'];
+        if (!isset($firewall['guard'])) {
+            $firewall['guard'] = [];
+        }
+
+        if (!isset($firewall['guard']['authenticators'])) {
+            $firewall['guard']['authenticators'] = [];
+        }
+
+        $firewall['guard']['authenticators'][] = $authenticatorFQCN;
+
+        $newData['security']['firewalls']['main'] = $firewall;
+        $this->manipulator->setData($newData);
+
+        $contents = $this->manipulator->getContents();
+
+        return $contents;
+    }
+
+    private function normalizeSecurityYamlFile()
+    {
+        if (!isset($this->manipulator->getData()['security'])) {
+            $newData = $this->manipulator->getData();
+            $newData['security'] = [];
+            $this->manipulator->setData($newData);
+        }
+    }
+
     private function updateProviders(UserClassConfiguration $userConfig, string $userClass)
     {
         if ($this->isSingleInMemoryProviderConfigured()) {