Improve step count for spectre imports.

app/Http/Controllers/Import/StatusController.php

@@ -113,6 +113,7 @@ class StatusController extends Controller
             $result['started'] = true;
             $result['running'] = true;
         }
+        $result['percentage'] = $result['percentage'] > 100 ? 100 : $result['percentage'];
 
         return Response::json($result);
     }

app/Import/Configuration/SpectreConfigurator.php

@@ -216,7 +216,7 @@ class SpectreConfigurator implements ConfiguratorInterface
 
         // set default extended status:
         $extendedStatus          = $this->repository->getExtendedStatus($job);
-        $extendedStatus['steps'] = 100;
+        $extendedStatus['steps'] = 6;
 
         // save to job:
         $job       = $this->repository->setConfiguration($job, $finalConfig);

app/Import/Routine/SpectreRoutine.php

@@ -191,6 +191,7 @@ class SpectreRoutine implements RoutineInterface
             }
         }
 
+
         Preferences::setForUser($this->job->user, 'spectre_customer', $customer->toArray());
 
         return $customer;
@@ -254,17 +255,11 @@ class SpectreRoutine implements RoutineInterface
         $customer = $this->getCustomer();
         Log::debug(sprintf('Customer ID is %s', $customer->getId()));
 
-        // add some steps done
-        $this->repository->addStepsDone($this->job, 2);
-
         // use customer to request a token:
         $uri   = route('import.status', [$this->job->key]);
         $token = $this->getToken($customer, $uri);
         Log::debug(sprintf('Token is %s', $token->getToken()));
 
-        // add some steps done
-        $this->repository->addStepsDone($this->job, 2);
-
         // update job, give it the token:
         $config                  = $this->getConfig();
         $config['has-token']     = true;
@@ -279,6 +274,7 @@ class SpectreRoutine implements RoutineInterface
         // update job, set status to "configuring".
         $this->setStatus('configuring');
         Log::debug(sprintf('Job status is now %s', $this->job->status));
+        $this->addStep();
     }
 
     /**
@@ -294,9 +290,6 @@ class SpectreRoutine implements RoutineInterface
         $request->setCustomer($customer);
         $request->call();
 
-        // add some steps done
-        $this->repository->addStepsDone($this->job, 2);
-
         $logins = $request->getLogins();
         /** @var Login $final */
         $final = null;
@@ -314,15 +307,11 @@ class SpectreRoutine implements RoutineInterface
         if (is_null($final)) {
             Log::error('Could not find a valid login for this user.');
             $this->repository->addError($this->job, 0, 'Spectre connection failed. Did you use invalid credentials, press Cancel or failed the 2FA challenge?');
-            $this->repository->setTotalSteps($this->job, 1);
-            $this->repository->setStepsDone($this->job, 1);
             $this->repository->setStatus($this->job, 'error');
 
             return;
         }
-
-        // add some steps done
-        $this->repository->addStepsDone($this->job, 2);
+        $this->addStep();
 
         // list the users accounts using this login.
         $accountRequest = new ListAccountsRequest($this->job->user);
@@ -330,9 +319,6 @@ class SpectreRoutine implements RoutineInterface
         $accountRequest->call();
         $accounts = $accountRequest->getAccounts();
 
-        // add some steps done
-        $this->repository->addStepsDone($this->job, 2);
-
         // store accounts in job:
         $all = [];
         /** @var Account $account */
@@ -348,13 +334,29 @@ class SpectreRoutine implements RoutineInterface
 
         $this->setConfig($config);
         $this->setStatus('configuring');
-
-        // add some steps done
-        $this->repository->addStepsDone($this->job, 2);
+        $this->addStep();
 
         return;
     }
 
+    /**
+     * Shorthand method.
+     */
+    private function addStep()
+    {
+        $this->repository->addStepsDone($this->job, 1);
+    }
+
+    /**
+     * Shorthand
+     *
+     * @param int $steps
+     */
+    private function addTotalSteps(int $steps)
+    {
+        $this->repository->addTotalSteps($this->job, $steps);
+    }
+
     /**
      * @return array
      */
@@ -364,6 +366,8 @@ class SpectreRoutine implements RoutineInterface
     }
 
     /**
+     * Shorthand method.
+     *
      * @return array
      */
     private function getExtendedStatus(): array
@@ -462,8 +466,11 @@ class SpectreRoutine implements RoutineInterface
                 $collection->push($importJournal);
             }
         }
+        $this->addStep();
         Log::debug(sprintf('Going to try and store all %d them.', $collection->count()));
-        // try to store them:
+
+        $this->addTotalSteps(7 * $collection->count());
+        // try to store them (seven steps per transaction)
         $storage = new ImportStorage;
 
         $storage->setJob($this->job);
@@ -494,15 +501,19 @@ class SpectreRoutine implements RoutineInterface
         Log::debug('Looping journals...');
         $journalIds = $storage->journals->pluck('id')->toArray();
         $tagId      = $tag->id;
+        $this->addTotalSteps(count($journalIds));
+
         foreach ($journalIds as $journalId) {
             Log::debug(sprintf('Linking journal #%d to tag #%d...', $journalId, $tagId));
             DB::table('tag_transaction_journal')->insert(['transaction_journal_id' => $journalId, 'tag_id' => $tagId]);
+            $this->addStep();
         }
         Log::info(sprintf('Linked %d journals to tag #%d ("%s")', $storage->journals->count(), $tag->id, $tag->tag));
 
         // set status to "finished"?
         // update job:
         $this->setStatus('finished');
+        $this->addStep();
 
         return;
     }
@@ -537,14 +548,9 @@ class SpectreRoutine implements RoutineInterface
                 'transactions' => $transactions,
             ];
             $count                  += count($transactions);
-
-            // add some steps done
-            $this->repository->addStepsDone($this->job, 2);
         }
-        // update number of steps:
-        $this->repository->setTotalSteps($this->job, $count * 5);
-        $this->repository->setStepsDone($this->job, 1);
         Log::debug(sprintf('Total number of transactions: %d', $count));
+        $this->addStep();
 
 
         $this->importTransactions($all);
@@ -563,6 +569,8 @@ class SpectreRoutine implements RoutineInterface
     }
 
     /**
+     * Shorthand method.
+     *
      * @param array $extended
      */
     private function setExtendedStatus(array $extended): void