Skip to content

Rename variables if the name matches class name #5454

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 13 commits into from
May 31, 2025
Merged

Rename variables if the name matches class name #5454

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
8551095
Rename variables if the name matches class name
radoslaw-panuszewski May 19, 2025
8e132e2
Prevent variable name conflicts; support unicode characters
radoslaw-panuszewski May 20, 2025
0820d5c
Fix tests in rewrite-kotlin
radoslaw-panuszewski May 20, 2025
f2f61a6
Add test verifying that random variables with matching name are not r…
radoslaw-panuszewski May 20, 2025
5bad96a
Use 𐐅/𐐭 instead of Ą/ą in test since Ą is in BMP
radoslaw-panuszewski May 21, 2025
9c4ea8f
Revert unicode characters handling in decapitalize method as it seems…
radoslaw-panuszewski May 21, 2025
255bbb3
Follow naming convention for ctx
timtebeek May 27, 2025
364f961
Move all logic into `visitVariable` and leverage `RenameVariable`
timtebeek May 31, 2025
9ffa9cf
Swap order of checks
timtebeek May 31, 2025
2eb0536
No changes to variable name without change to type
timtebeek May 31, 2025
263f59c
Merge branch 'main' into rename-variables-when-changing-type
timtebeek May 31, 2025
1381045
Update cursor before `generateVariableName` to avoid conflicts
timtebeek May 31, 2025
cc64311
Do not expect Kotlin changes just yet
timtebeek May 31, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
99 changes: 97 additions & 2 deletions rewrite-java-test/src/test/java/org/openrewrite/java/ChangeTypeTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1592,8 +1592,8 @@ public A1 method(A1 a1) {
import a.A2;

public class Example {
public A2 method(A2 a1) {
return a1;
public A2 method(A2 a2) {
return a2;
Comment on lines +1595 to +1596
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Neat idea indeed to rename these as well!

}
}
"""
Expand Down Expand Up @@ -1630,6 +1630,101 @@ public class Test {
);
}

@Test
void doNotRenameRandomVariablesMatchingClassName() {
rewriteRun(
spec -> spec.recipe(new ChangeType("a.A1", "a.A2", false)),
java(
"""
package a;
public class A1 {
}
""",
"""
package a;
public class A2 {
}
"""
),
java(
"""
package org.foo;

import a.A1;

public class Example {
public String method(A1 a, String a1) {
return a1;
}
}
""",
"""
package org.foo;

import a.A2;

public class Example {
public String method(A2 a, String a1) {
return a1;
}
}
"""
)
);
}

@Test
void noChangeToVariableNameWithoutChangeToType() {
rewriteRun(
spec -> spec.recipe(new ChangeType("a.A1", "a.A2", true)),
java(
"""
package a;
public class A1 {
}
"""
),
java(
"""
package a;
public class A2 {
}
"""
),
java(
"""
package org.foo;

import a.A1;
import a.A2;

public class Example {
public A1 method1(A1 a1) {
return a1;
}
public A2 method2(A2 a1) {
return a1; // Unchanged
}
}
""",
"""
package org.foo;

import a.A2;

public class Example {
public A2 method1(A2 a2) {
return a2;
}
public A2 method2(A2 a1) {
return a1; // Unchanged
}
}
"""
)
);
}

@Test
void updateVariableType() {
rewriteRun(
Expand Down
28 changes: 28 additions & 0 deletions rewrite-java/src/main/java/org/openrewrite/java/ChangeType.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
import org.jspecify.annotations.Nullable;
import org.openrewrite.*;
import org.openrewrite.internal.ListUtils;
import org.openrewrite.java.VariableNameUtils.GenerationStrategy;
import org.openrewrite.java.search.UsesType;
import org.openrewrite.java.tree.*;
import org.openrewrite.marker.Markers;
Expand Down Expand Up @@ -387,6 +388,26 @@ public J visitIdentifier(J.Identifier ident, ExecutionContext ctx) {
return visitAndCast(ident, ctx, super::visitIdentifier);
}

@Override
public J.VariableDeclarations.NamedVariable visitVariable(J.VariableDeclarations.NamedVariable variable, ExecutionContext ctx) {
J.VariableDeclarations.NamedVariable v = (J.VariableDeclarations.NamedVariable) super.visitVariable(variable, ctx);
if (v != variable) {
if (v.getSimpleName().equals(decapitalize(originalType.getClassName()))) {
if (targetType instanceof JavaType.FullyQualified) {
if (v.getVariableType() != null && TypeUtils.isOfType(targetType, v.getVariableType().getType())) {
String newName = VariableNameUtils.generateVariableName(
decapitalize(((JavaType.FullyQualified) targetType).getClassName()),
updateCursor(v),
GenerationStrategy.INCREMENT_NUMBER
);
doAfterVisit(new RenameVariable<>(v, newName));
}
}
}
}
return v;
}

@Override
public J visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
if (method.getMethodType() != null && method.getMethodType().hasFlags(Flag.Static)) {
Expand Down Expand Up @@ -778,4 +799,11 @@ private static boolean hasSameFQN(J.Import import_, JavaType targetType) {

return fqn != null && fqn.equals(curFqn);
}

private static String decapitalize(@Nullable String string) {
if (string != null && !string.isEmpty()) {
return Character.toLowerCase(string.charAt(0)) + string.substring(1);
}
return "";
}
}