From 301032c905f1c226f3be2a542a2e2441e4a30bff Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Fri, 28 Oct 2016 07:48:34 +1100
Subject: [PATCH] Expense categories

---
 app/Http/Controllers/ExpenseApiController.php | 25 ++++++++++++++++---
 app/Http/routes.php                           |  3 ++-
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/app/Http/Controllers/ExpenseApiController.php b/app/Http/Controllers/ExpenseApiController.php
index 5734d577feb7..5c7ee79cd3da 100644
--- a/app/Http/Controllers/ExpenseApiController.php
+++ b/app/Http/Controllers/ExpenseApiController.php
@@ -1,6 +1,8 @@
 <?php namespace App\Http\Controllers;
 
+use App\Http\Requests\Request;
 use App\Models\Expense;
+use App\Ninja\Repositories\ExpenseCategoryRepository;
 use App\Ninja\Repositories\ExpenseRepository;
 use App\Services\ExpenseService;
 use App\Http\Requests\ExpenseRequest;
@@ -12,15 +14,17 @@ class ExpenseApiController extends BaseAPIController
     // Expenses
     protected $expenseRepo;
     protected $expenseService;
+    protected $categoryRepo;
 
     protected $entityType = ENTITY_EXPENSE;
 
-    public function __construct(ExpenseRepository $expenseRepo, ExpenseService $expenseService)
+    public function __construct(ExpenseCategoryRepository $categoryRepo, ExpenseRepository $expenseRepo, ExpenseService $expenseService)
     {
         parent::__construct();
 
         $this->expenseRepo = $expenseRepo;
         $this->expenseService = $expenseService;
+        $this->categoryRepo = $categoryRepo;
     }
 
     /**
@@ -75,8 +79,8 @@ class ExpenseApiController extends BaseAPIController
         $expense = $this->expenseRepo->save($request->input());
 
         $expense = Expense::scope($expense->public_id)
-                    ->with('client', 'invoice', 'vendor')
-                    ->first();
+            ->with('client', 'invoice', 'vendor')
+            ->first();
 
         return $this->itemResponse($expense);
     }
@@ -146,4 +150,19 @@ class ExpenseApiController extends BaseAPIController
     }
 
 
+    public function updateCategory(UpdateExpenseCategoryRequest $request)
+    {
+        $category = $this->categoryRepo->save($request->input(), $request->entity());
+
+        return $this->itemResponse($category);
+    }
+
+    public function addCategory(CreateExpenseCategoryRequest $request)
+    {
+        $category = $this->categoryRepo->save($request->input());
+
+        return $this->itemResponse($category);
+
+    }
+
 }
diff --git a/app/Http/routes.php b/app/Http/routes.php
index dbb061e0ae29..07a29b090345 100644
--- a/app/Http/routes.php
+++ b/app/Http/routes.php
@@ -309,7 +309,8 @@ Route::group(['middleware' => 'api', 'prefix' => 'api/v1'], function()
     Route::get('dashboard', 'DashboardApiController@index');
     Route::resource('documents', 'DocumentAPIController');
     Route::resource('vendors', 'VendorApiController');
-    Route::resource('expenses', 'ExpenseApiController');
+    Route::post('createExpenseCategory', 'ExpenseApiController@addCategory');
+    Route::put('updateExpenseCategory/{expense_category_id}', 'ExpenseApiController@updateCategory');
 });
 
 // Redirects for legacy links