Adding logged/right/failwith annotation to junction tables

Author: moufmouf

src/TdbmFluidJunctionTableGraphqlOptions.php

@@ -0,0 +1,65 @@
+<?php
+
+
+namespace TheCodingMachine\FluidSchema;
+
+
+use function addslashes;
+use function var_export;
+
+class TdbmFluidJunctionTableGraphqlOptions
+{
+    /**
+     * @var TdbmFluidJunctionTableOptions
+     */
+    private $tdbmFluidJunctionTableOptions;
+    /**
+     * @var TdbmFluidTable
+     */
+    private $tdbmFluidTable;
+
+    public function __construct(TdbmFluidJunctionTableOptions $tdbmFluidJunctionTableOptions, TdbmFluidTable $tdbmFluidTable)
+    {
+        $this->tdbmFluidJunctionTableOptions = $tdbmFluidJunctionTableOptions;
+        $this->tdbmFluidTable = $tdbmFluidTable;
+    }
+
+    public function logged(bool $mustBeLogged = true): self
+    {
+        if ($mustBeLogged) {
+            $this->addAnnotation('TheCodingMachine\\GraphQLite\\Annotations\\Logged');
+        } else {
+            $this->removeAnnotation('TheCodingMachine\\GraphQLite\\Annotations\\Logged');
+        }
+        return $this;
+    }
+
+    public function right(string $rightName): self
+    {
+        $this->addAnnotation('TheCodingMachine\\GraphQLite\\Annotations\\Right', '(name="'.addslashes($rightName).'")');
+        return $this;
+    }
+
+    public function failWith($value): self
+    {
+        $this->addAnnotation('TheCodingMachine\\GraphQLite\\Annotations\\FailWith', '('.var_export($value, true).')');
+        return $this;
+    }
+
+    public function addAnnotation(string $annotation, string $content = '', bool $replaceExisting = true): self
+    {
+        $this->tdbmFluidTable->addAnnotation($annotation, $content, $replaceExisting);
+        return $this;
+    }
+
+    public function removeAnnotation(string $annotation): self
+    {
+        $this->tdbmFluidTable->removeAnnotation($annotation);
+        return $this;
+    }
+
+    public function endGraphql(): TdbmFluidJunctionTableOptions
+    {
+        return $this->tdbmFluidJunctionTableOptions;
+    }
+}

src/TdbmFluidJunctionTableOptions.php

@@ -20,8 +20,9 @@ public function __construct(TdbmFluidTable $tdbmFluidTable)
         $this->tdbmFluidTable = $tdbmFluidTable;
     }
 
-    public function graphql(): void
+    public function graphql(): TdbmFluidJunctionTableGraphqlOptions
     {
         $this->tdbmFluidTable->addAnnotation('TheCodingMachine\\GraphQLite\\Annotations\\Field');
+        return new TdbmFluidJunctionTableGraphqlOptions($this, $this->tdbmFluidTable);
     }
 }

src/TdbmFluidTable.php

@@ -131,6 +131,13 @@ public function addAnnotation(string $annotation, string $content = '', bool $re
         return $this;
     }
 
+    public function removeAnnotation(string $annotation): self
+    {
+        $comment = $this->getComment()->removeAnnotation($annotation);
+        $this->saveComment($comment);
+        return $this;
+    }
+
     public function getDbalTable(): Table
     {
         return $this->fluidTable->getDbalTable();

tests/TdbmFluidJunctionTableGraphqlOptionsTest.php

@@ -0,0 +1,39 @@
+<?php
+
+namespace TheCodingMachine\FluidSchema;
+
+use Doctrine\DBAL\Schema\Schema;
+use Doctrine\DBAL\Types\Type;
+use PHPUnit\Framework\TestCase;
+
+class TdbmFluidJunctionTableGraphqlOptionsTest extends TestCase
+{
+    public function testGraphql()
+    {
+        $schema = new Schema();
+        $fluid = new TdbmFluidSchema($schema);
+
+        $fluid->table('posts')->uuid();
+        $fluid->table('users')->uuid();
+
+        $junctionTableOptions = $fluid->junctionTable('posts', 'users');
+        $graphqlOptions = $junctionTableOptions->graphql();
+
+        $graphqlOptions->logged(true)
+                       ->right('CAN_EDIT')
+                       ->failWith(null);
+
+        $this->assertSame("\n@TheCodingMachine\GraphQLite\Annotations\Field
+@TheCodingMachine\GraphQLite\Annotations\Logged
+@TheCodingMachine\GraphQLite\Annotations\Right(name=\"CAN_EDIT\")
+@TheCodingMachine\GraphQLite\Annotations\FailWith(NULL)", $schema->getTable('posts_users')->getOptions()['comment']);
+
+        $graphqlOptions->logged(false);
+
+        $this->assertSame("\n@TheCodingMachine\GraphQLite\Annotations\Field
+@TheCodingMachine\GraphQLite\Annotations\Right(name=\"CAN_EDIT\")
+@TheCodingMachine\GraphQLite\Annotations\FailWith(NULL)", $schema->getTable('posts_users')->getOptions()['comment']);
+
+        $this->assertSame($junctionTableOptions, $graphqlOptions->endGraphql());
+    }
+}