From 4b42dd14340567c564d4dfb58dabac5ccbc266d0 Mon Sep 17 00:00:00 2001
From: Steven Robertson <srtrumpetaggie@gmail.com>
Date: Fri, 8 May 2020 18:58:09 -0700
Subject: [PATCH] able to load things from sys.modules but not everything loads
 that way

---
 mitogen/master.py | 32 ++++++++++++++++++++++++--------
 1 file changed, 24 insertions(+), 8 deletions(-)

diff --git a/mitogen/master.py b/mitogen/master.py
index f3c66c0e..e1d0ba00 100644
--- a/mitogen/master.py
+++ b/mitogen/master.py
@@ -138,7 +138,7 @@ def is_stdlib_path(path):
     )
 
 
-def get_child_modules(path):
+def get_child_modules(path, fullname):
     """
     Return the suffixes of submodules directly neated beneath of the package
     directory at `path`.
@@ -148,11 +148,29 @@ def get_child_modules(path):
         equivalent. Usually this is the module's ``__file__`` attribute, but
         is specified explicitly to avoid loading the module.
 
+    :param str fullname:
+        Full name of a module path. Only used with collections because
+        its modules can't be loaded with iter_modules()
+
     :return:
         List of submodule name suffixes.
     """
-    it = pkgutil.iter_modules([os.path.dirname(path)])
-    return [to_text(name) for _, name, _ in it]
+    if fullname.startswith("ansible_collections"):
+        # taco
+        # import epdb; epdb.set_trace()
+        # ISSUE: not everything is being loaded via sys.modules *facepalm*
+        # only `action` and `modules` show up from here: https://github.com/alikins/collection_inspect/tree/master/plugins 
+        # so we aren't able to load things like `module_utils`
+        # gonna have to go the file path route it looks like, or leverage other *method classes
+        submodules = []
+        import epdb; epdb.set_trace()
+        for each in dir(sys.modules[fullname]):
+            if not each.startswith("__"):
+                submodules.append(to_text(each))
+        return submodules
+    else:
+        it = pkgutil.iter_modules([os.path.dirname(path)])
+        return [to_text(name) for _, name, _ in it]
 
 
 def _looks_like_script(path):
@@ -560,8 +578,8 @@ class SysModulesMethod(FinderMethod):
                       fullname, alleged_name, module)
             return
 
-        if fullname == "ansible_collections":
-            # ansible named the fake __file__ for collections `__synthetic__` with no extension
+        if fullname.startswith("ansible_collections"):
+            # ansible names the fake __file__ for collections `__synthetic__` with no extension
             module.__file__ = module.__file__ + ".py"
             # import epdb; epdb.set_trace()
             # taco
@@ -1006,9 +1024,7 @@ class ModuleResponder(object):
         if is_pkg:
             # taco
             # child modules are empty...
-            # if fullname == "ansible_collections":
-            #     import epdb; epdb.set_trace()
-            pkg_present = get_child_modules(path)
+            pkg_present = get_child_modules(path, fullname)
             self._log.debug('%s is a package at %s with submodules %r',
                             fullname, path, pkg_present)
         else: