Add support for wildcard within directory name

Author: rhart

src/main/java/com/anacoders/cookbook/CreateYamlFilesByPattern.java

@@ -42,13 +42,14 @@ public String getDisplayName() {
 
     @Override
     public String getDescription() {
-        return "Creates YAML files in directories matching a glob-like pattern. Supports * (single segment) and ** (zero or more segments). " +
+        return "Creates YAML files in directories matching a glob-like pattern. Supports * (single segment or within segment) and ** (zero or more segments). " +
                 "Files are only created if they don't already exist. Example: 'projects/*/config.yaml' creates config.yaml in each direct child of projects/. " +
+                "Example: 'projects/project-*/config.yaml' matches directories like 'project-a', 'project-b'. " +
                 "Example: 'src/**/config/app.yaml' matches any 'config' directory under src at any depth.";
     }
 
     @Option(displayName = "File pattern",
-            description = "Pattern with wildcards for where to create files. Use * for single directory wildcard, ** for recursive. " +
+            description = "Pattern with wildcards for where to create files. Use * for single directory wildcard or within segment (e.g., 'project-*'), ** for recursive. " +
                     "Example: 'projects/*/config.yaml' creates config.yaml in each direct child of projects/",
             example = "projects/*/config.yaml")
     String filePattern;
@@ -174,6 +175,14 @@ private boolean matchSegments(List<String> pathSegs, List<String> patternSegs, i
                 pathIdx++;
                 patternIdx++;
 
+            } else if (pattern.contains("*") || pattern.contains("?")) {
+                // Glob pattern within segment (e.g., "project-*")
+                if (!matchGlobPattern(pathSegs.get(pathIdx), pattern)) {
+                    return false;
+                }
+                pathIdx++;
+                patternIdx++;
+
             } else {
                 // Literal segment must match exactly
                 if (!pattern.equals( pathSegs.get( pathIdx ) )) {
@@ -192,6 +201,32 @@ private boolean matchSegments(List<String> pathSegs, List<String> patternSegs, i
         return patternIdx == patternSegs.size() && pathIdx == pathSegs.size();
     }
 
+    /**
+     * Matches a path segment against a glob pattern that may contain wildcards.
+     * Supports * for matching any sequence of characters within a segment.
+     * Supports ? for matching any single character.
+     * Examples: "project-*" matches "project-a", "project-foo", etc.
+     */
+    private boolean matchGlobPattern(String segment, String pattern) {
+        // Convert glob pattern to regex
+        StringBuilder regex = new StringBuilder("^");
+        for (int i = 0; i < pattern.length(); i++) {
+            char c = pattern.charAt(i);
+            if (c == '*') {
+                regex.append(".*");
+            } else if (c == '?') {
+                regex.append(".");
+            } else if ("[]{}().+^$|\\".indexOf(c) != -1) {
+                // Escape regex special characters
+                regex.append("\\").append(c);
+            } else {
+                regex.append(c);
+            }
+        }
+        regex.append("$");
+        return segment.matches(regex.toString());
+    }
+
     private Set<Path> resolveTargetFilePaths(Set<Path> existingDirectories) {
         List<String> patternSegs = splitPattern(filePattern);
         if (patternSegs.isEmpty()) {
@@ -220,3 +255,4 @@ private Set<Path> resolveTargetFilePaths(Set<Path> existingDirectories) {
         return targets;
     }
 }
+

src/test/java/com/anacoders/cookbook/CreateYamlFilesByPatternTest.java

@@ -14,24 +14,19 @@
  * limitations under the License.
  */
 package com.anacoders.cookbook;
-
 import org.junit.jupiter.api.Test;
 import org.openrewrite.DocumentExample;
 import org.openrewrite.test.RecipeSpec;
 import org.openrewrite.test.RewriteTest;
-
 import static org.openrewrite.yaml.Assertions.yaml;
-
 class CreateYamlFilesByPatternTest implements RewriteTest {
-
     @Override
     public void defaults(RecipeSpec spec) {
         spec.recipe(new CreateYamlFilesByPattern(
                 "projects/*/config.yaml",
                 "apiVersion: v1\nkind: Config\nmetadata:\n  name: example"
         ));
     }
-
     @DocumentExample
     @Test
     void createsFilesInMatchingDirectoriesAndLeavesExistingFilesUntouched() {
@@ -92,7 +87,6 @@ void createsFilesInMatchingDirectoriesAndLeavesExistingFilesUntouched() {
                 )
         );
     }
-
     @Test
     void doesNothingWhenNoMatchingDirectories() {
         rewriteRun(
@@ -109,7 +103,6 @@ void doesNothingWhenNoMatchingDirectories() {
                 )
         );
     }
-
     @Test
     void worksWithNestedAndDeeplyNestedDirectories() {
         rewriteRun(
@@ -172,7 +165,6 @@ void worksWithNestedAndDeeplyNestedDirectories() {
                 )
         );
     }
-
     @Test
     void handlesMultipleWildcardsWithStarAndDoubleStar() {
         rewriteRun(
@@ -232,7 +224,6 @@ void handlesMultipleWildcardsWithStarAndDoubleStar() {
                 )
         );
     }
-
     @Test
     void skipsPartiallyMatchingDirectories() {
         rewriteRun(
@@ -264,7 +255,6 @@ void skipsPartiallyMatchingDirectories() {
                 )
         );
     }
-
     @Test
     void supportsDoubleStarRecursiveMatching() {
         rewriteRun(
@@ -315,7 +305,6 @@ void supportsDoubleStarRecursiveMatching() {
                 )
         );
     }
-
     @Test
     void exactPathAtRoot() {
         rewriteRun(
@@ -332,7 +321,6 @@ void exactPathAtRoot() {
                 )
         );
     }
-
     @Test
     void multipleDoubleStarsWithZeroSegmentMatching() {
         rewriteRun(
@@ -383,7 +371,6 @@ void multipleDoubleStarsWithZeroSegmentMatching() {
                 )
         );
     }
-
     @Test
     void filenameWithDots() {
         rewriteRun(
@@ -406,4 +393,45 @@ void filenameWithDots() {
                 )
         );
     }
+    @Test
+    void wildcardWithinDirectoryName() {
+        rewriteRun(
+                spec -> spec.recipe(new CreateYamlFilesByPattern(
+                        "projects/project-*/config.yaml",
+                        "content: value"
+                .expectedCyclesThatMakeChanges(2)),
+                yaml(
+                        """
+                        existing: file
+                        """,
+                        s -> s.path("projects/project-a/existing.yaml")
+                ),
+                yaml(
+                        """
+                        existing: file
+                        """,
+                        s -> s.path("projects/project-b/existing.yaml")
+                ),
+                yaml(
+                        """
+                        existing: file
+                        """,
+                        s -> s.path("projects/other/existing.yaml")
+                ),
+                yaml(
+                        doesNotExist(),
+                        """
+                        content: value
+                        """,
+                        s -> s.path("projects/project-a/config.yaml")
+                ),
+                yaml(
+                        doesNotExist(),
+                        """
+                        content: value
+                        """,
+                        s -> s.path("projects/project-b/config.yaml")
+                )
+        );
+    }
 }