tiny/spring-state-machine-renderer
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve REST path binding for enum path variables and valueOf

Browse Source 
Treat enum @PathVariable parameters as internal triggers, expand multiple path placeholders via cartesian bindings, derive event cases from proven Enum.valueOf chains, and add enterprise HTML regression coverage.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Kamil Patryk Kozakowski
2026-07-14 06:52:27 +02:00
parent 24fe8dfe00
commit 5d432e8893
5 changed files with 295 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/EntryPointBindingExpander.java
View File

@@ -50,6 +50,8 @@ public final class EntryPointBindingExpander {
if (entryPoint == null || entryPoint.getParameters() == null || entryPoint.getParameters().isEmpty()) {
return List.of();
}
List<Map<String, String>> variants = new ArrayList<>();
variants.add(new LinkedHashMap<>());
for (EntryPoint.Parameter parameter : entryPoint.getParameters()) {
if (!isPathOrQueryVariable(parameter)) {
continue;
@@ -57,7 +59,7 @@ public final class EntryPointBindingExpander {
if (isPlaceholderResolvedInPath(entryPoint, parameter.getName())) {
continue;
}
List<String> cases = resolveStringCasesForParameter(
List<String> cases = resolveBindingCasesForParameter(
entryPoint.getClassName() + "." + entryPoint.getMethodName(),
parameter.getName(),
context,
@@ -65,13 +67,17 @@ public final class EntryPointBindingExpander {
if (cases.isEmpty()) {
continue;
}
List<Map<String, String>> variants = new ArrayList<>();
for (String caseValue : cases) {
variants.add(Map.of(parameter.getName(), caseValue));
List<Map<String, String>> expanded = new ArrayList<>();
for (Map<String, String> base : variants) {
for (String caseValue : cases) {
Map<String, String> merged = new LinkedHashMap<>(base);
merged.put(parameter.getName(), caseValue);
expanded.add(merged);
}
}
return variants;
variants = expanded;
}
return List.of();
return variants.stream().filter(map -> !map.isEmpty()).toList();
}
public static EntryPoint withResolvedPath(EntryPoint entryPoint, Map<String, String> bindings) {
@@ -260,6 +266,18 @@ public final class EntryPointBindingExpander {
.anyMatch(a -> "PathVariable".equals(a) || "RequestParam".equals(a));
}
private static List<String> resolveBindingCasesForParameter(
String startMethod,
String paramName,
CodebaseContext context,
Map<String, List<CallEdge>> callGraph) {
List<String> stringCases = resolveStringCasesForParameter(startMethod, paramName, context, callGraph);
if (!stringCases.isEmpty()) {
return stringCases;
}
return resolveEnumValueOfCasesForParameter(startMethod, paramName, context, callGraph);
}
private static List<String> resolveStringCasesForParameter(
String startMethod,
String paramName,
@@ -293,6 +311,122 @@ public final class EntryPointBindingExpander {
return List.of();
}
private static List<String> resolveEnumValueOfCasesForParameter(
String startMethod,
String paramName,
CodebaseContext context,
Map<String, List<CallEdge>> callGraph) {
ArrayDeque<String> queue = new ArrayDeque<>();
Set<String> visited = new HashSet<>();
queue.add(startMethod);
while (!queue.isEmpty()) {
String methodFqn = queue.poll();
if (!visited.add(methodFqn)) {
continue;
}
if (visited.size() > 12) {
break;
}
List<String> localCases = extractEnumValueOfCases(methodFqn, paramName, context);
if (!localCases.isEmpty()) {
return localCases;
}
List<CallEdge> edges = callGraph.get(methodFqn);
if (edges == null) {
continue;
}
for (CallEdge edge : edges) {
if (edge.getTargetMethod() != null) {
queue.add(edge.getTargetMethod());
}
}
}
return List.of();
}
static List<String> extractEnumValueOfCases(String methodFqn, String paramName, CodebaseContext context) {
MethodDeclaration method = findMethodDeclaration(methodFqn, context);
if (method == null || method.getBody() == null || context == null) {
return List.of();
}
if (!hasParameter(method, paramName)) {
return List.of();
}
LinkedHashSet<String> enumTypes = new LinkedHashSet<>();
method.getBody().accept(new ASTVisitor() {
@Override
public boolean visit(MethodInvocation node) {
if (!"valueOf".equals(node.getName().getIdentifier()) || node.arguments().size() != 1) {
return super.visit(node);
}
Expression argument = (Expression) node.arguments().get(0);
if (!argumentReferencesParameter(argument, paramName)) {
return super.visit(node);
}
String enumType = resolveValueOfEnumType(node, context);
if (enumType != null) {
enumTypes.add(enumType);
}
return super.visit(node);
}
});
if (enumTypes.size() != 1) {
return List.of();
}
List<String> constants = context.getEnumValues(enumTypes.iterator().next());
if (constants == null || constants.isEmpty()) {
return List.of();
}
LinkedHashSet<String> pathValues = new LinkedHashSet<>();
for (String constant : constants) {
if (constant == null || constant.isBlank()) {
continue;
}
int lastDot = constant.lastIndexOf('.');
pathValues.add(lastDot >= 0 ? constant.substring(lastDot + 1) : constant);
}
return List.copyOf(pathValues);
}
private static boolean argumentReferencesParameter(Expression argument, String paramName) {
if (argument instanceof SimpleName simpleName) {
return paramName.equals(simpleName.getIdentifier());
}
if (argument instanceof MethodInvocation invocation
&& "toUpperCase".equals(invocation.getName().getIdentifier())
&& invocation.arguments().isEmpty()
&& invocation.getExpression() instanceof SimpleName simpleName) {
return paramName.equals(simpleName.getIdentifier());
}
if (argument instanceof MethodInvocation invocation
&& "toLowerCase".equals(invocation.getName().getIdentifier())
&& invocation.arguments().isEmpty()
&& invocation.getExpression() instanceof SimpleName simpleName) {
return paramName.equals(simpleName.getIdentifier());
}
return false;
}
private static String resolveValueOfEnumType(MethodInvocation valueOfInvocation, CodebaseContext context) {
Expression receiver = valueOfInvocation.getExpression();
if (receiver != null) {
org.eclipse.jdt.core.dom.ITypeBinding binding = receiver.resolveTypeBinding();
if (binding != null && binding.isEnum()) {
return binding.getQualifiedName();
}
}
if (receiver instanceof QualifiedName qualifiedName) {
return qualifiedName.getFullyQualifiedName();
}
if (receiver instanceof SimpleName simpleName && context != null) {
List<String> values = context.getEnumValues(simpleName.getIdentifier());
if (values != null && !values.isEmpty()) {
return simpleName.getIdentifier();
}
}
return null;
}
static List<String> extractStringSwitchCases(String methodFqn, String paramName, CodebaseContext context) {
MethodDeclaration method = findMethodDeclaration(methodFqn, context);
if (method == null || method.getBody() == null) {

10
state_machine_exporter/src/main/java/click/kamil/springstatemachineexporter/analysis/service/ExternalTriggerPolicy.java
View File

@@ -39,7 +39,7 @@ public final class ExternalTriggerPolicy {
if (paramName != null) {
RestParamBinding binding = findRestParameterBinding(entryPoint, paramName, context);
if (binding != null) {
if (binding.pathOrQueryVariable()) {
if (binding.pathOrQueryVariable() && !binding.enumType()) {
return true;
}
if (binding.requestBodyEnum()) {
@@ -56,13 +56,19 @@ public final class ExternalTriggerPolicy {
}
if (entryPoint.getName() != null && entryPoint.getName().contains("{")
&& MachineEnumCanonicalizer.isDynamicTriggerExpression(trigger.getEvent())) {
if (paramName != null) {
RestParamBinding pathBinding = findRestParameterBinding(entryPoint, paramName, context);
if (pathBinding != null && pathBinding.pathVariable() && pathBinding.enumType()) {
return false;
}
}
return true;
}
}
if (entryMethodFqn != null && resolvedEventParamName != null) {
RestParamBinding binding = findMethodParameterBinding(entryMethodFqn, resolvedEventParamName, context);
if (binding != null) {
if (binding.pathOrQueryVariable()) {
if (binding.pathOrQueryVariable() && !binding.enumType()) {
return true;
}
if (binding.requestBodyEnum()) {

46
state_machine_exporter/src/test/java/click/kamil/springstatemachineexporter/analysis/service/EntryPointBindingExpanderTest.java
View File

@@ -198,4 +198,50 @@ class EntryPointBindingExpanderTest {
assertThat(variants).extracting(map -> map.get("machineType"))
.containsExactlyInAnyOrder("ORDER", "DOCUMENT");
}
@Test
void shouldExpandEventPathVariableFromPreservedEnumValueOfParam(@TempDir Path tempDir) throws Exception {
String source = """
package com.example;
public class OrderController {
OrderGateway gateway;
public void transition(String event) {
gateway.trigger(event);
}
}
class OrderGateway {
void trigger(String event) {
OrderEvent.valueOf(event.toUpperCase());
}
}
enum OrderEvent { PAY, SHIP }
""";
Files.writeString(tempDir.resolve("App.java"), source);
CodebaseContext context = new CodebaseContext();
context.setResolveBindings(true);
context.scan(tempDir);
HeuristicCallGraphEngine engine = new HeuristicCallGraphEngine(context);
Map<String, List<click.kamil.springstatemachineexporter.analysis.model.CallEdge>> graph = engine.buildCallGraph();
EntryPoint entryPoint = EntryPoint.builder()
.className("com.example.OrderController")
.methodName("transition")
.name("POST /api/order/{event}")
.parameters(List.of(EntryPoint.Parameter.builder()
.name("event")
.type("String")
.annotations(List.of("PathVariable"))
.build()))
.build();
List<Map<String, String>> variants =
EntryPointBindingExpander.expandPathVariableBindings(entryPoint, context, graph);
assertThat(variants).extracting(map -> map.get("event"))
.containsExactlyInAnyOrder("PAY", "SHIP");
assertThat(EntryPointBindingExpander.withResolvedPath(entryPoint, variants.get(0)).getName())
.isEqualTo("POST /api/order/PAY");
}
}

37
state_machine_exporter/src/test/java/click/kamil/springstatemachineexporter/analysis/service/ExternalTriggerPolicyTest.java
View File

@@ -35,6 +35,43 @@ class ExternalTriggerPolicyTest {
entryPoint, trigger, "com.example.Api.trigger", "event", null)).isTrue();
}
@Test
void pathVariableEnumParameterShouldBeInternal(@TempDir Path tempDir) throws Exception {
String source = """
package com.example;
import org.springframework.web.bind.annotation.*;
public class Api {
@PostMapping("/api/order/{event}")
public void transition(@PathVariable OrderEvent event) {
fire(event);
}
void fire(OrderEvent event) {}
}
enum OrderEvent { PAY, SHIP }
""";
Files.writeString(tempDir.resolve("Api.java"), source);
CodebaseContext context = new CodebaseContext();
context.scan(tempDir);
EntryPoint entryPoint = EntryPoint.builder()
.type(EntryPoint.Type.REST)
.name("POST /api/order/{event}")
.className("com.example.Api")
.methodName("transition")
.parameters(List.of(EntryPoint.Parameter.builder()
.name("event")
.type("com.example.OrderEvent")
.annotations(List.of("PathVariable"))
.build()))
.build();
TriggerPoint trigger = TriggerPoint.builder()
.event("event")
.build();
assertThat(ExternalTriggerPolicy.isExternalFromSource(
entryPoint, trigger, "com.example.Api.transition", "event", context)).isFalse();
}
@Test
void requestBodyEnumShouldBeInternal(@TempDir Path tempDir) throws Exception {
String source = """