Add rekey feature

[airbundle] / Repository / LocationRepository.php

diff --git a/Repository/LocationRepository.php b/Repository/LocationRepository.php
index 80e099605112431951deb320875645a59bc31c0a..129a1763ee497dafc6b64bc339fbe20ab878fcc0 100644 (file)
--- a/Repository/LocationRepository.php
+++ b/Repository/LocationRepository.php
@@ -3,17 +3,79 @@
 namespace Rapsys\AirBundle\Repository;
 
 use Symfony\Component\Translation\TranslatorInterface;
+use Doctrine\ORM\Query\ResultSetMapping;
 
 /**
  * LocationRepository
  */
 class LocationRepository extends \Doctrine\ORM\EntityRepository {
        /**
-        * Fetch translated location with session by date period
+        * Find complementary locations by session id
+        *
+        * @param $id The session id
+        * @return array The other locations
+        */
+       public function findComplementBySessionId($id) {
+               //Fetch complement locations
+               $ret = $this->getEntityManager()
+                         ->createQuery('SELECT l.id, l.title FROM RapsysAirBundle:Session s LEFT JOIN RapsysAirBundle:Session s2 WITH s2.id != s.id AND s2.slot = s.slot AND s2.date = s.date LEFT JOIN RapsysAirBundle:Location l WITH l.id != s.location AND (l.id != s2.location OR s2.location IS NULL) WHERE s.id = :sid GROUP BY l.id ORDER BY l.id')
+                       ->setParameter('sid', $id)
+                       ->getArrayResult();
+
+               //Rekey array
+               $ret = array_column($ret, 'id', 'title');
+
+               return $ret;
+       }
+
+       /**
+        * Find translated location title sorted by date period
+        *
+        * @param $translator The TranslatorInterface instance
+        * @param $period The date period
+        * @param $granted The session is granted
+        */
+       public function findTranslatedSortedByPeriod(TranslatorInterface $translator, $period, $userId = null) {
+               //Fetch sessions
+               $ret = $this->getEntityManager()
+                       ->createQuery(
+'SELECT l.id, l.title
+FROM RapsysAirBundle:Location l
+LEFT JOIN RapsysAirBundle:Session s WITH s.location = l.id AND s.date BETWEEN :begin AND :end
+LEFT JOIN RapsysAirBundle:Application a WITH a.id = s.application'.(!empty($userId)?' AND a.user = :uid':'').'
+GROUP BY l.id
+ORDER BY '.(!empty($userId)?'COUNT(a.id) DESC, ':'').'COUNT(s.id) DESC, l.id'
+                       )
+                       ->setParameter('begin', $period->getStartDate())
+                       ->setParameter('end', $period->getEndDate());
+
+               //Set optional user id
+               if (!empty($userId)) {
+                       $ret->setParameter('uid', $userId);
+               }
+
+               //Get Result
+               $ret = $ret->getResult();
+
+               //Rekey array
+               $ret = array_column($ret, 'title', 'id');
+
+               //Filter array
+               foreach($ret as $k => $v) {
+                       $ret[$k] = $translator->trans($v);
+               }
+
+               //Send result
+               return $ret;
+       }
+
+       /**
+        * Fetch translated location title with session by date period
         *
         * @param $translator The TranslatorInterface instance
         * @param $period The date period
         * @param $granted The session is granted
+        * TODO: a dropper
         */
        public function fetchTranslatedLocationByDatePeriod(TranslatorInterface $translator, $period, $granted = false) {
                //Fetch sessions
@@ -34,4 +96,79 @@ class LocationRepository extends \Doctrine\ORM\EntityRepository {
                //Send result
                return $ret;
        }
+
+       /**
+        * Fetch translated location title with user session by date period
+        *
+        * @param $translator The TranslatorInterface instance
+        * @param $period The date period
+        * @param $userId The user uid
+        * TODO: a dropper
+        */
+       public function fetchTranslatedUserLocationByDatePeriod(TranslatorInterface $translator, $period, $userId) {
+               //Fetch sessions
+               $ret = $this->getEntityManager()
+                       ->createQuery('SELECT l.id, l.title FROM RapsysAirBundle:Application a JOIN RapsysAirBundle:Session s JOIN RapsysAirBundle:Location l WHERE a.user = :uid AND a.session = s.id AND s.date BETWEEN :begin AND :end AND s.location = l.id GROUP BY l.id ORDER BY l.id')
+                       ->setParameter('begin', $period->getStartDate())
+                       ->setParameter('end', $period->getEndDate())
+                       ->setParameter('uid', $userId)
+                       ->getResult();
+
+               //Rekey array
+               $ret = array_column($ret, 'title', 'id');
+
+               //Filter array
+               foreach($ret as $k => $v) {
+                       $ret[$k] = $translator->trans($v);
+               }
+
+               //Send result
+               return $ret;
+       }
+
+       /**
+        * Find locations by user id
+        *
+        * @param $id The user id
+        * @return array The user locations
+        */
+       public function findByUserId($userId) {
+               //Get entity manager
+               $em = $this->getEntityManager();
+
+               //Get quote strategy
+               $qs = $em->getConfiguration()->getQuoteStrategy();
+               $dp = $em->getConnection()->getDatabasePlatform();
+
+               //Get quoted table names
+               //XXX: this allow to make this code table name independent
+               $tables = [
+                       'RapsysAirBundle:UserLocation' => $qs->getJoinTableName($em->getClassMetadata('RapsysAirBundle:User')->getAssociationMapping('locations'), $em->getClassMetadata('RapsysAirBundle:User'), $dp),
+                       'RapsysAirBundle:Location' => $qs->getTableName($em->getClassMetadata('RapsysAirBundle:Location'), $dp)
+               ];
+
+               //Set the request
+               $req = 'SELECT l.id, l.title
+FROM RapsysAirBundle:UserLocation AS ul
+JOIN RapsysAirBundle:Location AS l ON (l.id = ul.location_id)
+WHERE ul.user_id = :uid';
+
+               //Replace bundle entity name by table name
+               $req = str_replace(array_keys($tables), array_values($tables), $req);
+
+               //Get result set mapping instance
+               //XXX: DEBUG: see ../blog.orig/src/Rapsys/BlogBundle/Repository/ArticleRepository.php
+               $rsm = new ResultSetMapping();
+
+               //Declare result set for our request
+               $rsm->addEntityResult('RapsysAirBundle:Location', 'l');
+               $rsm->addFieldResult('l', 'id', 'id');
+               $rsm->addFieldResult('l', 'title', 'title');
+
+               //Send result
+               return $em
+                       ->createNativeQuery($req, $rsm)
+                       ->setParameter('uid', $userId)
+                       ->getResult();
+       }
 }