478 lines
15 KiB
PHP
478 lines
15 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Kimai time-tracking app.
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace App\Repository;
|
|
|
|
use App\Entity\Activity;
|
|
use App\Entity\Project;
|
|
use App\Entity\ProjectComment;
|
|
use App\Entity\Timesheet;
|
|
use App\Entity\User;
|
|
use App\Model\ProjectStatistic;
|
|
use App\Repository\Loader\ProjectLoader;
|
|
use App\Repository\Paginator\LoaderPaginator;
|
|
use App\Repository\Paginator\PaginatorInterface;
|
|
use App\Repository\Query\ProjectFormTypeQuery;
|
|
use App\Repository\Query\ProjectQuery;
|
|
use Doctrine\ORM\EntityRepository;
|
|
use Doctrine\ORM\ORMException;
|
|
use Doctrine\ORM\Query;
|
|
use Doctrine\ORM\QueryBuilder;
|
|
use Pagerfanta\Pagerfanta;
|
|
|
|
class ProjectRepository extends EntityRepository
|
|
{
|
|
/**
|
|
* @param mixed $id
|
|
* @param null $lockMode
|
|
* @param null $lockVersion
|
|
* @return Project|null
|
|
*/
|
|
public function find($id, $lockMode = null, $lockVersion = null)
|
|
{
|
|
/** @var Project|null $project */
|
|
$project = parent::find($id, $lockMode, $lockVersion);
|
|
if (null === $project) {
|
|
return null;
|
|
}
|
|
|
|
$loader = new ProjectLoader($this->getEntityManager());
|
|
$loader->loadResults([$project]);
|
|
|
|
return $project;
|
|
}
|
|
|
|
/**
|
|
* @param Project $project
|
|
* @throws ORMException
|
|
* @throws \Doctrine\ORM\OptimisticLockException
|
|
*/
|
|
public function saveProject(Project $project)
|
|
{
|
|
$entityManager = $this->getEntityManager();
|
|
$entityManager->persist($project);
|
|
$entityManager->flush();
|
|
}
|
|
|
|
/**
|
|
* @param null|bool $visible
|
|
* @return int
|
|
*/
|
|
public function countProject($visible = null)
|
|
{
|
|
if (null !== $visible) {
|
|
return $this->count(['visible' => (bool) $visible]);
|
|
}
|
|
|
|
return $this->count([]);
|
|
}
|
|
|
|
public function getProjectStatistics(Project $project): ProjectStatistic
|
|
{
|
|
$stats = new ProjectStatistic($project);
|
|
|
|
$qb = $this->getEntityManager()->createQueryBuilder();
|
|
|
|
$qb
|
|
->addSelect('COUNT(t.id) as recordAmount')
|
|
->addSelect('SUM(t.duration) as recordDuration')
|
|
->addSelect('SUM(t.rate) as recordRate')
|
|
->from(Timesheet::class, 't')
|
|
->andWhere('t.project = :project')
|
|
;
|
|
$timesheetResult = $qb->getQuery()->execute(['project' => $project], Query::HYDRATE_ARRAY);
|
|
|
|
if (isset($timesheetResult[0])) {
|
|
$stats->setRecordAmount($timesheetResult[0]['recordAmount']);
|
|
$stats->setRecordDuration($timesheetResult[0]['recordDuration']);
|
|
$stats->setRecordRate($timesheetResult[0]['recordRate']);
|
|
}
|
|
|
|
$qb = $this->getEntityManager()->createQueryBuilder();
|
|
$qb->select('COUNT(a.id) as activityAmount')
|
|
->from(Activity::class, 'a')
|
|
->andWhere('a.project = :project')
|
|
;
|
|
$resultActivities = $qb->getQuery()->execute(['project' => $project], Query::HYDRATE_ARRAY);
|
|
|
|
if (isset($resultActivities[0])) {
|
|
$resultActivities = $resultActivities[0];
|
|
|
|
$stats->setActivityAmount($resultActivities['activityAmount']);
|
|
}
|
|
|
|
return $stats;
|
|
}
|
|
|
|
private function addPermissionCriteria(QueryBuilder $qb, ?User $user = null, array $teams = [])
|
|
{
|
|
// make sure that all queries without a user see all projects
|
|
if (null === $user && empty($teams)) {
|
|
return;
|
|
}
|
|
|
|
// make sure that admins see all projects
|
|
if (null !== $user && ($user->isSuperAdmin() || $user->isAdmin())) {
|
|
return;
|
|
}
|
|
|
|
if (null !== $user) {
|
|
$teams = array_merge($teams, $user->getTeams()->toArray());
|
|
}
|
|
|
|
$qb->leftJoin('p.teams', 'teams')
|
|
->leftJoin('c.teams', 'c_teams');
|
|
|
|
if (empty($teams)) {
|
|
$qb->andWhere($qb->expr()->isNull('c_teams'));
|
|
$qb->andWhere($qb->expr()->isNull('teams'));
|
|
|
|
return;
|
|
}
|
|
|
|
$orProject = $qb->expr()->orX(
|
|
$qb->expr()->isNull('teams'),
|
|
$qb->expr()->isMemberOf(':teams', 'p.teams')
|
|
);
|
|
$qb->andWhere($orProject);
|
|
|
|
$orCustomer = $qb->expr()->orX(
|
|
$qb->expr()->isNull('c_teams'),
|
|
$qb->expr()->isMemberOf(':teams', 'c.teams')
|
|
);
|
|
$qb->andWhere($orCustomer);
|
|
|
|
$qb->setParameter('teams', $teams);
|
|
}
|
|
|
|
/**
|
|
* @deprecated since 1.1 - don't use this method, it ignores team permission checks
|
|
*/
|
|
public function builderForEntityType($project, $customer)
|
|
{
|
|
$query = new ProjectFormTypeQuery();
|
|
$query->setProject($project);
|
|
$query->setCustomer($customer);
|
|
|
|
return $this->getQueryBuilderForFormType($query);
|
|
}
|
|
|
|
/**
|
|
* Returns a query builder that is used for ProjectType and your own 'query_builder' option.
|
|
*
|
|
* @param ProjectFormTypeQuery $query
|
|
* @return QueryBuilder
|
|
*/
|
|
public function getQueryBuilderForFormType(ProjectFormTypeQuery $query): QueryBuilder
|
|
{
|
|
$qb = $this->getEntityManager()->createQueryBuilder();
|
|
|
|
$qb
|
|
->select('p')
|
|
->from(Project::class, 'p')
|
|
->leftJoin('p.customer', 'c')
|
|
->addOrderBy('c.name', 'ASC')
|
|
->addOrderBy('p.name', 'ASC')
|
|
;
|
|
|
|
$qb->andWhere($qb->expr()->eq('p.visible', ':visible'));
|
|
$qb->andWhere($qb->expr()->eq('c.visible', ':customer_visible'));
|
|
|
|
if (!$query->isIgnoreDate()) {
|
|
$now = new \DateTime();
|
|
$qb->andWhere(
|
|
$qb->expr()->andX(
|
|
$qb->expr()->orX(
|
|
$qb->expr()->lte('p.start', ':start'),
|
|
$qb->expr()->isNull('p.start')
|
|
),
|
|
$qb->expr()->orX(
|
|
$qb->expr()->gte('p.end', ':start'),
|
|
$qb->expr()->isNull('p.end')
|
|
)
|
|
)
|
|
)->setParameter('start', $now);
|
|
|
|
$qb->andWhere(
|
|
$qb->expr()->andX(
|
|
$qb->expr()->orX(
|
|
$qb->expr()->gte('p.end', ':end'),
|
|
$qb->expr()->isNull('p.end')
|
|
),
|
|
$qb->expr()->orX(
|
|
$qb->expr()->lte('p.start', ':end'),
|
|
$qb->expr()->isNull('p.start')
|
|
)
|
|
)
|
|
)->setParameter('end', $now);
|
|
}
|
|
|
|
$qb->setParameter('visible', true, \PDO::PARAM_BOOL);
|
|
$qb->setParameter('customer_visible', true, \PDO::PARAM_BOOL);
|
|
|
|
if (null !== $query->getProject()) {
|
|
$qb->orWhere('p.id = :project')->setParameter('project', $query->getProject());
|
|
}
|
|
|
|
if (null !== $query->getCustomer()) {
|
|
$qb->andWhere('p.customer = :customer')
|
|
->setParameter('customer', $query->getCustomer());
|
|
}
|
|
|
|
if (null !== $query->getProjectToIgnore()) {
|
|
$qb->andWhere($qb->expr()->neq('p.id', ':ignored'));
|
|
$qb->setParameter('ignored', $query->getProjectToIgnore());
|
|
}
|
|
|
|
$this->addPermissionCriteria($qb, $query->getUser(), $query->getTeams());
|
|
|
|
return $qb;
|
|
}
|
|
|
|
private function getQueryBuilderForQuery(ProjectQuery $query): QueryBuilder
|
|
{
|
|
$qb = $this->getEntityManager()->createQueryBuilder();
|
|
|
|
$qb
|
|
->select('p')
|
|
->from(Project::class, 'p')
|
|
->leftJoin('p.customer', 'c')
|
|
;
|
|
|
|
$orderBy = $query->getOrderBy();
|
|
switch ($orderBy) {
|
|
case 'customer':
|
|
$orderBy = 'c.name';
|
|
break;
|
|
default:
|
|
$orderBy = 'p.' . $orderBy;
|
|
break;
|
|
}
|
|
|
|
$qb->addOrderBy($orderBy, $query->getOrder());
|
|
|
|
if (in_array($query->getVisibility(), [ProjectQuery::SHOW_VISIBLE, ProjectQuery::SHOW_HIDDEN])) {
|
|
$qb
|
|
->andWhere($qb->expr()->eq('p.visible', ':visible'))
|
|
->andWhere($qb->expr()->eq('c.visible', ':customer_visible'))
|
|
;
|
|
|
|
if (ProjectQuery::SHOW_VISIBLE === $query->getVisibility()) {
|
|
$qb->setParameter('visible', true, \PDO::PARAM_BOOL);
|
|
} elseif (ProjectQuery::SHOW_HIDDEN === $query->getVisibility()) {
|
|
$qb->setParameter('visible', false, \PDO::PARAM_BOOL);
|
|
}
|
|
|
|
$qb->setParameter('customer_visible', true, \PDO::PARAM_BOOL);
|
|
}
|
|
|
|
if (null !== $query->getCustomer()) {
|
|
$qb->andWhere('p.customer = :customer')
|
|
->setParameter('customer', $query->getCustomer());
|
|
}
|
|
|
|
// this is far from being perfect, possible enhancements:
|
|
// there could also be a range selection to be able to select all projects that were active between from and to
|
|
// begin = null and end = null
|
|
// begin = null and end <= to
|
|
// begin < to and end = null
|
|
// begin > from and end < to
|
|
// ... and more ...
|
|
|
|
$begin = $query->getProjectStart();
|
|
$end = $query->getProjectEnd();
|
|
|
|
if (null !== $begin) {
|
|
$qb->andWhere(
|
|
$qb->expr()->andX(
|
|
$qb->expr()->orX(
|
|
$qb->expr()->lte('p.start', ':start'),
|
|
$qb->expr()->isNull('p.start')
|
|
),
|
|
$qb->expr()->orX(
|
|
$qb->expr()->gte('p.end', ':start'),
|
|
$qb->expr()->isNull('p.end')
|
|
)
|
|
)
|
|
)->setParameter('start', $query->getProjectStart());
|
|
}
|
|
|
|
if (null !== $end) {
|
|
$qb->andWhere(
|
|
$qb->expr()->andX(
|
|
$qb->expr()->orX(
|
|
$qb->expr()->gte('p.end', ':end'),
|
|
$qb->expr()->isNull('p.end')
|
|
),
|
|
$qb->expr()->orX(
|
|
$qb->expr()->lte('p.start', ':end'),
|
|
$qb->expr()->isNull('p.start')
|
|
)
|
|
)
|
|
)->setParameter('end', $query->getProjectEnd());
|
|
}
|
|
|
|
$this->addPermissionCriteria($qb, $query->getCurrentUser());
|
|
|
|
if ($query->hasSearchTerm()) {
|
|
$searchAnd = $qb->expr()->andX();
|
|
$searchTerm = $query->getSearchTerm();
|
|
|
|
foreach ($searchTerm->getSearchFields() as $metaName => $metaValue) {
|
|
$qb->leftJoin('p.meta', 'meta');
|
|
$searchAnd->add(
|
|
$qb->expr()->andX(
|
|
$qb->expr()->eq('meta.name', ':metaName'),
|
|
$qb->expr()->like('meta.value', ':metaValue')
|
|
)
|
|
);
|
|
$qb->setParameter('metaName', $metaName);
|
|
$qb->setParameter('metaValue', '%' . $metaValue . '%');
|
|
}
|
|
|
|
if ($searchTerm->hasSearchTerm()) {
|
|
$searchAnd->add(
|
|
$qb->expr()->orX(
|
|
$qb->expr()->like('p.name', ':searchTerm'),
|
|
$qb->expr()->like('p.comment', ':searchTerm'),
|
|
$qb->expr()->like('p.orderNumber', ':searchTerm')
|
|
)
|
|
);
|
|
$qb->setParameter('searchTerm', '%' . $searchTerm->getSearchTerm() . '%');
|
|
}
|
|
|
|
if ($searchAnd->count() > 0) {
|
|
$qb->andWhere($searchAnd);
|
|
}
|
|
}
|
|
|
|
// this will make sure, that we do not accidentally create results with multiple rows
|
|
// => which would result in a wrong LIMIT / pagination results
|
|
// the second group by is needed due to SQL standard (even though logically not really required for this query)
|
|
$qb->addGroupBy('p.id')->addGroupBy($orderBy);
|
|
|
|
return $qb;
|
|
}
|
|
|
|
public function countProjectsForQuery(ProjectQuery $query): int
|
|
{
|
|
$qb = $this->getQueryBuilderForQuery($query);
|
|
$qb
|
|
->resetDQLPart('select')
|
|
->resetDQLPart('orderBy')
|
|
->resetDQLPart('groupBy')
|
|
->select($qb->expr()->countDistinct('p.id'))
|
|
;
|
|
|
|
return (int) $qb->getQuery()->getSingleScalarResult();
|
|
}
|
|
|
|
public function getPagerfantaForQuery(ProjectQuery $query): Pagerfanta
|
|
{
|
|
$paginator = new Pagerfanta($this->getPaginatorForQuery($query));
|
|
$paginator->setMaxPerPage($query->getPageSize());
|
|
$paginator->setCurrentPage($query->getPage());
|
|
|
|
return $paginator;
|
|
}
|
|
|
|
private function getPaginatorForQuery(ProjectQuery $query): PaginatorInterface
|
|
{
|
|
$counter = $this->countProjectsForQuery($query);
|
|
$qb = $this->getQueryBuilderForQuery($query);
|
|
|
|
return new LoaderPaginator(new ProjectLoader($qb->getEntityManager()), $qb, $counter);
|
|
}
|
|
|
|
/**
|
|
* @param ProjectQuery $query
|
|
* @return Project[]
|
|
*/
|
|
public function getProjectsForQuery(ProjectQuery $query): iterable
|
|
{
|
|
$qb = $this->getQueryBuilderForQuery($query);
|
|
$results = $qb->getQuery()->execute();
|
|
$loader = new ProjectLoader($qb->getEntityManager());
|
|
$loader->loadResults($results);
|
|
|
|
return $results;
|
|
}
|
|
|
|
/**
|
|
* @param Project $delete
|
|
* @param Project|null $replace
|
|
* @throws \Doctrine\ORM\ORMException
|
|
*/
|
|
public function deleteProject(Project $delete, ?Project $replace = null)
|
|
{
|
|
$em = $this->getEntityManager();
|
|
$em->beginTransaction();
|
|
|
|
try {
|
|
if (null !== $replace) {
|
|
$qb = $em->createQueryBuilder();
|
|
$qb
|
|
->update(Timesheet::class, 't')
|
|
->set('t.project', ':replace')
|
|
->where('t.project = :delete')
|
|
->setParameter('delete', $delete)
|
|
->setParameter('replace', $replace)
|
|
->getQuery()
|
|
->execute();
|
|
|
|
$qb = $em->createQueryBuilder();
|
|
$qb
|
|
->update(Activity::class, 'a')
|
|
->set('a.project', ':replace')
|
|
->where('a.project = :delete')
|
|
->setParameter('delete', $delete)
|
|
->setParameter('replace', $replace)
|
|
->getQuery()
|
|
->execute();
|
|
}
|
|
|
|
$em->remove($delete);
|
|
$em->flush();
|
|
$em->commit();
|
|
} catch (ORMException $ex) {
|
|
$em->rollback();
|
|
throw $ex;
|
|
}
|
|
}
|
|
|
|
public function getComments(Project $project): array
|
|
{
|
|
$qb = $this->getEntityManager()->createQueryBuilder();
|
|
$qb
|
|
->select('comments')
|
|
->from(ProjectComment::class, 'comments')
|
|
->andWhere($qb->expr()->eq('comments.project', ':project'))
|
|
->addOrderBy('comments.pinned', 'DESC')
|
|
->addOrderBy('comments.createdAt', 'DESC')
|
|
->setParameter('project', $project)
|
|
;
|
|
|
|
return $qb->getQuery()->getResult();
|
|
}
|
|
|
|
public function saveComment(ProjectComment $comment)
|
|
{
|
|
$entityManager = $this->getEntityManager();
|
|
$entityManager->persist($comment);
|
|
$entityManager->flush();
|
|
}
|
|
|
|
public function deleteComment(ProjectComment $comment)
|
|
{
|
|
$entityManager = $this->getEntityManager();
|
|
$entityManager->remove($comment);
|
|
$entityManager->flush();
|
|
}
|
|
}
|