Merge pull request #10 from LuigiPulcini/main

Add support for `prefix` option
pronamic · Jan 27, 2022 · 70f8543 · 70f8543
2 parents 8478326 + bc88207
commit 70f8543
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/bin/wp-documentor b/bin/wp-documentor
@@ -49,6 +49,7 @@ $application->register( 'parse' )
 		null
 	)
 	->addOption( 'relative', null, InputOption::VALUE_REQUIRED, 'Relative to directory.', null )
+	->addOption( 'prefix', null, InputOption::VALUE_REQUIRED, 'Comma-separated list of prefixes hooks must match.', null )
 	->addOption( 'output', 'o', InputOption::VALUE_REQUIRED, 'Write output to file instead of stdout.', null )
 	->addOption( 'memory-limit', null, InputOption::VALUE_REQUIRED, 'Memory limit for the run.', null )
 	->addOption(
@@ -125,6 +126,10 @@ $application->register( 'parse' )
 
 		$documentor->relative = ( null === $relative ) ? $source : $relative;
 
+		$prefix = $input->getOption( 'prefix' );
+
+		$documentor->prefix = ( null === $prefix ) ? null : $prefix;
+
 		/**
 		 * Output.
 		 *

diff --git a/src/Documentor.php b/src/Documentor.php
@@ -173,6 +173,15 @@ function( Node $node ) {
 				);
 			}
 
+			if ( ! is_null( $this->prefix ) ) {
+				$pattern = sprintf( '/^(%s)/', implode( '|', explode( ',', $this->prefix ) ) );
+				preg_match( $pattern, $tag_name, $matches );
+
+				if ( empty( $matches ) ) {
+					continue;
+				}
+			}
+
 			$tag = new Tag( $tag_name, $tag_arg );
 
 			$doc_comment = $statement->getDocComment();