src/Wallabag/CoreBundle/Helper/TagsAssigner.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Helper;
   4
   5 use Wallabag\CoreBundle\Entity\Entry;
   6 use Wallabag\CoreBundle\Entity\Tag;
   7 use Wallabag\CoreBundle\Repository\TagRepository;
   8
   9 class TagsAssigner
  10 {
  11
  12     /**
  13      * @var TagRepository $tagRepository
  14      */
  15     protected $tagRepository;
  16
  17     public function __construct(TagRepository $tagRepository)
  18     {
  19         $this->tagRepository = $tagRepository;
  20     }
  21
  22     /**
  23      * Assign some tags to an entry.
  24      *
  25      * @param Entry        $entry
  26      * @param array|string $tags          An array of tag or a string coma separated of tag
  27      * @param array        $entitiesReady Entities from the EntityManager which are persisted but not yet flushed
  28      *                                    It is mostly to fix duplicate tag on import @see http://stackoverflow.com/a/7879164/569101
  29      *
  30      * @return Tag[]
  31      */
  32     public function assignTagsToEntry(Entry $entry, $tags, array $entitiesReady = [])
  33     {
  34         $tagsEntities = [];
  35
  36         if (!is_array($tags)) {
  37             $tags = explode(',', $tags);
  38         }
  39
  40         // keeps only Tag entity from the "not yet flushed entities"
  41         $tagsNotYetFlushed = [];
  42         foreach ($entitiesReady as $entity) {
  43             if ($entity instanceof Tag) {
  44                 $tagsNotYetFlushed[$entity->getLabel()] = $entity;
  45             }
  46         }
  47
  48         foreach ($tags as $label) {
  49             $label = trim($label);
  50
  51             // avoid empty tag
  52             if (0 === strlen($label)) {
  53                 continue;
  54             }
  55
  56             if (isset($tagsNotYetFlushed[$label])) {
  57                 $tagEntity = $tagsNotYetFlushed[$label];
  58             } else {
  59                 $tagEntity = $this->tagRepository->findOneByLabel($label);
  60
  61                 if (null === $tagEntity) {
  62                     $tagEntity = new Tag();
  63                     $tagEntity->setLabel($label);
  64                 }
  65             }
  66
  67             // only add the tag on the entry if the relation doesn't exist
  68             if (false === $entry->getTags()->contains($tagEntity)) {
  69                 $entry->addTag($tagEntity);
  70                 $tagsEntities[] = $tagEntity;
  71             }
  72         }
  73
  74         return $tagsEntities;
  75     }
  76 }