TruthSource in Login

app/Http/Controllers/Auth/LoginController.php

@@ -217,6 +217,12 @@ class LoginController extends BaseController
             $cu = CompanyUser::query()
                   ->where('user_id', auth()->user()->id);
 
+            $truth = app()->make(TruthSource::class);
+
+            $truth->setCompanyUser($cu->first());
+            $truth->setUser(auth()->user());
+            $truth->setCompany($user->account->default_company);
+
             if(!$cu->exists())
                 return response()->json(['message' => 'User not linked to any companies'], 403);
 

app/Http/Middleware/TokenAuth.php

@@ -32,6 +32,7 @@ class TokenAuth
     public function handle($request, Closure $next)
     {
         if ($request->header('X-API-TOKEN') && ($company_token = CompanyToken::with(['user', 'company', 'cu'])->where('token', $request->header('X-API-TOKEN'))->first())) {
+            
             $user = $company_token->user;
 
             $error = [