rollback include datatable file

WeLabs · Sep 19, 2016 · db10204 · db10204
1 parent 24ed332
commit db10204
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/Generators/Scaffold/ControllerGenerator.php b/src/Generators/Scaffold/ControllerGenerator.php
@@ -86,7 +86,7 @@ private function generateDataTable()
 
         FileUtil::createFile($path, $fileName, $templateData);
 
-        $this->commandData->commandComment("\n$fileName created: ");
+        $this->commandData->commandComment("\nDataTable created: ");
         $this->commandData->commandInfo($fileName);
     }
 
@@ -95,5 +95,11 @@ public function rollback()
         if ($this->rollbackFile($this->path, $this->fileName)) {
             $this->commandData->commandComment('Controller file deleted: '.$this->fileName);
         }
+
+        if ($this->commandData->getAddOn('datatables')) {
+            if ($this->rollbackFile($this->commandData->config->pathDataTables, $this->commandData->modelName.'DataTable.php')) {
+                $this->commandData->commandComment('DataTable file deleted: '.$this->fileName);
+            }
+        }
     }
 }