From 2ab59402046bb61611e2cf3efecddbfae8856625 Mon Sep 17 00:00:00 2001
From: Kamil Patryk Kozakowski <kamil.p.kozakowski@gmail.com>
Date: Mon, 14 Jul 2025 21:05:31 +0200
Subject: [PATCH] fetches more classes

---
 .../app/StateMachineTransitionConfigurationMethodFinder.java | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java b/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java
index eb53942..7204327 100644
--- a/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java
+++ b/src/main/java/click/kamil/springstatemachineexporter/ast/app/StateMachineTransitionConfigurationMethodFinder.java
@@ -8,6 +8,7 @@ import java.util.List;
 public class StateMachineTransitionConfigurationMethodFinder {
     private final CompilationUnit cu;
     private final ConfigureMethodVisitor visitor = new ConfigureMethodVisitor();
+    private static final List<String> stateMachineConfigurationAdapterSuperclasses = List.of("EnumStateMachineConfigurerAdapter", "StateMachineConfigurerAdapter");
 
     public StateMachineTransitionConfigurationMethodFinder(String source) {
         ASTParser parser = ASTParser.newParser(AST.getJLSLatest());
@@ -47,9 +48,9 @@ public class StateMachineTransitionConfigurationMethodFinder {
 
             if (superclass.isParameterizedType()) {
                 ParameterizedType pt = (ParameterizedType) superclass;
-                return pt.getType().toString().equals("StateMachineConfigurerAdapter");
+                return stateMachineConfigurationAdapterSuperclasses.contains(pt.getType().toString());
             }
-            return superclass.toString().equals("StateMachineConfigurerAdapter");
+            return stateMachineConfigurationAdapterSuperclasses.contains(superclass.toString());
         }
 
         private boolean isConfigureMethod(MethodDeclaration method) {