Allow passing only the template path, as long as it contains .phtml files

Simplifies usage for the default use case. If developers want to specify
an alternate extension, they can use file globbing.

Author: weierophinney

bin/templatemap_generator.php

@@ -11,13 +11,13 @@
 
 Usage:
 
-templatemap_generator.php [-h|--help] templatepath files...
+templatemap_generator.php [-h|--help] templatepath <files...>
 
 --help|-h                    Print this usage message.
 templatepath                 Path to templates relative to current working
                              path; used to identify what to strip from
                              template names. Must be a directory.
-files...                     List of files to include in the template
+<files...>                   List of files to include in the template
                              map, relative to the current working path.
 
 The script assumes that paths included in the template map are relative
@@ -30,7 +30,12 @@
 need to edit the file after generation to ensure it contains valid
 PHP).
 
-To provide a list of files, we recommend using one of the following.
+If only the templatepath argument is provided, the script will look for
+all .phtml files under that directory, creating a map for you.
+
+If you want to specify a specific list of files -- for instance, if you
+are using an extension other than .phtml -- we recommend one of the
+following constructs:
 
 For any shell, you can pipe the results of `find`:
 
@@ -48,16 +53,21 @@
 
 Examples:
 
+  # Using only a templatepath argument, which will match any .phtml
+  # files found under the provided path:
+  $ cd module/Application/config/
+  $ ../../../vendor/bin/templatemap_generator.php ../view > template_map.config.php
+
   # Create a template_map.config.php file in the Application module's
   # config directory, relative to the view directory, and only containing
-  # .phtml files; overwrite any existing files:
+  # .html.php files; overwrite any existing files:
   $ cd module/Application/config/
-  $ ../../../vendor/bin/templatemap_generator.php ../view ../view/**/*.phtml > template_map.config.php
+  $ ../../../vendor/bin/templatemap_generator.php ../view ../view/**/*.html.php > template_map.config.php
 
   # OR using find:
   $ ../../../vendor/bin/templatemap_generator.php \
   > ../view \
-  > $(find ../view -name '*.phtml') > template_map.config.php
+  > $(find ../view -name '*.html.php') > template_map.config.php
 EOH;
 
 // Called without arguments
@@ -80,20 +90,23 @@
     exit(2);
 }
 
-// Not enough arguments
-if ($argc < 3) {
+$basePath = $argv[1];
+$files = ($argc < 3)
+    ? findTemplateFilesInTemplatePath($basePath)
+    : array_slice($argv, 2);
+
+// No files provided
+if (empty($files)) {
     fwrite(STDERR, 'No files specified.' . PHP_EOL . PHP_EOL);
     fwrite(STDERR, $help . PHP_EOL);
     exit(2);
 }
 
-$basePath = $argv[1];
-$files    = array_slice($argv, 2);
-$map      = [];
+$map = [];
 $realPath = realpath($basePath);
 
 $entries = array_map(function ($file) use ($basePath, $realPath) {
-    $file     = str_replace('\\', '/', $file);
+    $file = str_replace('\\', '/', $file);
 
     $template = (0 === strpos($file, $realPath))
         ? substr($file, strlen($realPath))
@@ -118,3 +131,24 @@
     . '];';
 
 exit(0);
+
+function findTemplateFilesInTemplatePath($templatePath)
+{
+    $rdi = new RecursiveDirectoryIterator($templatePath, RecursiveDirectoryIterator::FOLLOW_SYMLINKS);
+    $rii = new RecursiveIteratorIterator($rdi, RecursiveIteratorIterator::LEAVES_ONLY);
+
+    $files = [];
+    foreach ($rii as $file) {
+        if (! $file instanceof SplFileInfo) {
+            continue;
+        }
+
+        if (! preg_match('#^phtml$#i', $file->getExtension())) {
+            continue;
+        }
+
+        $files[] = $file->getPathname();
+    }
+
+    return $files;
+}