Support CSRF protection token of Symfony's form component (implements #…

…2120)
nelmio · Sep 12, 2023 · b7e3636 · b7e3636
1 parent bd10c34
commit b7e3636
Show file tree

Hide file tree

Showing 2 changed files with 92 additions and 0 deletions.
diff --git a/ModelDescriber/FormModelDescriber.php b/ModelDescriber/FormModelDescriber.php
@@ -134,6 +134,20 @@ private function parseForm(OA\Schema $schema, FormInterface $form)
 
             $this->findFormType($config, $property);
         }
+
+        if ($form->getConfig()->getOption('csrf_protection', false)) {
+            $tokenFieldName = $form->getConfig()->getOption('csrf_field_name');
+
+            $property = Util::getProperty($schema, $tokenFieldName);
+            $property->type = 'string';
+            $property->description = 'CSRF token';
+
+            if (Generator::isDefault($schema->required)) {
+                $schema->required = [];
+            }
+
+            $schema->required[] = $tokenFieldName;
+        }
     }
 
     /**

diff --git a/Tests/ModelDescriber/FormModelDescriberTest.php b/Tests/ModelDescriber/FormModelDescriberTest.php
@@ -0,0 +1,78 @@
+<?php
+
+namespace Nelmio\ApiDocBundle\Tests\ModelDescriber;
+
+use Doctrine\Common\Annotations\Reader;
+use Nelmio\ApiDocBundle\Model\Model;
+use Nelmio\ApiDocBundle\Model\ModelRegistry;
+use Nelmio\ApiDocBundle\ModelDescriber\FormModelDescriber;
+use Nelmio\ApiDocBundle\Tests\ModelDescriber\Fixtures\CsrfEnablingFormType;
+use OpenApi\Annotations\Property;
+use OpenApi\Attributes\OpenApi;
+use OpenApi\Attributes\Schema;
+use OpenApi\Generator;
+use PHPUnit\Framework\TestCase;
+use Symfony\Component\Form\Extension\Core\Type\FormType;
+use Symfony\Component\Form\FormConfigInterface;
+use Symfony\Component\Form\FormFactoryInterface;
+use Symfony\Component\Form\FormInterface;
+use Symfony\Component\PropertyInfo\Type;
+
+class FormModelDescriberTest extends TestCase
+{
+    /**
+     * @dataProvider provideCsrfProtectionOptions
+     */
+    public function testDescribeCreatesTokenPropertyDependingOnOptions(bool $csrfProtectionEnabled, string $tokenName, bool $expectProperty): void
+    {
+        $formConfigMock = $this->createMock(FormConfigInterface::class);
+        $formConfigMock->expects($this->exactly($csrfProtectionEnabled ? 2 : 1))
+            ->method('getOption')
+            ->willReturnMap([
+                ['csrf_protection', false, $csrfProtectionEnabled],
+                ['csrf_field_name', null, $tokenName],
+            ]);
+
+        $formMock = $this->createMock(FormInterface::class);
+        $formMock->expects($this->exactly($csrfProtectionEnabled ? 2 : 1))
+            ->method('getConfig')
+            ->willReturn($formConfigMock);
+
+        $formFactoryMock = $this->createMock(FormFactoryInterface::class);
+        $formFactoryMock->expects($this->once())
+            ->method('create')
+            ->willReturn($formMock);
+
+        $annotationReader = $this->createMock(Reader::class);
+
+        $api = new OpenApi();
+        $model = new Model(new Type(Type::BUILTIN_TYPE_OBJECT, false, FormType::class));
+        $schema = new Schema();
+        $modelRegistry = new ModelRegistry([], $api);
+
+        $describer = new FormModelDescriber($formFactoryMock, $annotationReader, []);
+        $describer->setModelRegistry($modelRegistry);
+
+        $describer->describe($model, $schema);
+
+
+        if ($expectProperty) {
+            $filteredProperties = array_filter($schema->properties, function (Property $property) use ($tokenName) {
+                return $property->property === $tokenName;
+            });
+
+            $this->assertCount(1, $filteredProperties);
+        } else {
+            $this->assertSame(Generator::UNDEFINED, $schema->properties);
+        }
+    }
+
+    public function provideCsrfProtectionOptions(): array
+    {
+        return [
+            [true, '_token', true],
+            [true, '_another_token', true],
+            [false, '_token', false],
+        ];
+    }
+}