I think... I think it's done.

YTG1234 · Nov 1, 2020 · 854e6a5 · 854e6a5
1 parent f7c14ed
commit 854e6a5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/main/java/io/github/protonmc/ProtonCheckstylePlugin/FromModuleCheck.java b/src/main/java/io/github/protonmc/ProtonCheckstylePlugin/FromModuleCheck.java
@@ -1,6 +1,8 @@
 package io.github.protonmc.ProtonCheckstylePlugin;
 
-import com.puppycrawl.tools.checkstyle.api.*;
+import com.puppycrawl.tools.checkstyle.api.AbstractCheck;
+import com.puppycrawl.tools.checkstyle.api.DetailAST;
+import com.puppycrawl.tools.checkstyle.api.TokenTypes;
 
 public class FromModuleCheck extends AbstractCheck {
     @Override
@@ -27,10 +29,11 @@ public void visitToken(DetailAST ast) {
             boolean foundFromModule = false;
             boolean foundMixinMethod = false;
             for (int i = 0; i < modifier.getChildCount(); i++) {
-                if (modifier.findFirstToken(TokenTypes.IDENT).getText().equals("Inject") ||
-                    modifier.findFirstToken(TokenTypes.IDENT).getText().equals("Redirect") ||
-                    modifier.findFirstToken(TokenTypes.IDENT).getText().equals("ModifyArg") ||
-                    modifier.findFirstToken(TokenTypes.IDENT).getText().equals("ModifyArgs")) {
+                if (modifier.findFirstToken(TokenTypes.IDENT).getText().contains("Inject") ||
+                    modifier.findFirstToken(TokenTypes.IDENT).getText().contains("Redirect") ||
+                    modifier.findFirstToken(TokenTypes.IDENT).getText().contains("ModifyArg") ||
+                    modifier.findFirstToken(TokenTypes.IDENT).getText().contains("ModifyConstant") ||
+                    modifier.findFirstToken(TokenTypes.IDENT).getText().contains("ModifyVariable")) {
                     foundMixinMethod = true;
                 }
                 if (modifier.findFirstToken(TokenTypes.IDENT).getText().equals("FromModule")) {