monthly budget, monthly report, unified report calculation (#2684)

src/Customer/CustomerStatisticService.php

@@ -10,9 +10,17 @@
 namespace App\Customer;
 
 use App\Entity\Customer;
+use App\Entity\Project;
 use App\Event\CustomerStatisticEvent;
+use App\Model\CustomerBudgetStatisticModel;
 use App\Model\CustomerStatistic;
 use App\Repository\CustomerRepository;
+use App\Repository\TimesheetRepository;
+use App\Timesheet\DateTimeFactory;
+use DateTime;
+use Doctrine\DBAL\Types\Types;
+use Doctrine\ORM\Query;
+use Doctrine\ORM\QueryBuilder;
 use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 
 /**
@@ -21,20 +29,120 @@ use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 class CustomerStatisticService
 {
     private $repository;
+    private $timesheetRepository;
     private $dispatcher;
 
-    public function __construct(CustomerRepository $customerRepository, EventDispatcherInterface $dispatcher)
+    public function __construct(CustomerRepository $customerRepository, TimesheetRepository $timesheetRepository, EventDispatcherInterface $dispatcher)
     {
         $this->repository = $customerRepository;
+        $this->timesheetRepository = $timesheetRepository;
         $this->dispatcher = $dispatcher;
     }
 
-    public function getCustomerStatistics(Customer $customer): CustomerStatistic
+    /**
+     * WARNING: this method does not respect the budget type. Your results will always be wither the "full lifetime data" or the "selected date-range".
+     *
+     * @param Customer $customer
+     * @param DateTime|null $begin
+     * @param DateTime|null $end
+     * @return CustomerStatistic
+     */
+    public function getCustomerStatistics(Customer $customer, ?DateTime $begin = null, ?DateTime $end = null): CustomerStatistic
     {
-        $statistic = $this->repository->getCustomerStatistics($customer);
-        $event = new CustomerStatisticEvent($customer, $statistic);
+        $statistics = $this->getBudgetStatistic([$customer], $begin, $end);
+        $event = new CustomerStatisticEvent($customer, array_pop($statistics), $begin, $end);
         $this->dispatcher->dispatch($event);
 
-        return $statistic;
+        return $event->getStatistic();
+    }
+
+    public function getBudgetStatisticModel(Customer $customer, DateTime $today): CustomerBudgetStatisticModel
+    {
+        $stats = new CustomerBudgetStatisticModel($customer);
+        $stats->setStatisticTotal($this->getCustomerStatistics($customer));
+
+        $begin = null;
+        $end = $today;
+
+        if ($customer->isMonthlyBudget()) {
+            $dateFactory = new DateTimeFactory($today->getTimezone());
+            $begin = $dateFactory->getStartOfMonth($today);
+            $end = $dateFactory->getEndOfMonth($today);
+        }
+
+        $stats->setStatistic($this->getCustomerStatistics($customer, $begin, $end));
+
+        return $stats;
+    }
+
+    /**
+     * @param Customer[] $customers
+     * @param DateTime|null $begin
+     * @param DateTime|null $end
+     * @return array<int, CustomerStatistic>
+     */
+    private function getBudgetStatistic(array $customers, ?DateTime $begin = null, ?DateTime $end = null): array
+    {
+        $statistics = [];
+        foreach ($customers as $customer) {
+            $statistics[$customer->getId()] = new CustomerStatistic();
+        }
+
+        $qb = $this->createStatisticQueryBuilder($customers, $begin, $end);
+
+        $result = $qb->getQuery()->getResult();
+
+        if (null !== $result) {
+            foreach ($result as $resultRow) {
+                $statistic = $statistics[$resultRow['id']];
+                $statistic->setDuration($statistic->getDuration() + $resultRow['duration']);
+                $statistic->setRate($statistic->getRate() + $resultRow['rate']);
+                $statistic->setInternalRate($statistic->getInternalRate() + $resultRow['internalRate']);
+                $statistic->setCounter($statistic->getCounter() + $resultRow['counter']);
+                if ($resultRow['billable']) {
+                    $statistic->setDurationBillable($resultRow['duration']);
+                    $statistic->setRateBillable($resultRow['rate']);
+                    $statistic->setInternalRateBillable($resultRow['internalRate']);
+                    $statistic->setCounterBillable($resultRow['counter']);
+                }
+            }
+        }
+
+        return $statistics;
+    }
+
+    private function createStatisticQueryBuilder(array $customers, DateTime $begin = null, ?DateTime $end = null): QueryBuilder
+    {
+        $qb = $this->timesheetRepository->createQueryBuilder('t');
+        $qb
+            ->select('IDENTITY(p.customer) AS id')
+            ->join(Project::class, 'p', Query\Expr\Join::WITH, 't.project = p.id')
+            ->addSelect('COALESCE(SUM(t.duration), 0) as duration')
+            ->addSelect('COALESCE(SUM(t.rate), 0) as rate')
+            ->addSelect('COALESCE(SUM(t.internalRate), 0) as internalRate')
+            ->addSelect('COUNT(t.id) as counter')
+            ->addSelect('t.billable as billable')
+            ->andWhere($qb->expr()->isNotNull('t.end'))
+            ->groupBy('id')
+            ->addGroupBy('billable')
+            ->andWhere($qb->expr()->in('p.customer', ':customer'))
+            ->setParameter('customer', $customers)
+        ;
+
+        if ($begin !== null) {
+            $qb
+                ->andWhere($qb->expr()->gte('t.begin', ':begin'))
+                ->setParameter('begin', $begin, Types::DATETIME_MUTABLE)
+            ;
+        }
+
+        if ($end !== null) {
+            $qb
+                ->andWhere($qb->expr()->lte('t.begin', ':end'))
+                ->setParameter('end', $end, Types::DATETIME_MUTABLE)
+            ;
+        }
+
+        return $qb;
     }
 }