Make "files" argument optional (#14)

Nyholm · web-flow · commit 973a8ace746e · 2021-04-19T09:49:29.000+02:00
diff --git a/composer.json b/composer.json
@@ -11,6 +11,7 @@
         "symfony/docs-builder": "^0.15.0",
         "symfony/dotenv": "^5.2",
         "symfony/event-dispatcher": "^5.2",
+        "symfony/finder": "^5.2",
         "symfony/flex": "^1.3.1",
         "symfony/framework-bundle": "^5.2",
         "symfony/process": "^5.2",
diff --git a/src/Command/CheckDocsCommand.php b/src/Command/CheckDocsCommand.php
@@ -20,6 +20,7 @@
 use Symfony\Component\Console\Output\OutputInterface;
 use Symfony\Component\Console\Style\SymfonyStyle;
 use Symfony\Component\Filesystem\Filesystem;
+use Symfony\Component\Finder\Finder;
 use SymfonyDocsBuilder\BuildConfig;
 
 class CheckDocsCommand extends Command
@@ -43,7 +44,7 @@ protected function configure()
     {
         $this
             ->addArgument('source-dir', InputArgument::REQUIRED, 'RST files Source directory')
-            ->addArgument('files', InputArgument::IS_ARRAY + InputArgument::REQUIRED, 'RST files that should be verified.')
+            ->addArgument('files', InputArgument::IS_ARRAY, 'RST files that should be verified.', [])
             ->addOption('output-format', null, InputOption::VALUE_REQUIRED, 'Valid options are github and console', 'github')
             ->addOption('generate-baseline', null, InputOption::VALUE_REQUIRED, 'Generate a new baseline', false)
             ->addOption('baseline', null, InputOption::VALUE_REQUIRED, 'Use a baseline', false)
@@ -77,6 +78,10 @@ protected function initialize(InputInterface $input, OutputInterface $output)
     protected function execute(InputInterface $input, OutputInterface $output): int
     {
         $files = $input->getArgument('files');
+        if ([] === $files) {
+            $files = $this->findFiles($input->getArgument('source-dir'));
+        }
+
         $parseQueue = new ParseQueue();
         foreach ($files as $filename) {
             // Remove ".rst"
@@ -121,4 +126,17 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 
         return Command::SUCCESS;
     }
+
+    private function findFiles(string $directory): array
+    {
+        $files = [];
+        $finder = new Finder();
+        $finder->in($directory)
+            ->name('*.rst');
+        foreach ($finder as $file) {
+            $files[] = $file->getRelativePathname();
+        }
+
+        return $files;
+    }
 }
