public async Task <IActionResult> PutBudget(int id, ApiBudget budget)
{
if (id != budget.BudgetId)
{
return(BadRequest("Budget does not exist."));
}
var resource = new CoreBudget
{
BudgetId = budget.BudgetId,
TotalAmtBills = budget.TotalAmtBills,
User = (await _userRepo.GetUserById(budget.UserId))
};
try
{
await _repo.UpdateBudgetAsync(resource);
}
catch (DbUpdateConcurrencyException)
{
if (!await _repo.BudgetExistAsync(id))
{
return(NotFound("Budget not found."));
}
else
{
throw;
}
}
return(Ok("Budget updated!"));
}
public async Task UpdateBudgetAsync(CoreBudget budget)
{
var newbudget = Mapper.MapBudget(budget);
_context.Entry(newbudget).State = EntityState.Modified;
await Save();
}
public async Task <CoreBudget> AddBudgetAsync(CoreBudget budget)
{
var newBudget = Mapper.MapBudget(budget);
_context.Budget.Add(newBudget);
await Save();
return(Mapper.MapBudget(newBudget));
}
/* ************************
* * *
* * Budgets *
* * *
* ************************
*/
public static Budget MapBudget(CoreBudget budget)
{
return(new Budget
{
BudgetId = budget.BudgetId,
TotalAmtBills = budget.TotalAmtBills,
UserId = budget.User.UserId
});
}
/* ************************
* * *
* * Budgets *
* * *
* ************************
*/
public static ApiBudget MapBudgets(CoreBudget budget)
{
return(new ApiBudget
{
BudgetId = budget.BudgetId,
TotalAmtBills = budget.TotalAmtBills,
UserId = budget.User.UserId,
UserFirstName = budget.User.FirstName,
UserLastName = budget.User.LastName
});
}
public async Task <ActionResult> PostBudget(ApiBudget budget)
{
try
{
var resource = new CoreBudget
{
BudgetId = budget.BudgetId,
TotalAmtBills = budget.TotalAmtBills,
User = (await _userRepo.GetUserById(budget.UserId))
};
await _repo.AddBudgetAsync(resource);
return(Ok("Budget has been added!"));
}
catch (Exception e)
{
return(BadRequest(e));
}
}
