From 0a56dbc26970991f73987fd8434cde00dcb5ad13 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Kr=C3=A4mer?= <florian.kraemer@cakedc.com>
Date: Fri, 12 Aug 2016 11:44:18 +0200
Subject: [PATCH] Fixing the shell.

---
 src/Shell/PurifierShell.php | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/src/Shell/PurifierShell.php b/src/Shell/PurifierShell.php
index 68b8d26..c576240 100644
--- a/src/Shell/PurifierShell.php
+++ b/src/Shell/PurifierShell.php
@@ -35,9 +35,10 @@ protected function _getTable()
         $table = TableRegistry::get($this->args[0]);
         $connection = $table->connection();
         $tables = $connection->schemaCollection()->listTables();
-        if (!in_array($this->args[0], $tables)) {
-            $this->abort(__d('Burzum/HtmlPurifier', 'Table `{0}` does not exist in connection `{1}`!', $this->args[0], $connection->configName()));
+        if (!in_array($table->table(), $tables)) {
+            $this->abort(__d('Burzum/HtmlPurifier', 'Table `{0}` does not exist in connection `{1}`!', $table->table(), $connection->configName()));
         }
+        return $table;
     }
 
     /**
@@ -54,6 +55,7 @@ protected function _getFields(Table $table)
                 $this->abort(sprintf('Table `%s` is missing the field `%s`.', $table->table(), $field));
             }
         }
+
         return $fields;
     }
 
@@ -95,11 +97,11 @@ public function purify()
 
         $this->helper('progress')->output([
             'total' => $total,
-            'callback' => function ($progress) use ($total, $table) {
+            'callback' => function ($progress) use ($total, $table, $fields) {
                 $chunkSize = 25;
                 $chunkCount = 0;
                 while ($chunkCount <= $total) {
-                    $this->_process($table, $chunkCount, $chunkSize);
+                    $this->_process($table, $chunkCount, $chunkSize, $fields);
                     $chunkCount = $chunkCount + $chunkSize;
                     $progress->increment($chunkSize);
                     $progress->draw();
@@ -117,14 +119,12 @@ public function purify()
      * @param int $chunkSize
      * @return void
      */
-    protected function _process(Table $table, $chunkCount, $chunkSize)
-    {
+    protected function _process(Table $table, $chunkCount, $chunkSize, $fields) {
         $query = $table->find();
         if ($table->hasFinder('purifier')) {
             $query->find('purifier');
         }
 
-        $fields = explode(',', $this->param('fields'));
         $fields[] = $table->primaryKey();
 
         $results = $query
@@ -143,7 +143,7 @@ protected function _process(Table $table, $chunkCount, $chunkSize)
                 $table->save($result);
                 $chunkCount++;
             } catch (\Exception $e) {
-                $this->error($e->getMessage());
+                $this->abort($e->getMessage());
             }
         }
     }
@@ -151,8 +151,7 @@ protected function _process(Table $table, $chunkCount, $chunkSize)
     /**
      * {@inheritDoc}
      */
-    public function getOptionParser()
-    {
+    public function getOptionParser() {
         $parser = parent::getOptionParser();
 
         $parser->description([