From 5e88be6760270382c16f69197c19bc5c924c75cf Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Tue, 20 Jun 2023 10:38:09 +0300
Subject: [PATCH] Update the index page to show only city's companies.

---
 app/Http/Controllers/API/v2/CompanyController.php | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/app/Http/Controllers/API/v2/CompanyController.php b/app/Http/Controllers/API/v2/CompanyController.php
index aaeb2ee..94fd907 100644
--- a/app/Http/Controllers/API/v2/CompanyController.php
+++ b/app/Http/Controllers/API/v2/CompanyController.php
@@ -55,9 +55,11 @@ class CompanyController extends Controller
         $companyAPI = new CompanyAPI($fetcher);
 
         $companyMap = $companyAPI->getMap($city_uuid);
-        $companyAPI->saveMap($companyMap);
+        $companyAPI->saveMap($companyMap, $city);
 
-        return new CompanyCollection(Company::all());
+        $companies = $city->getCompanies();
+
+        return new CompanyCollection($companies);
     }
 
     /**