🚧 file entities now get their classifications added

Author: carbontwelve

src/Entities/Classification.php

@@ -21,7 +21,7 @@ class Classification
      *
      * @ManyToMany(targetEntity="Taxonomy", inversedBy="classification")
      * @JoinTable(
-     *  name="taxonomy_classifications",
+     *  name="classifications_taxonomy",
      *  joinColumns={
      *      @JoinColumn(name="classification_id", referencedColumnName="id")
      *  },
@@ -35,11 +35,7 @@ class Classification
     /**
      * @var \Doctrine\Common\Collections\Collection|File[]
      *
-     * @ManyToMany(targetEntity="Classification")
-     * @JoinTable(name="file_classifications",
-     *      joinColumns={@JoinColumn(name="classification_id", referencedColumnName="id")},
-     *      inverseJoinColumns={@JoinColumn(name="file_id", referencedColumnName="id")}
-     * )
+     * @ManyToMany(targetEntity="File", mappedBy="classifications")
      */
     private $files;
 

src/Entities/File.php

@@ -1,6 +1,8 @@
 <?php
 
 namespace TapestryCloud\Database\Entities;
+use Doctrine\Common\Collections\ArrayCollection;
+use Doctrine\Common\Collections\Collection;
 
 /**
  * @Entity
@@ -18,11 +20,41 @@ class File
      * @ManyToOne(targetEntity="Environment") */
     private $environment;
 
+    /**
+     * @var Collection|Classification[]
+     *
+     * @ManyToMany(targetEntity="Classification", inversedBy="files")
+     * @JoinTable(
+     *  name="classifications_files",
+     *  joinColumns={
+     *      @JoinColumn(name="file_id", referencedColumnName="id")
+     *  },
+     *  inverseJoinColumns={
+     *      @JoinColumn(name="classification_id", referencedColumnName="id")
+     *  }
+     * )
+     */
+    private $classifications;
+
     /**
      * @var string
      * @Column(type="string") */
     private $uid;
 
+    /**
+     * File constructor.
+     */
+    public function __construct()
+    {
+        $this->classifications = new ArrayCollection();
+    }
+
+    /**
+     * File Hydration.
+     *
+     * @param \Tapestry\Entities\File $file
+     * @param Environment|null $environment
+     */
     public function hydrate(\Tapestry\Entities\File $file, Environment $environment = null) {
         $this->uid = $file->getUid();
 
@@ -59,4 +91,28 @@ public function setEnvironment(Environment $environment)
     {
         $this->environment = $environment;
     }
+
+    /**
+     * @param Classification $classification
+     */
+    public function addClassification(Classification $classification)
+    {
+        if ($this->classifications->contains($classification)) {
+            return;
+        }
+
+        $this->classifications->add($classification);
+    }
+
+    /**
+     * @param Classification $classification
+     */
+    public function removeClassification(Classification $classification)
+    {
+        if (!$this->classifications->contains($classification)) {
+            return;
+        }
+
+        $this->classifications->removeElement($classification);
+    }
 }

src/Synchronizes/ContentTypes.php

@@ -8,6 +8,7 @@
 use TapestryCloud\Database\Entities\Classification;
 use TapestryCloud\Database\Entities\ContentType;
 use TapestryCloud\Database\Entities\Environment;
+use TapestryCloud\Database\Entities\File;
 use TapestryCloud\Database\Entities\Taxonomy;
 
 class ContentTypes
@@ -35,7 +36,7 @@ public function sync(ContentTypeFactory $contentTypeFactory, Environment $enviro
                 $this->em->persist($record);
                 $this->em->flush();
 
-                $this->syncTaxonomyToContentType($record, $contentType->getTaxonomies());
+                $this->syncTaxonomyToContentType($record, $contentType->getTaxonomies(), $environment);
                 continue;
             }
 
@@ -46,11 +47,9 @@ public function sync(ContentTypeFactory $contentTypeFactory, Environment $enviro
     /**
      * @param ContentType $contentType
      * @param TapestryTaxonomy[] $taxonomies
+     * @param Environment $environment
      */
-    private function syncTaxonomyToContentType(ContentType $contentType, array $taxonomies) {
-
-        $n = $contentType->getTaxonomy();
-        $p = 1;
+    private function syncTaxonomyToContentType(ContentType $contentType, array $taxonomies, Environment $environment) {
 
         /** @var TapestryTaxonomy $taxonomy */
         foreach($taxonomies as $taxonomy) {
@@ -65,7 +64,6 @@ private function syncTaxonomyToContentType(ContentType $contentType, array $taxo
                 $this->em->flush();
 
                 foreach ($taxonomy->getFileList() as $classification => $files) {
-
                     if (! $classificationRecord = $this->em->getRepository(Classification::class)->findOneBy(['name' => $classification])){
                         $classificationRecord = new Classification();
                         $classificationRecord->setName($classification);
@@ -74,6 +72,15 @@ private function syncTaxonomyToContentType(ContentType $contentType, array $taxo
 
                     $classificationRecord->addTaxonomy($record);
                     $this->em->flush();
+
+                    foreach(array_keys($files) as $filename) {
+                        /** @var File $file */
+                        if ($file = $this->em->getRepository(File::class)->findOneBy(['uid' => $filename, 'environment' => $environment->getId()])){
+                            $file->addClassification($classificationRecord);
+                            $this->em->flush();
+                        }
+                    }
+
                 }
             }
         }

tests/PluginTest.php

@@ -10,10 +10,17 @@
 use Tapestry\Entities\Project;
 use Tapestry\Generator;
 use Tapestry\Tapestry;
+use TapestryCloud\Database\Entities\Classification;
+use TapestryCloud\Database\Entities\ContentType;
+use TapestryCloud\Database\Entities\File;
+use TapestryCloud\Database\Entities\Taxonomy;
 
 class PluginTest extends \PHPUnit_Framework_TestCase
 {
 
+    /**
+     * @var EntityManager
+     */
     protected static $em;
 
     /**
@@ -35,6 +42,7 @@ public static function setUpBeforeClass()
             )
         );
         $tool = new SchemaTool($em);
+        $tool->dropDatabase();
         $tool->createSchema($em->getMetadataFactory()->getAllMetadata());
 
         self::$em = $em;
@@ -66,8 +74,34 @@ public function testPlugin(){
         $project = $tapestry->getContainer()->get(Project::class);
         $generator->generate($project, new NullOutput());
 
-        /** @var Project $project */
-        //$project = $tapestry->getContainer()->get(Project::class);
+        $contentTypes = self::$em->getRepository(ContentType::class)->findAll();
+
+        $this->assertCount(2, $contentTypes);
+
+        /** @var ContentType $contentType */
+        foreach ($contentTypes as $contentType) {
+            $env = $contentType->getEnvironment();
+            $this->assertEquals('testing', $env->getName());
+
+            if ($contentType->getName() === 'blog'){
+                $taxonomies = $contentType->getTaxonomy();
+                $this->assertCount(2, $taxonomies);
+
+                /** @var Taxonomy $taxonomy */
+                foreach ($taxonomies as $taxonomy) {
+                    if ($taxonomy->getName() === 'tag') {
+                        $classifications = $taxonomy->getClassifications();
+                        $this->assertCount(2, $classifications);
+                    }
+                }
+            }
+        }
+
+        $files = self::$em->getRepository(File::class)->findAll();
+        $this->assertCount(3, $files);
+
+        $classifications = self::$em->getRepository(Classification::class)->findAll();
+        $this->assertCount(4, $classifications);
 
         $n = 1;
     }