API Update API to reflect changes to CLI interaction

silverstripe · Sep 4, 2024 · ddc910c · ddc910c
1 parent d9c5f8c
commit ddc910c
Show file tree

Hide file tree

Showing 5 changed files with 128 additions and 101 deletions.
diff --git a/_config/extensions.yml b/_config/extensions.yml
@@ -28,17 +28,17 @@ SilverStripe\FrameworkTest\Fields\NestedGridField\NonRelationalData:
     extensions:
     - SilverStripe\FrameworkTest\Extension\TestDataObjectExtension
 
-SilverStripe\ORM\DatabaseAdmin:
+SilverStripe\Dev\Command\DbBuild:
   extensions:
-    - SilverStripe\FrameworkTest\GridFieldArbitraryData\DatabaseBuildExtension
+    - SilverStripe\FrameworkTest\GridFieldArbitraryData\DbBuildExtension
 
 ---
 Only:
   moduleexists: 'silverstripe/testsession'
 ---
 SilverStripe\TestSession\TestSessionEnvironment:
   extensions:
-    - SilverStripe\FrameworkTest\GridFieldArbitraryData\DatabaseBuildExtension
+    - SilverStripe\FrameworkTest\GridFieldArbitraryData\DbBuildExtension
 
 ---
 Only:

diff --git a/...dArbitraryData/DatabaseBuildExtension.php → ...idFieldArbitraryData/DbBuildExtension.php b/...dArbitraryData/DatabaseBuildExtension.php → ...idFieldArbitraryData/DbBuildExtension.php
@@ -4,7 +4,8 @@
 
 use SilverStripe\Control\Director;
 use SilverStripe\Core\Extension;
-use SilverStripe\ORM\DatabaseAdmin;
+use SilverStripe\Dev\Command\DbBuild;
+use SilverStripe\HybridExecution\HybridOutput;
 use SilverStripe\ORM\DataObject;
 use SilverStripe\ORM\DB;
 use SilverStripe\ORM\FieldType\DBDatetime;
@@ -13,7 +14,7 @@
 /**
  * Builds the table and adds default records for the ArbitraryDataModel.
  */
-class DatabaseBuildExtension extends Extension
+class DbBuildExtension extends Extension
 {
     /**
      * This extension hook is on TestSessionEnvironment, which is used by behat but not by phpunit.
@@ -22,78 +23,57 @@ class DatabaseBuildExtension extends Extension
      */
     protected function onAfterStartTestSession()
     {
-        $this->buildTable(true);
+        $output = HybridOutput::create(
+            Director::is_cli() ? HybridOutput::FORMAT_ANSI : HybridOutput::FORMAT_HTML,
+            HybridOutput::VERBOSITY_QUIET
+        );
+        $output->startList();
+        $this->buildTable($output);
+        $output->stopList();
         $this->populateData();
     }
 
     /**
-     * This extension hook is on DatabaseAdmin, after dev/build has finished building the database.
+     * This extension hook is on DbBuild, after dev/build has finished building the database.
      */
-    protected function onAfterBuild(bool $quiet, bool $populate, bool $testMode): void
+    protected function onAfterBuild(HybridOutput $output, bool $populate, bool $testMode): void
     {
         if ($testMode) {
             return;
         }
 
-        if (!$quiet) {
-            if (Director::is_cli()) {
-                echo "\nCREATING TABLE FOR FRAMEWORKTEST ARBITRARY DATA\n\n";
-            } else {
-                echo "\n<p><b>Creating table for frameworktest arbitrary data</b></p><ul>\n\n";
-            }
-        }
-
-        $this->buildTable($quiet);
-
-        if (!$quiet && !Director::is_cli()) {
-            echo '</ul>';
-        }
-
+        $output->writeln('<options=bold>Creating table for frameworktest arbitrary data</>');
+        $output->startList();
+        $this->buildTable($output);
+        $output->stopList();
         if ($populate) {
-            if (!$quiet) {
-                if (Director::is_cli()) {
-                    echo "\nCREATING DATABASE RECORDS FOR FRAMEWORKTEST ARBITRARY DATA\n\n";
-                } else {
-                    echo "\n<p><b>Creating database records arbitrary data</b></p><ul>\n\n";
-                }
-            }
-
+            $output->writeln('<options=bold>Creating database records arbitrary data</>');
+            $output->startList();
             $this->populateData();
-
-            if (!$quiet && !Director::is_cli()) {
-                echo '</ul>';
-            }
-        }
-
-        if (!$quiet) {
-            echo (Director::is_cli()) ? "\n Frameworktest database build completed!\n\n" : '<p>Frameworktest database build completed!</p>';
+            $output->stopList();
         }
+        $output->writeln(['<options=bold>Frameworktest database build completed!</>', '']);
     }
 
-    private function buildTable(bool $quiet): void
+    private function buildTable(HybridOutput $output): void
     {
         $tableName = ArbitraryDataModel::TABLE_NAME;
 
         // Log data
-        if (!$quiet) {
-            $showRecordCounts = DatabaseAdmin::config()->get('show_record_counts');
-            if ($showRecordCounts && DB::get_schema()->hasTable($tableName)) {
-                try {
-                    $count = SQLSelect::create()->setFrom($tableName)->count();
-                    $countSuffix = " ($count records)";
-                } catch (\Exception $e) {
-                    $countSuffix = ' (error getting record count)';
-                }
-            } else {
-                $countSuffix = "";
-            }
-
-            if (Director::is_cli()) {
-                echo " * $tableName$countSuffix\n";
-            } else {
-                echo "<li>$tableName$countSuffix</li>\n";
+        $showRecordCounts = DbBuild::config()->get('show_record_counts');
+        if ($showRecordCounts && DB::get_schema()->hasTable($tableName)) {
+            try {
+                $count = SQLSelect::create()->setFrom($tableName)->count();
+                $countSuffix = " ($count records)";
+            } catch (\Exception $e) {
+                $countSuffix = ' (error getting record count)';
             }
+        } else {
+            $countSuffix = "";
         }
+        // We're adding one list item, but we need to do it this way for consistency
+        // with the rest of dev/build.
+        $output->writeListItem($tableName . $countSuffix);
 
         // Get field schema
         $fields = [

diff --git a/code/tasks/FTFileMakerTask.php b/code/tasks/FTFileMakerTask.php
@@ -11,6 +11,10 @@
 use SilverStripe\Security\Security;
 use SilverStripe\Core\Path;
 use SilverStripe\Core\Manifest\ModuleResourceLoader;
+use SilverStripe\HybridExecution\HybridOutput;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
 
 /**
  * Creates sample folder and file structure, useful to test performance,
@@ -24,10 +28,6 @@
  * recursively delete any generated ones through the following bash command in `assets/`:
  * `find . -name '*Resampled*' -print0 | xargs -0 rm`
  *
- * Parameters:
- *  - reset=1: Optionally truncate ALL files and folders in the database, plus delete
- *    the entire `assets/` directory.
- *
  *
  * Configuration
  *
@@ -180,8 +180,6 @@ class FTFileMakerTask extends BuildTask
         'video.m4v' => 'SilverStripe\Assets\File',
     ];
 
-    protected $lineBreak = "\n<br>";
-
     /** @var Member */
     protected $anonymousMember = null;
 
@@ -197,7 +195,7 @@ class FTFileMakerTask extends BuildTask
      */
     protected $folderCounts = [];
 
-    public function run($request)
+    protected function execute(InputInterface $input, HybridOutput $output): int
     {
         set_time_limit(0);
 
@@ -213,15 +211,11 @@ public function run($request)
         }
         Security::setCurrentUser($this->anonymousMember);
 
-        if (php_sapi_name() == "cli") {
-            $this->lineBreak = "\n";
-        }
-
-        if ($request->getVar('reset')) {
-            $this->reset();
+        if ($input->getOption('reset')) {
+            $this->reset($output);
         }
 
-        $fileCounts = $request->getVar('fileCounts');
+        $fileCounts = $input->getOption('fileCounts');
         if ($fileCounts) {
             $counts = explode(',', $fileCounts ?? '');
             $this->fileCounts = array_map(function ($int) {
@@ -231,7 +225,7 @@ public function run($request)
             $this->fileCounts = FTFileMakerTask::config()->get('fileCountByDepth');
         }
 
-        $folderCounts = $request->getVar('folderCounts');
+        $folderCounts = $input->getOption('folderCounts');
         if ($folderCounts) {
             $counts = explode(',', $folderCounts ?? '');
             $this->folderCounts = array_map(function ($int) {
@@ -241,26 +235,27 @@ public function run($request)
             $this->folderCounts = FTFileMakerTask::config()->get('folderCountByDepth');
         }
 
-        echo "Downloading fixtures" . $this->lineBreak;
-        $fixtureFilePaths = $this->downloadFixtureFiles();
+        $output->writeln('Downloading fixtures');
+        $fixtureFilePaths = $this->downloadFixtureFiles($output);
 
         if (!FTFileMakerTask::config()->get('documentsOnly')) {
-            echo "Generate thumbnails" . $this->lineBreak;
+            $output->writeln('Generate thumbnails');
             $this->generateThumbnails($fixtureFilePaths);
         }
 
-        echo "Generate files" . $this->lineBreak;
+        $output->writeln('Generate files');
         $this->generateFiles($fixtureFilePaths);
 
         if (!FTFileMakerTask::config()->get('doPutProtectedFilesInPublicStore')) {
-            echo "Incorrectly putting protected files into public asset store on purpose" . $this->lineBreak;
+            $output->writeln('Incorrectly putting protected files into public asset store on purpose');
             $this->putProtectedFilesInPublicAssetStore();
         }
+        return Command::SUCCESS;
     }
 
-    protected function reset()
+    protected function reset(HybridOutput $output)
     {
-        echo "Resetting assets" . $this->lineBreak;
+        $output->writeln('Resetting assets');
 
         DB::query('TRUNCATE "File"');
         DB::query('TRUNCATE "File_Live"');
@@ -271,7 +266,7 @@ protected function reset()
         }
     }
 
-    protected function downloadFixtureFiles()
+    protected function downloadFixtureFiles(HybridOutput $output)
     {
         $client = new Client(['base_uri' => $this->fixtureFileBaseUrl]);
 
@@ -293,7 +288,7 @@ protected function downloadFixtureFiles()
                 $promises[$filename] = $client->getAsync($filename, [
                     'sink' => $path
                 ]);
-                echo "Downloading $url" . $this->lineBreak;
+                $output->writeln("Downloading $url");
             }
         }
 
@@ -497,4 +492,28 @@ protected function watermarkImage(string $stampPath, string $targetPath): ?strin
         return $targetPath;
     }
 
+    public function getOptions(): array
+    {
+        return [
+            new InputOption(
+                'reset',
+                null,
+                InputOption::VALUE_NONE,
+                'Optionally truncate ALL files and folders in the database,'
+                . ' plus delete the entire `assets/` directory',
+            ),
+            new InputOption(
+                'fileCounts',
+                null,
+                InputOption::VALUE_REQUIRED,
+                'Comma separated string'
+            ),
+            new InputOption(
+                'folderCounts',
+                null,
+                InputOption::VALUE_REQUIRED,
+                'Comma separated string'
+            ),
+        ];
+    }
 }