New: Support WrongOperationParametersInspection in @WrapMethods. (#…

…2303)
minecraft-dev · Jun 1, 2024 · efcfd5d · efcfd5d
1 parent 10bd8d7
commit efcfd5d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/main/kotlin/platform/mixin/inspection/mixinextras/WrongOperationParametersInspection.kt b/src/main/kotlin/platform/mixin/inspection/mixinextras/WrongOperationParametersInspection.kt
@@ -61,7 +61,9 @@ class WrongOperationParametersInspection : MixinInspection() {
                 PsiField::class.java
             ) ?: return
 
-            if (!containingMethod.hasAnnotation(MixinConstants.MixinExtras.WRAP_OPERATION)) {
+            if (!containingMethod.hasAnnotation(MixinConstants.MixinExtras.WRAP_OPERATION) &&
+                !containingMethod.hasAnnotation(MixinConstants.MixinExtras.WRAP_METHOD)
+            ) {
                 return
             }
 

diff --git a/src/main/kotlin/platform/mixin/util/MixinConstants.kt b/src/main/kotlin/platform/mixin/util/MixinConstants.kt
@@ -86,6 +86,7 @@ object MixinConstants {
     object MixinExtras {
         const val OPERATION = "com.llamalad7.mixinextras.injector.wrapoperation.Operation"
         const val WRAP_OPERATION = "com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation"
+        const val WRAP_METHOD = "com.llamalad7.mixinextras.injector.wrapmethod.WrapMethod"
         const val LOCAL = "com.llamalad7.mixinextras.sugar.Local"
         const val LOCAL_REF_PACKAGE = "com.llamalad7.mixinextras.sugar.ref."