feat: add support for new mitogen. Older version with patching is sup…

…ported.

README.md

@@ -3,19 +3,23 @@
 This collection allows to use [Mitogen](https://github.com/mitogen-hq/mitogen)
 strategy without the need to specify absolute path to the strategy file.
 
-It also performs live patching for Mitogen restrictions in Ansible
-versions, making it possible to use Mitogen with the current Ansible version.
+It also performs live patching for oler versions of Mitogen (3.4), removing
+Ansible version restrictions, making it possible to use Mitogen
+on unsupported versions.
 
 Tested versions of ansible-core:
 
 * 2.14
 * 2.15
 * 2.16
+* 2.17
 
-It patches `ansible_mitogen` code and unpatches it
+When needed, it patches `ansible_mitogen` code and unpatches it
 back right after module import, so the original
 files are kept intact.
 
+If no patching needed, it just passes mitogen content as it is.
+
 ## Install
 To use this collection, you need to install Mitogen:
 

plugins/strategy/mitogen_free.py

@@ -31,8 +31,17 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+import importlib.metadata
 
-from . import patching
+try:
+    mitogen_version = importlib.metadata.version("mitogen")
+except importlib.metadata.PackageNotFoundError:
+    raise ImportError("Unable to load mitogen module.")
 
-with patching.patch_version():
+if mitogen_version == "0.3.4":
+    from . import patching
+
+    with patching.patch_version():
+        from ansible_mitogen.plugins.strategy.mitogen_free import *
+else:
     from ansible_mitogen.plugins.strategy.mitogen_free import *

plugins/strategy/mitogen_host_pinned.py

@@ -15,7 +15,17 @@
 short_description: Mitogen patching strategy
 """
 
-from . import patching
+import importlib.metadata
 
-with patching.patch_version():
+try:
+    mitogen_version = importlib.metadata.version("mitogen")
+except importlib.metadata.PackageNotFoundError:
+    raise ImportError("Unable to load mitogen module.")
+
+if mitogen_version == "0.3.4":
+    from . import patching
+
+    with patching.patch_version():
+        from ansible_mitogen.plugins.strategy.mitogen_host_pinned import *
+else:
     from ansible_mitogen.plugins.strategy.mitogen_host_pinned import *

plugins/strategy/mitogen_linear.py

@@ -15,7 +15,17 @@
 short_description: Mitogen patching strategy
 """
 
-from . import patching
+import importlib.metadata
 
-with patching.patch_version():
+try:
+    mitogen_version = importlib.metadata.version("mitogen")
+except importlib.metadata.PackageNotFoundError:
+    raise ImportError("Unable to load mitogen module.")
+
+if mitogen_version == "0.3.4":
+    from . import patching
+
+    with patching.patch_version():
+        from ansible_mitogen.plugins.strategy.mitogen_linear import *
+else:
     from ansible_mitogen.plugins.strategy.mitogen_linear import *