diff --git a/src/main/kotlin/platform/mixin/inspection/mixinextras/WrongOperationParametersInspection.kt b/src/main/kotlin/platform/mixin/inspection/mixinextras/WrongOperationParametersInspection.kt index 34491a9fa..12416ea77 100644 --- 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 index 7d74f6cdb..93ae8408d 100644 --- 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."