From 4536bc056686697e9c073a5d70c4b5e169f80505 Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Wed, 11 Sep 2019 20:15:44 +1000
Subject: [PATCH] API fixes

---
 app/Http/Controllers/Auth/LoginController.php | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/app/Http/Controllers/Auth/LoginController.php b/app/Http/Controllers/Auth/LoginController.php
index f38fc15330..3047e88357 100644
--- a/app/Http/Controllers/Auth/LoginController.php
+++ b/app/Http/Controllers/Auth/LoginController.php
@@ -16,7 +16,9 @@ use App\Http\Controllers\Controller;
 use App\Jobs\Account\CreateAccount;
 use App\Libraries\MultiDB;
 use App\Libraries\OAuth\OAuth;
+use App\Models\CompanyUser;
 use App\Models\User;
+use App\Transformers\CompanyUserTransformer;
 use App\Transformers\UserTransformer;
 use App\Utils\Traits\UserSessionAttributes;
 use Illuminate\Foundation\Auth\AuthenticatesUsers;
@@ -41,9 +43,9 @@ class LoginController extends BaseController
     use AuthenticatesUsers;
     use UserSessionAttributes;
 
-    protected $entity_type = User::class;
+    protected $entity_type = CompanyUser::class;
 
-    protected $entity_transformer = UserTransformer::class;
+    protected $entity_transformer = CompanyUserTransformer::class;
 
     /**
      * Where to redirect users after login.
@@ -105,7 +107,10 @@ class LoginController extends BaseController
             
             $user->setCompany($user->user_companies->first()->account->default_company);
 
-            return $this->itemResponse($this->guard()->user());
+            $ct = CompanyUser::whereUserId($user->id);
+
+            return $this->listResponse($ct);
+            //return $this->itemResponse($this->guard()->user());
         }
         else {