Phraseanet/lib/Doctrine/Repositories/BasketRepository.php

<?php

/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2010 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Repositories;

use Doctrine\ORM\EntityRepository;
use DoctrineExtensions\Paginate\Paginate;

/**
 *
 * @package
 * @license     http://opensource.org/licenses/gpl-3.0 GPLv3
 * @link        www.phraseanet.com
 */
class BasketRepository extends EntityRepository
{

  const MYBASKETS = 'my baskets';
  const RECEIVED = 'received';
  const VALIDATION_SENT = 'validation_sent';
  const VALIDATION_DONE = 'validation_done';

  /**
   * Returns all basket for a given user that are not marked as archived
   *
   * @param \User_Adapter $user
   * @return \Doctrine\Common\Collections\ArrayCollection
   */
  public function findActiveByUser(\User_Adapter $user)
  {
    $dql = 'SELECT b FROM Entities\Basket b
            WHERE b.usr_id = :usr_id AND b.archived = false';

    $query = $this->_em->createQuery($dql);
    $query->setParameters(array('usr_id' => $user->get_id()));

    return $query->getResult();
  }

  /**
   * Returns all unread basket for a given user that are not marked as archived
   *
   * @param \User_Adapter $user
   * @return \Doctrine\Common\Collections\ArrayCollection
   */
  public function findUnreadActiveByUser(\User_Adapter $user)
  {
    $dql = 'SELECT b FROM Entities\Basket b
            WHERE b.usr_id = :usr_id
              AND b.archived = false AND b.is_read = false';

    $query = $this->_em->createQuery($dql);
    $query->setParameters(array('usr_id' => $user->get_id()));

    return $query->getResult();
  }

  /**
   * Returns all baskets that are in validation session not expired  and
   * where a specified user is participant (not owner)
   *
   * @param \User_Adapter $user
   * @return \Doctrine\Common\Collections\ArrayCollection
   */
  public function findActiveValidationByUser(\User_Adapter $user)
  {
    $dql = 'SELECT b FROM Entities\Basket b
              JOIN b.validation s
              JOIN s.participants p
            WHERE b.usr_id != ?1 AND p.usr_id = ?2
                  AND s.expires > CURRENT_TIMESTAMP()';

    $query = $this->_em->createQuery($dql);
    $query->setParameters(array(1 => $user->get_id(), 2 => $user->get_id()));

    return $query->getResult();
  }

  /**
   * Find a basket specified by his basket_id and his owner
   *
   * @throws \Exception_NotFound
   * @throws \Exception_Forbidden
   * @param type $basket_id
   * @param \User_Adapter $user
   * @return \Entities\Basket
   */
  public function findUserBasket($basket_id, \User_Adapter $user)
  {
    $basket = $this->find($basket_id);

    /* @var $basket \Entities\Basket */
    if (null === $basket)
    {
      throw new \Exception_NotFound(_('Basket is not found'));
    }

    if ($basket->getowner()->get_id() != $user->get_id())
    {
      throw new \Exception_Forbidden(_('You have not access to this basket'));
    }

    return $basket;
  }

  public function findContainingRecord(\record_adapter $record)
  {

    $dql = 'SELECT b FROM Entities\Basket b
              JOIN b.elements e
            WHERE e.record_id = :record_id AND e.sbas_id = e.sbas_id';

    $params = array(
        'record_id' => $record->get_record_id()
    );

    $query = $this->_em->createQuery($dql);
    $query->setParameters($params);

    return $query->getResult();
  }

  public function findWorkzoneBasket(\User_Adapter $user, $query, $year, $type, $offset, $perPage)
  {
    $params = array();

    switch ($type)
    {
      case self::RECEIVED:
        $dql = 'SELECT b FROM Entities\Basket b
                  WHERE b.usr_id = :usr_id AND b.pusher_id IS NOT NULL';
        $params = array(
            'usr_id' => $user->get_id()
        );
        break;
      case self::VALIDATION_DONE:
        $dql = 'SELECT b FROM Entities\Basket b
                  JOIN b.validation s
                  JOIN s.participants p
                WHERE b.usr_id != ?1 AND p.usr_id = ?2';
        $params = array(
            1 => $user->get_id()
            , 2 => $user->get_id()
        );
        break;
      case self::VALIDATION_SENT:
        $dql = 'SELECT b FROM Entities\Basket b
                JOIN b.validation v
                WHERE b.usr_id = :usr_id';
        $params = array(
            'usr_id' => $user->get_id()
        );
        break;
      default:
        $dql = 'SELECT b FROM Entities\Basket b
                LEFT JOIN b.validation s LEFT JOIN s.participants p
                WHERE (b.usr_id = :usr_id OR p.usr_id = :validating_usr_id)';
        $params = array(
            'usr_id' => $user->get_id(),
            'validating_usr_id' => $user->get_id()
        );
        break;
    }

    if (ctype_digit($year) && strlen($year) == 4)
    {
      $dql .= ' AND b.created >= :min_date AND b.created <= :max_date ';

      $params['min_date'] = sprintf('%d-01-01 00:00:00', $year);
      $params['max_date'] = sprintf('%d-12-31 23:59:59', $year);
    }

    if (trim($query) !== '')
    {
      $dql .= ' AND (b.name LIKE :name OR b.description LIKE :description) ';

      $params['name'] = '%'.$query.'%';
      $params['description'] = '%'.$query.'%';
    }

    $query = $this->_em->createQuery($dql);
    $query->setParameters($params);

    $count = Paginate::getTotalQueryResults($query);
    $paginateQuery = Paginate::getPaginateQuery($query, $offset, $perPage);
    $result = $paginateQuery->getResult();

    return array('count' => $count, 'result' => $result);
  }

}