improve class pattern (#85)

openpilot-hub · Sep 1, 2024 · a63721e · a63721e
1 parent 5ca10ff
commit a63721e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/zhongan/devpilot/util/NewFileUtils.java b/src/main/java/com/zhongan/devpilot/util/NewFileUtils.java
@@ -121,13 +121,13 @@ private static PsiDirectory createPsiDirectory(Project project, String dirPath)
     }
 
     private static String extraClassNameFromGeneratedText(String generatedText) {
-        String regex = "public class (\\w+) \\{";
+        String regex = "(?:public\\s+)?(?:abstract\\s+|final\\s+)?(class|interface|enum)\\s+(\\w+)";
         Pattern pattern = Pattern.compile(regex);
         Matcher matcher = pattern.matcher(generatedText);
 
         String generatedClassName = null;
         if (matcher.find()) {
-            generatedClassName = matcher.group(1);
+            generatedClassName = matcher.group(2);
         }
         return generatedClassName;
     }