PsiElement (@NotNull GrExpression expression) { if (expression instanceof GrUnaryExpression unary) { return unary.getOperationToken(); } if (expression instanceof GrBinaryExpression binary) { return binary.getOperationToken(); } if (expression instanceof GrSafeCastExpression safeCast) { return safeCast.getOperationToken(); } return null; }