/** * @param BudgetForm $budgetForm * @param AuthUser $user * * @return Budget|null */ public function save(BudgetForm $budgetForm, AuthUser $user) { if (empty($budgetForm->getBudgetId())) { $budgetForm->setCreateUserId($user->getUserId()); $family = $this->familyRepository->getFamilyByUserId($user->getUserId()); $budgetForm->setFamilyId($family->getFamilyId()); } $budgetForm->setIsClosed(false); $budgetId = $this->budgetRepository->save($budgetForm); if (!$budgetId) { return null; } return $this->budgetRepository->getBudgetById($budgetId); }
/** * @param int $budgetId * @return BudgetReport|null */ public function generate($budgetId) { $budget = $this->budgetRepository->getBudgetById($budgetId); if (!$budget) { // @todo need throw exception return null; } $user = $this->userRepository->getUserById($budget->getCreateUserId()); if (!$user) { // @todo need throw exception return null; } $estimates = $this->estimateRepository->getEstimatesByBudgetId($budget->getBudgetId()); $categoryIds = []; foreach ($estimates as $estimate) { $categoryIds[] = $estimate->getCategoryId(); } $budgetEstimates = []; if (!empty($categoryIds)) { $categories = $this->categoryRepository->getListOfCategoriesByIds($categoryIds); foreach ($estimates as $estimate) { if (empty($categories[$estimate->getCategoryId()])) { // @todo need throw exception continue; } $category = $categories[$estimate->getCategoryId()]; $filterCriteria = new FilterCriteria(); $filterCriteria->setFamilyId($budget->getFamilyId()); $filterCriteria->setCategoryId($category->getCategoryId()); $filterCriteria->setOperationType(Operation::OPERATION_TYPE_EXPENSE); $filterCriteria->setStartDate($budget->getStartDate()); $filterCriteria->setEndDate($budget->getEndDate()); $budgetEstimates[] = new BudgetEstimate($estimate, $category, $this->operationFilter->filter($filterCriteria)); } } return new BudgetReport($budget, $user, $budgetEstimates); }