From dfbaa1afd35a2d479faaed64b5495589952cefa6 Mon Sep 17 00:00:00 2001
From: Nabeel S <nabeelio@users.noreply.github.com>
Date: Sun, 23 Feb 2020 19:48:28 -0500
Subject: [PATCH] Fix rowmapper generator, check for fields, map users #443
 (#583)

* Fix rowmapper generator, check for fields, map users #443

* Formatting

* Remove value store at the end of the import

* Update the notes for the importer about users

* Uncomment importers disabled during testing
---
 app/Services/UserService.php                  |   7 ++
 .../Resources/views/step1-configure.blade.php |   2 +
 modules/Importer/Services/BaseImporter.php    |  12 +-
 modules/Importer/Services/ImporterService.php |   8 +-
 .../Services/Importers/AircraftImporter.php   |   5 +-
 .../Services/Importers/AirlineImporter.php    |   3 +-
 .../Services/Importers/AirportImporter.php    |   3 +-
 .../Services/Importers/ClearDatabase.php      |  18 ++-
 .../Services/Importers/FinalizeImporter.php   |  10 ++
 .../Services/Importers/FlightImporter.php     |   5 +-
 .../Services/Importers/GroupImporter.php      |  11 +-
 .../Services/Importers/PirepImporter.php      |  11 +-
 .../Services/Importers/RankImport.php         |   4 +-
 .../Services/Importers/UserImport.php         |  35 +++++-
 modules/Importer/Utils/IdMapper.php           |   8 ++
 modules/Importer/Utils/ImporterDB.php         | 117 ++++++++++++++----
 resources/views/system/acars/config.blade.php |   4 +-
 17 files changed, 214 insertions(+), 49 deletions(-)

diff --git a/app/Services/UserService.php b/app/Services/UserService.php
index 9be4e8a4..42b955e2 100644
--- a/app/Services/UserService.php
+++ b/app/Services/UserService.php
@@ -135,6 +135,13 @@ class UserService extends Service
         return $user;
     }
 
+    /**
+     * Return true or false if a pilot ID already exists
+     *
+     * @param int $pilot_id
+     *
+     * @return bool
+     */
     public function isPilotIdAlreadyUsed(int $pilot_id): bool
     {
         return User::where('pilot_id', '=', $pilot_id)->exists();
diff --git a/modules/Importer/Resources/views/step1-configure.blade.php b/modules/Importer/Resources/views/step1-configure.blade.php
index 138df6f8..41f01ce3 100644
--- a/modules/Importer/Resources/views/step1-configure.blade.php
+++ b/modules/Importer/Resources/views/step1-configure.blade.php
@@ -10,6 +10,8 @@
         <td colspan="2">
           <h4>IMPORTANT NOTES</h4>
           <ul>
+            <li>The first user's password (admin) will be "admin". Please change it after logging in</li>
+            <li>User passwords will be reset and they will need to use "Forgot Password" to reset it</li>
             <li>If you have more than 1000 PIREPs or flights, it's best to use the command-line importer!
               <a href="http://docs.phpvms.net/setup/importing-from-v2-v5" target="_blank">Click here</a> to
               see the documentation of how to use it.
diff --git a/modules/Importer/Services/BaseImporter.php b/modules/Importer/Services/BaseImporter.php
index 6790af52..8d97e380 100644
--- a/modules/Importer/Services/BaseImporter.php
+++ b/modules/Importer/Services/BaseImporter.php
@@ -8,6 +8,7 @@ use Illuminate\Bus\Queueable;
 use Illuminate\Contracts\Queue\ShouldQueue;
 use Illuminate\Foundation\Bus\Dispatchable;
 use Illuminate\Queue\InteractsWithQueue;
+use Illuminate\Support\Facades\Log;
 use Modules\Importer\Utils\IdMapper;
 use Modules\Importer\Utils\ImporterDB;
 
@@ -27,7 +28,7 @@ abstract class BaseImporter implements ShouldQueue
     /**
      * The mapper class used for old IDs to new IDs
      *
-     * @var \Illuminate\Contracts\Foundation\Application|mixed
+     * @var IdMapper
      */
     protected $idMapper;
 
@@ -38,6 +39,13 @@ abstract class BaseImporter implements ShouldQueue
      */
     protected $table;
 
+    /**
+     * The column used for the ID, used for the ORDER BY
+     *
+     * @var string
+     */
+    protected $idField = 'id';
+
     public function __construct()
     {
         $importerService = app(ImporterService::class);
@@ -66,6 +74,8 @@ abstract class BaseImporter implements ShouldQueue
 
         $start = 0;
         $total_rows = $this->db->getTotalRows($this->table);
+        Log::info('Found '.$total_rows.' rows for '.$this->table);
+
         do {
             $end = $start + $this->db->batchSize;
             if ($end > $total_rows) {
diff --git a/modules/Importer/Services/ImporterService.php b/modules/Importer/Services/ImporterService.php
index 3f72ab21..f7e79371 100644
--- a/modules/Importer/Services/ImporterService.php
+++ b/modules/Importer/Services/ImporterService.php
@@ -6,6 +6,7 @@ use App\Contracts\Service;
 use App\Repositories\KvpRepository;
 use Exception;
 use Illuminate\Http\Request;
+use Illuminate\Support\Facades\Log;
 use Modules\Importer\Services\Importers\AircraftImporter;
 use Modules\Importer\Services\Importers\AirlineImporter;
 use Modules\Importer\Services\Importers\AirportImporter;
@@ -130,6 +131,11 @@ class ImporterService extends Service
 
         /** @var $importerInst \Modules\Importer\Services\BaseImporter */
         $importerInst = new $importer();
-        $importerInst->run($start);
+
+        try {
+            $importerInst->run($start);
+        } catch (Exception $e) {
+            Log::error('Error running importer: '.$e->getMessage());
+        }
     }
 }
diff --git a/modules/Importer/Services/Importers/AircraftImporter.php b/modules/Importer/Services/Importers/AircraftImporter.php
index 55e70574..a00ac7af 100644
--- a/modules/Importer/Services/Importers/AircraftImporter.php
+++ b/modules/Importer/Services/Importers/AircraftImporter.php
@@ -9,7 +9,7 @@ use Modules\Importer\Services\BaseImporter;
 
 class AircraftImporter extends BaseImporter
 {
-    public $table = 'aircraft';
+    protected $table = 'aircraft';
 
     /**
      * CONSTANTS
@@ -25,7 +25,8 @@ class AircraftImporter extends BaseImporter
         $this->info('Subfleet ID is '.$subfleet->id);
 
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start);
+        foreach ($rows as $row) {
             $where = [
                 'name'         => $row->fullname,
                 'registration' => $row->registration,
diff --git a/modules/Importer/Services/Importers/AirlineImporter.php b/modules/Importer/Services/Importers/AirlineImporter.php
index 939a6b3f..6b5bcc42 100644
--- a/modules/Importer/Services/Importers/AirlineImporter.php
+++ b/modules/Importer/Services/Importers/AirlineImporter.php
@@ -18,7 +18,8 @@ class AirlineImporter extends BaseImporter
         $this->comment('--- AIRLINE IMPORT ---');
 
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start);
+        foreach ($rows as $row) {
             $attrs = [
                 'iata'   => $row->code,
                 'name'   => $row->name,
diff --git a/modules/Importer/Services/Importers/AirportImporter.php b/modules/Importer/Services/Importers/AirportImporter.php
index f37183d5..e0f91880 100644
--- a/modules/Importer/Services/Importers/AirportImporter.php
+++ b/modules/Importer/Services/Importers/AirportImporter.php
@@ -25,7 +25,8 @@ class AirportImporter extends BaseImporter
         ];
 
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start, $fields) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start, $fields);
+        foreach ($rows as $row) {
             $attrs = [
                 'id'      => trim($row->icao),
                 'icao'    => trim($row->icao),
diff --git a/modules/Importer/Services/Importers/ClearDatabase.php b/modules/Importer/Services/Importers/ClearDatabase.php
index a4e01da6..e637e4ca 100644
--- a/modules/Importer/Services/Importers/ClearDatabase.php
+++ b/modules/Importer/Services/Importers/ClearDatabase.php
@@ -3,6 +3,7 @@
 namespace Modules\Importer\Services\Importers;
 
 use App\Models\Acars;
+use App\Models\Aircraft;
 use App\Models\Airline;
 use App\Models\Airport;
 use App\Models\Bid;
@@ -15,6 +16,7 @@ use App\Models\Journal;
 use App\Models\JournalTransaction;
 use App\Models\News;
 use App\Models\Pirep;
+use App\Models\Role;
 use App\Models\Subfleet;
 use App\Models\User;
 use App\Models\UserAward;
@@ -68,12 +70,7 @@ class ClearDatabase extends BaseImporter
         FlightFieldValue::truncate();
         Flight::truncate();
         Subfleet::truncate();
-
-        // Clear permissions
-        //        DB::table('permission_role')->truncate();
-        //        DB::table('permission_user')->truncate();
-        //        DB::table('role_user')->truncate();
-        //        Role::truncate();
+        Aircraft::truncate();
 
         Airline::truncate();
         Airport::truncate();
@@ -83,6 +80,15 @@ class ClearDatabase extends BaseImporter
         UserAward::truncate();
         User::truncate();
 
+        // Clear permissions
+        DB::table('permission_role')->truncate();
+        DB::table('permission_user')->truncate();
+        DB::table('role_user')->truncate();
+
+        // Role::truncate();
+
         DB::statement('SET FOREIGN_KEY_CHECKS=1');
+
+        $this->idMapper->clear();
     }
 }
diff --git a/modules/Importer/Services/Importers/FinalizeImporter.php b/modules/Importer/Services/Importers/FinalizeImporter.php
index b3b56b25..c711664b 100644
--- a/modules/Importer/Services/Importers/FinalizeImporter.php
+++ b/modules/Importer/Services/Importers/FinalizeImporter.php
@@ -34,6 +34,7 @@ class FinalizeImporter extends BaseImporter
     {
         $this->findLastPireps();
         $this->recalculateUserStats();
+        $this->clearValueStore();
     }
 
     /**
@@ -41,6 +42,7 @@ class FinalizeImporter extends BaseImporter
      */
     protected function findLastPireps()
     {
+        // TODO
     }
 
     /**
@@ -55,4 +57,12 @@ class FinalizeImporter extends BaseImporter
             $userSvc->recalculateStats($user);
         });
     }
+
+    /**
+     * Clear the value store of any old value mappings
+     */
+    protected function clearValueStore()
+    {
+        $this->idMapper->clear();
+    }
 }
diff --git a/modules/Importer/Services/Importers/FlightImporter.php b/modules/Importer/Services/Importers/FlightImporter.php
index 2e1a7be1..affb0d96 100644
--- a/modules/Importer/Services/Importers/FlightImporter.php
+++ b/modules/Importer/Services/Importers/FlightImporter.php
@@ -24,13 +24,14 @@ class FlightImporter extends BaseImporter
             'flightlevel',
             'deptime',
             'arrtime',
-            'flightttime',
+            'flighttime',
             'notes',
             'enabled',
         ];
 
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start, $fields) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start, $fields);
+        foreach ($rows as $row) {
             $airline_id = $this->idMapper->getMapping('airlines', $row->code);
 
             $flight_num = trim($row->flightnum);
diff --git a/modules/Importer/Services/Importers/GroupImporter.php b/modules/Importer/Services/Importers/GroupImporter.php
index 97aa9ebf..07e5673e 100644
--- a/modules/Importer/Services/Importers/GroupImporter.php
+++ b/modules/Importer/Services/Importers/GroupImporter.php
@@ -12,6 +12,7 @@ use Modules\Importer\Services\BaseImporter;
 class GroupImporter extends BaseImporter
 {
     protected $table = 'groups';
+    protected $idField = 'groupid';
 
     /**
      * Permissions in the legacy system, mapping them to the current system
@@ -68,7 +69,15 @@ class GroupImporter extends BaseImporter
         $roleSvc = app(RoleService::class);
 
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start);
+        foreach ($rows as $row) {
+            // Legacy "administrator" role is now "admin", just map that 1:1
+            if (strtolower($row->name) === 'administrators') {
+                $role = Role::where('name', 'admin')->first();
+                $this->idMapper->addMapping('group', $row->groupid, $role->id);
+                continue;
+            }
+
             $name = str_slug($row->name);
             $role = Role::firstOrCreate(
                 ['name' => $name],
diff --git a/modules/Importer/Services/Importers/PirepImporter.php b/modules/Importer/Services/Importers/PirepImporter.php
index b4486774..146beee0 100644
--- a/modules/Importer/Services/Importers/PirepImporter.php
+++ b/modules/Importer/Services/Importers/PirepImporter.php
@@ -11,6 +11,7 @@ use Modules\Importer\Services\BaseImporter;
 class PirepImporter extends BaseImporter
 {
     protected $table = 'pireps';
+    protected $idField = 'pirepid';
 
     public function run($start = 0)
     {
@@ -32,11 +33,17 @@ class PirepImporter extends BaseImporter
             'distance',
             'flighttime_stamp',
             'flighttype',
-            'flightlevel',
         ];
 
+        // See if there's a flightlevel column, export that if there is
+        $columns = $this->db->getColumns($this->table);
+        if (in_array('flightlevel', $columns)) {
+            $fields[] = 'flightlevel';
+        }
+
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start, $fields) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start, $fields);
+        foreach ($rows as $row) {
             $pirep_id = $row->pirepid;
             $user_id = $this->idMapper->getMapping('users', $row->pilotid);
             $airline_id = $this->idMapper->getMapping('airlines', $row->code);
diff --git a/modules/Importer/Services/Importers/RankImport.php b/modules/Importer/Services/Importers/RankImport.php
index 47e173ac..c90181a4 100644
--- a/modules/Importer/Services/Importers/RankImport.php
+++ b/modules/Importer/Services/Importers/RankImport.php
@@ -8,13 +8,15 @@ use Modules\Importer\Services\BaseImporter;
 class RankImport extends BaseImporter
 {
     protected $table = 'ranks';
+    protected $idField = 'rankid';
 
     public function run($start = 0)
     {
         $this->comment('--- RANK IMPORT ---');
 
         $count = 0;
-        foreach ($this->db->readRows($this->table, $start) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start);
+        foreach ($rows as $row) {
             $rank = Rank::firstOrCreate(['name' => $row->rank], [
                 'image_url' => $row->rankimage,
                 'hours'     => $row->minhours,
diff --git a/modules/Importer/Services/Importers/UserImport.php b/modules/Importer/Services/Importers/UserImport.php
index 187946c5..6e19fbe0 100644
--- a/modules/Importer/Services/Importers/UserImport.php
+++ b/modules/Importer/Services/Importers/UserImport.php
@@ -3,6 +3,7 @@
 namespace Modules\Importer\Services\Importers;
 
 use App\Models\Enums\UserState;
+use App\Models\Role;
 use App\Models\User;
 use App\Services\UserService;
 use App\Support\Units\Time;
@@ -15,6 +16,7 @@ use Modules\Importer\Services\BaseImporter;
 class UserImport extends BaseImporter
 {
     protected $table = 'pilots';
+    protected $idField = 'pilotid';
 
     /**
      * @var UserService
@@ -29,7 +31,8 @@ class UserImport extends BaseImporter
 
         $count = 0;
         $first_row = true;
-        foreach ($this->db->readRows($this->table, $start) as $row) {
+        $rows = $this->db->readRows($this->table, $this->idField, $start);
+        foreach ($rows as $row) {
             $pilot_id = $row->pilotid; // This isn't their actual ID
             $name = $row->firstname.' '.$row->lastname;
 
@@ -49,8 +52,13 @@ class UserImport extends BaseImporter
 
             // Look for a user with that pilot ID already. If someone has it
             if ($this->userSvc->isPilotIdAlreadyUsed($pilot_id)) {
-                Log::info('User with pilot id '.$pilot_id.' exists, reassigning');
-                $pilot_id = $this->userSvc->getNextAvailablePilotId();
+                Log::info('User with pilot id '.$pilot_id.' exists');
+
+                // Is this the same user? If not, get a new pilot ID
+                $user_exist = User::where('pilot_id', $pilot_id)->first();
+                if ($user_exist->email !== $row->email) {
+                    $pilot_id = $this->userSvc->getNextAvailablePilotId();
+                }
             }
 
             $attrs = [
@@ -91,9 +99,26 @@ class UserImport extends BaseImporter
     {
         // Be default add them to the user role, and then determine if they
         // belong to any other groups, and add them to that
-        $this->userSvc->addUserToRole($user, 'user');
+        $roleMappings = [];
+        $newRoles = [];
 
-        // Figure out what other groups they belong to
+        // Figure out what other groups they belong to... read from the old table, and map
+        // them to the new group(s)
+        $old_user_groups = $this->db->findBy('groupmembers', ['pilotid' => $old_pilot_id]);
+        foreach ($old_user_groups as $oldGroup) {
+            $newRoleId = $this->idMapper->getMapping('group', $oldGroup->groupid);
+
+            // Only lookup a new role ID if found
+            // if (!in_array($newRoleId, $roleMappings)) {
+            //     $roleMappings[$newRoleId] = Role::where(['id' => $newRoleId])->first();
+            // }
+
+            // $newRoles[] = $roleMappings[$newRoleId];
+            $newRoles[] = $newRoleId;
+        }
+
+        // Assign the groups to the new user
+        $user->attachRoles($newRoles);
     }
 
     /**
diff --git a/modules/Importer/Utils/IdMapper.php b/modules/Importer/Utils/IdMapper.php
index 14399b06..4a8e0278 100644
--- a/modules/Importer/Utils/IdMapper.php
+++ b/modules/Importer/Utils/IdMapper.php
@@ -46,4 +46,12 @@ class IdMapper extends Service
 
         return $this->valueStore->get($key_name);
     }
+
+    /**
+     * Clear the value store
+     */
+    public function clear()
+    {
+        $this->valueStore->flush();
+    }
 }
diff --git a/modules/Importer/Utils/ImporterDB.php b/modules/Importer/Utils/ImporterDB.php
index 60ce6010..3716966c 100644
--- a/modules/Importer/Utils/ImporterDB.php
+++ b/modules/Importer/Utils/ImporterDB.php
@@ -85,6 +85,28 @@ class ImporterDB
         return $table;
     }
 
+    /**
+     * Get the names of the columns for a particular table
+     *
+     * @param $table
+     *
+     * @return mixed
+     */
+    public function getColumns($table)
+    {
+        $this->connect();
+
+        $sql = 'SHOW COLUMNS FROM '.$this->tableName($table);
+        $result = $this->conn->query($sql)->fetchAll();
+
+        $rows = [];
+        foreach ($result as $row) {
+            $rows[] = $row->Field;
+        }
+
+        return $rows;
+    }
+
     /**
      * @param $table
      *
@@ -102,50 +124,95 @@ class ImporterDB
         return (int) $rows;
     }
 
+    /**
+     * Read rows from a table with a given assoc array. Simple
+     *
+     * @param string $table
+     * @param array  $attrs
+     *
+     * @return false|\PDOStatement
+     */
+    public function findBy($table, array $attrs)
+    {
+        $this->connect();
+
+        $where = [];
+        foreach ($attrs as $col => $value) {
+            $where[] = $col.'=\''.$value.'\'';
+        }
+
+        $where = implode(' AND ', $where);
+
+        $sql = implode(' ', [
+            'SELECT',
+            '*',
+            'FROM',
+            $this->tableName($table),
+            'WHERE',
+            $where,
+        ]);
+
+        return $this->conn->query($sql);
+    }
+
     /**
      * Read all the rows in a table, but read them in a batched manner
      *
-     * @param string $table The name of the table
-     * @param int    [$start_offset]
-     * @param string [$fields]
+     * @param string $table        The name of the table
+     * @param string $order_by     Column to order by
+     * @param int    $start_offset
+     * @param string $fields
      *
-     * @return \Generator
+     * @return array
      */
-    public function readRows($table, $start_offset = 0, $fields = '*')
+    public function readRows($table, $order_by = 'id', $start_offset = 0, $fields = '*')
     {
         $this->connect();
 
         $offset = $start_offset;
-        $total_rows = $this->getTotalRows($table);
+        // $total_rows = $this->getTotalRows($table);
 
-        while ($offset < $total_rows) {
-            $rows_to_read = $offset + $this->batchSize;
-            if ($rows_to_read > $total_rows) {
-                $rows_to_read = $total_rows;
-            }
-
-            // Log::info('Reading '.$offset.' to '.$rows_to_read.' of '.$total_rows);
-            yield from $this->readRowsOffset($table, $this->batchSize, $offset, $fields);
-
-            $offset += $this->batchSize;
+        $rows = [];
+        $result = $this->readRowsOffset($table, $this->batchSize, $offset, $order_by, $fields);
+        if ($result === false) {
+            return [];
         }
+
+        try {
+            foreach ($result as $row) {
+                $rows[] = $row;
+            }
+        } catch (Exception $e) {
+            Log::error('foreach rows error: '.$e->getMessage());
+        }
+
+        return $rows;
     }
 
     /**
      * @param string $table
-     * @param int    $limit  Number of rows to read
-     * @param int    $offset Where to start from
-     * @param string [$fields]
+     * @param int    $limit    Number of rows to read
+     * @param int    $offset   Where to start from
+     * @param        $order_by
+     * @param string $fields
      *
-     * @return \Generator
+     * @return false|\PDOStatement|void
      */
-    public function readRowsOffset($table, $limit, $offset, $fields = '*')
+    public function readRowsOffset($table, $limit, $offset, $order_by, $fields = '*')
     {
         if (is_array($fields)) {
             $fields = implode(',', $fields);
         }
 
-        $sql = 'SELECT '.$fields.' FROM '.$this->tableName($table).' LIMIT '.$limit.' OFFSET '.$offset;
+        $sql = implode(' ', [
+            'SELECT',
+            $fields,
+            'FROM',
+            $this->tableName($table),
+            'ORDER BY '.$order_by.' ASC',
+            'LIMIT '.$limit,
+            'OFFSET '.$offset,
+        ]);
 
         try {
             $result = $this->conn->query($sql);
@@ -153,9 +220,7 @@ class ImporterDB
                 return;
             }
 
-            foreach ($result as $row) {
-                yield $row;
-            }
+            return $result;
         } catch (PDOException $e) {
             // Without incrementing the offset, it should re-run the same query
             Log::error('Error readRowsOffset: '.$e->getMessage());
@@ -163,6 +228,8 @@ class ImporterDB
             if (strpos($e->getMessage(), 'server has gone away') !== false) {
                 $this->connect();
             }
+        } catch (\Exception $e) {
+            Log::error('Error readRowsOffset: '.$e->getMessage());
         }
     }
 }
diff --git a/resources/views/system/acars/config.blade.php b/resources/views/system/acars/config.blade.php
index df925f87..36534251 100644
--- a/resources/views/system/acars/config.blade.php
+++ b/resources/views/system/acars/config.blade.php
@@ -1,4 +1,6 @@
-<?xml version = "1.0" encoding = "utf-8"?>
+@php
+echo '<?xml version = "1.0" encoding = "utf-8"?>';
+@endphp
 <AcarsConfiguration xmlns:i="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://schemas.phpvms.net/acars/config">
   <AirlineUrl>{{ config('app.url') }}</AirlineUrl>
   <ApiKey>{{ $user->api_key }}</ApiKey>