fixing user specific rates in kimai v1 importer (#1521)

phpstan.neon

@@ -21,5 +21,4 @@ parameters:
         - '#Access to an undefined property Faker\\Generator::\$bs.#'
         - '#Method Symfony\\Contracts\\EventDispatcher\\EventDispatcherInterface::dispatch\(\) invoked with 2 parameters, 1 required.#'
     excludes_analyse:
-        - %rootDir%/../../../src/Command/KimaiImporterCommand.php
         - %rootDir%/../../../src/Ldap/LdapDriver.php

src/Command/KimaiImporterCommand.php

@@ -12,10 +12,12 @@ namespace App\Command;
 use App\Doctrine\TimesheetSubscriber;
 use App\Entity\Activity;
 use App\Entity\ActivityMeta;
+use App\Entity\ActivityRate;
 use App\Entity\Customer;
 use App\Entity\CustomerMeta;
 use App\Entity\Project;
 use App\Entity\ProjectMeta;
+use App\Entity\ProjectRate;
 use App\Entity\Timesheet;
 use App\Entity\User;
 use App\Entity\UserPreference;
@@ -58,52 +60,54 @@ final class KimaiImporterCommand extends Command
      * Create the user default passwords
      * @var UserPasswordEncoderInterface
      */
-    protected $encoder;
+    private $encoder;
     /**
      * Validates the entities before they will be created
      * @var ValidatorInterface
      */
-    protected $validator;
+    private $validator;
     /**
      * Connection to the Kimai v2 database to write imported data to
      * @var ManagerRegistry
      */
-    protected $doctrine;
+    private $doctrine;
     /**
      * Connection to the old database to import data from
      * @var Connection
      */
-    protected $connection;
+    private $connection;
     /**
      * Prefix for the v1 database tables.
      * @var string
      */
-    protected $dbPrefix = '';
+    private $dbPrefix = '';
     /**
+     * Old UserID => new User()
      * @var User[]
      */
-    protected $users = [];
+    private $users = [];
     /**
      * @var Customer[]
      */
-    protected $customers = [];
+    private $customers = [];
     /**
+     * Old Project ID => new Project()
      * @var Project[]
      */
-    protected $projects = [];
+    private $projects = [];
     /**
      * id => [projectId => Activity]
-     * @var Activity[]
+     * @var array<Activity[]>
      */
-    protected $activities = [];
+    private $activities = [];
     /**
      * @var bool
      */
-    protected $debug = false;
+    private $debug = false;
     /**
      * @var array
      */
-    protected $oldActivities = [];
+    private $oldActivities = [];
 
     public function __construct(UserPasswordEncoderInterface $encoder, ManagerRegistry $registry, ValidatorInterface $validator)
     {
@@ -710,31 +714,12 @@ final class KimaiImporterCommand extends Command
 
             $project->setMetaField($metaField);
 
-            foreach ($fixedRates as $fixedRow) {
-                if ($fixedRow['activityID'] !== null || $fixedRow['projectID'] === null) {
-                    continue;
-                }
-                if ($fixedRow['projectID'] == $oldProject['projectID']) {
-                    $project->setFixedRate($fixedRow['rate']);
-                }
-            }
-
-            foreach ($rates as $ratesRow) {
-                if ($ratesRow['userID'] !== null || $ratesRow['activityID'] !== null || $ratesRow['projectID'] === null) {
-                    continue;
-                }
-                if ($ratesRow['projectID'] == $oldProject['projectID']) {
-                    $project->setHourlyRate($ratesRow['rate']);
-                }
-            }
-
             if (!$this->validateImport($io, $project)) {
                 throw new Exception('Failed to validate project: ' . $project->getName());
             }
 
             try {
                 $entityManager->persist($project);
-                $entityManager->flush();
                 if ($this->debug) {
                     $io->success('Created project: ' . $project->getName() . ' for customer: ' . $customer->getName());
                 }
@@ -744,6 +729,54 @@ final class KimaiImporterCommand extends Command
                 $io->error('Reason: ' . $ex->getMessage());
             }
 
+            foreach ($fixedRates as $fixedRow) {
+                // activity rates a re assigned in createActivity()
+                if ($fixedRow['activityID'] !== null || $fixedRow['projectID'] === null) {
+                    continue;
+                }
+                if ($fixedRow['projectID'] == $oldProject['projectID']) {
+                    $projectRate = new ProjectRate();
+                    $projectRate->setProject($project);
+                    $projectRate->setRate($fixedRow['rate']);
+                    $projectRate->setIsFixed(true);
+
+                    try {
+                        $entityManager->persist($projectRate);
+                        if ($this->debug) {
+                            $io->success('Created fixed project rate: ' . $project->getName() . ' for customer: ' . $customer->getName());
+                        }
+                    } catch (Exception $ex) {
+                        $io->error(sprintf('Failed to create fixed project rate for %s: %s' . $project->getName(), $ex->getMessage()));
+                    }
+                }
+            }
+
+            foreach ($rates as $ratesRow) {
+                if ($ratesRow['activityID'] !== null || $ratesRow['projectID'] === null) {
+                    continue;
+                }
+                if ($ratesRow['projectID'] == $oldProject['projectID']) {
+                    $projectRate = new ProjectRate();
+                    $projectRate->setProject($project);
+                    $projectRate->setRate($ratesRow['rate']);
+
+                    if ($ratesRow['userID'] !== null) {
+                        $projectRate->setUser($this->users[$ratesRow['userID']]);
+                    }
+
+                    try {
+                        $entityManager->persist($projectRate);
+                        if ($this->debug) {
+                            $io->success('Created project rate: ' . $project->getName() . ' for customer: ' . $customer->getName());
+                        }
+                    } catch (Exception $ex) {
+                        $io->error(sprintf('Failed to create project rate for %s: %s' . $project->getName(), $ex->getMessage()));
+                    }
+                }
+            }
+
+            $entityManager->flush();
+
             $this->projects[$oldProject['projectID']] = $project;
         }
 
@@ -826,7 +859,7 @@ final class KimaiImporterCommand extends Command
      * @param array $oldActivity
      * @param array $fixedRates
      * @param array $rates
-     * @param int $projectId
+     * @param int $oldProjectId
      * @return Activity
      * @throws Exception
      */
@@ -836,12 +869,12 @@ final class KimaiImporterCommand extends Command
         array $oldActivity,
         array $fixedRates,
         array $rates,
-        $projectId
+        $oldProjectId
     ) {
-        $activityId = $oldActivity['activityID'];
+        $oldActivityId = $oldActivity['activityID'];
 
-        if (isset($this->activities[$activityId][$projectId])) {
-            return $this->activities[$activityId][$projectId];
+        if (isset($this->activities[$oldActivityId][$oldProjectId])) {
+            return $this->activities[$oldActivityId][$oldProjectId];
         }
 
         $isActive = (bool) $oldActivity['visible'] && !(bool) $oldActivity['trash'];
@@ -851,9 +884,9 @@ final class KimaiImporterCommand extends Command
             $io->warning('Found empty activity name, setting it to: ' . $name);
         }
 
-        if (null !== $projectId && !isset($this->projects[$projectId])) {
+        if (null !== $oldProjectId && !isset($this->projects[$oldProjectId])) {
             throw new Exception(
-                sprintf('Did not find project [%s], skipping activity creation [%s] %s', $projectId, $activityId, $name)
+                sprintf('Did not find project [%s], skipping activity creation [%s] %s', $oldProjectId, $oldActivityId, $name)
             );
         }
 
@@ -865,8 +898,8 @@ final class KimaiImporterCommand extends Command
             ->setBudget($oldActivity['budget'] ?? 0)
         ;
 
-        if (null !== $projectId) {
-            $project = $this->projects[$projectId];
+        if (null !== $oldProjectId) {
+            $project = $this->projects[$oldProjectId];
             $activity->setProject($project);
         }
 
@@ -877,39 +910,12 @@ final class KimaiImporterCommand extends Command
 
         $activity->setMetaField($metaField);
 
-        foreach ($fixedRates as $fixedRow) {
-            if ($fixedRow['activityID'] === null) {
-                continue;
-            }
-            if ($fixedRow['projectID'] !== null && $fixedRow['projectID'] !== $projectId) {
-                continue;
-            }
-
-            if ($fixedRow['activityID'] == $oldActivity['activityID']) {
-                $activity->setFixedRate($fixedRow['rate']);
-            }
-        }
-
-        foreach ($rates as $ratesRow) {
-            if ($ratesRow['userID'] !== null || $ratesRow['activityID'] === null) {
-                continue;
-            }
-            if ($ratesRow['projectID'] !== null && $ratesRow['projectID'] !== $projectId) {
-                continue;
-            }
-
-            if ($ratesRow['activityID'] == $oldActivity['activityID']) {
-                $activity->setHourlyRate($ratesRow['rate']);
-            }
-        }
-
         if (!$this->validateImport($io, $activity)) {
             throw new Exception('Failed to validate activity: ' . $activity->getName());
         }
 
         try {
             $entityManager->persist($activity);
-            $entityManager->flush();
             if ($this->debug) {
                 $io->success('Created activity: ' . $activity->getName());
             }
@@ -918,10 +924,65 @@ final class KimaiImporterCommand extends Command
             $io->error('Reason: ' . $ex->getMessage());
         }
 
-        if (!isset($this->activities[$activityId])) {
-            $this->activities[$activityId] = [];
+        if (!isset($this->activities[$oldActivityId])) {
+            $this->activities[$oldActivityId] = [];
         }
-        $this->activities[$activityId][$projectId] = $activity;
+        $this->activities[$oldActivityId][$oldProjectId] = $activity;
+
+        foreach ($fixedRates as $fixedRow) {
+            if ($fixedRow['activityID'] === null) {
+                continue;
+            }
+            if ($fixedRow['projectID'] !== null && $fixedRow['projectID'] !== $oldProjectId) {
+                continue;
+            }
+
+            if ($fixedRow['activityID'] == $oldActivityId) {
+                $activityRate = new ActivityRate();
+                $activityRate->setActivity($activity);
+                $activityRate->setRate($fixedRow['rate']);
+                $activityRate->setIsFixed(true);
+
+                try {
+                    $entityManager->persist($activityRate);
+                    if ($this->debug) {
+                        $io->success('Created fixed activity rate: ' . $activity->getName());
+                    }
+                } catch (Exception $ex) {
+                    $io->error(sprintf('Failed to create fixed activity rate for %s: %s' . $activity->getName(), $ex->getMessage()));
+                }
+            }
+        }
+
+        foreach ($rates as $ratesRow) {
+            if ($ratesRow['activityID'] === null) {
+                continue;
+            }
+            if ($ratesRow['projectID'] !== null && $ratesRow['projectID'] !== $oldProjectId) {
+                continue;
+            }
+
+            if ($ratesRow['activityID'] == $oldActivityId) {
+                $activityRate = new ActivityRate();
+                $activityRate->setActivity($activity);
+                $activityRate->setRate($ratesRow['rate']);
+
+                if ($ratesRow['userID'] !== null) {
+                    $activityRate->setUser($this->users[$ratesRow['userID']]);
+                }
+
+                try {
+                    $entityManager->persist($activityRate);
+                    if ($this->debug) {
+                        $io->success('Created activity rate: ' . $activity->getName());
+                    }
+                } catch (Exception $ex) {
+                    $io->error(sprintf('Failed to create activity rate for %s: %s' . $activity->getName(), $ex->getMessage()));
+                }
+            }
+        }
+
+        $entityManager->flush();
 
         return $activity;
     }